feat: clean up zone processing

Author: BrendanParmer

NodeToPython/geometry/node_tree.py

@@ -1,5 +1,5 @@
 import bpy
-from bpy.types import GeometryNodeTree
+from bpy.types import GeometryNodeTree, GeometryNode
 
 if bpy.app.version >= (3, 6, 0):
     from bpy.types import GeometryNodeSimulationInput
@@ -15,9 +15,10 @@
 class NTP_GeoNodeTree(NTP_NodeTree):
     def __init__(self, node_tree: GeometryNodeTree, var: str):
         super().__init__(node_tree, var)
+        self.zone_inputs_: dict[list[GeometryNode]] = {}
         if bpy.app.version >= (3, 6, 0):
-            self.sim_inputs: list[GeometryNodeSimulationInput] = []
+            self.zone_inputs_["GeometryNodeSimulationInput"] = []
         if bpy.app.version >= (4, 0, 0):
-            self.repeat_inputs: list[GeometryNodeRepeatInput] = []
+            self.zone_inputs_["GeometryNodeRepeatInput"] = []
         if bpy.app.version >= (4, 3, 0):
-            self.foreach_element_inputs: list[GeometryNodeForeachGeometryElementInput] = []
+            self.zone_inputs_["GeometryNodeForeachGeometryElementInput"] = []

NodeToPython/geometry/operator.py

@@ -49,12 +49,8 @@ def _process_node(self, node: Node, ntp_nt: NTP_GeoNodeTree) -> None:
                 self._group_io_settings(node, "output", ntp_nt)
                 ntp_nt.outputs_set = True
 
-        if node.bl_idname == 'GeometryNodeSimulationInput':
-            ntp_nt.sim_inputs.append(node)
-        elif node.bl_idname == 'GeometryNodeRepeatInput':
-            ntp_nt.repeat_inputs.append(node)
-        elif node.bl_idname == 'GeometryNodeForeachGeometryElementInput':
-            ntp_nt.foreach_element_inputs.append(node)
+        if node.bl_idname in ntp_nt.zone_inputs_:
+            ntp_nt.zone_inputs_[node.bl_idname].append(node)
 
         self._hide_hidden_sockets(node)
 
@@ -141,12 +137,8 @@ def _process_node_tree(self, node_tree: GeometryNodeTree) -> None:
         for node in node_tree.nodes:
             self._process_node(node, ntp_nt)
 
-        if bpy.app.version >= (3, 6, 0):
-            self._process_zones(ntp_nt.sim_inputs)
-        if bpy.app.version >= (4, 0, 0):
-            self._process_zones(ntp_nt.repeat_inputs)
-        if bpy.app.version >= (4, 3, 0):
-            self._process_zones(ntp_nt.foreach_element_inputs)
+        for zone_list in ntp_nt.zone_inputs_.values():
+            self._process_zones(zone_list)
 
         #set look of nodes
         self._set_parents(node_tree)