diff --git a/src/madgui/plot/twissfigure.yml b/src/madgui/plot/twissfigure.yml index 109ab92a..ba871416 100644 --- 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 index b6ae3a89..bc36261e 100644 --- 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))