nodebpy 0.1.1__py3-none-any.whl → 0.2.1__py3-none-any.whl

nodebpy/nodes/vector.py

@@ -0,0 +1,528 @@
+from typing import Literal
+
+import bpy
+
+from ..builder import NodeBuilder, SocketLinker
+from ..types import (
+    TYPE_INPUT_VALUE,
+    TYPE_INPUT_VECTOR,
+)
+
+
+class RadialTiling(NodeBuilder):
+    """Transform Coordinate System for Radial Tiling"""
+
+    _bl_idname = "ShaderNodeRadialTiling"
+    node: bpy.types.ShaderNodeRadialTiling
+
+    def __init__(
+        self,
+        vector: TYPE_INPUT_VECTOR = None,
+        sides: TYPE_INPUT_VALUE = 5.0,
+        roundness: TYPE_INPUT_VALUE = 0.0,
+        *,
+        normalize: bool = False,
+    ):
+        super().__init__()
+        key_args = {"Vector": vector, "Sides": sides, "Roundness": roundness}
+        self.normalize = normalize
+        self._establish_links(**key_args)
+
+    @property
+    def i_vector(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector")
+
+    @property
+    def i_sides(self) -> SocketLinker:
+        """Input socket: Sides"""
+        return self._input("Sides")
+
+    @property
+    def i_roundness(self) -> SocketLinker:
+        """Input socket: Roundness"""
+        return self._input("Roundness")
+
+    @property
+    def o_segment_coordinates(self) -> SocketLinker:
+        """Output socket: Segment Coordinates"""
+        return self._output("Segment Coordinates")
+
+    @property
+    def o_segment_id(self) -> SocketLinker:
+        """Output socket: Segment ID"""
+        return self._output("Segment ID")
+
+    @property
+    def o_segment_width(self) -> SocketLinker:
+        """Output socket: Segment Width"""
+        return self._output("Segment Width")
+
+    @property
+    def o_segment_rotation(self) -> SocketLinker:
+        """Output socket: Segment Rotation"""
+        return self._output("Segment Rotation")
+
+    @property
+    def normalize(self) -> bool:
+        return self.node.normalize
+
+    @normalize.setter
+    def normalize(self, value: bool):
+        self.node.normalize = value
+
+
+class VectorCurves(NodeBuilder):
+    """Map input vector components with curves"""
+
+    _bl_idname = "ShaderNodeVectorCurve"
+    node: bpy.types.ShaderNodeVectorCurve
+
+    def __init__(
+        self,
+        fac: TYPE_INPUT_VALUE = 1.0,
+        vector: TYPE_INPUT_VECTOR = None,
+    ):
+        super().__init__()
+        key_args = {"Fac": fac, "Vector": vector}
+
+        self._establish_links(**key_args)
+
+    @property
+    def i_fac(self) -> SocketLinker:
+        """Input socket: Factor"""
+        return self._input("Fac")
+
+    @property
+    def i_vector(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector")
+
+    @property
+    def o_vector(self) -> SocketLinker:
+        """Output socket: Vector"""
+        return self._output("Vector")
+
+
+class VectorMath(NodeBuilder):
+    """Perform vector math operation"""
+
+    _bl_idname = "ShaderNodeVectorMath"
+    node: bpy.types.ShaderNodeVectorMath
+
+    def __init__(
+        self,
+        vector: TYPE_INPUT_VECTOR = None,
+        vector_001: TYPE_INPUT_VECTOR = None,
+        vector_002: TYPE_INPUT_VECTOR = None,
+        scale: TYPE_INPUT_VALUE = 1.0,
+        *,
+        operation: Literal[
+            "ADD",
+            "SUBTRACT",
+            "MULTIPLY",
+            "DIVIDE",
+            "MULTIPLY_ADD",
+            "CROSS_PRODUCT",
+            "PROJECT",
+            "REFLECT",
+            "REFRACT",
+            "FACEFORWARD",
+            "DOT_PRODUCT",
+            "DISTANCE",
+            "LENGTH",
+            "SCALE",
+            "NORMALIZE",
+            "ABSOLUTE",
+            "POWER",
+            "SIGN",
+            "MINIMUM",
+            "MAXIMUM",
+            "FLOOR",
+            "CEIL",
+            "FRACTION",
+            "MODULO",
+            "WRAP",
+            "SNAP",
+            "SINE",
+            "COSINE",
+            "TANGENT",
+        ] = "ADD",
+    ):
+        super().__init__()
+        key_args = {
+            "Vector": vector,
+            "Vector_001": vector_001,
+            "Vector_002": vector_002,
+            "Scale": scale,
+        }
+        self.operation = operation
+        self._establish_links(**key_args)
+
+    @classmethod
+    def add(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Add'."""
+        return cls(operation="ADD", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def subtract(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Subtract'."""
+        return cls(operation="SUBTRACT", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def multiply(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Multiply'."""
+        return cls(operation="MULTIPLY", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def divide(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Divide'."""
+        return cls(operation="DIVIDE", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def project(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Project'."""
+        return cls(operation="PROJECT", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def reflect(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Reflect'."""
+        return cls(operation="REFLECT", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def refract(
+        cls,
+        vector: TYPE_INPUT_VECTOR = None,
+        vector_001: TYPE_INPUT_VECTOR = None,
+        scale: TYPE_INPUT_VALUE = 1.0,
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Refract'."""
+        return cls(
+            operation="REFRACT", vector=vector, vector_001=vector_001, scale=scale
+        )
+
+    @classmethod
+    def faceforward(
+        cls,
+        vector: TYPE_INPUT_VECTOR = None,
+        vector_001: TYPE_INPUT_VECTOR = None,
+        vector_002: TYPE_INPUT_VECTOR = None,
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Faceforward'."""
+        return cls(
+            operation="FACEFORWARD",
+            vector=vector,
+            vector_001=vector_001,
+            vector_002=vector_002,
+        )
+
+    @classmethod
+    def distance(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Distance'."""
+        return cls(operation="DISTANCE", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def length(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Length'."""
+        return cls(operation="LENGTH", vector=vector)
+
+    @classmethod
+    def scale(
+        cls, vector: TYPE_INPUT_VECTOR = None, scale: TYPE_INPUT_VALUE = 1.0
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Scale'."""
+        return cls(operation="SCALE", vector=vector, scale=scale)
+
+    @classmethod
+    def normalize(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Normalize'."""
+        return cls(operation="NORMALIZE", vector=vector)
+
+    @classmethod
+    def absolute(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Absolute'."""
+        return cls(operation="ABSOLUTE", vector=vector)
+
+    @classmethod
+    def power(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Power'."""
+        return cls(operation="POWER", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def sign(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Sign'."""
+        return cls(operation="SIGN", vector=vector)
+
+    @classmethod
+    def minimum(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Minimum'."""
+        return cls(operation="MINIMUM", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def maximum(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Maximum'."""
+        return cls(operation="MAXIMUM", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def floor(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Floor'."""
+        return cls(operation="FLOOR", vector=vector)
+
+    @classmethod
+    def ceil(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Ceil'."""
+        return cls(operation="CEIL", vector=vector)
+
+    @classmethod
+    def fraction(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Fraction'."""
+        return cls(operation="FRACTION", vector=vector)
+
+    @classmethod
+    def modulo(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Modulo'."""
+        return cls(operation="MODULO", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def wrap(
+        cls,
+        vector: TYPE_INPUT_VECTOR = None,
+        vector_001: TYPE_INPUT_VECTOR = None,
+        vector_002: TYPE_INPUT_VECTOR = None,
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Wrap'."""
+        return cls(
+            operation="WRAP",
+            vector=vector,
+            vector_001=vector_001,
+            vector_002=vector_002,
+        )
+
+    @classmethod
+    def snap(
+        cls, vector: TYPE_INPUT_VECTOR = None, vector_001: TYPE_INPUT_VECTOR = None
+    ) -> "VectorMath":
+        """Create Vector Math with operation 'Snap'."""
+        return cls(operation="SNAP", vector=vector, vector_001=vector_001)
+
+    @classmethod
+    def sine(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Sine'."""
+        return cls(operation="SINE", vector=vector)
+
+    @classmethod
+    def cosine(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Cosine'."""
+        return cls(operation="COSINE", vector=vector)
+
+    @classmethod
+    def tangent(cls, vector: TYPE_INPUT_VECTOR = None) -> "VectorMath":
+        """Create Vector Math with operation 'Tangent'."""
+        return cls(operation="TANGENT", vector=vector)
+
+    @property
+    def i_vector(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector")
+
+    @property
+    def i_vector_001(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector_001")
+
+    @property
+    def i_vector_002(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector_002")
+
+    @property
+    def i_scale(self) -> SocketLinker:
+        """Input socket: Scale"""
+        return self._input("Scale")
+
+    @property
+    def o_vector(self) -> SocketLinker:
+        """Output socket: Vector"""
+        return self._output("Vector")
+
+    @property
+    def o_value(self) -> SocketLinker:
+        """Output socket: Value"""
+        return self._output("Value")
+
+    @property
+    def operation(
+        self,
+    ) -> Literal[
+        "ADD",
+        "SUBTRACT",
+        "MULTIPLY",
+        "DIVIDE",
+        "MULTIPLY_ADD",
+        "CROSS_PRODUCT",
+        "PROJECT",
+        "REFLECT",
+        "REFRACT",
+        "FACEFORWARD",
+        "DOT_PRODUCT",
+        "DISTANCE",
+        "LENGTH",
+        "SCALE",
+        "NORMALIZE",
+        "ABSOLUTE",
+        "POWER",
+        "SIGN",
+        "MINIMUM",
+        "MAXIMUM",
+        "FLOOR",
+        "CEIL",
+        "FRACTION",
+        "MODULO",
+        "WRAP",
+        "SNAP",
+        "SINE",
+        "COSINE",
+        "TANGENT",
+    ]:
+        return self.node.operation
+
+    @operation.setter
+    def operation(
+        self,
+        value: Literal[
+            "ADD",
+            "SUBTRACT",
+            "MULTIPLY",
+            "DIVIDE",
+            "MULTIPLY_ADD",
+            "CROSS_PRODUCT",
+            "PROJECT",
+            "REFLECT",
+            "REFRACT",
+            "FACEFORWARD",
+            "DOT_PRODUCT",
+            "DISTANCE",
+            "LENGTH",
+            "SCALE",
+            "NORMALIZE",
+            "ABSOLUTE",
+            "POWER",
+            "SIGN",
+            "MINIMUM",
+            "MAXIMUM",
+            "FLOOR",
+            "CEIL",
+            "FRACTION",
+            "MODULO",
+            "WRAP",
+            "SNAP",
+            "SINE",
+            "COSINE",
+            "TANGENT",
+        ],
+    ):
+        self.node.operation = value
+
+
+class VectorRotate(NodeBuilder):
+    """Rotate a vector around a pivot point (center)"""
+
+    _bl_idname = "ShaderNodeVectorRotate"
+    node: bpy.types.ShaderNodeVectorRotate
+
+    def __init__(
+        self,
+        vector: TYPE_INPUT_VECTOR = None,
+        center: TYPE_INPUT_VECTOR = None,
+        axis: TYPE_INPUT_VECTOR = None,
+        angle: TYPE_INPUT_VALUE = 0.0,
+        rotation: TYPE_INPUT_VECTOR = None,
+        *,
+        rotation_type: Literal[
+            "AXIS_ANGLE", "X_AXIS", "Y_AXIS", "Z_AXIS", "EULER_XYZ"
+        ] = "AXIS_ANGLE",
+        invert: bool = False,
+    ):
+        super().__init__()
+        key_args = {
+            "Vector": vector,
+            "Center": center,
+            "Axis": axis,
+            "Angle": angle,
+            "Rotation": rotation,
+        }
+        self.rotation_type = rotation_type
+        self.invert = invert
+        self._establish_links(**key_args)
+
+    @property
+    def i_vector(self) -> SocketLinker:
+        """Input socket: Vector"""
+        return self._input("Vector")
+
+    @property
+    def i_center(self) -> SocketLinker:
+        """Input socket: Center"""
+        return self._input("Center")
+
+    @property
+    def i_axis(self) -> SocketLinker:
+        """Input socket: Axis"""
+        return self._input("Axis")
+
+    @property
+    def i_angle(self) -> SocketLinker:
+        """Input socket: Angle"""
+        return self._input("Angle")
+
+    @property
+    def i_rotation(self) -> SocketLinker:
+        """Input socket: Rotation"""
+        return self._input("Rotation")
+
+    @property
+    def o_vector(self) -> SocketLinker:
+        """Output socket: Vector"""
+        return self._output("Vector")
+
+    @property
+    def rotation_type(
+        self,
+    ) -> Literal["AXIS_ANGLE", "X_AXIS", "Y_AXIS", "Z_AXIS", "EULER_XYZ"]:
+        return self.node.rotation_type
+
+    @rotation_type.setter
+    def rotation_type(
+        self, value: Literal["AXIS_ANGLE", "X_AXIS", "Y_AXIS", "Z_AXIS", "EULER_XYZ"]
+    ):
+        self.node.rotation_type = value
+
+    @property
+    def invert(self) -> bool:
+        return self.node.invert
+
+    @invert.setter
+    def invert(self, value: bool):
+        self.node.invert = value