Register pema (#71)

* auto register pema * empty line cleanujp
XENONnT · Dec 18, 2023 · 8ec17b8 · 8ec17b8
1 parent 3ee593c
commit 8ec17b8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/saltax/contexts.py b/saltax/contexts.py
@@ -3,6 +3,7 @@
 import cutax
 import strax
 from immutabledict import immutabledict
+import pema
 
 # straxen XENONnT options/configuration
 XNT_COMMON_OPTS = straxen.contexts.xnt_common_opts.copy()
@@ -180,6 +181,9 @@ def xenonnt_salted(runid,
     st.set_config(dict(fax_file=instr_file_name))
     st.set_config(dict(saltax_mode=saltax_mode))
 
+    # Register pema plugins
+    st.register_all(pema.match_plugins)
+
     return st
 
 def sxenonnt(runid=None,

diff --git a/saltax/match/visual.py b/saltax/match/visual.py
@@ -188,4 +188,5 @@ def plot_wf(ind, st_salt, st_simu, st_data, runid, matched_simu,
                       matched_simu_s2_timerange_i[1]+s2_ext_window_samples)
     axs[1,1].set_xlabel("Time [10ns]")
     axs[1,1].legend()
-    fig.show()
+    fig.show()
+
diff --git a/saltax/plugins/s_peaklets.py b/saltax/plugins/s_peaklets.py
@@ -876,4 +876,3 @@ def sum_waveform_salted(peaks, hits, records, record_links, adc_to_pe, n_top_cha
 
         p['n_saturated_channels'] = p['saturated_channel'].sum()
         p['area_per_channel'][:] = area_per_channel
-