Skip to content

New utility function to change RunIDs in a ROOT file #633

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
acoussat wants to merge 2 commits into
base: master
Choose a base branch
from
Open

New utility function to change RunIDs in a ROOT file #633

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
70e41e7
New utility function to change RunIDs in a ROOT file
acoussat Dec 9, 2024
d810461
[pre-commit.ci] Automatic python and c++ formatting
pre-commit-ci[bot] Dec 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions opengate/utility.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -604,3 +604,24 @@ def standard_error_c4_correction(n):
return (
np.sqrt(2 / (n - 1)) * sc.special.gamma(n / 2) / sc.special.gamma((n - 1) / 2)
)


def update_runid(input_path, run_id, tree, output_path):
"""
Update a ROOT file by setting the RunID to some constant value.

When a GATE simulation with multiple runs was parallelized by splitting it into several one-run simulations, all output ROOT files have RunIDs equal to one.
This can be a problem for post-processing ROOT files based on their RunID.
By using this piece of code, one can update the RunID from a ROOT output file to the actual RunID.

This creates a new ROOT file, and does not operate in-place. The original ROOT file is preserved.
"""
import uproot

input_tree = uproot.open(input_path)[tree]
input_array = input_tree.arrays(library="np")
input_array["RunID"] = np.full(
input_array["RunID"].shape, run_id, dtype=input_array["RunID"].dtype
)
with uproot.recreate(output_path) as output_file:
output_file[tree] = input_array
Loading