petal-qc 0.0.0__py3-none-any.whl

petal_qc/thermal/IRPetalParam.py

@@ -0,0 +1,71 @@
+"""Parameters needed for the thermal analysis of the petal."""
+
+
+class IRPetalParam(object):
+    """Default values for IR image handling."""
+
+    def __init__(self, values=None):
+        """Initialize.
+
+        Args:
+        ----
+            values: ArgParser or dict with user values-
+
+        """
+        self.institute = 'IFIC'   # Either IFIC or DESY to treat the different files
+        self.thrs = -20.0         # the threshold
+        self.gauss_size = 15      # Radius of gausian filtering
+        self.grad_sigma = 2.5     # Sigma of grading calculation
+        self.distance = 5         # Distance in contour between slices
+        self.npoints = 15         # Number of points per segment
+        self.min_area = 2500      # minumum area of a valid contour.
+        self.contour_cut = 0.2    # Fraction of IR image range to define contour.
+        self.contour_smooth = 25  # Value to smooth contour
+        self.width = 2            # half widh of rectangle around point in path when getting T.
+        self.do_fit = True        # True to fit the segment points.
+        self.rotate = True        # Rotate to have a vertical petal in mirror image
+        self.debug = False        # To debug
+
+        if values is not None:
+            self.set_values(values)
+
+    def set_values(self, values):
+        """Set parameters from input values."""
+        if isinstance(values, dict):
+            for key in self.__dict__.keys():
+                if key in values:
+                    setattr(self, key, values[key])
+
+        else:
+            for key in self.__dict__.keys():
+                if hasattr(values, key):
+                    setattr(self, key, getattr(values, key))
+
+    def print(self):
+        """Print all values."""
+        print("\nParameters:")
+        for key, val in self.__dict__.items():
+            print("{}: {}".format(key, val))
+        print("")
+
+    @staticmethod
+    def add_parameters(parser):
+        """Add parameters to the ArgumentParser."""
+        # Get the default value for the parameters.
+        P = IRPetalParam()
+
+        parser.add_argument("--institute", type=str,
+                            default=P.institute,
+                            help="Either IFIC or DESY to treat the different files")
+        parser.add_argument("--thrs", type=float, default=P.thrs, help="Temperature threshold")
+        parser.add_argument("--gauss_size", type=int, default=P.gauss_size, help="Radius of gausian filtering")
+        parser.add_argument("--distance", type=float, default=P.distance, help="Distance in contour beteween slices")
+        parser.add_argument("--npoints", type=int, default=P.npoints, help="Number of points per segment")
+        parser.add_argument("--min_area", type=float, default=P.min_area, help="minumum area of a valid contour")
+        parser.add_argument("--width", type=int, default=P.width,
+                            help="width of average rectangle en get_T_along_path.")
+        parser.add_argument("--contour_cut", type=float, default=P.contour_cut,
+                            help="Fraction of IR image range to define contour.")
+        parser.add_argument("--contour_smooth", type=float, default=P.contour_smooth,
+                            help="Value to smooth contour")
+        parser.add_argument("--debug", action="store_true", default=False, help="Show additional information.")

petal_qc/thermal/PetalColorMaps.py

