petal-qc 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

petal_qc/__init__.py

@@ -1,5 +1,5 @@
 """petal_qc python module."""
-__version__ = "0.0.9"
+__version__ = "0.0.10"
 
 
 def coreMetrology():
@@ -23,6 +23,11 @@ def bustapeReport():
     # from .BTreport.bustapeReport import main
     main()
 
+def uploadPetalInformation():
+    """Read files from AVS nd create Petal core in PDB."""
+    from .metrology.uploadPetalInformation import main
+    main()
+
 def dashBoard():
     """Launches the Core thermal analysis ahd PDB script."""
     from .dashBoard import main

petal_qc/metrology/compare_Cores.py

@@ -77,17 +77,21 @@ def read_data_files(options):
         if not label in labels:
             labels.append(label)
 
-        val = get_value(data, options.value)
-        if "front" in tmp[1].lower():
-            front[label] = val
-        else:
-            back[label] = val
+        try:
+            val = get_value(data, options.value)
+            if "front" in tmp[1].lower():
+                front[label] = val
+            else:
+                back[label] = val
+        except KeyError as E:
+            print("Error in {}:\n{}".format(fnam, E))
+            continue
 
     labels.sort()
 
     return front, back, labels
 
-def draw_deltas(data, keys, fnam=None, title="Front"):
+def draw_deltas(data, keys, fnam=None, title="Front", draw_text=True):
     """Plot the  position deltas."""
     key_table = {"Bot.": "PL01", "Slot": "PL02", "Top": "PL03",
                  "Bot-FD01": "PL01-FD01", "Bot-FD02": "PL01-FD02", "FD01-FD02": "FD01-FD02" }
@@ -123,8 +127,9 @@ def draw_deltas(data, keys, fnam=None, title="Front"):
                 LBL[k].append(label.split('.')[1].lstrip('0'))
 
         ax[i].scatter(P[i][:,0], P[i][:,1])
-        for j in range(len(LBL[i])):
-            ax[i].text(P[i][j,0], P[i][j,1], LBL[i][j]) #, ha='center', va='top')
+        if draw_text:
+            for j in range(len(LBL[i])):
+                ax[i].text(P[i][j,0], P[i][j,1], LBL[i][j]) #, ha='center', va='top')
 
     save_figure(fig, fnam, prefix=title)
 
@@ -142,10 +147,12 @@ def show_positions(options):
 
     front, back, labels = read_data_files(options)
     val_name = options.value.split('/')[-1]
-    draw_deltas(front, keys, fnam=options.out, title="{} - Front".format(val_name))
-    draw_deltas(back, keys, fnam=options.out, title="{} - Back".format(val_name))
+    draw_text = not options.no_legend
+    draw_deltas(front, keys, fnam=options.out, title="{} - Front".format(val_name), draw_text=draw_text)
+    draw_deltas(back, keys, fnam=options.out, title="{} - Back".format(val_name), draw_text=draw_text)
 
-    plt.show()
+    if not options.no_show:
+        plt.show()
 
 def show_flatness(options):
     """Show flatness plots."""
@@ -177,9 +184,10 @@ def show_flatness(options):
         a.set_ylim(0, 1.2*max(y_lim[0][1], y_lim[1][1]))
         x_lim = a.get_xlim()
         a.fill_between(x_lim, 0, 0.050, facecolor="darkseagreen", alpha=0.1)
-        a.legend(ncol=3, fontsize="x-small")
+        if not options.no_legend:
+            a.legend(ncol=3, fontsize="x-small")
 
-    save_figure(fig, options.out, prefix=opts.prefix)
+    save_figure(fig, options.out, prefix=options.prefix)
 
 def main(options):
     """Main entry."""
@@ -204,11 +212,13 @@ def main(options):
         vback = [back[x] for x in labels]
         ax.plot(X, vfront, '*', label="Front")
         ax.plot(X, vback, 'o', label="Back")
-        ax.legend()
+        if not options.no_legend:
+            ax.legend()
 
         save_figure(fig, options.out, prefix=options.prefix)
 
-    plt.show()
+    if not options.no_show:
+        plt.show()
 
 
 if __name__ == "__main__":
@@ -218,6 +228,8 @@ if __name__ == "__main__":
     parser.add_argument("--value", default=None, help="Value to plot")
     parser.add_argument("--prefix", default=None, help="prefix for out file")
     parser.add_argument("--out", default=None, help="File to store the figure.")
+    parser.add_argument("--no-legend", dest="no_legend", default=False, action="store_true", help="Do not draw the legend")
+    parser.add_argument("--no-show", dest="no_show", default=False, action="store_true", help="Do not show the figure")
 
     opts = parser.parse_args()
     if len(opts.files) == 0:

petal_qc/metrology/coreMetrology.py

@@ -16,7 +16,7 @@ import gi
 gi.require_version("Gtk", "3.0")
 from gi.repository import Gtk, GObject, Gio, GLib
 
-__HELP__ = "https://petal-qc.docs.cern.ch"
+__HELP__ = "https://petal-qc.docs.cern.ch/metrology.html"
 
 class CommaSeparatedListAction(Action):
     """Create a list from the comma sepparated numbers at imput."""