Copy db info in dorothea and sophronia

next-exp · Feb 18, 2025 · 658d8bb · 658d8bb
1 parent b729768
commit 658d8bb
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/invisible_cities/cities/components.py b/invisible_cities/cities/components.py
@@ -1503,3 +1503,9 @@ def mask_sensors(rwf_pmt, rwf_sipm):
                , rwf_sipm[active_sipms])
 
     return mask_sensors
+
+
+def copy_db_info(file_in, file_out):
+    with tb.open_file(file_in) as file:
+        if "DB" in file.root:
+            file.copy_node(file.root.DB, file_out.root, recursive=True)
diff --git a/invisible_cities/cities/dorothea.py b/invisible_cities/cities/dorothea.py
@@ -55,6 +55,7 @@
 from .  components import city
 from .  components import collect
 from .  components import copy_mc_info
+from .  components import copy_db_info
 from .  components import print_every
 from .  components import pmap_from_files
 from .  components import peak_classifier
@@ -143,4 +144,5 @@ def dorothea( files_in         :  OneOrManyFiles
             copy_mc_info(files_in, h5out, result.evtnum_list,
                          detector_db, run_number)
 
+        copy_db_info(files_in[0], h5out)
         return result
diff --git a/invisible_cities/cities/sophronia.py b/invisible_cities/cities/sophronia.py
@@ -55,6 +55,7 @@
 
 from .  components import                  city
 from .  components import          copy_mc_info
+from .  components import          copy_db_info
 from .  components import           print_every
 from .  components import       peak_classifier
 from .  components import   compute_xy_position
@@ -187,4 +188,5 @@ def sophronia( files_in           : OneOrManyFiles
             copy_mc_info(files_in, h5out, result.evtnum_list,
                          detector_db, run_number)
 
+        copy_db_info(files_in[0], h5out)
         return result