mantis/primitives_nodes.py

from .utilities.node_common import *
from .base_definitions import MantisNode, NodeSocket
from .primitives_sockets import *

def TellClasses():
    return [
             # Primitives
             CirclePrimitive,
             GeometryLattice,
            ]

#*#-------------------------------#++#-------------------------------#*#
# P R I M I T I V E S
#*#-------------------------------#++#-------------------------------#*#

class PrimitiveNode(MantisNode):
    __slots__ = ()
    node_type = "UTILITY"
    
    def __init__(self, signature, base_tree, socket_templates=[],
                 additional_parameters = {}):
        super().__init__(signature, base_tree, socket_templates,
                         additional_parameters)
        self.prepared = True
    
    def reset_execution(self):
        super().reset_execution()
        self.prepared=True

class CirclePrimitive(PrimitiveNode):
    '''A node representing a Circle Primitive mesh'''
    __slots__ = ()
    socket_templates = CirclePrimitiveSockets
    passes=frozenset({'bPrepare',})

    # mode is unused but needed for interface consistency
    def bGetObject(self, mode=''):
        from bpy import data
        # first try Curve, then try Mesh
        bObject = data.curves.get(self.evaluate_input("Name"))
        if not bObject:
            bObject = data.meshes.get(self.evaluate_input("Name"))
        return bObject
        
    def bPrepare(self, bContext = None,):
        # Get the datablock
        data = self.bGetObject()
        import bpy
        if not data:
            data = bpy.data.meshes.new( self.evaluate_input("Name") )
        # make the circle
        import bmesh; bm = bmesh.new()
        bmesh.ops.create_circle( # lazy but easy
            bm,
            cap_ends=False,
            radius=max(self.evaluate_input("Radius"), 0.0001),
            segments=min( max( self.evaluate_input("Number of Points"), 3), 1024),
            )
        # this is rotated 90 degrees, we need Y-up instead of Z-up
        from mathutils import Matrix
        from math import pi
        for v in bm.verts:
            v.co = Matrix.Rotation(pi/2, 4, 'X') @ v.co
        # done with this, push it to the data and free the bmesh.
        bm.to_mesh(data); bm.free()
        self.executed = True

from .primitives_sockets import LatticeSockets
class GeometryLattice(PrimitiveNode):
    '''A node representing a Circle Primitive mesh'''
    __slots__ = ()
    socket_templates = LatticeSockets

    def __init__(self, signature, base_tree):
        super().__init__(signature, base_tree, self.socket_templates)
        self.prepared = False

    def reset_execution(self):
        super().reset_execution()
        self.prepared=False
        self.executed=False
        
    # mode is unused but needed for interface consistency
    def bGetObject(self, mode=''):
        from bpy import data
        bObject = data.lattices.get(self.evaluate_input("Name"))
        return bObject
        
    def bPrepare(self, bContext = None,):
        # Get the datablock
        data = self.bGetObject()
        import bpy
        if not data:
            data = bpy.data.lattices.new( self.evaluate_input("Name") )
        props_sockets = self.property_socket_map
        evaluate_sockets(self, {"DEFAULT":data}, props_sockets)
        self.prepared = True; self.executed = True