blob: 573469f556f327eed24bde344d3d042e570f3757 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
from copy import deepcopy
from .stitch import Stitch
from ..utils import cut_path
from ..stitches import running_stitch
def add_tie(stitches, tie_path):
if stitches[-1].no_ties:
# It's from a manual stitch block, so don't add tie stitches. The user
# will add them if they want them.
return
tie_path = cut_path(tie_path, 0.6)
tie_stitches = running_stitch(tie_path, 0.3)
tie_stitches = [Stitch(stitch.x, stitch.y) for stitch in tie_stitches]
stitches.extend(deepcopy(tie_stitches[1:]))
stitches.extend(deepcopy(list(reversed(tie_stitches))[1:]))
def add_tie_off(stitches):
add_tie(stitches, list(reversed(stitches)))
def add_tie_in(stitches, upcoming_stitches):
add_tie(stitches, upcoming_stitches)
def add_ties(stitch_plan):
"""Add tie-off before and after trims, jumps, and color changes."""
need_tie_in = True
for color_block in stitch_plan:
new_stitches = []
for i, stitch in enumerate(color_block.stitches):
is_special = stitch.trim or stitch.jump or stitch.color_change or stitch.stop
# see stop.py for an explanation of the fake color change
is_fake = stitch.fake_color_change
if is_special and not is_fake and not need_tie_in:
add_tie_off(new_stitches)
new_stitches.append(stitch)
need_tie_in = True
elif need_tie_in and not is_special:
new_stitches.append(stitch)
add_tie_in(new_stitches, upcoming_stitches=color_block.stitches[i:])
need_tie_in = False
else:
new_stitches.append(stitch)
color_block.replace_stitches(new_stitches)
if not need_tie_in:
# tie off at the end if we haven't already
add_tie_off(color_block.stitches)
|
