Register truth plugins in the context

XENONnT · Feb 1, 2024 · 02520a1 · 02520a1
1 parent 2711b37
commit 02520a1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fuse/context.py b/fuse/context.py
@@ -36,6 +36,11 @@
                        fuse.pmt_and_daq.PMTResponseAndDAQ,
                        ]
 
+#Plugins to get truth information
+truth_information_plugins = [fuse.truth_information.PeakTruth,
+                             fuse.truth_information.EventTruth,
+                             fuse.truth_information.SurvivingClusters,]
+
 def microphysics_context(output_folder = "./fuse_data"
                          ):
     """
@@ -92,6 +97,10 @@ def full_chain_context(output_folder = "./fuse_data",
     for plugin in pmt_and_daq_plugins:
         st.register(plugin)
 
+    #Register truth plugins
+    for plugin in truth_information_plugins:
+        st.register(plugin)
+
     if corrections_version is not None:
         st.apply_xedocs_configs(version=corrections_version)