node_ListRepeater.py

import bpy, bmesh, mathutils
from mathutils import Vector, Matrix
from node_s import *
from util import *

class ListRepeaterNode(Node, SverchCustomTreeNode):
    ''' List repeater '''
    bl_idname = 'ListRepeaterNode'
    bl_label = 'List repeater'
    bl_icon = 'OUTLINER_OB_EMPTY'
    
    level = bpy.props.IntProperty(name = 'level', default=1, min=0, update=updateNode)
    number = bpy.props.IntProperty(name = 'number', default=1, min=1, update=updateNode)
    unwrap = bpy.props.BoolProperty(name = 'unwrap', default=False, update=updateNode)
    typ = bpy.props.StringProperty(name='typ', default='')
    newsock = bpy.props.BoolProperty(name='newsock', default=False)
    
    def draw_buttons(self, context, layout):
        layout.prop(self, "level", text="level")
        layout.prop(self, "number", text="number")
        layout.prop(self, "unwrap", text="unwrap")
        
    def init(self, context):
        self.inputs.new('StringsSocket', "Data", "Data")
        self.inputs.new('StringsSocket', "Number","Number")
        self.outputs.new('StringsSocket',"Data", "Data")
        
    def update(self):
        # достаём два слота - вершины и полики
        if 'Data' in self.inputs and self.inputs['Data'].links:
            inputsocketname = 'Data'
            outputsocketname = ['Data',]
            changable_sockets(self, inputsocketname, outputsocketname)
            
            data = SvGetSocketAnyType(self, self.inputs['Data'])
            
            if 'Number' in self.inputs and self.inputs['Number'].links:
                tmp = SvGetSocketAnyType(self,self.inputs['Number'])
                Number = tmp[0]
            else:
                Number = [self.number]
            
            if 'Data' in self.outputs and self.outputs['Data'].links:
                out_ = self.count(data, self.level, Number)
                if self.unwrap:
                    if len(out_)>0:
                        out = []
                        for o in out_:
                            out.extend(o)
                else:
                    out = out_
                    
                SvSetSocketAnyType(self, 'Data', out)
            
    def count(self, data, level, number, cou=0):
        if level:
            out = []
            for idx, obj in enumerate(data):
                out.append(self.count(obj, level-1, number, idx))
                
        else:
            out = []
            indx = min(cou,len(number)-1)
            for i in range(int(number[indx])):
                out.append(data)
        return out
            

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

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

if __name__ == "__main__":
    register()