reboost 0.3.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

reboost/optmap/cli.py

@@ -188,7 +188,7 @@ def optical_cli() -> None:
 
     # STEP 1: build evt file from hit tier
     if args.command == "evt":
-        from reboost.optmap.evt import build_optmap_evt
+        from .evt import build_optmap_evt
 
         _check_input_file(parser, args.detectors)
         _check_input_file(parser, args.input)
@@ -202,7 +202,7 @@ def optical_cli() -> None:
 
     # STEP 2a: build map file from evt tier
     if args.command == "createmap":
-        from reboost.optmap.create import create_optical_maps
+        from .create import create_optical_maps
 
         _check_input_file(parser, args.input)
         _check_output_file(parser, args.output)
@@ -230,7 +230,7 @@ def optical_cli() -> None:
 
     # STEP 2b: view maps
     if args.command == "viewmap":
-        from reboost.optmap.mapview import view_optmap
+        from .mapview import view_optmap
 
         _check_input_file(parser, args.input)
         if args.divide is not None:
@@ -247,7 +247,7 @@ def optical_cli() -> None:
 
     # STEP 2c: merge maps
     if args.command == "mergemap":
-        from reboost.optmap.create import merge_optical_maps
+        from .create import merge_optical_maps
 
         # load settings for binning from config file.
         _check_input_file(parser, args.input, "settings")
@@ -262,14 +262,14 @@ def optical_cli() -> None:
 
     # STEP 2d: check maps
     if args.command == "checkmap":
-        from reboost.optmap.create import check_optical_map
+        from .create import check_optical_map
 
         _check_input_file(parser, args.input)
         check_optical_map(args.input)
 
     # STEP 3: convolve with hits from non-optical simulations
     if args.command == "convolve":
-        from reboost.optmap.convolve import convolve
+        from .convolve import convolve
 
         _check_input_file(parser, [args.map, args.edep])
         _check_output_file(parser, args.output, optional=True)
@@ -285,7 +285,7 @@ def optical_cli() -> None:
 
     # STEP X: rebin maps
     if args.command == "rebin":
-        from reboost.optmap.create import rebin_optical_maps
+        from .create import rebin_optical_maps
 
         _check_input_file(parser, args.input)
         _check_output_file(parser, args.output)

reboost/shape/group.py

