fix handling of large images

bunburya · Feb 6, 2023 · fc2629e · fc2629e
1 parent 53600cd
commit fc2629e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/bother_utils/heightmap.py b/bother_utils/heightmap.py
@@ -315,6 +315,12 @@ def crop_image(im: Image, width: int, height: int, mode: str) -> Image:
         box[0] = (im.width - width) // 2
     box[2] = box[0] + width
 
+
+    # Image.crop chokes on large images unless we increase the max size
+    size = width * height
+    if size > Image.MAX_IMAGE_PIXELS:
+        Image.MAX_IMAGE_PIXELS = size
+
     return im.crop(box)
 
 def scale_image(im: Image, width: int, height: int) -> Image: