feat: new geo node tree flags

Author: BrendanParmer

geometry/node_settings.py

@@ -43,16 +43,16 @@
     'NodeGroupInput' : [],
 
     # Input > Scene
+    'GeometryNodeTool3DCursor'   : [],
+
     'GeometryNodeCollectionInfo' : [("transform_space", ST.ENUM)],
 
     'GeometryNodeImageInfo'      : [],
-
     'GeometryNodeIsViewport'     : [],
 
     'GeometryNodeObjectInfo'     : [("transform_space", ST.ENUM)],
 
     'GeometryNodeSelfObject'     : [],
-
     'GeometryNodeInputSceneTime' : [],
 
 
@@ -64,21 +64,18 @@
 
     # GEOMETRY
     'GeometryNodeJoinGeometry'       : [],
-
     'GeometryNodeGeometryToInstance' : [],
 
     # Geometry > Read
     'GeometryNodeInputID'             : [],
-
     'GeometryNodeInputIndex'          : [],
 
     'GeometryNodeInputNamedAttribute' : [("data_type", ST.ENUM)],
 
     'GeometryNodeInputNormal'         : [],
-
     'GeometryNodeInputPosition'       : [],
-
     'GeometryNodeInputRadius'         : [],
+    'GeometryNodeToolSelection'       : [],
 
     # Geometry > Sample
     'GeometryNodeProximity'      : [("target_element", ST.ENUM)],
@@ -95,13 +92,12 @@
     'GeometryNodeSampleNearest'  : [("domain", ST.ENUM)],
 
     # Geometry > Write
-    'GeometryNodeSetID'       : [],
-
-    'GeometryNodeSetPosition' : [],
+    'GeometryNodeSetID'            : [],
+    'GeometryNodeSetPosition'      : [],
+    'GeometryNodeToolSetSelection' : [],
 
     # Geometry > Operations
     'GeometryNodeBoundBox'           : [],
-
     'GeometryNodeConvexHull'         : [],
 
     'GeometryNodeDeleteGeometry'     : [("domain", ST.ENUM),
@@ -112,7 +108,6 @@
     'GeometryNodeMergeByDistance'    : [("mode",   ST.ENUM)],
 
     'GeometryNodeTransform'          : [],
-
     'GeometryNodeSeparateComponents' : [],
 
     'GeometryNodeSeparateGeometry'   : [("domain", ST.ENUM)],
@@ -121,24 +116,17 @@
     # CURVE
     # Curve > Read
     'GeometryNodeInputCurveHandlePositions' : [],
-
     'GeometryNodeCurveLength'               : [],
-
     'GeometryNodeInputTangent'              : [],
-
     'GeometryNodeInputCurveTilt'            : [],
-
     'GeometryNodeCurveEndpointSelection'    : [],
 
     'GeometryNodeCurveHandleTypeSelection'  : [("handle_type", ST.ENUM),
                                                ("mode",        ST.ENUM_SET)],
 
     'GeometryNodeInputSplineCyclic'         : [],
-
     'GeometryNodeSplineLength'              : [],
-
     'GeometryNodeSplineParameter'           : [],
-
     'GeometryNodeInputSplineResolution'     : [],
 
     # Curve > Sample
@@ -150,7 +138,6 @@
     'GeometryNodeSetCurveNormal'          : [("mode", ST.ENUM)],
 
     'GeometryNodeSetCurveRadius'          : [],
-
     'GeometryNodeSetCurveTilt'            : [],
 
     'GeometryNodeSetCurveHandlePositions' : [("mode", ST.ENUM)],
@@ -159,7 +146,6 @@
                                              ("mode",        ST.ENUM_SET)],
 
     'GeometryNodeSetSplineCyclic'         : [],
-
     'GeometryNodeSetSplineResolution'     : [],
 
     'GeometryNodeCurveSplineType'         : [("spline_type", ST.ENUM)],
@@ -180,7 +166,6 @@
     'GeometryNodeResampleCurve'         : [("mode", ST.ENUM)],
 
     'GeometryNodeReverseCurve'          : [],
-
     'GeometryNodeSubdivideCurve'        : [],
 
     'GeometryNodeTrimCurve'             : [("mode", ST.ENUM)],
@@ -195,7 +180,6 @@
     'GeometryNodeCurvePrimitiveLine'          : [("mode", ST.ENUM)],
 
     'GeometryNodeCurveSpiral'                 : [],
