legend-pygeom-hades 0.1.0a0__py3-none-any.whl

pygeomhades/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from pygeomhades._version import version as __version__
+from pygeomhades.core import construct
+
+__all__ = ["__version__", "construct"]

pygeomhades/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '0.1.0a0'
+__version_tuple__ = version_tuple = (0, 1, 0, 'a0')
+
+__commit_id__ = commit_id = None

pygeomhades/cli.py

@@ -0,0 +1,169 @@
+from __future__ import annotations
+
+import argparse
+import logging
+
+from dbetto import utils
+from pyg4ometry import config as meshconfig
+from pygeomtools import geometry, write_pygeom
+
+from . import _version, core
+
+log = logging.getLogger(__name__)
+
+
+def dump_gdml_cli(argv: list[str] | None = None) -> None:
+    args, config = _parse_cli_args(argv)
+
+    logging.basicConfig()
+    if args.verbose:
+        logging.getLogger("pygeomhades").setLevel(logging.DEBUG)
+    if args.debug:
+        logging.root.setLevel(logging.DEBUG)
+
+    vis_scene = {}
+    if isinstance(args.visualize, str):
+        vis_scene = utils.load_dict(args.visualize)
+
+    if args.clip_geometry:
+        vis_scene["clipper"] = [{"origin": [0, 0, 0], "normal": [1, 0, 0], "close_cuts": False}]
+
+    if vis_scene.get("fine_mesh", False) or args.check_overlaps:
+        meshconfig.setGlobalMeshSliceAndStack(100)
+
+    registry = core.construct(
+        args.hpge_name,
+        args.measurement,
+        assemblies=args.assemblies,
+        config=config,
+        public_geometry=args.public_geom,
+    )
+
+    if args.print_volumes:
+        geometry.print_volumes(registry, args.print_volumes)
+
+    if args.check_overlaps:
+        msg = "checking for overlaps"
+        log.info(msg)
+        registry.worldVolume.checkOverlaps(recursive=True)
+
+    # commit auxvals, and write to GDML file if requested.
+    if args.filename is not None:
+        log.info("exporting GDML geometry to %s", args.filename)
+    write_pygeom(registry, args.filename)
+
+    if args.visualize:
+        log.info("visualizing...")
+        from pygeomtools import viewer
+
+        viewer.visualize(registry, vis_scene)
+
+
+def _parse_cli_args(argv: list[str] | None = None) -> tuple[argparse.Namespace, dict]:
+    parser = argparse.ArgumentParser(
+        prog="legend-pygeom-hades",
+        description="%(prog)s command line interface",
+    )
+
+    # global options
+    parser.add_argument(
+        "--version",
+        action="version",
+        help="""Print %(prog)s version and exit""",
+        version=_version.__version__,
+    )
+    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(
+        "--visualize",
+        "-V",
+        nargs="?",
+        const=True,
+        help="""Open a VTK visualization of the generated geometry (with optional scene file)""",
+    )
+    parser.add_argument(
+        "--clip-geometry",
+        action="store_true",
+        help="""Clip the geometry for visualization purposes""",
+    )
+    parser.add_argument(
+        "--check-overlaps",
+        action="store_true",
+        help="""Check for overlaps with pyg4ometry (note: this might not be accurate)""",
+    )
+    parser.add_argument(
+        "--print-volumes",
+        action="store",
+        choices=("logical", "physical", "detector"),
+        help="""Print a list of logical or physical volume names (from the pyg4ometry registry)""",
+    )
+
+    # options for geometry generation.
+    #
+    # geometry options can also be specified in the config file, so the "default" argument of the argparse
+    # options cannot be used - we need to distinguish between an unspecified option and an explicitly set
+    # default option.
+    geom_opts = parser.add_argument_group("geometry options")
+    geom_opts.add_argument(
+        "--public-geom",
+        action="store_true",
+        default=None,
+        help="""Create a geometry from public testdata only.""",
+    )
+    geom_opts.add_argument(
+        "-c",
+        "--config",
+        action="store",
+        help="""Select a config file to read geometry information from. """,
+    )
+    geom_opts.add_argument(
+        "--hpge-name",
+        action="store",
+        required=True,
+        help="""Name of the detector eg "V07302A".""",
+    )
+    geom_opts.add_argument(
+        "--assemblies",
+        action="store",
+        default=["hpge", "lead_castle"],
+        help=(
+            """Select the assemblies to generate in the output.
+            (default: hpge and lead_castle)"""
+        ),
+    )
+
+    geom_opts.add_argument(
+        "--measurement",
+        action="store",
+        required=True,
+        help="""Name of the measurement eg "am_HS1_top_dlt".""",
+    )
+
+    parser.add_argument(
+        "filename",
+        default=None,
+        nargs="?",
+        help="""File name for the output GDML geometry.""",
+    )
+
+    args = parser.parse_args(argv)
+
+    config = {}
+
+    if args.config is not None:
+        config = utils.load_dict(args.config)
+
+    if not args.visualize and args.filename == "":
+        parser.error("no output file and no visualization specified")
+
+    return args, config

pygeomhades/configs/dummy_geom/B99000A.yaml

@@ -0,0 +1,25 @@
+name: B99000A
+type: bege
+production:
+  manufacturer: Test
+  enrichment:
+    val: 0.75
+    unc: 0.05
+geometry:
+  height_in_mm: 40.0
+  radius_in_mm: 35.0
+  groove:
+    depth_in_mm: 2.0
+    radius_in_mm:
+      outer: 12.0
+      inner: 10.0
+  pp_contact:
+    radius_in_mm: 7.5
+    depth_in_mm: 0
+  taper:
+    top:
+      angle_in_deg: 0.0
+      height_in_mm: 0.0
+    bottom:
+      angle_in_deg: 0.0
+      height_in_mm: 0.0

