reboost 0.8.3__tar.gz → 0.8.4__tar.gz

{reboost-0.8.3/src/reboost.egg-info → reboost-0.8.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reboost
-Version: 0.8.3
+Version: 0.8.4
 Summary: New LEGEND Monte-Carlo simulation post-processing
 Author-email: Manuel Huber <info@manuelhu.de>, Toby Dixon <toby.dixon.23@ucl.ac.uk>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration
@@ -28,7 +28,7 @@ Requires-Dist: scipy
 Requires-Dist: numba>=0.60
 Requires-Dist: legend-pydataobj>=1.15.1
 Requires-Dist: legend-pygeom-optics>=0.15.0
-Requires-Dist: legend-pygeom-tools>=0.0.25
+Requires-Dist: legend-pygeom-tools>=0.0.26
 Requires-Dist: legend-pygeom-hpges
 Requires-Dist: hist
 Requires-Dist: dbetto

{reboost-0.8.3 → reboost-0.8.4}/pyproject.toml

@@ -39,7 +39,7 @@ dependencies = [
     "numba >=0.60",
     "legend-pydataobj >=1.15.1",
     "legend-pygeom-optics >=0.15.0",
-    "legend-pygeom-tools >=0.0.25",
+    "legend-pygeom-tools >=0.0.26",
     "legend-pygeom-hpges",
     "hist",
     "dbetto",

reboost-0.8.4/src/reboost/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from .cli import cli
+
+if __name__ == "__main__":
+    cli()

{reboost-0.8.3 → reboost-0.8.4}/src/reboost/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.8.3'
-__version_tuple__ = version_tuple = (0, 8, 3)
+__version__ = version = '0.8.4'
+__version_tuple__ = version_tuple = (0, 8, 4)
 
-__commit_id__ = commit_id = 'g2085bdbed'
+__commit_id__ = commit_id = 'g974fc1b03'

reboost-0.8.4/src/reboost/optmap/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from .cli import optical_cli
+
+if __name__ == "__main__":
+    optical_cli()

{reboost-0.8.3 → reboost-0.8.4}/src/reboost/optmap/convolve.py

@@ -32,7 +32,7 @@ class OptmapForConvolve(NamedTuple):
 
 def open_optmap(optmap_fn: str) -> OptmapForConvolve:
     dets = lh5.ls(optmap_fn, "/channels/")
-    detidx = np.arange(0, dets.shape[0])
+    detidx = np.arange(0, len(dets))
 
     optmap_all = lh5.read("/all/prob", optmap_fn)
     assert isinstance(optmap_all, Histogram)
@@ -61,14 +61,14 @@ def open_optmap(optmap_fn: str) -> OptmapForConvolve:
             raise ValueError(msg)
     else:
         detidx = np.array([OPTMAP_ANY_CH])
-        dets = np.array(["all"])
+        dets = ["all"]
 
     # check the exponent from the optical map file
     if "_hitcounts_exp" in lh5.ls(optmap_fn):
         msg = "found _hitcounts_exp which is not supported any more"
         raise RuntimeError(msg)
 
-    dets = [d.replace("/channels/", "") for d in dets]
+    dets = np.array([d.replace("/channels/", "") for d in dets])
 
     return OptmapForConvolve(dets, detidx, optmap_edges, ow)
 

{reboost-0.8.3 → reboost-0.8.4}/src/reboost/optmap/evt.py

@@ -131,9 +131,9 @@ def get_optical_detectors_from_geom(geom_fn) -> dict[int, str]:
     import pygeomtools
 
     geom_registry = pyg4ometry.gdml.Reader(geom_fn).getRegistry()
-    detectors = pygeomtools.get_all_sensvols(geom_registry)
+    detectors = pygeomtools.get_all_sensvols(geom_registry, type_filter="optical")
     return OrderedDict(
-        [(d.uid, name) for name, d in detectors.items() if d.detector_type == "optical"]
+        [(d.uid, d.ntuple_name if d.ntuple_name else name) for name, d in detectors.items()]
     )
 
 

{reboost-0.8.3 → reboost-0.8.4}/src/reboost/optmap/mapview.py

@@ -95,7 +95,7 @@ def _read_data(
     histogram_choice: str = "prob",
 ) -> tuple[tuple[NDArray], NDArray]:
     histogram = histogram_choice if histogram_choice != "prob_unc_rel" else "prob"
-    detid = f"channels/{detid}" if detid != all and not detid.startswith("channels/") else detid
+    detid = f"channels/{detid}" if detid != "all" and not detid.startswith("channels/") else detid
 
     optmap_all = lh5.read(f"/{detid}/{histogram}", optmap_fn)
     optmap_edges = tuple([b.edges for b in optmap_all.binning])

reboost-0.8.4/src/reboost/spms/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+from .pe import detected_photoelectrons, emitted_scintillation_photons, load_optmap, load_optmap_all
+
+__all__ = [
+    "detected_photoelectrons",
+    "emitted_scintillation_photons",
+    "load_optmap",
+    "load_optmap_all",
+]

{reboost-0.8.3 → reboost-0.8.4}/src/reboost/spms/pe.py

@@ -140,8 +140,8 @@ def detected_photoelectrons(
     """
     hits = ak.Array(
         {
-            "num_scint_ph": num_scint_ph,
-            "particle": particle,
+            "num_scint_ph": units_conv_ak(num_scint_ph, "dimensionless"),
+            "particle": units_conv_ak(particle, "dimensionless"),
             "time": units_conv_ak(time, "ns"),
             "xloc": units_conv_ak(xloc, "m"),
             "yloc": units_conv_ak(yloc, "m"),
@@ -171,7 +171,12 @@ def emitted_scintillation_photons(
     material
         scintillating material name.
     """
-    hits = ak.Array({"edep": units_conv_ak(edep, "keV"), "particle": particle})
+    hits = ak.Array(
+        {
+            "edep": units_conv_ak(edep, "keV"),
+            "particle": units_conv_ak(particle, "dimensionless"),
+        }
+    )
 
     scint_mat_params = convolve._get_scint_params(material)
     ph = convolve.iterate_stepwise_depositions_scintillate(hits, scint_mat_params)

{reboost-0.8.3 → reboost-0.8.4/src/reboost.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reboost
-Version: 0.8.3
+Version: 0.8.4
 Summary: New LEGEND Monte-Carlo simulation post-processing
 Author-email: Manuel Huber <info@manuelhu.de>, Toby Dixon <toby.dixon.23@ucl.ac.uk>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration
@@ -28,7 +28,7 @@ Requires-Dist: scipy
 Requires-Dist: numba>=0.60
 Requires-Dist: legend-pydataobj>=1.15.1
 Requires-Dist: legend-pygeom-optics>=0.15.0
-Requires-Dist: legend-pygeom-tools>=0.0.25
+Requires-Dist: legend-pygeom-tools>=0.0.26
 Requires-Dist: legend-pygeom-hpges
 Requires-Dist: hist
 Requires-Dist: dbetto

{reboost-0.8.3 → reboost-0.8.4}/src/reboost.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 src/reboost/__init__.py
+src/reboost/__main__.py
 src/reboost/_version.py
 src/reboost/build_evt.py
 src/reboost/build_glm.py
@@ -31,6 +32,7 @@ src/reboost/math/__init__.py
 src/reboost/math/functions.py
 src/reboost/math/stats.py
 src/reboost/optmap/__init__.py
+src/reboost/optmap/__main__.py
 src/reboost/optmap/cli.py
 src/reboost/optmap/convolve.py
 src/reboost/optmap/create.py

{reboost-0.8.3 → reboost-0.8.4}/src/reboost.egg-info/requires.txt

@@ -5,7 +5,7 @@ scipy
 numba>=0.60
 legend-pydataobj>=1.15.1
 legend-pygeom-optics>=0.15.0
-legend-pygeom-tools>=0.0.25
+legend-pygeom-tools>=0.0.26
 legend-pygeom-hpges
 hist
 dbetto

{reboost-0.8.3 → reboost-0.8.4}/tests/hit/configs/spms.yaml

@@ -1,5 +1,6 @@
 objects:
   spms: "['S001', 'S002']"
+  optmap_test: reboost.spms.load_optmap_all(ARGS.optmap_path)
 
 processing_groups:
   - name: spms

{reboost-0.8.3 → reboost-0.8.4}/tests/hit/test_build_hit.py

@@ -333,6 +333,7 @@ def test_spms(test_gen_lh5_scint, tmptestdir):
     m.create_probability()
     m.write_lh5(map_file, "channels/S001", "overwrite_file")
     m.write_lh5(map_file, "channels/S002", "write_safe")
+    m.write_lh5(map_file, "all", "write_safe")
 
     outfile = f"{tmptestdir}/spms_hit.lh5"
     reboost.build_hit(

reboost-0.8.3/src/reboost/spms/__init__.py

@@ -1,5 +0,0 @@
-from __future__ import annotations
-
-from .pe import detected_photoelectrons, emitted_scintillation_photons, load_optmap
-
-__all__ = ["detected_photoelectrons", "emitted_scintillation_photons", "load_optmap"]