mantis/misc_nodes_socket_templates.py

from .mantis_dataclasses import MantisSocketTemplate as SockTemplate
from dataclasses import replace

SplineIndexTemplate = SockTemplate(name="Spline Index",
        bl_idname='UnsignedIntSocket', is_input=True, default_value=0,)

InputWidgetSockets = [
    WidgetName := SockTemplate(name='Name', is_input=True,
        bl_idname="EnumWidgetLibrarySocket",),
    Scale := SockTemplate(name='Scale', is_input=True,
        bl_idname="VectorScaleSocket", default_value=(1.0, 1.0, 1.0)),
    FlipAxes := SockTemplate(name='Flip Axes', is_input=True,
        bl_idname="BooleanThreeTupleSocket",),
    xFormOutput := SockTemplate(name='Widget',
        bl_idname='xFormSocket',is_input=False),
]

MatrixFromCurveSockets=[
    CurveTemplate := SockTemplate(name="Curve", bl_idname='EnumCurveSocket', 
        is_input=True,),
    SplineIndexTemplate,
    TotalDivisionsTemplate := SockTemplate(name="Total Divisions",
        bl_idname='UnsignedIntSocket', is_input=True, default_value=0,),
    MatrixIndexTemplate := SockTemplate(name="Matrix Index",
        bl_idname='UnsignedIntSocket', is_input=True, default_value=0,),
    MatrixOutTemplate := SockTemplate(
        name="Matrix", is_input=False,  bl_idname='MatrixSocket', ),
]

MatricesOutTemplate = replace(MatrixOutTemplate,
        name='Matrices', use_multi_input=True)
MatricesFromCurveSockets=MatrixFromCurveSockets.copy()
MatricesFromCurveSockets[4] = MatricesOutTemplate
MatricesFromCurveSockets.pop(3)

MatrixFromCurveSegmentSockets=[
    CurveTemplate,
    SplineIndexTemplate,
    SegmentIndexTemplate := replace(SplineIndexTemplate, name="Segment Index"),
    MatrixOutTemplate,
]

PointFromCurveSockets=[
    CurveTemplate,
    SplineIndexTemplate,
    FactorTemplate := SockTemplate(name="Factor", bl_idname='FloatFactorSocket', 
        is_input=True,),
    PointOutTemplate := SockTemplate(name="Point", bl_idname="VectorSocket",
        is_input=False, )
]

GetCurvePointSockets=[
    CurveTemplate,
    SplineIndexTemplate,
    IndexTemplate := SockTemplate(name="Index",
        bl_idname='UnsignedIntSocket', is_input=True, default_value=0,),
    OutputPointTemplate := SockTemplate(name="Point",
        bl_idname='VectorSocket', is_input=False,),
    OutputLeftHandleTemplate := SockTemplate(name="Left Handle",
        bl_idname='VectorSocket', is_input=False, hide=True),
    OutputRightHandleTemplate := SockTemplate(name="Right Handle",
        bl_idname='VectorSocket', is_input=False, hide=True),
]

GetNearestFactorOnCurveSockets=[
    CurveTemplate,
    SplineIndexTemplate,
    ReferencePointTemplate := SockTemplate(name="Reference Point",
        bl_idname='VectorSocket', is_input=True,),
    OutputFactorTemplate := SockTemplate(name="Factor",
        bl_idname='FloatSocket', is_input=False,),
]

MatrixInvertSockets=[
    Matrix1Template := SockTemplate(
    name="Matrix 1", is_input=True,  bl_idname='MatrixSocket', ),
    MatrixOutTemplate,
]

MatrixComposeSockets=[
    XBasisVector := SockTemplate(
    name="X Basis Vector", is_input=True,  bl_idname='VectorSocket', 
        default_value = (1.0, 0.0, 0.0) ),
    YBasisVector := replace(XBasisVector, name="Y Basis Vector",
        default_value = (0.0, 1.0, 0.0) ),
    ZBasisVector := replace(XBasisVector, name="Z Basis Vector",
        default_value = (0.0, 0.0, 1.0) ),
    Translation := replace(XBasisVector, name="Translation"),
    MatrixOutTemplate,
]

