Merge pull request #48 from hibtc/matchFeature

Add dynamical color change to snapshots

Author: eccortes95

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:

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))