Fix Pixel Sort Default + Add Divide

Author: EllangoK

post_processing/arithmetic_blend.py

@@ -10,7 +10,7 @@ def INPUT_TYPES(s):
             "required": {
                 "image1": ("IMAGE",),
                 "image2": ("IMAGE",),
-                "blend_mode": (["add", "subtract", "difference"],),
+                "blend_mode": (["add", "subtract", "difference", "divide"],),
             },
         }
 
@@ -26,6 +26,8 @@ def arithmetic_blend_images(self, image1: torch.Tensor, image2: torch.Tensor, bl
             blended_image = self.subtract(image1, image2)
         elif blend_mode == "difference":
             blended_image = self.difference(image1, image2)
+        elif blend_mode == "divide":
+            blended_image = self.divide(image1, image2)
         else:
             raise ValueError(f"Unsupported arithmetic blend mode: {blend_mode}")
 
@@ -41,6 +43,10 @@ def subtract(self, img1, img2):
     def difference(self, img1, img2):
         return torch.abs(img1 - img2)
 
+    def divide(self, img1, img2):
+        img2_safe = torch.where(img1 == 0, torch.tensor(1e-10), img1)
+        return img1 / img2_safe
+
 NODE_CLASS_MAPPINGS = {
     "ArithmeticBlend": ArithmeticBlend,
 }

post_processing/pixel_sort.py

@@ -14,7 +14,7 @@ def INPUT_TYPES(cls):
                 "mask": ("IMAGE",),
                 "direction": (["horizontal", "vertical"],),
                 "span_limit": ("INT", {
-                    "default": None,
+                    "default": 50,
                     "min": 0,
                     "max": 100,
                     "step": 5

post_processing_nodes.py

@@ -17,7 +17,7 @@ def INPUT_TYPES(s):
             "required": {
                 "image1": ("IMAGE",),
                 "image2": ("IMAGE",),
-                "blend_mode": (["add", "subtract", "difference"],),
+                "blend_mode": (["add", "subtract", "difference", "divide"],),
             },
         }
 
@@ -33,6 +33,8 @@ def arithmetic_blend_images(self, image1: torch.Tensor, image2: torch.Tensor, bl
             blended_image = self.subtract(image1, image2)
         elif blend_mode == "difference":
             blended_image = self.difference(image1, image2)
+        elif blend_mode == "divide":
+            blended_image = self.divide(image1, image2)
         else:
             raise ValueError(f"Unsupported arithmetic blend mode: {blend_mode}")
 
@@ -48,6 +50,10 @@ def subtract(self, img1, img2):
     def difference(self, img1, img2):
         return torch.abs(img1 - img2)
 
+    def divide(self, img1, img2):
+        img2_safe = torch.where(img1 == 0, torch.tensor(1e-10), img1)
+        return img1 / img2_safe
+
 class AsciiArt:
     def __init__(self):
         pass
@@ -1051,7 +1057,7 @@ def INPUT_TYPES(cls):
                 "mask": ("IMAGE",),
                 "direction": (["horizontal", "vertical"],),
                 "span_limit": ("INT", {
-                    "default": None,
+                    "default": 50,
                     "min": 0,
                     "max": 100,
                     "step": 5