Warning, /detector_benchmarks/benchmarks/calo_pid/Snakefile is written in an unsupported language. File is not indexed.
0001 def format_energy_for_dd4hep(s):
0002 return s.rstrip("kMGeV") + "*" + s.lstrip("0123456789")
0003
0004 rule calo_pid_sim:
0005 input:
0006 warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
0007 geometry_lib=find_epic_libraries(),
0008 output:
0009 "sim_output/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}/{ENERGY_MIN}to{ENERGY_MAX}/{THETA_MIN}to{THETA_MAX}deg/{PARTICLE}_{ENERGY}_{THETA_MIN}to{THETA_MAX}deg.{INDEX}.edm4hep.root",
0010 log:
0011 "sim_output/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}/{ENERGY_MIN}to{ENERGY_MAX}/{THETA_MIN}to{THETA_MAX}deg/{PARTICLE}_{ENERGY}_{THETA_MIN}to{THETA_MAX}deg.{INDEX}.edm4hep.root.log",
0012 wildcard_constraints:
0013 PARTICLE="(e-|pi-)",
0014 ENERGY_MIN="[0-9]+[kMG]eV",
0015 ENERGY_MAX="[0-9]+[kMG]eV",
0016 THETA_MIN="[0-9]+",
0017 THETA_MAX="[0-9]+",
0018 INDEX=r"\d{4}",
0019 params:
0020 N_EVENTS=1000,
0021 SEED=lambda wildcards: "1" + wildcards.INDEX,
0022 DETECTOR_PATH=os.environ["DETECTOR_PATH"],
0023 DETECTOR_CONFIG=lambda wildcards: wildcards.DETECTOR_CONFIG,
0024 ENERGY_MIN=lambda wildcards: format_energy_for_dd4hep(wildcards.ENERGY_MIN),
0025 ENERGY_MAX=lambda wildcards: format_energy_for_dd4hep(wildcards.ENERGY_MAX),
0026 THETA_MIN=lambda wildcards: wildcards.THETA_MIN,
0027 THETA_MAX=lambda wildcards: wildcards.THETA_MAX,
0028 DD4HEP_HASH=get_spack_package_hash("dd4hep"),
0029 NPSIM_HASH=get_spack_package_hash("npsim"),
0030 cache: True
0031 shell:
0032 """
0033 set -m # monitor mode to prevent lingering processes
0034 exec ddsim \
0035 --runType batch \
0036 --enableGun \
0037 --gun.momentumMin "{params.ENERGY_MIN}" \
0038 --gun.momentumMax "{params.ENERGY_MAX}" \
0039 --gun.thetaMin "{wildcards.THETA_MIN}*deg" \
0040 --gun.thetaMax "{wildcards.THETA_MAX}*deg" \
0041 --gun.particle {wildcards.PARTICLE} \
0042 --gun.distribution eta \
0043 --random.seed {params.SEED} \
0044 --filter.tracker edep0 \
0045 -v WARNING \
0046 --numberOfEvents {params.N_EVENTS} \
0047 --compactFile {params.DETECTOR_PATH}/{params.DETECTOR_CONFIG}.xml \
0048 --outputFile {output}
0049 """
0050
0051
0052 rule calo_pid_recon:
0053 input:
0054 "sim_output/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}/{ENERGY}/{PHASE_SPACE}/{PARTICLE}_{ENERGY}_{PHASE_SPACE}.{INDEX}.edm4hep.root",
0055 output:
0056 "sim_output/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}/{ENERGY}/{PHASE_SPACE}/{PARTICLE}_{ENERGY}_{PHASE_SPACE}.{INDEX}.eicrecon.tree.edm4eic.root",
0057 log:
0058 "sim_output/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}/{ENERGY}/{PHASE_SPACE}/{PARTICLE}_{ENERGY}_{PHASE_SPACE}.{INDEX}.eicrecon.tree.edm4eic.root.log",
0059 wildcard_constraints:
0060 INDEX=r"\d{4}",
0061 shell: """
0062 set -m # monitor mode to prevent lingering processes
0063 exec env DETECTOR_CONFIG={wildcards.DETECTOR_CONFIG} \
0064 eicrecon {input} -Ppodio:output_file={output} \
0065 -Ppodio:output_collections=MCParticles,EcalEndcapNRecHits,EcalEndcapNClusters,EcalEndcapNParticleIDInput_features,EcalEndcapNParticleIDTarget,EcalEndcapNParticleIDOutput_probability_tensor
0066 """
0067
0068
0069 rule calo_pid_input_list:
0070 input:
0071 electrons=expand(
0072 "sim_output/calo_pid/{{DETECTOR_CONFIG}}/{{PARTICLE}}/{ENERGY}/{PHASE_SPACE}/{{PARTICLE}}_{ENERGY}_{PHASE_SPACE}.{INDEX:04d}.eicrecon.tree.edm4eic.root",
0073 ENERGY=["100MeVto20GeV"],
0074 PHASE_SPACE=["130to177deg"],
0075 INDEX=range(100),
0076 ),
0077 output:
0078 "listing/calo_pid/{DETECTOR_CONFIG}/{PARTICLE}.lst",
0079 run:
0080 with open(output[0], "wt") as fp:
0081 fp.write("\n".join(input))
0082
0083
0084 rule calo_pid:
0085 input:
0086 electrons="listing/calo_pid/{DETECTOR_CONFIG}/e-.lst",
0087 pions="listing/calo_pid/{DETECTOR_CONFIG}/pi-.lst",
0088 matplotlibrc=".matplotlibrc",
0089 script="benchmarks/calo_pid/calo_pid.py",
0090 output:
0091 directory("results/{DETECTOR_CONFIG}/calo_pid")
0092 shell:
0093 """
0094 env \
0095 MATPLOTLIBRC={input.matplotlibrc} \
0096 DETECTOR_CONFIG={wildcards.DETECTOR_CONFIG} \
0097 PLOT_TITLE={wildcards.DETECTOR_CONFIG} \
0098 INPUT_ELECTRONS="{input.electrons}" \
0099 INPUT_PIONS="{input.pions}" \
0100 OUTPUT_DIR={output} \
0101 python {input.script}
0102 """
