From 8a451234b3090db488fbee9740a5f6be2f989253 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 18 Nov 2023 04:44:17 -0500
Subject: [PATCH] Add ImageCrop node.

---
 comfy_extras/nodes_images.py | 29 +++++++++++++++++++++++++++++
 nodes.py                     |  1 +
 2 files changed, 30 insertions(+)
 create mode 100644 comfy_extras/nodes_images.py

diff --git a/comfy_extras/nodes_images.py b/comfy_extras/nodes_images.py
new file mode 100644
index 00000000..2b8e9300
--- /dev/null
+++ b/comfy_extras/nodes_images.py
@@ -0,0 +1,29 @@
+import nodes
+MAX_RESOLUTION = nodes.MAX_RESOLUTION
+
+class ImageCrop:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": { "image": ("IMAGE",),
+                              "width": ("INT", {"default": 512, "min": 1, "max": MAX_RESOLUTION, "step": 1}),
+                              "height": ("INT", {"default": 512, "min": 1, "max": MAX_RESOLUTION, "step": 1}),
+                              "x": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1}),
+                              "y": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1}),
+                              }}
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "crop"
+
+    CATEGORY = "image/transform"
+
+    def crop(self, image, width, height, x, y):
+        x = min(x, image.shape[2] - 1)
+        y = min(y, image.shape[1] - 1)
+        to_x = width + x
+        to_y = height + y
+        img = image[:,y:to_y, x:to_x, :]
+        return (img,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "ImageCrop": ImageCrop,
+}
diff --git a/nodes.py b/nodes.py
index f9d2d7f6..2adc5e07 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1800,6 +1800,7 @@ def init_custom_nodes():
         "nodes_hypertile.py",
         "nodes_model_advanced.py",
         "nodes_model_downscale.py",
+        "nodes_images.py",
     ]
 
     for node_file in extras_files: