Merge pull request #48 from hibtc/matchFeature

Add dynamical color change to snapshots
hibtc · Apr 25, 2020 · 100ccc3 · 100ccc3
2 parents ea0ef8e + b1e7963
commit 100ccc3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/madgui/plot/twissfigure.yml b/src/madgui/plot/twissfigure.yml
@@ -36,7 +36,8 @@ select_style:
 # ``matplotlib.axes.Axes.plot()``
 reference_style:
   linestyle: "-"
-  color: "black"
+  color: "C0"
+  alpha: 0.5
 
 # Style for monitor markers:
 readouts_style:

diff --git a/src/madgui/widget/curvemanager.py b/src/madgui/widget/curvemanager.py
@@ -65,6 +65,9 @@ def data(self):
         return self.curveTable.rows
 
     def on_btn_save(self):
+        """
+        Add a Snapshot of the current twiss figure
+        """
         scene = self.scene
         twiss = scene.model.twiss()
         data = {
@@ -75,6 +78,7 @@ def on_btn_save(self):
         style = scene.config['reference_style']
         scene.snapshot_num += 1
         name = "snapshot {}".format(scene.snapshot_num)
+        style['color'] = 'C{}'.format(scene.snapshot_num)
         scene.add_curve(name, data, style)
         self.curveTable.edit(
             self.curveTable.model().index(len(self.available)-1, 0))