From e1b2f65419f0e024994daeaf86fc0b1ee9c3f580 Mon Sep 17 00:00:00 2001
From: Rodri <rgruiz0001@gmail.com>
Date: Tue, 8 Feb 2022 17:06:52 +0100
Subject: [PATCH] blackened
---
km3io/utils/extract_dom_rates.py | 29 ++++++++++++++++++++---------
setup.py | 7 ++++++-
2 files changed, 26 insertions(+), 10 deletions(-)
diff --git a/km3io/utils/extract_dom_rates.py b/km3io/utils/extract_dom_rates.py
index 1a0084b..550fe8e 100644
--- a/km3io/utils/extract_dom_rates.py
+++ b/km3io/utils/extract_dom_rates.py
@@ -15,21 +15,25 @@ import os
import numpy as np
import h5py
+
def get_doms_rates(slices, frame):
- raw_rates = [km3io.online.get_rate(getattr(slices.rates[frame],f"ch{i}")) for i in range(31)]
+ raw_rates = [
+ km3io.online.get_rate(getattr(slices.rates[frame], f"ch{i}")) for i in range(31)
+ ]
return np.sum(raw_rates, axis=0)
+
def main():
arguments = docopt(__doc__)
data = {}
for key in arguments:
- data[key.replace('-','')] = arguments[key]
+ data[key.replace("-", "")] = arguments[key]
- reader = km3io.OnlineReader(data['input_file'])
+ reader = km3io.OnlineReader(data["input_file"])
summary_slices = reader.summaryslices
- detector = km3pipe.hardware.Detector(data['detector_file'])
+ detector = km3pipe.hardware.Detector(data["detector_file"])
n_slices = len(summary_slices.slices)
@@ -37,18 +41,25 @@ def main():
for dom in detector.dom_ids:
doms_rates[dom] = np.zeros(n_slices)
- for i in range (n_slices):
+ for i in range(n_slices):
rates = get_doms_rates(summary_slices, i)
dom_ids = summary_slices.slices[i].dom_id
- for j in range(len(dom_ids)-1):
+ for j in range(len(dom_ids) - 1):
doms_rates[dom_ids[j]][i] = rates[j]
- f = h5py.File(data['output_file'], "a")
- f.create_dataset("frame_indices",data=np.array(summary_slices.headers.frame_index))
- f.create_dataset("frame_times" ,data=np.array(summary_slices.headers.UTC_seconds * 1e9 + summary_slices.headers.UTC_16nanosecondcycles * 16))
+ f = h5py.File(data["output_file"], "a")
+ f.create_dataset("frame_indices", data=np.array(summary_slices.headers.frame_index))
+ f.create_dataset(
+ "frame_times",
+ data=np.array(
+ summary_slices.headers.UTC_seconds * 1e9
+ + summary_slices.headers.UTC_16nanosecondcycles * 16
+ ),
+ )
for key, value in doms_rates.items():
f.create_dataset(str(key), data=value)
+
if __name__ == "__main__":
main()
diff --git a/setup.py b/setup.py
index 561548a..8e84f1e 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,12 @@ setup(
install_requires=read_requirements("install"),
extras_require={kind: read_requirements(kind) for kind in ["dev"]},
python_requires=">=3.6",
- entry_points={"console_scripts": ["KPrintTree=km3io.utils.kprinttree:main","ExtractDomRates=km3io.utils.ExtractDomRates:main"]},
+ entry_points={
+ "console_scripts": [
+ "KPrintTree=km3io.utils.kprinttree:main",
+ "ExtractDomRates=km3io.utils.extract_dom_rates:main",
+ ]
+ },
classifiers=[
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
--
GitLab