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

nodebpy/nodes/experimental.py

@@ -1,85 +1,116 @@
 from typing import Literal
-from ..builder import NodeBuilder, SocketLinker
+
 import bpy
 
-from .types import TYPE_INPUT_INT, TYPE_INPUT_VALUE
+from ..builder import NodeBuilder, SocketLinker
+from ..types import (
+    TYPE_INPUT_BOOLEAN,
+    TYPE_INPUT_INT,
+    TYPE_INPUT_MENU,
+    TYPE_INPUT_ROTATION,
+    TYPE_INPUT_COLOR,
+    TYPE_INPUT_MATRIX,
+    TYPE_INPUT_VALUE,
+    TYPE_INPUT_VECTOR,
+)
 
 
-class List(NodeBuilder):
-    """Create a list of values"""
+class GetListItem(NodeBuilder):
+    """Retrieve a value from a list"""
 
-    name = "GeometryNodeList"
-    node: bpy.types.GeometryNodeList
+    _bl_idname = "GeometryNodeListGetItem"
+    node: bpy.types.GeometryNodeListGetItem
 
     def __init__(
         self,
-        count: TYPE_INPUT_INT = 1,
-        value: TYPE_INPUT_VALUE = 0.0,
+        list: TYPE_INPUT_VALUE = 0.0,
+        index: TYPE_INPUT_INT = 0,
+        *,
         data_type: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ] = "FLOAT",
-        **kwargs,
     ):
         super().__init__()
-        key_args = {"Count": count, "Value": value}
-        key_args.update(kwargs)
+        key_args = {"List": list, "Index": index}
         self.data_type = data_type
         self._establish_links(**key_args)
 
