Add functions to get 3D krypton maps

invisible_cities/icaros/correction_functions.py

@@ -0,0 +1,53 @@
+import pandas as pd
+import numpy  as np
+from scipy.interpolate import griddata
+
+
+def normalization(krmap, method, xy_params = None):
+
+    mu_values = krmap.mu.dropna()
+
+    if method == 'max':
+        E_reference_max = mu_values.max()
+        return E_reference_max
+
+    if method == 'mean chamber':
+        E_reference_chamber = mu_values.mean()
+        return E_reference_chamber
+
+    if method == 'mean anode':
+        mu_values_anode = krmap[krmap.k == 0].mu
+        E_reference_anode = mu_values_anode.mean()
+        return E_reference_anode
+
+    mask_region = (krmap['x'] <= xy_params['x_high']) & (krmap['x'] >= xy_params['x_low']) & (krmap['y'] <= xy_params['y_high']) & (krmap['y'] >= xy_params['y_low'])
+
+    if method == 'mean region anode':
+        region = krmap[krmap.k == 0][mask_region]
+        E_reference_slice_anode = region.mu.mean()
+        return E_reference_slice_anode
+
+    if method == 'mean region chamber':
+        region = krmap[mask_region]
+        E_reference_region = region.mu.mean()
+        return E_reference_region
+
+
+def apply_3Dmap(krmap, norm_method, dt, x, y, E, xy_params = None, keV = False):
+
+    map_points = krmap['dt x y'.split()].values
+    norm = normalization(krmap, norm_method, xy_params)
+
+    data_points = np.stack([dt, x, y], axis = 1)
+    E_interpolated_data = griddata(map_points, krmap.mu.values, data_points, method = 'nearest')
+
+    correction_factor = norm/E_interpolated_data
+    Ec = E * correction_factor
+
+    if keV:
+        Ec = Ec * (41.55 / norm)
+
+    return Ec
+
+
+

invisible_cities/icaros/correction_functions_test.py

@@ -0,0 +1,43 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
+from correction_functions import normalization
+
+
+def test_method_norm():
+
+    k_vals = np.arange(10)
+    i_vals = np.arange(12)
+    j_vals = np.arange(12)
+
+    k, i, j = np.meshgrid(k_vals, i_vals, j_vals, indexing='ij')
+    k = k.ravel()
+    i = i.ravel()
+    j = j.ravel()
+
+    x = i * 25
+    y = j * 25
+    dt = k * 45
+
+    mu = (k*i + 800)*j
+
+    map_test = pd.DataFrame({
+        'k': k,
+        'i': i,
+        'j': j,
+        'x': x,
+        'y': y,
+        'dt': dt,
+        'mu': mu
+    })
+
+    region_mask = (
+        (map_test.x <= 100) & (map_test.x >= -100) &
+        (map_test.y <= 100) & (map_test.y >= -100)
+    )
+
+    assert normalization(map_test, 'max', None, None, None, None) == 9889
+    assert normalization(map_test, 'mean chamber', None, None, None, None) == 4536.125
+    assert normalization(map_test, 'mean anode', None, None, None, None) == 4400
+    assert normalization(map_test, 'mean region anode', -100, 100, -100, 100) == map_test.loc[(map_test.k == 0) & region_mask,'mu'].mean()
+    assert normalization(map_test, 'mean region chamber', -100, 100, -100, 100) == map_test.loc[region_mask,'mu'].mean()