feat: foreach element zones

Author: BrendanParmer

NodeToPython/geometry/node_tree.py

@@ -4,9 +4,12 @@
 if bpy.app.version >= (3, 6, 0):
     from bpy.types import GeometryNodeSimulationInput
 
-if bpy.app.version > (4, 0, 0):
+if bpy.app.version >= (4, 0, 0):
     from bpy.types import GeometryNodeRepeatInput
 
+if bpy.app.version >= (4, 3, 0):
+    from bpy.types import GeometryNodeForeachGeometryElementInput
+
 from ..ntp_node_tree import NTP_NodeTree
 
 class NTP_GeoNodeTree(NTP_NodeTree):
@@ -16,3 +19,5 @@ def __init__(self, node_tree: GeometryNodeTree, var: str):
             self.sim_inputs: list[GeometryNodeSimulationInput] = []
         if bpy.app.version >= (4, 0, 0):
             self.repeat_inputs: list[GeometryNodeRepeatInput] = []
+        if bpy.app.version >= (4, 3, 0):
+            self.foreach_element_inputs: list[GeometryNodeForeachGeometryElementInput] = []

NodeToPython/geometry/operator.py

@@ -51,14 +51,16 @@ def _process_node(self, node: Node, ntp_nt: NTP_GeoNodeTree) -> None:
 
         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)
 
         self._hide_hidden_sockets(node)
 
         if node.bl_idname not in {'GeometryNodeSimulationInput', 
-                                  'GeometryNodeRepeatInput'}:
+                                  'GeometryNodeRepeatInput',
+                                  'GeometryNodeForeachGeometryElementInput'}:
             self._set_socket_defaults(node)
 
     if bpy.app.version >= (3, 6, 0):
@@ -143,6 +145,8 @@ def _process_node_tree(self, node_tree: GeometryNodeTree) -> None:
             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)
 
         #set look of nodes
         self._set_parents(node_tree)

NodeToPython/node_settings.py

@@ -9,6 +9,7 @@ class ST(Enum):
 	COLOR = auto()
 	ENUM = auto()
 	ENUM_SET = auto()
+	EULER = auto()
 	FLOAT = auto()
 	INT = auto()
 	STRING = auto()
@@ -17,26 +18,28 @@ class ST(Enum):
 	VEC3 = auto()
 	VEC4 = auto()
 	BAKE_ITEMS = auto()
+	CAPTURE_ATTRIBUTE_ITEMS = auto()
 	COLOR_RAMP = auto()
 	CURVE_MAPPING = auto()
 	ENUM_DEFINITION = auto()
+	ENUM_ITEM = auto()
+	FOREACH_GEO_ELEMENT_GENERATION_ITEMS = auto()
+	FOREACH_GEO_ELEMENT_INPUT_ITEMS = auto()
+	FOREACH_GEO_ELEMENT_MAIN_ITEMS = auto()
 	INDEX_SWITCH_ITEMS = auto()
+	MENU_SWITCH_ITEMS = auto()
 	NODE_TREE = auto()
 	REPEAT_OUTPUT_ITEMS = auto()
 	SIM_OUTPUT_ITEMS = auto()
 	IMAGE = auto()
 	IMAGE_USER = auto()
-	CAPTURE_ATTRIBUTE_ITEMS = auto()
 	CRYPTOMATTE_ENTRIES = auto()
-	ENUM_ITEM = auto()
-	EULER = auto()
 	FILE_SLOTS = auto()
 	FONT = auto()
 	IMAGE_FORMAT_SETTINGS = auto()
 	LAYER_SLOTS = auto()
 	MASK = auto()
 	MATERIAL = auto()
-	MENU_SWITCH_ITEMS = auto()
 	MOVIE_CLIP = auto()
 	OBJECT = auto()
 	PARTICLE_SYSTEM = auto()
