dagmc-h5m-file-inspector 0.4.3__py3-none-any.whl → 0.6.0__py3-none-any.whl

dagmc_h5m_file_inspector/__init__.py

@@ -1,3 +1,7 @@
 from .core import get_volumes_from_h5m
 from .core import get_materials_from_h5m
 from .core import get_volumes_and_materials_from_h5m
+from .core import get_bounding_box_from_h5m
+from .core import get_volumes_sizes_from_h5m_by_cell_id
+from .core import get_volumes_sizes_from_h5m_by_material_name
+from .core import set_openmc_material_volumes_from_h5m

dagmc_h5m_file_inspector/_version.py

@@ -1,5 +1,34 @@
-# coding: utf-8
-# file generated by setuptools_scm
+# file generated by setuptools-scm
 # don't change, don't track in version control
-version = '0.4.3'
-version_tuple = (0, 4, 3)
+
+__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.6.0'
+__version_tuple__ = version_tuple = (0, 6, 0)
+
+__commit_id__ = commit_id = None

dagmc_h5m_file_inspector/cli.py

@@ -0,0 +1,106 @@
+import argparse
+import json
+import pprint
+
+from dagmc_h5m_file_inspector import (
+    get_volumes_from_h5m,
+    get_materials_from_h5m,
+    get_volumes_and_materials_from_h5m,
+)
+
+
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        "-i",
+        "--input",
+        type=str,
+        help="The filename of the h5m file",
+        required=True,
+    )
+
+    parser.add_argument(
+        "-v",
+        "--volumes",
+        help="Returns volume ids from a h5m file",
+        required=False,
+        action="store_true",
+    )
+
+    parser.add_argument(
+        "-m",
+        "--materials",
+        help="Returns materials tags from a h5m file",
+        required=False,
+        action="store_true",
+    )
+
+    parser.add_argument(
+        "-b",
+        "--both",
+        help="Returns volume ids with materials tags from a h5m file",
+        required=False,
+        action="store_true",
+    )
+
+    parser.add_argument(
+        "-o",
+        "--output",
+        type=str,
+        default="inspector_results.txt",
+        help="Returns volume ids with materials tags from a h5m file",
+        required=False,
+    )
+
+    parser.add_argument(
+        "--backend",
+        type=str,
+        choices=["h5py", "pymoab"],
+        default="h5py",
+        help="Backend to use for reading h5m files (default: h5py)",
+        required=False,
+    )
+
+    args = parser.parse_args()
+
+    volumes = None
+    if args.volumes:
+        volumes = get_volumes_from_h5m(filename=args.input, backend=args.backend)
+        print(f"\nVolume IDs ={volumes}")
+
+    materials = None
+    if args.materials:
+        materials = get_materials_from_h5m(filename=args.input, backend=args.backend)
+        print(f"\nMaterial tags ={materials}")
+
+    both = None
+    if args.both:
+        both = get_volumes_and_materials_from_h5m(filename=args.input, backend=args.backend)
+        pp = pprint.PrettyPrinter(indent=4)
+        print("\nVolume IDs and material tags=")
+        pp.pprint(both)
+
+    if args.volumes is False and args.materials is False and args.both is False:
+        print(
+            f"\nNo inspection of {args.input} carried out as outputs were not "
+            "specified. Output options include -v, --volume, -m, --materials"
+            ", -b, --both"
+        )
+    elif args.output:
+        with open(args.output, "w") as output_file:
+            text_to_write = {}
+            if volumes is not None:
+                text_to_write["volumes"] = volumes
+            if materials is not None:
+                text_to_write["materials"] = materials
+            if both is not None:
+                text_to_write["both"] = both
+            print(f"writing file {args.output}")
+            json.dump(text_to_write, output_file, indent=4)
+
+    print("\n")
+
+
+if __name__ == "__main__":
+    main()