From 49004bb3d1d82f107fae0b0264b7816c7c08f575 Mon Sep 17 00:00:00 2001
From: chflame163 <chflame@163.com>
Date: Thu, 27 Jun 2024 09:37:44 +0800
Subject: [PATCH] rename color_correct_brightness&contrast.py to
 color_correct_brightness_and_contrast.py; color_correct_shadow&highlight.py
 to color_correct_shadow_and_highlight.py; rembg_ultra.py to rmbg_ultra.py;
 rembg_ultra_v2.py to rmbg_ultra.py

---
 ... color_correct_brightness_and_contrast.py} | 118 +++++++++---------
 ... => color_correct_shadow_and_highlight.py} |   0
 py/{rembg_ultra.py => rmbg_ultra.py}          |   0
 py/{rembg_ultra_v2.py => rmbg_ultra_v2.py}    |   0
 4 files changed, 59 insertions(+), 59 deletions(-)
 rename py/{color_correct_brightness&contrast.py => color_correct_brightness_and_contrast.py} (97%)
 rename py/{color_correct_shadow&highlight.py => color_correct_shadow_and_highlight.py} (100%)
 rename py/{rembg_ultra.py => rmbg_ultra.py} (100%)
 rename py/{rembg_ultra_v2.py => rmbg_ultra_v2.py} (100%)

diff --git a/py/color_correct_brightness&contrast.py b/py/color_correct_brightness_and_contrast.py
similarity index 97%
rename from py/color_correct_brightness&contrast.py
rename to py/color_correct_brightness_and_contrast.py
index 1ead640..7a1bbb9 100644
--- a/py/color_correct_brightness&contrast.py
+++ b/py/color_correct_brightness_and_contrast.py
@@ -1,60 +1,60 @@
-from .imagefunc import *
-
-NODE_NAME = 'Brightness & Contrast'
-
-class ColorCorrectBrightnessAndContrast:
-
-    def __init__(self):
-        pass
-
-    @classmethod
-    def INPUT_TYPES(self):
-
-        return {
-            "required": {
-                "image": ("IMAGE", ),  #
-                "brightness": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
-                "contrast": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
-                "saturation": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
-            },
-            "optional": {
-            }
-        }
-
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("image",)
-    FUNCTION = 'color_correct_brightness_and_contrast'
-    CATEGORY = '😺dzNodes/LayerColor'
-
-    def color_correct_brightness_and_contrast(self, image, brightness, contrast, saturation):
-
-        ret_images = []
-
-        for i in image:
-            i = torch.unsqueeze(i,0)
-            __image = tensor2pil(i)
-            ret_image = __image.convert('RGB')
-            if brightness != 1:
-                brightness_image = ImageEnhance.Brightness(ret_image)
-                ret_image = brightness_image.enhance(factor=brightness)
-            if contrast != 1:
-                contrast_image = ImageEnhance.Contrast(ret_image)
-                ret_image = contrast_image.enhance(factor=contrast)
-            if saturation != 1:
-                color_image = ImageEnhance.Color(ret_image)
-                ret_image = color_image.enhance(factor=saturation)
-
-            if __image.mode == 'RGBA':
-                ret_image = RGB2RGBA(ret_image, __image.split()[-1])
-            ret_images.append(pil2tensor(ret_image))
-
-        log(f"{NODE_NAME} Processed {len(ret_images)} image(s).", message_type='finish')
-        return (torch.cat(ret_images, dim=0),)
-
-NODE_CLASS_MAPPINGS = {
-    "LayerColor: Brightness & Contrast": ColorCorrectBrightnessAndContrast
-}
-
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "LayerColor: Brightness & Contrast": "LayerColor: Brightness & Contrast"
+from .imagefunc import *
+
+NODE_NAME = 'Brightness & Contrast'
+
+class ColorCorrectBrightnessAndContrast:
+
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(self):
+
+        return {
+            "required": {
+                "image": ("IMAGE", ),  #
+                "brightness": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
+                "contrast": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
+                "saturation": ("FLOAT", {"default": 1, "min": 0.0, "max": 3, "step": 0.01}),
+            },
+            "optional": {
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("image",)
+    FUNCTION = 'color_correct_brightness_and_contrast'
+    CATEGORY = '😺dzNodes/LayerColor'
+
+    def color_correct_brightness_and_contrast(self, image, brightness, contrast, saturation):
+
+        ret_images = []
+
+        for i in image:
+            i = torch.unsqueeze(i,0)
+            __image = tensor2pil(i)
+            ret_image = __image.convert('RGB')
+            if brightness != 1:
+                brightness_image = ImageEnhance.Brightness(ret_image)
+                ret_image = brightness_image.enhance(factor=brightness)
+            if contrast != 1:
+                contrast_image = ImageEnhance.Contrast(ret_image)
+                ret_image = contrast_image.enhance(factor=contrast)
+            if saturation != 1:
+                color_image = ImageEnhance.Color(ret_image)
+                ret_image = color_image.enhance(factor=saturation)
+
+            if __image.mode == 'RGBA':
+                ret_image = RGB2RGBA(ret_image, __image.split()[-1])
+            ret_images.append(pil2tensor(ret_image))
+
+        log(f"{NODE_NAME} Processed {len(ret_images)} image(s).", message_type='finish')
+        return (torch.cat(ret_images, dim=0),)
+
+NODE_CLASS_MAPPINGS = {
+    "LayerColor: Brightness & Contrast": ColorCorrectBrightnessAndContrast
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "LayerColor: Brightness & Contrast": "LayerColor: Brightness & Contrast"
 }
\ No newline at end of file
diff --git a/py/color_correct_shadow&highlight.py b/py/color_correct_shadow_and_highlight.py
similarity index 100%
rename from py/color_correct_shadow&highlight.py
rename to py/color_correct_shadow_and_highlight.py
diff --git a/py/rembg_ultra.py b/py/rmbg_ultra.py
similarity index 100%
rename from py/rembg_ultra.py
rename to py/rmbg_ultra.py
diff --git a/py/rembg_ultra_v2.py b/py/rmbg_ultra_v2.py
similarity index 100%
rename from py/rembg_ultra_v2.py
rename to py/rmbg_ultra_v2.py