node_Series.py

import bpy
from node_s import *
from util import *

class GenSeriesNode(Node, SverchCustomTreeNode):
    ''' Generator series '''
    bl_idname = 'GenSeriesNode'
    bl_label = 'List Series'
    bl_icon = 'OUTLINER_OB_EMPTY'
    
    start_ = bpy.props.FloatProperty(name = 'start', description='start', default=0, options={'ANIMATABLE'}, update=updateNode)
    stop_ = bpy.props.FloatProperty(name = 'stop', description='stop', default=10, options={'ANIMATABLE'}, update=updateNode)
    step_ = bpy.props.FloatProperty(name = 'step', description='step', default=1, options={'ANIMATABLE'}, update=updateNode)
    
    def init(self, context):
        self.inputs.new('StringsSocket', "Start", "Start")
        self.inputs.new('StringsSocket', "Stop", "Stop")
        self.inputs.new('StringsSocket', "Step", "Step")
        self.outputs.new('StringsSocket', "Series", "Series")
    
    def draw_buttons(self, context, layout):
        layout.prop(self, "start_", text="start")
        layout.prop(self, "stop_", text="stop")
        layout.prop(self, "step_", text="step")

    def update(self):
        # inputs
        if 'Start' in self.inputs and self.inputs['Start'].links:
            tmp = SvGetSocketAnyType(self,self.inputs['Start'])
            Start = tmp[0][0]
        else:
            Start = self.start_
    
        if 'Stop' in self.inputs and self.inputs['Stop'].links:
            tmp = SvGetSocketAnyType(self,self.inputs['Stop'])
            Stop = tmp[0][0]
        else:
            Stop = self.stop_
        
        if 'Step' in self.inputs and self.inputs['Step'].links:
            tmp = SvGetSocketAnyType(self,self.inputs['Step'])
            Step = tmp[0][0]
        else:
            Step = self.step_
        
        # outputs
        if 'Series' in self.outputs and len(self.outputs['Series'].links)>0:
            if not self.outputs['Series'].node.socket_value_update:
                self.inputs['Series'].node.update()
            #print (Start, Stop, Step)
            if Step < 0:
                Step = 1
            if Stop < Start:
                Stop = Start+1
            series = [c for c in self.xfrange(Start, Stop, Step)]
            
            SvSetSocketAnyType(self,'Series', [series])

    def xfrange(self, start, stop, step):
        while start < stop:
            yield start
            start += step
    
    def update_socket(self, context):
        self.update()


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

if __name__ == "__main__":
    register()