-
     'GeometryNodeCurveQuadraticBezier'        : [],
 
     'GeometryNodeCurvePrimitiveQuadrilateral' : [("mode", ST.ENUM)],
@@ -204,54 +188,38 @@
 
     # Curve > Topology
     'GeometryNodeOffsetPointInCurve' : [],
-
     'GeometryNodeCurveOfPoint'       : [],
-
     'GeometryNodePointsOfCurve'      : [],
 
 
     # INSTANCES
     'GeometryNodeInstanceOnPoints'      : [],
-
     'GeometryNodeInstancesToPoints'     : [],
 
     'GeometryNodeRealizeInstances'      : [("legacy_behavior", ST.BOOL)],
 
     'GeometryNodeRotateInstances'       : [],
-
     'GeometryNodeScaleInstances'        : [],
-
     'GeometryNodeTranslateInstances'    : [],
-
     'GeometryNodeInputInstanceRotation' : [],
-
     'GeometryNodeInputInstanceScale'    : [],
 
 
     # MESH
     # Mesh > Read
     'GeometryNodeInputMeshEdgeAngle'       : [],
-
     'GeometryNodeInputMeshEdgeNeighbors'   : [],
-    
     'GeometryNodeInputMeshEdgeVertices'    : [],
-
     'GeometryNodeEdgesToFaceGroups'        : [],
-
     'GeometryNodeInputMeshFaceArea'        : [],
-
     'GeometryNodeInputMeshFaceNeighbors'   : [],
-
+    'GeometryNodeToolFaceSet'              : [],
     'GeometryNodeMeshFaceSetBoundaries'    : [],
-
     'GeometryNodeInputMeshFaceIsPlanar'    : [],
-
     'GeometryNodeInputShadeSmooth'         : [],
-
+    'GeometryNodeInputEdgeSmooth'          : [],
     'GeometryNodeInputMeshIsland'          : [],
-
     'GeometryNodeInputShortestEdgePaths'   : [],
-
     'GeometryNodeInputMeshVertexNeighbors' : [],
 
     # Mesh > Sample
@@ -260,13 +228,12 @@
     'GeometryNodeSampleUVSurface'      : [("data_type", ST.ENUM)],
 
     # Mesh > Write
+    'GeometryNodeToolSetFaceSet' : [],
     'GeometryNodeSetShadeSmooth' : [],
 
     # Mesh > Operations
     'GeometryNodeDualMesh'             : [],
-
     'GeometryNodeEdgePathsToCurves'    : [],
-
     'GeometryNodeEdgePathsToSelection' : [],
 
     'GeometryNodeExtrudeMesh'          : [("mode", ST.ENUM)],
@@ -285,7 +252,6 @@
                                           ("scale_mode", ST.ENUM)],
 
     'GeometryNodeSplitEdges'           : [],
-
     'GeometryNodeSubdivideMesh'        : [],
 
     'GeometryNodeSubdivisionSurface'   : [("boundary_smooth", ST.ENUM),
@@ -302,7 +268,6 @@
     'GeometryNodeMeshCylinder'  : [("fill_type", ST.ENUM)],
 
     'GeometryNodeMeshGrid'      : [],
-
     'GeometryNodeMeshIcoSphere' : [],
 
     'GeometryNodeMeshCircle'    : [("fill_type", ST.ENUM)],
@@ -314,17 +279,11 @@
 
     # Mesh > Topology
     'GeometryNodeCornersOfFace'      : [],
-
     'GeometryNodeCornersOfVertex'    : [],
-
     'GeometryNodeEdgesOfCorner'      : [],
-
     'GeometryNodeEdgesOfVertex'      : [],
-
     'GeometryNodeFaceOfCorner'       : [],
-
     'GeometryNodeOffsetCornerInFace' : [],
-
     'GeometryNodeVertexOfCorner'     : [],
 
     # Mesh > UV
@@ -340,7 +299,7 @@
                                               ("use_legacy_normal", ST.BOOL)],
 
     'GeometryNodePoints'                   : [],
-
+    'GeometryNodePointsToCurves'           : [],
     'GeometryNodePointsToVertices'         : [],
 
     'GeometryNodePointsToVolume'           : [("resolution_mode", ST.ENUM)],
