refactor socket types

backend/nodes/angle_measure.py

@@ -85,7 +85,10 @@ class AngleMeasure:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "input": ("ANNOTATION_SOURCE", {"label": "Input"}),
+                "input": ("ANNOTATION_SOURCE", {
+                    "label": "Input",
+                    "accepted_types": ["DATA_FIELD", "IMAGE"],
+                }),
                 "color": ("STRING", {"default": ANGLE_DEFAULT_COLOR, "color_picker": True}),
                 "stroke_width": ("FLOAT", {
                     "default": 1.35,

backend/nodes/annotations.py

@@ -23,7 +23,10 @@ class Annotations:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "input": ("ANNOTATION_SOURCE", {"label": "Input"}),
+                "input": ("ANNOTATION_SOURCE", {
+                    "label": "Input",
+                    "accepted_types": ["DATA_FIELD", "IMAGE"],
+                }),
                 "colormap": (["auto"] + list(COLORMAPS), {"hide_when_input_connected": "colormap_map"}),
                 "show_scale_bar": ("BOOLEAN", {"default": True}),
                 "show_color_map": ("BOOLEAN", {"default": True}),

backend/nodes/cursors.py

@@ -13,7 +13,10 @@ class Cursors:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "line": ("CURSOR_SOURCE", {"label": "input"}),
+                "line": ("LINE", {
+                    "label": "input",
+                    "accepted_types": ["DATA_FIELD"],
+                }),
                 "x1": ("FLOAT", {"default": 0.25, "min": 0.0, "max": 1.0, "step": 0.01, "hidden": True}),
                 "y1": ("FLOAT", {"default": 0.5,  "min": 0.0, "max": 1.0, "step": 0.01, "hidden": True}),
                 "x2": ("FLOAT", {"default": 0.75, "min": 0.0, "max": 1.0, "step": 0.01, "hidden": True}),

backend/nodes/markup.py

@@ -21,7 +21,10 @@ class Markup:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "input": ("ANNOTATION_SOURCE", {"label": "Input"}),
+                "input": ("ANNOTATION_SOURCE", {
+                    "label": "Input",
+                    "accepted_types": ["DATA_FIELD", "IMAGE"],
+                }),
                 "shape": (["line", "rectangle", "circle", "arrow"], {"default": "arrow"}),
                 "stroke_color": ("STRING", {"default": "#ff0000", "color_picker": True}),
                 "stroke_width": ("INT", {"default": 3, "min": 1, "max": 64, "step": 1}),

backend/nodes/preview_image.py

@@ -22,7 +22,10 @@ class PreviewImage:
                 "colormap": (["auto"] + list(COLORMAPS), {"hide_when_input_connected": "colormap_map"}),
             },
             "optional": {
-                "input": ("ANNOTATION_SOURCE", {"label": "Input"}),
+                "input": ("ANNOTATION_SOURCE", {
+                    "label": "Input",
+                    "accepted_types": ["DATA_FIELD", "IMAGE"],
+                }),
                 "colormap_map": ("COLORMAP", {"label": "colormap"}),
             }
         }

backend/nodes/print_table.py

@@ -9,7 +9,9 @@ class PrintTable:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "table": ("ANY_TABLE",),
+                "table": ("MEASURE_TABLE", {
+                    "accepted_types": ["RECORD_TABLE"],
+                }),
             }
         }
 

backend/nodes/save.py

@@ -29,7 +29,18 @@ class Save:
                     "hide_when_input_connected": "directory",
                     "top_socket_input": "directory",
                 }),
-                "value": ("SAVE_VALUE", {"label": "value"}),
+                "value": ("DATA_FIELD", {
+                    "label": "value",
+                    "accepted_types": [
+                        "IMAGE",
+                        "ANNOTATION_SOURCE",
+                        "LINE",
+                        "MEASURE_TABLE",
+                        "RECORD_TABLE",
+                        "MESH_MODEL",
+                        "FLOAT",
+                    ],
+                }),
                 "format": ("STRING", {
                     "default": "TIFF",
                     "choices_by_source_type": {

backend/nodes/save_image.py

@@ -17,7 +17,10 @@ class SaveImage:
             "directory": ("DIRECTORY", {"label": "directory"}),
         }
         for i in range(_MAX_SAVE_FIELDS):
-            optional[f"field_{i}"] = ("SAVE_LAYER", {"label": f"layer {i + 1}"})
+            optional[f"field_{i}"] = ("DATA_FIELD", {
+                "label": f"layer {i + 1}",
+                "accepted_types": ["IMAGE", "ANNOTATION_SOURCE"],
+            })
             optional[f"layer_name_{i}"] = ("STRING", {
                 "default": "",
                 "placeholder": "name",

backend/nodes/stats.py

@@ -26,7 +26,9 @@ class Stats:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "input": ("STATS_SOURCE",),
+                "input": ("DATA_FIELD", {
+                    "accepted_types": ["IMAGE", "LINE", "RECORD_TABLE"],
+                }),
                 "column": ("STRING", {
                     "default": "value",
                     "choices_from_table_input": "input",

backend/nodes/value_display.py

@@ -11,7 +11,9 @@ class ValueDisplay:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "value": ("VALUE_SOURCE",),
+                "value": ("FLOAT", {
+                    "accepted_types": ["MEASURE_TABLE"],
+                }),
                 "measurement": ("STRING", {
                     "default": "",
                     "choices_from_measure_input": "value",