From 6dcfa314d6763aa4dc1c47f576fd76de48e141a9 Mon Sep 17 00:00:00 2001
From: Stefan Koehler <st.d.koehler@gmx.de>
Date: Sat, 10 May 2025 00:04:23 +0200
Subject: [PATCH] adapt naming so that we are shown under "Color Transfer" in
 the Node Library

---
 __init__.py       |  4 ++--
 color_transfer.py | 18 +++++++++---------
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/__init__.py b/__init__.py
index b102236..e2d4076 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ColorTransferReinhard, ColorPaletteNode
+from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ReferenceTransferReinhard, ColorPaletteNode
 
 
 NODE_CLASS_MAPPINGS = {
@@ -8,7 +8,7 @@ NODE_CLASS_MAPPINGS = {
     "PalletteSoftTransfer": PaletteSoftTransfer,
     "PaletteRbfTransfer": PaletteRbfTransfer,
     "PaletteOptimalTransportTransfer": PaletteOptimalTransportTransfer,
-    "ColorTransferReinhard": ColorTransferReinhard,
+    "ColorTransferReinhard": ReferenceTransferReinhard,
     "ColorPalette": ColorPaletteNode,
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
diff --git a/color_transfer.py b/color_transfer.py
index 78f1426..2b44d34 100644
--- a/color_transfer.py
+++ b/color_transfer.py
@@ -252,7 +252,7 @@ def process_image_with_palette(image, target_colors, color_space, cluster_method
 
     return torch.cat(processedImages, dim=0)
 
-class ColorTransferReinhard(ComfyNodeABC):
+class ReferenceTransferReinhard(ComfyNodeABC):
     @classmethod
     def INPUT_TYPES(cls):
         data_in = {
@@ -265,7 +265,7 @@ class ColorTransferReinhard(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Reference Transfer"
 
     def color_transfer(self, image, image_reference):
 
@@ -321,7 +321,7 @@ class PaletteOptimalTransportTransfer(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_transfer(self, image, target_colors, palette_extension_method, palette_extension_points, blend_mode, blend_ratio):
 
@@ -394,7 +394,7 @@ class PaletteRbfTransfer(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_transfer(self, image, target_colors, rbf_function, epsilon):
         """
@@ -460,7 +460,7 @@ class PaletteSoftTransfer(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_transfer(self, image, target_colors, blend_mode, blend_ratio, softness):
         """
@@ -531,7 +531,7 @@ class PaletteTransferReinhard(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_transfer(self, image, target_colors):
         if len(target_colors) == 0:
@@ -596,7 +596,7 @@ class PalleteTransferClustering(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_transfer(self, image, target_colors, palette_extension_method, palette_extension_points, gaussian_blur):
         if len(target_colors) == 0:
@@ -635,7 +635,7 @@ class PaletteTransferNode(ComfyNodeABC):
 
     RETURN_TYPES = (IO.IMAGE,)
     FUNCTION = "color_transfer"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
 
     def color_transfer(self, image, target_colors, color_space, cluster_method, distance_method, gaussian_blur):
@@ -667,7 +667,7 @@ class ColorPaletteNode(ComfyNodeABC):
     RETURN_TYPES = ("COLOR_LIST", )
     RETURN_NAMES = ("Color palette", )
     FUNCTION = "color_list"
-    CATEGORY = "Palette Transfer"
+    CATEGORY = "Color Transfer/Palette Transfer"
 
     def color_list(self, color_palette):
         return (ast.literal_eval(color_palette), )