MatrixAlignRollSockets=[
    MatrixInTemplate := replace(Matrix1Template, name="Matrix"),
    AlignVector := SockTemplate(name="Alignment Vector", is_input=True,
       bl_idname='VectorSocket', default_value=(0.0,-1.0,0.0)),
    MatrixOutTemplate,
]

CompareSockets = [
    ComparisonOperation := SockTemplate( name='Comparison',
            is_input=True, bl_idname="EnumCompareOperation",
            default_value="EQUAL",),
    WildcardATemplate := SockTemplate(
        name="A", is_input=True,  bl_idname='WildcardSocket', ),
    WildcardBTemplate := replace(WildcardATemplate, name="B"),
    CompareOutputTemplate := SockTemplate(
        name="Result", is_input=False, bl_idname="BooleanSocket",),
]

NumberOfSplinesSockets=[
    CurveTemplate,
    NumberOfSplinesOut := SockTemplate(name="Number of Splines",
        bl_idname='UnsignedIntSocket', is_input=False),
]

CollectionJoinSockets = [
    JoinCollections := SockTemplate(name='Collections', is_input=True,
        bl_idname="BoneCollectionSocket", display_shape='SQUARE_DOT',
        use_multi_input=True,),
    CollectionDeclarationOutput := SockTemplate(name='Collection',
        bl_idname="BoneCollectionSocket", is_input=False,)
]

CollectionHierarchySockets = [
    ParentCollection := SockTemplate(name='Parent Collection', is_input=True,
        bl_idname="BoneCollectionSocket",),
    ChildCollection := SockTemplate(name='Child Collection', is_input=True,
        bl_idname="BoneCollectionSocket",),
    CollectionDeclarationOutput,
]


UtilityCustomPropertySockets = [
    Name := SockTemplate(name='Name', is_input=True,
        bl_idname="StringSocket", category='general'),
    
    CustomPropType := SockTemplate(name='Type', is_input=True,
        bl_idname="EnumCustomPropTypeSocket", category='general'),
    # Subtype - add this later

    # float properties
    DefaultFloatValue := SockTemplate(name='Default (Float)', is_input=True,
        bl_idname="FloatSocket",  category='float'),
    Min := SockTemplate(name='Min (Float)', is_input=True,
        bl_idname="FloatSocket", category='float'),
    Max := SockTemplate(name='Max (Float)', is_input=True,
        bl_idname="FloatSocket", category='float'),
    SoftMin := SockTemplate(name='Soft Min (Float)', is_input=True,
        bl_idname="FloatSocket", category='float'),
    SoftMax := SockTemplate(name='Soft Max (Float)', is_input=True,
        bl_idname="FloatSocket", category='float'),
    
    # bool properties
    DefaultVectorValue := SockTemplate(name='Default (Bool)', is_input=True,
        bl_idname="BooleanSocket", hide=True, category='bool'),
    # int properties
    DefaultIntValue := SockTemplate(name='Default (Int)', is_input=True,
        bl_idname="IntSocket", hide=True, category='int'),
    IntMin := SockTemplate(name='Min (Int)', is_input=True,
        bl_idname="IntSocket", hide=True, category='int'),
    IntMax := SockTemplate(name='Max (Int)', is_input=True,
        bl_idname="IntSocket", hide=True, category='int'),
    IntSoftMin := SockTemplate(name='Soft Min (Int)', is_input=True,
        bl_idname="IntSocket", hide=True, category='int'),
    IntSoftMax := SockTemplate(name='Soft Max (Int)', is_input=True,
        bl_idname="IntSocket", hide=True, category='int'),
    # vector properties
    DefaultVectorValue := SockTemplate(name='Default (Vector)', is_input=True,
        bl_idname="VectorSocket", hide=True, category='vector'),
    # String Properties
    DefaultStringValue := SockTemplate(name='Default (String)', is_input=True,
        bl_idname="StringSocket", hide=True, category='string'),
    # general:
    Description := SockTemplate(name='Description', is_input=True,
        bl_idname="StringSocket", category='general'),
    PropertyDeclaration := SockTemplate(name='Custom Property',
        is_input=False, bl_idname="CustomPropSocket", category='general'),
]