@@ -0,0 +1,62 @@
+"""Colormaps for Petal QC."""
+from matplotlib import colors
+
+cdict = {'blue': [[0.0, 0.17254901960784313, 0.17254901960784313],
+                  [0.4596774193548387, 0.9215686274509803, 0.9215686274509803],
+                  [0.49193548387096775, 0.9372549019607843, 0.9372549019607843],
+                  [0.6048387096774194, 1.0, 1.0],
+                  [0.6451612903225806, 0.9803921568627451, 0.9803921568627451],
+                  [0.6774193548387096, 0.8156862745098039, 0.8156862745098039],
+                  [0.7258064516129032, 0.35294117647058826, 0.35294117647058826],
+                  [0.75, 0.050980392156862744, 0.050980392156862744],
+                  [0.782258064516129, 0.0196078431372549, 0.0196078431372549],
+                  [0.7983870967741935, 0.01568627450980392, 0.01568627450980392],
+                  [0.8306451612903226, 0.00784313725490196, 0.00784313725490196],
+                  [0.8548387096774194, 0.00784313725490196, 0.00784313725490196],
+                  [0.9435483870967742, 0.00392156862745098, 0.00392156862745098],
+                  [0.9838709677419355, 0.2235294117647059, 0.2235294117647059],
+                  [1.0, 0.5019607843137255, 0.5019607843137255]],
+         'green': [[0.0, 0.16470588235294117, 0.16470588235294117],
+                   [0.4596774193548387, 0.9176470588235294, 0.9176470588235294],
+                   [0.49193548387096775, 0.9294117647058824, 0.9294117647058824],
+                   [0.6048387096774194, 0.4549019607843137, 0.4549019607843137],
+                   [0.6451612903225806, 0.35294117647058826, 0.35294117647058826],
+                   [0.6774193548387096, 0.403921568627451, 0.403921568627451],
+                   [0.7258064516129032, 0.6431372549019608, 0.6431372549019608],
+                   [0.75, 0.8274509803921568, 0.8274509803921568],
+                   [0.782258064516129, 1.0, 1.0],
+                   [0.7983870967741935, 0.984313725490196, 0.984313725490196],
+                   [0.8306451612903226, 0.9686274509803922, 0.9686274509803922],
+                   [0.8548387096774194, 0.8117647058823529, 0.8117647058823529],
+                   [0.9435483870967742, 0.21176470588235294, 0.21176470588235294],
+                   [0.9838709677419355, 0.01568627450980392, 0.01568627450980392],
+                   [1.0, 0.0, 0.0]],
+         'red': [[0.0, 0.16862745098039217, 0.16862745098039217],
+                 [0.4596774193548387, 0.9254901960784314, 0.9254901960784314],
+                 [0.49193548387096775, 0.9372549019607843, 0.9372549019607843],
+                 [0.6048387096774194, 0.25882352941176473, 0.25882352941176473],
+                 [0.6451612903225806, 0.10980392156862745, 0.10980392156862745],
+                 [0.6774193548387096, 0.0, 0.0],
+                 [0.7258064516129032, 0.0, 0.0],
+                 [0.75, 0.00392156862745098, 0.00392156862745098],
+                 [0.782258064516129, 0.07450980392156863, 0.07450980392156863],
+                 [0.7983870967741935, 0.47843137254901963, 0.47843137254901963],
+                 [0.8306451612903226, 0.796078431372549, 0.796078431372549],
+                 [0.8548387096774194, 0.9568627450980393, 0.9568627450980393],
+                 [0.9435483870967742, 0.9803921568627451, 0.9803921568627451],
+                 [0.9838709677419355, 1.0, 1.0],
+                 [1.0, 0.996078431372549, 0.996078431372549]]}
+
+HighContrast = colors.LinearSegmentedColormap('HighContrast', segmentdata=cdict)
+
+if __name__ == "__main__":
+    import matplotlib.pyplot as plt
+    from matplotlib import cm
+
+    fig, ax = plt.subplots()
+    fig.colorbar(cm.ScalarMappable(
+        norm=colors.Normalize(), cmap=HighContrast), ax=ax)
+    fig.colorbar(cm.ScalarMappable(norm=colors.Normalize(),
+                                   cmap=HighContrast.reversed()), ax=ax)
+
+    plt.show()

petal_qc/thermal/Petal_IR_Analysis.py