+    @classmethod
+    def float(
+        cls, list: TYPE_INPUT_VALUE = 0.0, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Float'."""
+        return cls(data_type="FLOAT", list=list, index=index)
+
+    @classmethod
+    def integer(
+        cls, list: TYPE_INPUT_INT = 0, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Integer'."""
+        return cls(data_type="INT", list=list, index=index)
+
+    @classmethod
+    def boolean(
+        cls, list: TYPE_INPUT_BOOLEAN = False, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Boolean'."""
+        return cls(data_type="BOOLEAN", list=list, index=index)
+
+    @classmethod
+    def vector(
+        cls, list: TYPE_INPUT_VECTOR = None, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Vector'."""
+        return cls(data_type="VECTOR", list=list, index=index)
+
+    @classmethod
+    def color(
+        cls, list: TYPE_INPUT_COLOR = None, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Color'."""
+        return cls(data_type="RGBA", list=list, index=index)
+
+    @classmethod
+    def rotation(
+        cls, list: TYPE_INPUT_ROTATION = None, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Rotation'."""
+        return cls(data_type="ROTATION", list=list, index=index)
+
+    @classmethod
+    def matrix(
+        cls, list: TYPE_INPUT_MATRIX = None, index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Matrix'."""
+        return cls(data_type="MATRIX", list=list, index=index)
+
+    @classmethod
+    def menu(
+        cls, list: TYPE_INPUT_MENU = "", index: TYPE_INPUT_INT = 0
+    ) -> "GetListItem":
+        """Create Get List Item with operation 'Menu'."""
+        return cls(data_type="MENU", list=list, index=index)
+
     @property
-    def i_count(self) -> SocketLinker:
-        """Input socket: Count"""
-        return self._input("Count")
+    def i_list(self) -> SocketLinker:
+        """Input socket: List"""
+        return self._input("List")
 
     @property
-    def i_value(self) -> SocketLinker:
-        """Input socket: Value"""
-        return self._input("Value")
+    def i_index(self) -> SocketLinker:
+        """Input socket: Index"""
+        return self._input("Index")
 
     @property
-    def o_list(self) -> SocketLinker:
-        """Output socket: List"""
-        return self._output("List")
+    def o_value(self) -> SocketLinker:
+        """Output socket: Value"""
+        return self._output("Value")
 
     @property
     def data_type(
         self,
     ) -> Literal[
-        "FLOAT",
-        "INT",
-        "BOOLEAN",
-        "VECTOR",
-        "RGBA",
-        "ROTATION",
-        "MATRIX",
-        "STRING",
-        "MENU",
-        "SHADER",
-        "OBJECT",
-        "IMAGE",
-        "GEOMETRY",
-        "COLLECTION",
-        "TEXTURE",
-        "MATERIAL",
-        "BUNDLE",
-        "CLOSURE",
+        "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
     ]:
         return self.node.data_type
 
@@ -87,104 +118,100 @@ class List(NodeBuilder):
     def data_type(
         self,
         value: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ],
     ):
         self.node.data_type = value
 
 
-class GetListItem(NodeBuilder):
-    """Retrieve a value from a list"""
+class List(NodeBuilder):
+    """Create a list of values"""
 
-    name = "GeometryNodeListGetItem"
-    node: bpy.types.GeometryNodeListGetItem
+    _bl_idname = "GeometryNodeList"
+    node: bpy.types.GeometryNodeList
 
     def __init__(
         self,
-        list: TYPE_INPUT_VALUE = 0.0,
-        index: TYPE_INPUT_INT = 0,
+        count: TYPE_INPUT_INT = 1,
+        value: TYPE_INPUT_VALUE = 0.0,
+        *,
         data_type: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ] = "FLOAT",
-        **kwargs,
     ):
         super().__init__()
-        key_args = {"List": list, "Index": index}
-        key_args.update(kwargs)
+        key_args = {"Count": count, "Value": value}
         self.data_type = data_type
         self._establish_links(**key_args)
 
+    @classmethod
+    def float(cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_VALUE = 0.0) -> "List":
+        """Create List with operation 'Float'."""
+        return cls(data_type="FLOAT", count=count, value=value)
+
+    @classmethod
+    def integer(cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_INT = 0) -> "List":
+        """Create List with operation 'Integer'."""
+        return cls(data_type="INT", count=count, value=value)
+
+    @classmethod
+    def boolean(
+        cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_BOOLEAN = False
+    ) -> "List":
+        """Create List with operation 'Boolean'."""
+        return cls(data_type="BOOLEAN", count=count, value=value)
+
+    @classmethod
+    def vector(
+        cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_VECTOR = None
+    ) -> "List":
+        """Create List with operation 'Vector'."""
+        return cls(data_type="VECTOR", count=count, value=value)
+
+    @classmethod
+    def color(cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_COLOR = None) -> "List":
+        """Create List with operation 'Color'."""
+        return cls(data_type="RGBA", count=count, value=value)
+
+    @classmethod
+    def rotation(
+        cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_ROTATION = None
+    ) -> "List":
+        """Create List with operation 'Rotation'."""
+        return cls(data_type="ROTATION", count=count, value=value)
+
+    @classmethod
+    def matrix(
+        cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_MATRIX = None
+    ) -> "List":
+        """Create List with operation 'Matrix'."""
+        return cls(data_type="MATRIX", count=count, value=value)
+
+    @classmethod
+    def menu(cls, count: TYPE_INPUT_INT = 1, value: TYPE_INPUT_MENU = "") -> "List":
+        """Create List with operation 'Menu'."""
+        return cls(data_type="MENU", count=count, value=value)
+
     @property
-    def i_list(self) -> SocketLinker:
-        """Input socket: List"""
-        return self._input("List")
+    def i_count(self) -> SocketLinker:
+        """Input socket: Count"""
+        return self._input("Count")
 
     @property
-    def i_index(self) -> SocketLinker:
-        """Input socket: Index"""
-        return self._input("Index")
+    def i_value(self) -> SocketLinker:
+        """Input socket: Value"""
+        return self._input("Value")
 
     @property
-    def o_value(self) -> SocketLinker:
-        """Output socket: Value"""
-        return self._output("Value")
+    def o_list(self) -> SocketLinker:
+        """Output socket: List"""
+        return self._output("List")
 
     @property
     def data_type(
         self,
     ) -> Literal[
-        "FLOAT",
-        "INT",
-        "BOOLEAN",
-        "VECTOR",
-        "RGBA",
-        "ROTATION",
-        "MATRIX",
-        "STRING",
-        "MENU",
-        "SHADER",
-        "OBJECT",
-        "IMAGE",
-        "GEOMETRY",
-        "COLLECTION",
-        "TEXTURE",
-        "MATERIAL",
-        "BUNDLE",
-        "CLOSURE",
+        "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
     ]:
         return self.node.data_type
 
@@ -192,24 +219,7 @@ class GetListItem(NodeBuilder):
     def data_type(
         self,
         value: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ],
     ):
         self.node.data_type = value
@@ -218,40 +228,62 @@ class GetListItem(NodeBuilder):
 class ListLength(NodeBuilder):
     """Count how many items are in a given list"""
 
-    name = "GeometryNodeListLength"
+    _bl_idname = "GeometryNodeListLength"
     node: bpy.types.GeometryNodeListLength
 
     def __init__(
         self,
         list: TYPE_INPUT_VALUE = 0.0,
+        *,
         data_type: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ] = "FLOAT",
-        **kwargs,
     ):
         super().__init__()
         key_args = {"List": list}
-        key_args.update(kwargs)
         self.data_type = data_type
         self._establish_links(**key_args)
 
+    @classmethod
+    def float(cls, list: TYPE_INPUT_VALUE = 0.0) -> "ListLength":
+        """Create List Length with operation 'Float'."""
+        return cls(data_type="FLOAT", list=list)
+
+    @classmethod
+    def integer(cls, list: TYPE_INPUT_INT = 0) -> "ListLength":
+        """Create List Length with operation 'Integer'."""
+        return cls(data_type="INT", list=list)
+
+    @classmethod
+    def boolean(cls, list: TYPE_INPUT_BOOLEAN = False) -> "ListLength":
+        """Create List Length with operation 'Boolean'."""
+        return cls(data_type="BOOLEAN", list=list)
+
+    @classmethod
+    def vector(cls, list: TYPE_INPUT_VECTOR = None) -> "ListLength":
+        """Create List Length with operation 'Vector'."""
+        return cls(data_type="VECTOR", list=list)
+
+    @classmethod
+    def color(cls, list: TYPE_INPUT_COLOR = None) -> "ListLength":
+        """Create List Length with operation 'Color'."""
+        return cls(data_type="RGBA", list=list)
+
+    @classmethod
+    def rotation(cls, list: TYPE_INPUT_ROTATION = None) -> "ListLength":
+        """Create List Length with operation 'Rotation'."""
+        return cls(data_type="ROTATION", list=list)
+
+    @classmethod
+    def matrix(cls, list: TYPE_INPUT_MATRIX = None) -> "ListLength":
+        """Create List Length with operation 'Matrix'."""
+        return cls(data_type="MATRIX", list=list)
+
+    @classmethod
+    def menu(cls, list: TYPE_INPUT_MENU = "") -> "ListLength":
+        """Create List Length with operation 'Menu'."""
+        return cls(data_type="MENU", list=list)
+
     @property
     def i_list(self) -> SocketLinker:
         """Input socket: List"""
@@ -266,24 +298,7 @@ class ListLength(NodeBuilder):
     def data_type(
         self,
     ) -> Literal[
-        "FLOAT",
-        "INT",
-        "BOOLEAN",
-        "VECTOR",
-        "RGBA",
-        "ROTATION",
-        "MATRIX",
-        "STRING",
-        "MENU",
-        "SHADER",
-        "OBJECT",
-        "IMAGE",
-        "GEOMETRY",
-        "COLLECTION",
-        "TEXTURE",
-        "MATERIAL",
-        "BUNDLE",
-        "CLOSURE",
+        "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
     ]:
         return self.node.data_type
 
@@ -291,24 +306,7 @@ class ListLength(NodeBuilder):
     def data_type(
         self,
         value: Literal[
-            "FLOAT",
-            "INT",
-            "BOOLEAN",
-            "VECTOR",
-            "RGBA",
-            "ROTATION",
-            "MATRIX",
-            "STRING",
-            "MENU",
-            "SHADER",
-            "OBJECT",
-            "IMAGE",
-            "GEOMETRY",
-            "COLLECTION",
-            "TEXTURE",
-            "MATERIAL",
-            "BUNDLE",
-            "CLOSURE",
+            "FLOAT", "INT", "BOOLEAN", "VECTOR", "RGBA", "ROTATION", "MATRIX", "MENU"
         ],
     ):
         self.node.data_type = value