node_Line.py

import bpy
from node_s import *
from util import *

class LineNode(Node, SverchCustomTreeNode):
    ''' Line '''
    bl_idname = 'LineNode'
    bl_label = 'Line'
    bl_icon = 'OUTLINER_OB_EMPTY'
    
    int_ = bpy.props.IntProperty(name = 'int_', description='Nº Vertices', default=2, min=2, options={'ANIMATABLE'}, update=updateNode)
    step_ = bpy.props.FloatProperty(name = 'step_', description='Step length', default=1.0, options={'ANIMATABLE'}, update=updateNode)

    def init(self, context):
        self.inputs.new('StringsSocket', "Nº Vertices", "Nº Vertices")
        self.inputs.new('StringsSocket', "Step", "Step length")
        self.outputs.new('VerticesSocket', "Vertices", "Vertices")
        self.outputs.new('StringsSocket', "Edges", "Edges")
    
    def draw_buttons(self, context, layout):
        layout.prop(self, "int_", text="Nº Vert")
        layout.prop(self, "step_", text="Step")

    def update(self):
        # inputs
        if 'Nº Vertices' in self.inputs and self.inputs['Nº Vertices'].links:
            Integer = int(SvGetSocketAnyType(self,self.inputs['Nº Vertices'])[0][0])
        else:
            Integer = self.int_

        if 'Step' in self.inputs and self.inputs['Step'].links:
            Step = SvGetSocketAnyType(self,self.inputs['Step'])[0]
            
            if len(Step) < Integer:
                fullList(Step, Integer)
            
            listVert = []
            for i in range(Integer):
                listVert.append(round((i*Step[i]),4))
            X = listVert

        else:
            Step = self.step_
            listVert = [Step*(i) for i in range(Integer)]
            X = listVert

        # outputs
        if 'Vertices' in self.outputs and self.outputs['Vertices'].links:

            Y = [0.0]
            Z = [0.0]
            max_num = len(X)
            fullList(Y,max_num)
            fullList(Z,max_num)

            points = list(zip(X,Y,Z))
            SvSetSocketAnyType(self, 'Vertices',[points])

        if 'Edges' in self.outputs and self.outputs['Edges'].links:

            listEdg = []
            for i in range(Integer-1):
                listEdg.append((i, i+1))

            edg = list(listEdg)
            SvSetSocketAnyType(self, 'Edges',[edg])

    def update_socket(self, context):
        self.update()


def register():
    bpy.utils.register_class(LineNode)
    
def unregister():
    bpy.utils.unregister_class(LineNode)

if __name__ == "__main__":
    register()