@@ -0,0 +1,142 @@
+"""Store resutls."""
+import matplotlib.pyplot as plt
+import matplotlib.ticker as ticker
+import numpy as np
+from petal_qc.thermal import IRPetal
+from petal_qc.thermal import contours
+from petal_qc.thermal.PetalColorMaps import HighContrast
+
+
+class AnalysisResult(object):
+    """Contains results of IR image analysis."""
+
+    def __init__(self) -> None:
+        """Initialization."""
+        self.path_length = []  # position within pipe (0 U-bend at bottom)
+        self.path_temp = []    # temperatrure at given positions
+        self.path_spread = []  # spread at given positions
+        self.sensor_avg = []   # average on sensors
+        self.sensor_std = []   # spread on sensors
+
+    def from_json(self, J):
+        """Get values from JSon."""
+        for key in self.__dict__.keys():
+            try:
+                if hasattr(self, key):
+                    setattr(self, key, np.array(J[key]))
+
+            except Exception:
+                print("AnalysisResult.from_json: Ignoring {} component".format(key))
+                continue
+
+
+def show_2D_image(img, title=None):
+    """Show a 2D image."""
+    if isinstance(img, list):
+        fig, ax = plt.subplots(1, len(img))
+        if title:
+            fig.suptitle(title)
+
+        if len(img) == 1:
+            ax = [ax, ]
+
+        for i, im in enumerate(img):
+            pcm = ax[i].imshow(im, origin='lower', cmap=HighContrast.reversed())  # cmap="jet")
+            fig.colorbar(pcm, ax=ax[i])
+
+    else:
+        fig, ax = plt.subplots(1, 1)
+        if title:
+            fig.suptitle(title)
+        pcm = ax.imshow(img, origin='lower', cmap=HighContrast.reversed())  # cmap="jet")
+        fig.colorbar(pcm, ax=ax)
+
+    plt.draw()
+    plt.pause(0.001)
+
+    return fig, ax
+
+
+def analyze_IR_image(img, pipe, sensors, iside, params) -> AnalysisResult:
+    """Analyze the IR image at input.
+
+    Args:
+    ----
+        img (ndarray): THe image
+        pipes: The pipe
+        sensors: The sensor areas
+        iside: petal image with EoS on the left(0) or right (1)
+        params: IRPetal parameters
+
+    Returns
+    -------
+        An AnalysisResult object.
+
+    """
+    sensor_order = [
+        [0, 1, 2, 4, 3, 6, 5, 9, 8, 7],
+        [0, 1, 2, 3, 4, 5, 6, 8, 9, 7]
+    ]
+    width = 25
+    if img.shape[0] > 700:
+        width = 2*width
+
+    result = AnalysisResult()
+    result.path_temp, result.path_length = IRPetal.get_T_along_path(pipe, img, params.width, True)
+    result.path_spread, pL = IRPetal.get_spread_along_path(pipe, img, width, True)
+    Savg = []
+    Sstd = []
+    for S in sensors:
+        avg, std = contours.get_average_in_contour(img, S)
+        Savg.append(avg)
+        Sstd.append(std)
+
+    result.sensor_avg = [Savg[sensor_order[iside][j]] for j in range(10)]
+    result.sensor_std = [Sstd[sensor_order[iside][j]] for j in range(10)]
+
+    return result
+
+
+def analyze_IR_image_list(imgs, pipes, sensors, params):
+    """Analyze a list of IR images at input.
+
+    Args:
+    ----
+        img (list(IRBImage)): THe images. Assumed to be type0, type1, type0, type1...
+        pipes: The pipes
+        sensors: The sensor areas
+        params: IRPetal parameters
+
+    Returns
+    -------
+        list of AnalysisResult objects
+
+    """
+    out = []
+    for i, img in enumerate(imgs):
+        j = i % 2
+        res = analyze_IR_image(img, pipes[j], sensors[j], j, params)
+        out.append(res)
+
+    return out
+
+
+def analyze_IR_mirrored_image(img, pipes, sensors, params):
+    """Analyze the mirrored IR image at input.
+
+    Args:
+    ----
+        img (IRBImage): THe image
+        pipes: The pipes
+        sensors: The sensor areas
+        params: IRPetal parameters
+
+    Returns
+    -------
+        list of AnalysisResult objects
+        images: the image data
+
+    """
+    images = IRPetal.get_mirrored_petal_images(img, params)
+    results = analyze_IR_image_list(images, pipes, sensors, params)
+    return results, images