Skip to content

Commit

Permalink
prepare the run to have magnetic deflection and plenoptics
Browse files Browse the repository at this point in the history
  • Loading branch information
relleums committed Dec 16, 2023
1 parent 631030e commit e9fec44
Show file tree
Hide file tree
Showing 2 changed files with 43 additions and 1 deletion.
35 changes: 35 additions & 0 deletions plenoirf/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,41 @@ def run(run_dir, pool, logger=None):
if logger is None:
logger = json_line_logger.LoggerStdout()

logger.debug("gather provenance")
prov = provenance.make_provenance()
provenance.tar_open_append_close(
path=opj(run_dir, "provenance", "init.tar"),
provenance=prov,
)

logger.debug("read config")
config = configurating.read(run_dir=run_dir)

while magnetic_deflection.needs_to_run(
work_dir=opj(run_dir, "magnetic_deflection"),
num_showers_target=config["magnetic_deflection"]["num_showers_target"],
):
logger.info("Produce more showers in magnetic_deflection.")
magnetic_deflection.run(
work_dir=opj(run_dir, "magnetic_deflection"),
pool=pool,
num_runs=config["magnetic_deflection"]["run"]["num_runs"],
num_showers_per_run=config["magnetic_deflection"]["run"][
"num_showers_per_run"
],
num_showers_target=config["magnetic_deflection"][
"num_showers_target"
],
)
logger.info("magnetic_deflection is complete")

plenoptics.run(
work_dir=opj(run_dir, "plenoptics"),
pool=pool,
logger=logger,
)
logger.info("plenoptics is complete")


def makesuredirs(path):
os.makedirs(path, exist_ok=True)
9 changes: 8 additions & 1 deletion plenoirf/configurating.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,14 @@ def make_particles():


def make_magnetic_deflection():
return {"energy_stop_GeV": 64.0}
return {
"energy_stop_GeV": 64.0,
"num_showers_target": 2 * 1000 * 1000,
"run": {
"num_runs": 192,
"num_showers_per_run": 1280,
},
}


def make_plenoptics():
Expand Down

0 comments on commit e9fec44

Please sign in to comment.