Class GreasePencilNode

nodes.py:253–294

class GreasePencilNode(Node, ModularTreeNode):
    bl_idname = "GreasePencilNode"
    bl_label = "Grease_Pencil"

    smooth_iterations = IntProperty(min=0, default=1)
    radius = FloatProperty(min=0, default=.7)
    branch_length = FloatProperty(min=.001, default=.6)
    radius_decrease = FloatProperty(min=0, max=.999, default=.97)
    grease_pencil_memory = StringProperty(default="")

    def init(self, context):
        self.outputs.new("TreeSocketType", "Tree")
        self.outputs.new("SelectionSocketType", "Selection")

    @property
    def selection(self):
        return ["gp_branch"]

    def draw_buttons(self, context, layout):
        properties = ["smooth_iterations", "radius", "radius_decrease", "branch_length"]
        # bpy.ops.mod_tree.connect_strokes(point_dist=self.branch_length, automatic=True, connect_all=True,
        #                                  child_stroke_index=1, parent_stroke_index=0,
        #                                  smooth_iterations=self.smooth_iterations)

        op_props = layout.operator("mod_tree.connect_strokes", text='update strokes')
        op_props.point_dist = self.branch_length
        op_props.connect_all = True
        op_props.child_stroke_index = 1
        op_props.parent_stroke_index = 0
        op_props.smooth_iterations = self.smooth_iterations
        for i in properties:
            layout.prop(self, i)

    def execute(self):

        gp = bpy.context.scene.grease_pencil
        if gp is not None and gp.layers.active is not None and gp.layers.active.active_frame is not None and len(
                gp.layers.active.active_frame.strokes) > 0 and len(gp.layers.active.active_frame.strokes[0].points) > 1:

            strokes = [[i.co for i in j.points] for j in gp.layers.active.active_frame.strokes]
            root = build_tree_from_strokes(strokes, self.radius, self.radius_decrease)
            return root


class SplitNode(Node, ModularTreeNode):