Feat/dss cimprove master rebasing

cal_tools/cal_tools/plotting.py

@@ -112,9 +112,9 @@ import tabulate
 def create_constant_overview(constant, name, cells, vmin=None, vmax=None, entries=3,
                             out_folder=None, infix=None):
     gmap = {0: 'High', 1: 'Medium', 2: 'Low'}
+    fig = plt.figure(figsize=(10, 5))
+    ax = fig.add_subplot(111)
     for g in range(entries):
-        fig = plt.figure(figsize=(10, 5))
-        ax = fig.add_subplot(111)
         table = []
 
         for qm in constant.keys():
@@ -130,12 +130,13 @@ def create_constant_overview(constant, name, cells, vmin=None, vmax=None, entrie
             #                                                          np.nanmedian(d),
             #                                                          np.nanstd(d)))
             table.append([name, qm, gmap[g], np.nanmean(d), np.nanmedian(d), np.nanstd(d)])
-            ax.step(np.arange(cells), np.nanmean(d, axis=(0,1)))
-        ax.set_xlabel("Memory cell")
-        ax.set_ylabel(name)
-        ax.set_title("{} Gain Median per Cell".format(gmap[g]))
-        if vmin and vmax:
-            ax.set_ylim(vmin, vmax)
+            ax.step(np.arange(cells), np.nanmean(d, axis=(0,1)), label=f'{gmap[g]}-Gain')
+    ax.set_xlabel("Memory cell")
+    ax.set_ylabel(name)
+    ax.legend()
+    ax.set_title(f"{name} Median per Cell".format(gmap[g]))
+    if vmin and vmax:
+        ax.set_ylim(vmin, vmax)
         #if out_folder and infix:
         #    fig.savefig("{}/dark_analysis_{}_{}_per_cell_gain{}.png".format(out_folder,
         #                                                                    infix,