From 8e49b4431e48b776eef5df465f5bb00f46e9b707 Mon Sep 17 00:00:00 2001
From: Kaalleen <36401965+kaalleen@users.noreply.github.com>
Date: Sat, 1 Mar 2025 09:50:36 +0100
Subject: display stop commands in simulator and print preview (#3545)

---
 lib/stitch_plan/color_block.py | 8 ++++++++
 lib/stitch_plan/stitch_plan.py | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/stitch_plan/color_block.py b/lib/stitch_plan/color_block.py
index 3cec826d..75b72a3b 100644
--- a/lib/stitch_plan/color_block.py
+++ b/lib/stitch_plan/color_block.py
@@ -73,6 +73,12 @@ class ColorBlock(object):
             previous_stitch = stitch
         return length
 
+    @property
+    def num_stops(self):
+        """Number of stops in this color block."""
+
+        return sum(1 for stitch in self if stitch.stop)
+
     @property
     def num_trims(self):
         """Number of trims in this color block."""
@@ -87,6 +93,8 @@ class ColorBlock(object):
 
     @property
     def stop_after(self):
+        # TODO: we do not add the stop command necessarily as the last stitch
+        # also we do not necessarily start a new color block when a stop command appears
         if self.last_stitch is not None:
             return self.last_stitch.stop
         else:
diff --git a/lib/stitch_plan/stitch_plan.py b/lib/stitch_plan/stitch_plan.py
index f6e3f0de..8439de18 100644
--- a/lib/stitch_plan/stitch_plan.py
+++ b/lib/stitch_plan/stitch_plan.py
@@ -172,7 +172,7 @@ class StitchPlan(object):
 
     @property
     def num_stops(self):
-        return sum(1 for block in self if block.stop_after)
+        return sum(block.num_stops for block in self)
 
     @property
     def num_trims(self):
-- 
cgit v1.2.3