legend-pydataobj 1.2.1__py3-none-any.whl → 1.4.0__py3-none-any.whl

{legend_pydataobj-1.2.1.dist-info → legend_pydataobj-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: legend-pydataobj
-Version: 1.2.1
+Version: 1.4.0
 Summary: LEGEND Python Data Objects
 Home-page: https://github.com/legend-exp/legend-pydataobj
 Author: The LEGEND Collaboration

{legend_pydataobj-1.2.1.dist-info → legend_pydataobj-1.4.0.dist-info}/RECORD

@@ -1,28 +1,29 @@
 lgdo/__init__.py,sha256=mY6pUNy2yJ2MYzq_ZdhVZZ7xItBW1KJ8h9qA29bOECU,2878
-lgdo/_version.py,sha256=2U0Gn26fYI3Vgj5hgkLM8I3wI6YEVdffJGllaVW-sSc,411
+lgdo/_version.py,sha256=R8-T9fmURjcuoxYpHTAjyNAhgJPDtI2jogCjqYYkfCU,411
+lgdo/cli.py,sha256=6o2vGwEq0Fq1y67RTxOHjkVNmN9XGhIBnb8DFFm8ANQ,1428
 lgdo/lgdo_utils.py,sha256=LvqE_eQZjKOuLrocbxc21rvWvh7NA4BSiaJh-jhlxVs,5598
-lgdo/lh5_store.py,sha256=lXfrvZQtqQuzNlOba1pPU156FNBb1LrAXX1dddH2y1U,68404
+lgdo/lh5_store.py,sha256=-ceGURWdu0jKTsDTL9bqQLxQ0T8USeeSXAk2cqd-2UU,74246
 lgdo/logging.py,sha256=Nu3wgIoWN7cyUxuzPom5rMwFvTlBu8p8d9uONHDquRg,965
 lgdo/compression/__init__.py,sha256=oT9OXiDDxC7BZciWrQVfHZNkOxXfj4p8EpF2tF04w84,1091
-lgdo/compression/base.py,sha256=82cQJujfvoAOKBFx761dEcx_xM02TBCBBuBo6i78tuI,838
-lgdo/compression/generic.py,sha256=zWoZqR5H4x-xpbvBYKDZ0rnNqUh9pC9aZeoc8Z98OjI,2252
-lgdo/compression/radware.py,sha256=ZfdebK_N1r1t2YVqqBu5B6RN87OoN9_deeiR97mP0Zo,19312
+lgdo/compression/base.py,sha256=ujQY2kYF4z3ZdAy7gXaoDPXFbG2Av1IQ1Nnx6UGLjmk,896
+lgdo/compression/generic.py,sha256=CyOg4xyPG1lUuE4oJLU-jReKwQQo5Xn5Im3vgRVLpUY,2485
+lgdo/compression/radware.py,sha256=K39y5v1E7DymrP2GMYsVrmREMp8hqlNU66DXvQrrLVU,23843
 lgdo/compression/utils.py,sha256=YhxfJ_zvDu9I1BnjOsALR7SYbNohZNdm8HZ8IqTjd6w,1060
-lgdo/compression/varlen.py,sha256=SJtfRfP5T_YM4YsyuAXXZgsRtSps2_ERBE985_Eli3w,14497
+lgdo/compression/varlen.py,sha256=FwZTyqKpSXQEL9cN0g3NtfC7E7i7hejVn8vIfxqSN6A,14874
 lgdo/types/__init__.py,sha256=5V2V_e8zLb7wmOGwnkSeuQBN_0yPq6zFyo06gBHa3_c,736
 lgdo/types/array.py,sha256=Z0RyxW3iXCKlnppN97hLde6Yoj5OFL_teOg5IhCmz68,4489
 lgdo/types/arrayofequalsizedarrays.py,sha256=vq8kjt7ckztX65duRkcSZTaFxz8AtdIRIvs4D9wmcCU,4492
 lgdo/types/encoded.py,sha256=M1Z4dKZ4Dq5VkB03-jYmK_jDEfrVi8O0u2w4lZaWHA0,11302
 lgdo/types/fixedsizearray.py,sha256=zDhj9GXcwPBcnwhlh3RTrJIdATJVZ-5If39PUWBy9mw,1250
 lgdo/types/lgdo.py,sha256=COil-sFRsUWe9jXN5PdOVF3w0h6KNw-CVnzFW91kYXM,1515
-lgdo/types/scalar.py,sha256=UnSfVqXOJyAMwInccWXFWpXKRyjMlpfgbMTTGJm1EWE,1651
+lgdo/types/scalar.py,sha256=LulBUzGZ75p_WA9OWf_nB_sENMzfRdYJReUXp0ljYZ4,1657
 lgdo/types/struct.py,sha256=UxV0wnCHoQM5rSmzEC9EIKWYV6drHVyK5Ab7UQztuj4,2984
-lgdo/types/table.py,sha256=h5E4Keu-a3l25WXf7m_hm2Yr4RRzGjS_WCQ_Pla7SAI,12531
+lgdo/types/table.py,sha256=kgJtI4Ea6jNhsQWS_R-9Ilt7Xm9n5B97sv4Cq6m5q7E,12667
 lgdo/types/vectorofvectors.py,sha256=1oxKJDX8VVWpmvUUDHHEzEYw0RRWJrMjOB-jHRY12N4,21859
 lgdo/types/waveform_table.py,sha256=52vqjGudX5_ZR1-b087jx3vuTxJ_yEPO-dO8Dpi0ceg,9407
-legend_pydataobj-1.2.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-legend_pydataobj-1.2.1.dist-info/METADATA,sha256=04PaGLGHYgrMlALV4jsT8sxg87zV32SxOO-LXrg_L4g,3577
-legend_pydataobj-1.2.1.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-legend_pydataobj-1.2.1.dist-info/entry_points.txt,sha256=j22HoS-1cVhTtKJkDnKB49uNH0nEVER2Tpw-lVh1aws,41
-legend_pydataobj-1.2.1.dist-info/top_level.txt,sha256=KyR-EUloqiXcQ62IWnzBmtInDtvsHl4q2ZJAZgTcLXE,5
-legend_pydataobj-1.2.1.dist-info/RECORD,,
+legend_pydataobj-1.4.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+legend_pydataobj-1.4.0.dist-info/METADATA,sha256=NIX9NkqNDbKopKc0EgUEzy7fBdALFOfiGRJjHh1b01w,3577
+legend_pydataobj-1.4.0.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+legend_pydataobj-1.4.0.dist-info/entry_points.txt,sha256=j22HoS-1cVhTtKJkDnKB49uNH0nEVER2Tpw-lVh1aws,41
+legend_pydataobj-1.4.0.dist-info/top_level.txt,sha256=KyR-EUloqiXcQ62IWnzBmtInDtvsHl4q2ZJAZgTcLXE,5
+legend_pydataobj-1.4.0.dist-info/RECORD,,

lgdo/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.2.1'
-__version_tuple__ = version_tuple = (1, 2, 1)
+__version__ = version = '1.4.0'
+__version_tuple__ = version_tuple = (1, 4, 0)

lgdo/cli.py

@@ -0,0 +1,56 @@
+"""legend-pydataobj's command line interface utilities."""
+import argparse
+import logging
+import sys
+
+import lgdo
+import lgdo.logging
+from lgdo import show
+
+
+def lh5ls():
+    """:func:`.show` command line interface."""
+    parser = argparse.ArgumentParser(
+        prog="lh5ls", description="Inspect LEGEND HDF5 (LH5) file contents"
+    )
+
+    # global options
+    parser.add_argument(
+        "--version", action="store_true", help="""Print lgdo version and exit"""
+    )
+    parser.add_argument(
+        "--verbose",
+        "-v",
+        action="store_true",
+        help="""Increase the program verbosity""",
+    )
+    parser.add_argument(
+        "--debug",
+        "-d",
+        action="store_true",
+        help="""Increase the program verbosity to maximum""",
+    )
+
+    parser.add_argument(
+        "lh5_file",
+        help="""Input LH5 file.""",
+    )
+    parser.add_argument("lh5_group", nargs="?", help="""LH5 group.""", default="/")
+    parser.add_argument(
+        "--attributes", "-a", action="store_true", help="""Print HDF5 attributes too"""
+    )
+
+    args = parser.parse_args()
+
+    if args.verbose:
+        lgdo.logging.setup(logging.DEBUG)
+    elif args.debug:
+        lgdo.logging.setup(logging.DEBUG, logging.root)
+    else:
+        lgdo.logging.setup()
+
+    if args.version:
+        print(lgdo.__version__)  # noqa: T201
+        sys.exit()
+
+    show(args.lh5_file, args.lh5_group, attrs=args.attributes)

lgdo/compression/base.py

@@ -3,6 +3,8 @@ from __future__ import annotations
 import re
 from dataclasses import asdict, dataclass
 
+numba_defaults: dict = {"nopython": True, "cache": True}
+
 
 @dataclass(frozen=True)
 class WaveformCodec:

lgdo/compression/generic.py

@@ -40,6 +40,7 @@ def encode(
 
 def decode(
     obj: lgdo.VectorOfEncodedVectors | lgdo.ArrayOfEncodedEqualSizedArrays,
+    out_buf: lgdo.ArrayOfEqualSizedArrays = None,
 ) -> lgdo.VectorOfVectors | lgdo.ArrayOfEqualsizedArrays:
     """Decode encoded LGDOs.
 
@@ -51,6 +52,9 @@ def decode(
     ----------
     obj
         LGDO array type.
+    out_buf
+        pre-allocated LGDO for the decoded signals. See documentation of
+        wrapped encoders for limitations.
     """
     if "codec" not in obj.attrs:
         raise RuntimeError(
@@ -61,9 +65,11 @@ def decode(
     log.debug(f"decoding {repr(obj)} with {codec}")
 
     if _is_codec(codec, radware.RadwareSigcompress):
-        return radware.decode(obj, shift=int(obj.attrs.get("codec_shift", 0)))
+        return radware.decode(
+            obj, sig_out=out_buf, shift=int(obj.attrs.get("codec_shift", 0))
+        )
     elif _is_codec(codec, varlen.ULEB128ZigZagDiff):
-        return varlen.decode(obj)
+        return varlen.decode(obj, sig_out=out_buf)
     else:
         raise ValueError(f"'{codec}' not supported")