Class ConnectStrokes

grease_pencil.py:83–130

translate a stroke onto another

class ConnectStrokes(Operator):
    """translate a stroke onto another"""
    bl_idname = "mod_tree.connect_strokes"
    bl_label = "connect strokes"
    bl_options = {"REGISTER", "UNDO"}

    point_dist = FloatProperty(min=0.001, default=.8)
    smooth_iterations = IntProperty(min=0, default=1)
    automatic = BoolProperty(default=True)
    connect_all = BoolProperty(default=True)
    child_stroke_index = IntProperty(
        default=-1)
    parent_stroke_index = IntProperty(
        default=0)

    def execute(self, context):
        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:

            smooth_distribute_gp_layer(gp.layers.active.active_frame, self.point_dist, self.smooth_iterations)

            if self.connect_all:
                moving_range = list(range(1, len(gp.layers.active.active_frame.strokes)))
            else:
                moving_range = [self.child_stroke_index]

            for self.child_stroke_index in moving_range:

                moving_stroke = [i.co for i in gp.layers.active.active_frame.strokes[self.child_stroke_index].points]
                if self.automatic:
                    pos = gp.layers.active.active_frame.strokes[self.child_stroke_index].points[0].co
                    min_dist = 10
                    min_index = 500
                    for index, stroke in enumerate(gp.layers.active.active_frame.strokes):
                        if index != self.child_stroke_index % len(gp.layers.active.active_frame.strokes):
                            dist = min([(i.co - pos).length for i in stroke.points])
                            if dist < min_dist:
                                min_index = index
                                min_dist = dist
                    # print(min_index)
                    self.parent_stroke_index = min_index
                parent_stroke = [i.co for i in gp.layers.active.active_frame.strokes[self.parent_stroke_index].points]
                new_locations = connect_strokes(moving_stroke, parent_stroke)
                for i, point in enumerate(gp.layers.active.active_frame.strokes[self.child_stroke_index].points):
                    point.co = new_locations[i]

        return {'FINISHED'}


def build_tree_from_strokes(strokes, radius, radius_dec):