summaryrefslogtreecommitdiff
path: root/lib/sew_stack/__init__.py
blob: 7f6b2edc7e1b0ab5aa886259d373043b8776ec05 (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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import wx

from . import stitch_layers
from ..debug.debug import debug
from ..elements import EmbroideryElement
from ..utils import DotDict


class SewStack(EmbroideryElement):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.config = DotDict(self.get_json_param('sew_stack', default=dict(layers=list())))
        self.sew_stack_only = self.get_boolean_param('sew_stack_only', False)

        self.layers = []
        for layer_config in self.config.layers:
            layer_class = stitch_layers.by_id[layer_config.layer_id]
            self.layers.append(layer_class(sew_stack=self, config=layer_config))
            debug.log(f"layer name: {self.layers[-1].name}")

    def move_layer(self, from_index, to_index):
        layer = self.layers.pop(from_index)
        self.layers.insert(to_index, layer)

    def append_layer(self, layer_class):
        new_layer = layer_class(sew_stack=self, config={})
        self.layers.append(new_layer)
        return new_layer

    def delete_layer(self, index):
        del self.layers[index]

    def save(self):
        """Save current configuration of layers to sew_stack SVG attribute"""
        self.config.layers = [layer.config for layer in self.layers]
        self.set_json_param("sew_stack", self.config)
        self.set_param("sew_stack_only", self.sew_stack_only)

    def uses_previous_stitch(self):
        if self.config.layers:
            return self.config.layers[0].uses_previous_stitch_group
        else:
            return False

    @property
    def first_stitch(self):
        # For now, we'll return None, but later on we might let the first
        # StitchLayer set the first_stitch.
        return None

    def uses_next_element(self):
        return True

    def get_cache_key_data(self, previous_stitch, next_element):
        return self.config.layers

    def get_default_random_seed(self):
        return self.node.get_id() or ""

    def to_stitch_groups(self, previous_stitch_group=None, next_element=None):
        stitch_groups = []
        for layer in self.layers:
            if layer.enabled:
                this_layer_previous_stitch_group = this_layer_next_element = None

                if layer is self.layers[0]:
                    this_layer_previous_stitch_group = previous_stitch_group
                else:
                    this_layer_previous_stitch_group = stitch_groups[-1]

                if layer is self.layers[-1]:
                    this_layer_next_element = next_element

                stitch_groups.extend(layer.embroider(this_layer_previous_stitch_group, this_layer_next_element))

        return stitch_groups