nodebpy 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

nodebpy/nodes/__init__.py

@@ -18,6 +18,9 @@ from .manual import (
     SimulationInput,
     SimulationOutput,
     SimulationZone,
+    ForEachGeometryElementInput,
+    ForEachGeometryElementOutput,
+    ForEachGeometryElementZone,
     FormatString,
     Value,
     AccumulateField,
@@ -414,6 +417,9 @@ __all__ = (
     "FlipFaces",
     "FloatCurve",
     "FloatToInteger",
+    "ForEachGeometryElementInput",
+    "ForEachGeometryElementOutput",
+    "ForEachGeometryElementZone",
     "FormatString",
     "GaborTexture",
     "Gamma",

nodebpy/nodes/attribute.py

@@ -11,13 +11,16 @@ from ..types import (
     TYPE_INPUT_STRING,
     TYPE_INPUT_ROTATION,
     TYPE_INPUT_COLOR,
+    TYPE_INPUT_MATRIX,
     TYPE_INPUT_VALUE,
     TYPE_INPUT_VECTOR,
 )
 
 
 class BlurAttribute(NodeBuilder):
-    """Mix attribute values of neighboring elements"""
+    """
+    Mix attribute values of neighboring elements
+    """
 
     _bl_idname = "GeometryNodeBlurAttribute"
     node: bpy.types.GeometryNodeBlurAttribute
@@ -109,7 +112,9 @@ class BlurAttribute(NodeBuilder):
 
 
 class DomainSize(NodeBuilder):
-    """Retrieve the number of elements in a geometry for each attribute domain"""
+    """
+    Retrieve the number of elements in a geometry for each attribute domain
+    """
 
     _bl_idname = "GeometryNodeAttributeDomainSize"
     node: bpy.types.GeometryNodeAttributeDomainSize
@@ -181,7 +186,9 @@ class DomainSize(NodeBuilder):
 
 
 class RemoveNamedAttribute(NodeBuilder):
-    """Delete an attribute with a specified name from a geometry. Typically used to optimize performance"""
+    """
+    Delete an attribute with a specified name from a geometry. Typically used to optimize performance
+    """
 
     _bl_idname = "GeometryNodeRemoveAttribute"
     node: bpy.types.GeometryNodeRemoveAttribute
@@ -219,7 +226,9 @@ class RemoveNamedAttribute(NodeBuilder):
 
 
 class StoreNamedAttribute(NodeBuilder):
-    """Store the result of a field on a geometry as an attribute with the specified name"""
+    """
+    Store the result of a field on a geometry as an attribute with the specified name
+    """
 
     _bl_idname = "GeometryNodeStoreNamedAttribute"
     node: bpy.types.GeometryNodeStoreNamedAttribute
@@ -360,6 +369,74 @@ class StoreNamedAttribute(NodeBuilder):
             value=value,
         )
 
+    @classmethod
+    def matrix(
+        cls,
+        geometry: TYPE_INPUT_GEOMETRY = None,
+        selection: TYPE_INPUT_BOOLEAN = True,
+        name: TYPE_INPUT_STRING = "",
+        value: TYPE_INPUT_MATRIX = None,
+    ) -> "StoreNamedAttribute":
+        """Create Store Named Attribute with operation '4x4 Matrix'."""
+        return cls(
+            data_type="FLOAT4X4",
+            geometry=geometry,
+            selection=selection,
+            name=name,
+            value=value,
+        )
+
+    @classmethod
+    def int8(
+        cls,
+        geometry: TYPE_INPUT_GEOMETRY = None,
+        selection: TYPE_INPUT_BOOLEAN = True,
+        name: TYPE_INPUT_STRING = "",
+        value: TYPE_INPUT_INT = 0,
+    ) -> "StoreNamedAttribute":
+        """Create Store Named Attribute with operation '8-Bit Integer'."""
+        return cls(
+            data_type="INT8",
+            geometry=geometry,
+            selection=selection,
+            name=name,
+            value=value,
+        )
+
+    @classmethod
+    def vector2(
+        cls,
+        geometry: TYPE_INPUT_GEOMETRY = None,
+        selection: TYPE_INPUT_BOOLEAN = True,
+        name: TYPE_INPUT_STRING = "",
+        value: TYPE_INPUT_VECTOR = None,
+    ) -> "StoreNamedAttribute":
+        """Create Store Named Attribute with operation '2D Vector'."""
+        return cls(
+            data_type="FLOAT2",
+            geometry=geometry,
+            selection=selection,
+            name=name,
+            value=value,
+        )
+
+    @classmethod
+    def byte_color(
+        cls,
+        geometry: TYPE_INPUT_GEOMETRY = None,
+        selection: TYPE_INPUT_BOOLEAN = True,
+        name: TYPE_INPUT_STRING = "",
+        value: TYPE_INPUT_COLOR = None,
+    ) -> "StoreNamedAttribute":
+        """Create Store Named Attribute with operation 'Byte Color'."""
+        return cls(
+            data_type="BYTE_COLOR",
+            geometry=geometry,
+            selection=selection,
+            name=name,
+            value=value,
+        )
+
     @classmethod
     def point(
         cls,
@@ -411,6 +488,23 @@ class StoreNamedAttribute(NodeBuilder):
             value=value,
         )
 
+    @classmethod
+    def face_corner(
+        cls,
+        geometry: TYPE_INPUT_GEOMETRY = None,
+        selection: TYPE_INPUT_BOOLEAN = True,
+        name: TYPE_INPUT_STRING = "",
+        value: TYPE_INPUT_COLOR = None,
+    ) -> "StoreNamedAttribute":
+        """Create Store Named Attribute with operation 'Face Corner'."""
+        return cls(
+            domain="CORNER",
+            geometry=geometry,
+            selection=selection,
+            name=name,
+            value=value,
+        )
+
     @classmethod
     def spline(
         cls,

nodebpy/nodes/color.py

@@ -1,4 +1,3 @@
-
 import bpy
 
 from ..builder import NodeBuilder, SocketLinker
@@ -9,7 +8,9 @@ from ..types import (
 
 
 class Gamma(NodeBuilder):
-    """Apply a gamma correction"""
+    """
+    Apply a gamma correction
+    """
 
     _bl_idname = "ShaderNodeGamma"
     node: bpy.types.ShaderNodeGamma
@@ -41,7 +42,9 @@ class Gamma(NodeBuilder):
 
 
 class RgbCurves(NodeBuilder):
-    """Apply color corrections for each color channel"""
+    """
+    Apply color corrections for each color channel
+    """
 
     _bl_idname = "ShaderNodeRGBCurve"
     node: bpy.types.ShaderNodeRGBCurve