From f123674d0218efea2042b1877d31c0a00ea8a892 Mon Sep 17 00:00:00 2001
From: ViaFerrata <michimoser@onlinehome.de>
Date: Mon, 14 Jan 2019 17:36:11 +0100
Subject: [PATCH] Add small tool to count number of events in a folder.
---
utilities/count_number_of_events_in_folder.py | 84 +++++++++++++++++++
1 file changed, 84 insertions(+)
create mode 100644 utilities/count_number_of_events_in_folder.py
diff --git a/utilities/count_number_of_events_in_folder.py b/utilities/count_number_of_events_in_folder.py
new file mode 100644
index 0000000..b4202e0
--- /dev/null
+++ b/utilities/count_number_of_events_in_folder.py
@@ -0,0 +1,84 @@
+
+import os
+import numpy as np
+import h5py
+
+
+def count_number_of_events_in_folder(dirpath):
+
+ fpath_list = []
+ for file in os.listdir(dirpath):
+ if file.endswith('.h5'):
+ fpath = os.path.join(dirpath, file)
+ fpath_list.append(fpath)
+
+ n_total = 0
+ for fpath in fpath_list:
+ f = h5py.File(fpath, 'r')
+ n_total += f['event_info'].shape[0]
+ f.close()
+
+ return n_total
+
+
+
+def main():
+ dirpath_rn = '/home/saturn/capn/mppi033h/Data/raw_data/random_noise'
+ rn_n_total = count_number_of_events_in_folder(dirpath_rn)
+ print('--------------------------------------------------------')
+ print('Total number of random_noise events: ' + str(rn_n_total))
+ print('--------------------------------------------------------')
+
+ dirpath_mupage = '/home/saturn/capn/mppi033h/Data/raw_data/mupage'
+ mupage_n_total = count_number_of_events_in_folder(dirpath_mupage)
+ print('--------------------------------------------------------')
+ print('Total number of mupage events: ' + str(mupage_n_total))
+ print('--------------------------------------------------------')
+
+ dirpath_muon_cc_low_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/1-5GeV/muon-CC'
+ dirpath_muon_cc_high_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/3-100GeV/muon-CC'
+ muon_cc_low_e_n_total = count_number_of_events_in_folder(dirpath_muon_cc_low_e)
+ muon_cc_high_e_n_total = count_number_of_events_in_folder(dirpath_muon_cc_high_e)
+ print('--------------------------------------------------------')
+ print('Total number of muon-CC 1-5GeV events: ' + str(muon_cc_low_e_n_total))
+ print('Total number of muon-CC 3-100GeV events: ' + str(muon_cc_high_e_n_total))
+ print('--------------------------------------------------------')
+
+ dirpath_elec_cc_low_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/1-5GeV/elec-CC'
+ dirpath_elec_cc_high_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/3-100GeV/elec-CC'
+ elec_cc_low_e_n_total = count_number_of_events_in_folder(dirpath_elec_cc_low_e)
+ elec_cc_high_e_n_total = count_number_of_events_in_folder(dirpath_elec_cc_high_e)
+ print('--------------------------------------------------------')
+ print('Total number of elec-CC 1-5GeV events: ' + str(elec_cc_low_e_n_total))
+ print('Total number of elec-CC 3-100GeV events: ' + str(elec_cc_high_e_n_total))
+ print('--------------------------------------------------------')
+
+ dirpath_elec_nc_low_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/1-5GeV/elec-NC'
+ dirpath_elec_nc_high_e = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/3-100GeV/elec-NC'
+ elec_nc_low_e_n_total = count_number_of_events_in_folder(dirpath_elec_nc_low_e)
+ elec_nc_high_e_n_total = count_number_of_events_in_folder(dirpath_elec_nc_high_e)
+ print('--------------------------------------------------------')
+ print('Total number of elec-NC 1-5GeV events: ' + str(elec_nc_low_e_n_total))
+ print('Total number of elec-NC 3-100GeV events: ' + str(elec_nc_high_e_n_total))
+ print('--------------------------------------------------------')
+
+ dirpath_tau_cc = '/home/saturn/capn/mppi033h/Data/raw_data/ORCA_JTE_NEMOWATER/calibrated/with_jte_times/3-100GeV/tau-CC'
+ tau_cc_n_total = count_number_of_events_in_folder(dirpath_tau_cc)
+ print('--------------------------------------------------------')
+ print('Total number of tau-CC 3-100GeV events: ' + str(tau_cc_n_total))
+ print('--------------------------------------------------------')
+
+
+ n_total_neutr = muon_cc_low_e_n_total + muon_cc_high_e_n_total + elec_cc_low_e_n_total + elec_cc_high_e_n_total\
+ + elec_nc_low_e_n_total + elec_nc_high_e_n_total + tau_cc_n_total
+ print('--------------------------------------------------------')
+ print('Total number of neutrino events: ' + str(n_total_neutr))
+ print('--------------------------------------------------------')
+
+
+
+
+
+
+if __name__ == '__main__':
+ main()
--
GitLab