Show start and end staff line

2 files changed, 10 insertions, 0 deletions

diff --git a/staff.py b/staff.py
index 7986921..3a5bb3b 100644
--- a/staff.py
+++ b/staff.py
@@ -24,6 +24,7 @@ class Staff:
     lmarg = 0
     rmarg = 0
     spacing = 0
+    ymax = 0
 
     def __init__(self):
         pass
@@ -36,4 +37,5 @@ class Staff:
             context.move_to(self.lmarg, line_y)
             context.line_to(width - self.rmarg, line_y)
             line_y += self.spacing
+        self.ymax = line_y - self.spacing
         context.stroke()
diff --git a/track.py b/track.py
index 837e929..c84d221 100644
--- a/track.py
+++ b/track.py
@@ -35,6 +35,14 @@ class Track:
         self.staff.y = self.linespacing * 3
         self.staff.draw(context, width)
 
+        #start and end line
+        context.move_to(self.staff.lmarg, self.staff.y)
+        context.line_to(self.staff.lmarg, self.staff.ymax)
+        context.stroke()
+        context.move_to(width - self.staff.rmarg, self.staff.y)
+        context.line_to(width - self.staff.rmarg, self.staff.ymax)
+        context.stroke()
+
     def set_y(self, y):
         self._y = y