@@ -122,7 +122,7 @@ def group_by_time(
 
     # get difference
     time_diffs = np.diff(obj[time_name])
-    index_diffs = np.array(np.diff(obj[evtid_name]), dtype=np.int32)
+    index_diffs = np.diff(obj[evtid_name])
 
     # index of the last element in each run
     time_change = (time_diffs > window * 1000) & (index_diffs == 0)

reboost/utils.py

@@ -4,12 +4,16 @@ import importlib
 import itertools
 import logging
 import re
+import time
 from collections.abc import Iterable, Mapping
 from contextlib import contextmanager
 from pathlib import Path
 
 from dbetto import AttrsDict
-from lgdo.types import Table, VectorOfVectors
+from lgdo import lh5
+from lgdo.types import Struct, Table, VectorOfVectors
+
+from .profile import ProfileDict
 
 log = logging.getLogger(__name__)
 
@@ -304,3 +308,49 @@ def _check_output_file(parser, file: str | Iterable[str] | None, optional: bool
     for f in file:
         if Path(f).exists():
             parser.error(f"output file {f} already exists")
+
+
+def write_lh5(
+    hit_table: Table,
+    file: str,
+    time_dict: ProfileDict,
+    out_field: str,
+    out_detector: str,
+    wo_mode: str,
+):
+    """Write the lh5 file. This function handles writing first the data as a struct and then appending to this.
+
+    Parameters
+    ----------
+    hit_table
+        the table to write
+    file
+        the file to write to
+    time_dict
+        the dictionary of timing information to update.
+    out_field
+        output field
+    out_detector
+        output detector name
+    wo_mode
+        the mode to pass to `lh5.write`
+    """
+    if time_dict is not None:
+        start_time = time.time()
+
+    if wo_mode != "a":
+        lh5.write(
+            Struct({out_detector: hit_table}),
+            out_field,
+            file,
+            wo_mode=wo_mode,
+        )
+    else:
+        lh5.write(
+            hit_table,
+            f"{out_field}/{out_detector}",
+            file,
+            wo_mode=wo_mode,
+        )
+    if time_dict is not None:
+        time_dict.update_field("write", start_time)

{reboost-0.3.0.dist-info → reboost-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reboost
-Version: 0.3.0
+Version: 0.4.2
 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

{reboost-0.3.0.dist-info → reboost-0.4.2.dist-info}/RECORD

@@ -1,25 +1,25 @@
-reboost/__init__.py,sha256=RVNl3Qgx_hTUeBGXaWYmiTcmXUDhTfvlAGGC8bo_jP8,316
-reboost/_version.py,sha256=AGmG_Lx0-9ztFw_7d9mYbaYuC-2abxE1oXOUNAY29YY,511
+reboost/__init__.py,sha256=2Lc9Idexf1S3IQvHQS1LPD-SCud7Mnmpr2dJ6U-ThvU,303
+reboost/_version.py,sha256=_F8vLxUxrAtC2alXNPGVa9l3P6_vLpQAzemS6QlnPGQ,511
 reboost/build_evt.py,sha256=zj3wG_kaV3EoRMQ33AkCNa_2Fv8cLtRuhyRyRmSrOYQ,4797
-reboost/build_glm.py,sha256=VIzRyCc53FQvcXAVbARszNH5wH5Pr0a9WWlSCZuUY5w,9380
-reboost/build_hit.py,sha256=Zu8WoeFYtFU_xUO3I8OZM5psNaiv1boMJPnBWC58nfQ,14958
-reboost/build_tcm.py,sha256=-PawBHoHj0zsm4XsZu5bco9d9a09STicZchduefSNfI,2951
-reboost/cli.py,sha256=swPJcYzvg18rSOMN-mpe0PCMf1-a9V7osIssX7JP7k0,6459
-reboost/core.py,sha256=7Nclc6RUCOSJ1CWVAX0rFNJGM1LEgqvc4tD04CxEAtg,10766
-reboost/iterator.py,sha256=D0aDCwtUKeo3_JQQKrmYmFc7jbqcKb5jrpMkUuaJ76s,4762
+reboost/build_glm.py,sha256=zRTXTmlvHePcFDJkJwBbIo5nMjW0ZP4-2CJvqN2QBe8,9525
+reboost/build_hit.py,sha256=tf7KBp8DYuAmT-k6oBF54rjlIu_y9tVuyuGldA2I0Jk,15247
+reboost/build_tcm.py,sha256=Ntf2nrS072SGlZzdTHETtaQdhl0tL3ikvdzrOQvlb-E,2944
+reboost/cli.py,sha256=YZvCaxkU0VA_xMAeyw1PN_x17RUkl8uzv19nkNkz3kE,6509
+reboost/core.py,sha256=B577b5KzAYpGI7c4eCxIRKgt9tq6hwVr2-DtYTml1e8,12826
+reboost/iterator.py,sha256=YFsNjC2Onsyhvkogbs79eD8iqstCdCAfFEaEOMr_ujg,6810
 reboost/log_utils.py,sha256=VqS_9OC5NeNU3jcowVOBB0NJ6ssYvNWnirEY-JVduEA,766
 reboost/profile.py,sha256=EOTmjmS8Rm_nYgBWNh6Rntl2XDsxdyed7yEdWtsZEeg,2598
 reboost/units.py,sha256=3EH8XlpbsObdu5vLgxhm1600L6UNYD5jng4SjJT_1QE,2202
-reboost/utils.py,sha256=VheyCMcaXT5VPXAotiI5YqLKTf45KhTMpwULg2xyB3o,8531
+reboost/utils.py,sha256=CL83IBkddFXdzD83JXGQeZ_N6WxJ8qwj7GgBt1vzmbg,9652
 reboost/hpge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reboost/hpge/psd.py,sha256=868OUJzO9TNja0YSrZ3NDGeEAbUtpDZnmvBDm0jCC9E,6856
-reboost/hpge/surface.py,sha256=SZyTmOCTipf27jYaJhtdInzGF1RZ2wKpbtf6HlOQYwM,3662
+reboost/hpge/psd.py,sha256=jAUAoQ_PMz76wyA1NXYHNKtOwoCnRT3My8_LCFrKi-U,13860
+reboost/hpge/surface.py,sha256=lbWcFnFFWKxtFKs755GyM9US_IfyxaoM6MpOIZgIMM0,7478
 reboost/hpge/utils.py,sha256=0Rx4HubCOm8JMECjWcAJXfAch9OkSlRpUkdsSlzwZ2E,2830
 reboost/math/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reboost/math/functions.py,sha256=OymiYTcA0NXxxm-MBDw5kqyNwHoLCmuv4J48AwnSrbU,5633
 reboost/math/stats.py,sha256=iiOEi87x93kqPWeSmlRiA5Oe-R8XR-plm6Z532PhC9M,1401
 reboost/optmap/__init__.py,sha256=imvuyld-GLw8qdwqW-lXCg2feptcTyQo3wIzPvDHwmY,93
-reboost/optmap/cli.py,sha256=TszAYZIwHTXwJYQOMKxreCZ6pXlGUb6q6YZ3iW3Bxf4,9670
+reboost/optmap/cli.py,sha256=SzbPRgsbR5Llm3aSJubH02Ym8FQyTH7kvuLjK7faLiY,9572
 reboost/optmap/convolve.py,sha256=jCH_d04yioB8hsJEPunm0zynA0ne4lx0ldSC1GJG_eY,14129
 reboost/optmap/create.py,sha256=Nm5-xEe8M9q2GFQnUv8oN8qpAz9nZArIrQcPboqRmCQ,17153
 reboost/optmap/evt.py,sha256=9rfAdN9MqL6UuUxUcMDGVwpcuqRVc2RwmEmd87jgYww,4698
@@ -28,11 +28,11 @@ reboost/optmap/numba_pdg.py,sha256=y8cXR5PWE2Liprp4ou7vl9do76dl84vXU52ZJD9_I7A,7
 reboost/optmap/optmap.py,sha256=j4rfbQ84PYSpE-BvP4Rdt96ZjPdwy8P4e4eZz1mATys,12817
 reboost/shape/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reboost/shape/cluster.py,sha256=RIvBlhHzp88aaUZGofp5SD9bimnoiqIOddhQ84jiwoM,8135
-reboost/shape/group.py,sha256=Q3DhEPxbhw3p4bwvpswSd0A-p224l5vRZnfQIEkOVJE,4475
+reboost/shape/group.py,sha256=_z2qCOret3E-kj-nrp1-J5j2lEwQpgfYdQp2pgpDHR8,4449
 reboost/shape/reduction.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reboost-0.3.0.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-reboost-0.3.0.dist-info/METADATA,sha256=GCn2W1jn_cplxbS1-bJCRjghoEaPgscoTuAhVkizXVY,44251
-reboost-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-reboost-0.3.0.dist-info/entry_points.txt,sha256=DxhD6BidSWNot9BrejHJjQ7RRLmrMaBIl52T75oWTwM,93
-reboost-0.3.0.dist-info/top_level.txt,sha256=q-IBsDepaY_AbzbRmQoW8EZrITXRVawVnNrB-_zyXZs,8
-reboost-0.3.0.dist-info/RECORD,,
+reboost-0.4.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+reboost-0.4.2.dist-info/METADATA,sha256=TnY5hEIWwoDYIwUEvTW5UeAKpI9t3hU5TCFzl1j8cQM,44251
+reboost-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+reboost-0.4.2.dist-info/entry_points.txt,sha256=DxhD6BidSWNot9BrejHJjQ7RRLmrMaBIl52T75oWTwM,93
+reboost-0.4.2.dist-info/top_level.txt,sha256=q-IBsDepaY_AbzbRmQoW8EZrITXRVawVnNrB-_zyXZs,8
+reboost-0.4.2.dist-info/RECORD,,