Merge branch 'examples/geant4' into 296-ensure-all-examples-are-up-to-date-and-verified

Author: kzscisoft

.github/workflows/examples.yml

@@ -2,7 +2,7 @@ name: Simvue Examples
 
 on:
   push:
-    branches: [ "main", "dev", "hotfix/update-ci", "296-*" ]
+    branches: [ "main", "dev", "hotfix/update-ci", "296-*", "examples/*" ]
   pull_request:
     branches: [ "main", "dev", "hotfix/update-ci" ]
 
@@ -153,3 +153,29 @@ jobs:
           export SIMVUE_URL=${{ secrets.SIMVUE_URL }}
           export SIMVUE_TOKEN=${{ secrets.SIMVUE_TOKEN }}
           python3.11 ./examples/Tensorflow/dynamic_rnn.py --ci
+
+  GEANT4:
+    runs-on: ubuntu-latest
+    container:
+      image: artemisbeta/geant4:11.2.1
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+      - name: Install Simvue
+        run: python3 -m pip install .
+      - name: Install Dependencies
+        run: python3 -m pip install -r examples/Geant4/requirements.txt
+      - name: Build Example
+        run: |
+          cmake -DCMAKE_PREFIX_PATH=/usr/local/share/geant4/install/4.11.2/ -Bbuild examples/Geant4/FixedTarget/
+          cmake --build build
+      - name: Run Example
+        run: |
+          export SIMVUE_URL=${{ secrets.SIMVUE_URL }}
+          export SIMVUE_TOKEN=${{ secrets.SIMVUE_TOKEN }}
+          python examples/Geant4/geant4_simvue.py build/MaterialTesting --ci --events 10

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "examples/Geant4/FixedTarget"]
+	path = examples/Geant4/FixedTarget
+	url = https://github.com/artemis-beta/Geant4-Example.git

examples/Geant4/FixedTarget

@@ -0,0 +1,119 @@
+"""
+Geant4 Simvue
+=============
+
+Example of repeating simulation of a proton fired at a target of beryllium
+monitoring the yield of key particles of interest
+"""
+
+import multiparser
+import multiparser.parsing.file as mp_file_parse
+import simvue
+import uproot
+import multiprocessing
+import typing
+import click
+import pathlib
+import os
+import tempfile
+
+from particle import Particle
+
+
+@click.command
+@click.argument("g4_binary", type=click.Path(exists=True))
+@click.option("--config", type=click.Path(exists=True), default=None)
+@click.option("--ci", is_flag=True, default=False)
+@click.option("--momentum", type=float, default=10)
+@click.option("--events", type=int, default=100)
+def geant4_simvue_example(
+    g4_binary: str, config: typing.Optional[str], ci: bool, momentum: float, events: int
+) -> None:
+    @mp_file_parse.file_parser
+    def root_file_parser(
+        file_name: str, *_, **__
+    ) -> tuple[dict[str, typing.Any], dict[str, typing.Any]]:
+        with uproot.open(file_name) as root_data:
+            hit_data: dict[str, uproot.TBranch]
+            if not (hit_data := root_data.get("Hits")):
+                raise RuntimeError("Expected key 'Hits' in ROOT file")
+
+        particles_of_interest = [2212, 211, 11, 22, 2112]
+
+        all_particles = hit_data["fID"].array(library="np").tolist()
+
+        out_data = {
+            Particle.from_pdgid(abs(identifier)).name: [
+                abs(i) for i in all_particles
+            ].count(abs(identifier))
+            for identifier in particles_of_interest
+        }
+
+        return {}, out_data
+
+    termination_trigger = multiprocessing.Event()
+
+    with simvue.Run() as run:
+        run.init(
+            "Geant4_simvue_demo",
+            folder="/simvue_client_demos",
+            tags=[
+                "Geant4",
+            ],
+            description="Geant4 fixed target scenario",
+            retention_period="1 hour" if ci else None,
+        )
+
+        kwargs: dict[str, typing.Any] = {}
+
+        if config:
+            kwargs["script"] = config
+        with tempfile.TemporaryDirectory() as tempd:
+            with multiparser.FileMonitor(
+                per_thread_callback=lambda metrics, *_: run.log_metrics(metrics),
+                exception_callback=run.log_event,
+                terminate_all_on_fail=False,
+                plain_logging=True,
+                flatten_data=True,
+                termination_trigger=termination_trigger,
+            ) as monitor:
+                monitor.track(
+                    path_glob_exprs=[f'{pathlib.Path(tempd).joinpath("*")}'],
+                    parser_func=root_file_parser,
+                    static=True,
+                )
+                monitor.run()
+
+                for i in range(events):
+                    if i % 10 == 0:
+                        click.secho(
+                            f"Running {i+1}/{events} with momentum {momentum} GeV",
+                            bold=True,
+                            fg="cyan",
+                        )
+                    running_simulation = multiprocessing.Event()
+                    run.add_process(
+                        identifier=f"Geant4_simulation_{momentum}GeV_{i}",
+                        executable=g4_binary,
+                        momentum=momentum,
+                        batch=True,
+                        output=pathlib.Path(tempd).joinpath(
+                            f"output_{momentum}GeV_{i+1}.root"
+                        ),
+                        completion_trigger=running_simulation
+                        if i == events - 1
+                        else None,
+                        **kwargs,
+                    )
+
+                termination_trigger.set()
+
+    for file in pathlib.Path().cwd().glob("Geant4*.err"):
+        os.remove(file)
+
+    for file in pathlib.Path().cwd().glob("Geant4*.out"):
+        os.remove(file)
+
+
+if __name__ in "__main__":
+    geant4_simvue_example()

examples/Geant4/geant4_simvue.py

@@ -0,0 +1,5 @@
+simvue
+uproot
+ukaea-multiparser
+click
+particle