Function build_tree_from_strokes_rec

grease_pencil.py:155–201

(points, module, head, curr_index, curr_stroke, splits, strokes, radius_dec)

def build_tree_from_strokes_rec(points, module, head, curr_index, curr_stroke, splits, strokes, radius_dec):
    if len(points)>1:
        pos = points.popleft()
        direction = points[0]-pos
        choice = 'branch'
        radius = module.head_1_radius if head == 0 else module.head_2_radius

        for parent_stroke, point_index, child_stroke in splits:
            if curr_stroke == parent_stroke and point_index == curr_index:
                child_points = deque(strokes[child_stroke])
                child_points.popleft()
                child_points.popleft()
                child_direction = (child_points[0] - pos)
                child_length = child_direction.length
                if child_length < 1.5*radius:
                    for i in range(int(1.5*radius/child_length)):
                        if len(child_points)>1:
                            child_points.popleft()
                child_direction = (child_points[0] - pos)
                child_length = child_direction.length
                child_direction.normalize()
                spin = directions_to_spin(direction, child_direction)
                choice = 'split'
                break
        if choice == 'branch':
            new_module = Branch(pos, direction.normalized(), radius, direction.length, head_radius=radius_dec,
                                resolution=0, spin=module.spin)
        elif choice == 'split':
            new_module = Split(pos, direction.normalized(), radius, 0, 0, spin, head_2_length=child_length)
            if direction.length < radius*.6:
                number_to_pop = int(radius*.6/direction.length)
                for i in range(number_to_pop):
                    if len(points)>1:
                        points.popleft()
                direction = points[0] - pos
            new_module.head_1_length = direction.length
            new_module.head_2_direction = child_direction
            new_module.secondary_angle = direction.angle(child_direction)

        if head == 0:
            module.head_module_1 = new_module
        else:
            module.head_module_2 = new_module
        new_module.creator = "gp_trunk" if curr_stroke == 0 else "gp_branch"
        build_tree_from_strokes_rec(points, new_module, 0, curr_index+1, curr_stroke, splits, strokes, radius_dec)
        if choice == 'split':
            build_tree_from_strokes_rec(child_points, new_module, 1, 2, child_stroke, splits, strokes, radius_dec)