@@ -1467,7 +1470,10 @@ class NodeInfo(NamedTuple):
 			NTPNodeSetting("active_input_index", ST.INT),
 			NTPNodeSetting("active_main_index", ST.INT),
 			NTPNodeSetting("domain", ST.ENUM),
+			NTPNodeSetting("generation_items", ST.FOREACH_GEO_ELEMENT_GENERATION_ITEMS),
+			NTPNodeSetting("input_items", ST.FOREACH_GEO_ELEMENT_INPUT_ITEMS),
 			NTPNodeSetting("inspection_index", ST.INT),
+			NTPNodeSetting("main_items", ST.FOREACH_GEO_ELEMENT_MAIN_ITEMS),
 		],
 		min_version_ = (4, 3, 0)
 	),

NodeToPython/ntp_operator.py

@@ -446,6 +446,12 @@ def _set_settings_defaults(self, node: Node) -> None:
                 self._capture_attribute_items(attr, f"{node_var}.{attr_name}")
             elif st == ST.MENU_SWITCH_ITEMS:
                 self._menu_switch_items(attr, f"{node_var}.{attr_name}")
+            elif st == ST.FOREACH_GEO_ELEMENT_GENERATION_ITEMS:
+                self._foreach_geo_element_generation_items(attr, f"{node_var}.{attr_name}")
+            elif st == ST.FOREACH_GEO_ELEMENT_INPUT_ITEMS:
+                self._foreach_geo_element_input_items(attr, f"{node_var}.{attr_name}")
+            elif st == ST.FOREACH_GEO_ELEMENT_MAIN_ITEMS:
+                self._foreach_geo_element_main_items(attr, f"{node_var}.{attr_name}")
 
     if bpy.app.version < (4, 0, 0):
         def _set_group_socket_defaults(self, socket_interface: NodeSocketInterface,
@@ -1277,6 +1283,43 @@ def _menu_switch_items(self, menu_switch_items: bpy.types.NodeMenuSwitchItems, m
                 desc_str = str_to_py_str(item.description)
                 self._write(f"{menu_switch_items_str}[{i}].description = {desc_str}")
 
+    if bpy.app.version >= (4, 3, 0):
+        def _foreach_geo_element_generation_items(self,
+            generation_items: bpy.types.NodeGeometryForeachGeometryElementGenerationItems,
+            generation_items_str: str
+        ) -> None:
+            self._write(f"{generation_items_str}.clear()")
+            for i, item in enumerate(generation_items):
+                socket_type = enum_to_py_str(item.socket_type)
+                name_str = str_to_py_str(item.name)
+                self._write(f"{generation_items_str}.new({socket_type}, {name_str})")
+                
+                item_str = f"{generation_items_str}[{i}]"
+                
+                ad = enum_to_py_str(item.domain)
+                self._write(f"{item_str}.domain = {ad}")
+
+        def _foreach_geo_element_input_items(self,
+            input_items: bpy.types.NodeGeometryForeachGeometryElementInputItems,
+            input_items_str: str
+        ) -> None:
+            self._write(f"{input_items_str}.clear()")
+            for i, item in enumerate(input_items):
+                socket_type = enum_to_py_str(item.socket_type)
+                name_str = str_to_py_str(item.name)
+                self._write(f"{input_items_str}.new({socket_type}, {name_str})")
+
+        def _foreach_geo_element_main_items(self,
+            main_items: bpy.types.NodeGeometryForeachGeometryElementMainItems,
+            main_items_str: str
+        ) -> None:
+            self._write(f"{main_items_str}.clear()")
+            for i, item in enumerate(main_items):
+                socket_type = enum_to_py_str(item.socket_type)
+                name_str = str_to_py_str(item.name)
+                self._write(f"{main_items_str}.new({socket_type}, {name_str})")
+
+
     def _set_parents(self, node_tree: NodeTree) -> None:
         """
         Sets parents for all nodes, mostly used to put nodes in frames