pygeomhades/configs/dummy_geom/C99000A.yaml

@@ -0,0 +1,31 @@
+name: C99000A
+type: coax
+production:
+  manufacturer: Test
+  enrichment:
+    val: 0.75
+    unc: 0.05
+geometry:
+  height_in_mm: 80
+  radius_in_mm: 40
+  borehole:
+    radius_in_mm: 7
+    depth_in_mm: 70
+  groove:
+    depth_in_mm: 2
+    radius_in_mm:
+      outer: 20
+      inner: 17
+  pp_contact:
+    radius_in_mm: 17
+    depth_in_mm: 0
+  taper:
+    top:
+      angle_in_deg: 45
+      height_in_mm: 5
+    bottom:
+      angle_in_deg: 45
+      height_in_mm: 2
+    borehole:
+      angle_in_deg: 0
+      height_in_mm: 0

pygeomhades/configs/dummy_geom/P99000A.yaml

@@ -0,0 +1,20 @@
+name: P99000A
+type: ppc
+production:
+  manufacturer: Test
+  enrichment:
+    val: 0.75
+    unc: 0.05
+geometry:
+  height_in_mm: 40.0
+  radius_in_mm: 35.0
+  pp_contact:
+    radius_in_mm: 2
+    depth_in_mm: 2
+  taper:
+    top:
+      angle_in_deg: 0
+      height_in_mm: 0
+    bottom:
+      angle_in_deg: 30
+      height_in_mm: 10

pygeomhades/configs/dummy_geom/V99000A.yaml

@@ -0,0 +1,31 @@
+name: V99000A
+type: icpc
+production:
+  manufacturer: Test
+  enrichment:
+    val: 0.75
+    unc: 0.05
+geometry:
+  height_in_mm: 60
+  radius_in_mm: 35
+  borehole:
+    radius_in_mm: 5
+    depth_in_mm: 45
+  groove:
+    depth_in_mm: 1
+    radius_in_mm:
+      outer: 10
+      inner: 9
+  pp_contact:
+    radius_in_mm: 3
+    depth_in_mm: 2
+  taper:
+    top:
+      angle_in_deg: 10
+      height_in_mm: 10
+    bottom:
+      angle_in_deg: 0
+      height_in_mm: 0
+    borehole:
+      angle_in_deg: 0
+      height_in_mm: 0

pygeomhades/configs/holder_wrap/B00000B.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00000B
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 41.0
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 39.5
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 41.5
+  position: 3.5
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 37.2
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 37.7
+  position: 3.5

pygeomhades/configs/holder_wrap/B00000D.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00000D
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 39.2
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 40.7
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 42.2
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 38.4
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 38.9
+  position: 4.0

pygeomhades/configs/holder_wrap/B00002C.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00002C
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 38.4
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 39.9
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 41.4
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 37.6
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 38.1
+  position: 4.0

pygeomhades/configs/holder_wrap/B00032B.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00032B
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 36.9
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 38.4
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 40.0
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 36.1
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 36.6
+  position: 4.0

pygeomhades/configs/holder_wrap/B00035A.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00035A
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 37.8
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 39.3
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 40.8
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 37.0
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 37.5
+  position: 4.0

pygeomhades/configs/holder_wrap/B00035B.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00035B
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 39.2
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 40.7
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 42.2
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 38.4
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 38.9
+  position: 4.0

pygeomhades/configs/holder_wrap/B00061C.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00061C
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 38.3
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 39.8
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 41.3
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 37.5
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 38.0
+  position: 4.0

pygeomhades/configs/holder_wrap/B00076C.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00076C
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 38.9
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 40.4
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 41.9
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 38.1
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 38.6
+  position: 4.0

pygeomhades/configs/holder_wrap/B00091B.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: B00091B
+  position: 5.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 0
+        radius_in_mm: 0
+      outer:
+        height_in_mm: 0
+        radius_in_mm: 0
+    cylinder:
+      inner:
+        height_in_mm: 62.5
+        radius_in_mm: 36.3
+      outer:
+        height_in_mm: 65.5
+        radius_in_mm: 37.8
+    edge:
+      height_in_mm: 0
+      radius_in_mm: 0
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 0
+      position_top_ring_in_mm: 14.7
+      radius_in_mm: 39.3
+  position: 4.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 49.0
+      radius_in_mm: 35.5
+    outer:
+      height_in_mm: 50.0
+      radius_in_mm: 36.0
+  position: 4.0

pygeomhades/configs/holder_wrap/V00048B.yaml

@@ -0,0 +1,37 @@
+detector:
+  name: V00048B
+  position: 7.0
+holder:
+  geometry:
+    bottom_cyl:
+      inner:
+        height_in_mm: 15.7
+        radius_in_mm: 8.0
+      outer:
+        height_in_mm: 25.4
+        radius_in_mm: 13.0
+    cylinder:
+      inner:
+        height_in_mm: 105.0
+        radius_in_mm: 37.5
+      outer:
+        height_in_mm: 108.1
+        radius_in_mm: 38.3
+    edge:
+      height_in_mm: 1.3
+      radius_in_mm: 40.5
+    rings:
+      height_in_mm: 8.6
+      position_bottom_ring_in_mm: 64.4
+      position_top_ring_in_mm: 14.0
+      radius_in_mm: 40.5
+  position: 6.0
+wrap:
+  geometry:
+    inner:
+      height_in_mm: 92.9
+      radius_in_mm: 36.3
+    outer:
+      height_in_mm: 94.3
+      radius_in_mm: 36.8
+  position: 4.5