@@ -350,25 +309,19 @@
 
     # VOLUME
     'GeometryNodeVolumeCube'   : [],
-
     'GeometryNodeVolumeToMesh' : [("resolution_mode", ST.ENUM)],
 
 
     # SIMULATION
     'GeometryNodeSimulationInput'  : [],
-
     'GeometryNodeSimulationOutput' : [],
 
 
     # MATERIAL
     'GeometryNodeReplaceMaterial'    : [],
-
     'GeometryNodeInputMaterialIndex' : [],
-
     'GeometryNodeMaterialSelection'  : [],
-
     'GeometryNodeSetMaterial'        : [],
-
     'GeometryNodeSetMaterialIndex'   : [],
 
 
@@ -413,7 +366,10 @@
 
     'FunctionNodeRandomValue' : [("data_type", ST.ENUM)],
 
-    'GeometryNodeSwitch'      : [("input_type", ST.ENUM)],
+    'GeometryNodeRepeatInput'  : [],
+    'GeometryNodeRepeatOutput' : [],
+
+    'GeometryNodeSwitch' : [("input_type", ST.ENUM)],
 
     # Utilities > Color
     'ShaderNodeValToRGB'        : [("color_ramp", ST.COLOR_RAMP)],
@@ -430,11 +386,8 @@
 
     # Utilities > Text
     'GeometryNodeStringJoin'             : [],
-
     'FunctionNodeReplaceString'          : [],
-
     'FunctionNodeSliceString'            : [],
-
     'FunctionNodeStringLength'           : [],
 
     'GeometryNodeStringToCurves'         : [("align_x",    ST.ENUM),
@@ -444,7 +397,6 @@
                                             ("pivot_mode", ST.ENUM)],
 
     'FunctionNodeValueToString'          : [],
-
     'FunctionNodeInputSpecialCharacters' : [],
 
     # Utilities > Vector
@@ -456,7 +408,6 @@
                                 ("rotation_type", ST.ENUM)],
 
     'ShaderNodeCombineXYZ'   : [],
-
     'ShaderNodeSeparateXYZ'  : [],
 
     # Utilities > Field
@@ -490,11 +441,21 @@
                                  ("use_clamp", ST.BOOL)],
 
     # Utilities > Rotation
-    'FunctionNodeAlignEulerToVector' : [("axis",       ST.ENUM),
-                                        ("pivot_axis", ST.ENUM)],
+    'FunctionNodeAlignEulerToVector'    : [("axis",       ST.ENUM),
+                                           ("pivot_axis", ST.ENUM)],
+
+    'FunctionNodeAxisAngleToRotation'   : [],
+    'FunctionNodeEulerToRotation'       : [],
+    'FunctionNodeInvertRotation'        : [],
 
-    'FunctionNodeRotateEuler'        : [("space", ST.ENUM),
-                                        ("type",  ST.ENUM)],
+    'FunctionNodeRotateEuler'           : [("space", ST.ENUM),
+                                           ("type",  ST.ENUM)],
+
+    'FunctionNodeRotateVector'          : [],
+    'FunctionNodeRotationToAxisAngle'   : [],
+    'FunctionNodeRotationToEuler'       : [],
+    'FunctionNodeRotationToQuaternion'  : [],
+    'FunctionNodeQuaternionToRotation'  : [],
 
     # MISC
     'NodeFrame' : [],

geometry/operator.py

@@ -119,9 +119,17 @@ def _process_node_tree(self, node_tree: GeometryNodeTree,
         self._write("\n")
 
         if bpy.app.version >= (4, 0, 0):
-            self._write(f"{inner}{nt_var}.is_modifier = {node_tree.is_modifier}\n")
-            self._write(f"{inner}{nt_var}.is_tool = {node_tree.is_tool}\n")
-
+            geo_node_tree_flags =  ["is_mode_edit", 
+                                    "is_mode_sculpt",
+                                    "is_modifier",
+                                    "is_tool",
+                                    "is_type_curve",
+                                    "is_type_mesh",
+                                    "is_type_point_cloud"]
+            for flag in geo_node_tree_flags:
+                self._write(f"{inner}{nt_var}.{flag} = {getattr(node_tree, flag)}\n")
+            self._write("\n")
+    
         #initialize nodes
         self._write(f"{inner}#initialize {nt_var} nodes\n")