Avoid dividing by zero when writing a blank image

yt-project · Sep 27, 2023 · ed3bdac · ed3bdac
1 parent 3760d58
commit ed3bdac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/yt/visualization/image_writer.py b/yt/visualization/image_writer.py
@@ -149,6 +149,9 @@ def write_bitmap(bitmap_array, filename, max_val=None, transpose=False):
             alpha_channel.shape = s1, s2, 1
         if max_val is None:
             max_val = bitmap_array[:, :, :3].max()
+            if max_val == 0.0:
+                # avoid dividing by zero for blank images
+                max_val = 1.0
         bitmap_array = np.clip(bitmap_array[:, :, :3] / max_val, 0.0, 1.0) * 255
         bitmap_array = np.concatenate(
             [bitmap_array.astype("uint8"), alpha_channel], axis=-1