Skip to content

Commit

Permalink
benchmarks/femc_*: scale event count with momentum
Browse files Browse the repository at this point in the history
  • Loading branch information
@veprbl
veprbl committed Oct 11, 2024
1 parent fee8b16 commit 6f48010
Show file tree
Hide file tree
Showing 3 changed files with 42 additions and 18 deletions.
20 changes: 14 additions & 6 deletions benchmarks/femc_electron/Snakefile
View file
Original file line number Diff line number Diff line change
@@ -1,32 +1,39 @@
def get_n_events(wildcards):
energy = float(wildcards.P)
n_events = 1000
n_events = int(n_events // ((energy / 20) ** 0.5))
return n_events


rule femc_electron_generate:
input:
script="benchmarks/femc_electron/analysis/gen_particles.cxx",
params:
NEVENTS_GEN=1000,
N_EVENTS=get_n_events,
th_max=28,
th_min=2.0
output:
GEN_FILE="sim_output/femc_electron/e-_{P}GeV.hepmc"
shell:
"""
mkdir -p sim_output/femc_electron
root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "e-", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
root -l -b -q '{input.script}({params.N_EVENTS},"{output.GEN_FILE}", "e-", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
"""

rule femc_electron_simulate:
input:
GEN_FILE="sim_output/femc_electron/e-_{P}GeV.hepmc"
params:
N_EVENTS=get_n_events,
PHYSICS_LIST="FTFP_BERT"
output:
SIM_FILE="sim_output/femc_electron/{DETECTOR_CONFIG}_sim_e-_{P}GeV.edm4hep.root"
shell:
"""
NEVENTS_SIM=1000
# Running simulation
npsim \
--compactFile $DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml \
--numberOfEvents $NEVENTS_SIM \
--numberOfEvents {params.N_EVENTS} \
--physicsList {params.PHYSICS_LIST} \
--inputFiles {input.GEN_FILE} \
--outputFile {output.SIM_FILE}
Expand All @@ -37,10 +44,11 @@ rule femc_electron_recon:
SIM_FILE="sim_output/femc_electron/{DETECTOR_CONFIG}_sim_e-_{P}GeV.edm4hep.root"
output:
REC_FILE="sim_output/femc_electron/{DETECTOR_CONFIG}_rec_e-_{P}GeV.edm4eic.root"
params:
N_EVENTS=get_n_events,
shell:
"""
NEVENTS_REC=1000
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents=$NEVENTS_REC
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents={params.N_EVENTS}
"""

rule femc_electron_analysis:
Expand Down
20 changes: 14 additions & 6 deletions benchmarks/femc_photon/Snakefile
View file
Original file line number Diff line number Diff line change
@@ -1,32 +1,39 @@
def get_n_events(wildcards):
energy = float(wildcards.P)
n_events = 1000
n_events = int(n_events // ((energy / 20) ** 0.5))
return n_events


rule femc_photon_generate:
input:
script="benchmarks/femc_photon/analysis/gen_particles.cxx",
params:
NEVENTS_GEN=1000,
N_EVENTS=get_n_events,
th_max=28,
th_min=2.0
output:
GEN_FILE="sim_output/femc_photon/photon_{P}GeV.hepmc"
shell:
"""
mkdir -p sim_output/femc_photon
root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "gamma", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
root -l -b -q '{input.script}({params.N_EVENTS},"{output.GEN_FILE}", "gamma", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
"""

rule femc_photon_simulate:
input:
GEN_FILE="sim_output/femc_photon/photon_{P}GeV.hepmc"
params:
N_EVENTS=get_n_events,
PHYSICS_LIST="FTFP_BERT"
output:
SIM_FILE="sim_output/femc_photon/{DETECTOR_CONFIG}_sim_photon_{P}GeV.edm4hep.root"
shell:
"""
NEVENTS_SIM=1000
# Running simulation
npsim \
--compactFile $DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml \
--numberOfEvents $NEVENTS_SIM \
--numberOfEvents {params.N_EVENTS} \
--physicsList {params.PHYSICS_LIST} \
--inputFiles {input.GEN_FILE} \
--outputFile {output.SIM_FILE}
Expand All @@ -37,10 +44,11 @@ rule femc_photon_recon:
SIM_FILE="sim_output/femc_photon/{DETECTOR_CONFIG}_sim_photon_{P}GeV.edm4hep.root"
output:
REC_FILE="sim_output/femc_photon/{DETECTOR_CONFIG}_rec_photon_{P}GeV.edm4eic.root"
params:
N_EVENTS=get_n_events,
shell:
"""
NEVENTS_REC=1000
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents=$NEVENTS_REC
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents={params.N_EVENTS}
"""

rule femc_photon_analysis:
Expand Down
20 changes: 14 additions & 6 deletions benchmarks/femc_pi0/Snakefile
View file
Original file line number Diff line number Diff line change
@@ -1,32 +1,39 @@
def get_n_events(wildcards):
energy = float(wildcards.P)
n_events = 1000
n_events = int(n_events // ((energy / 20) ** 0.5))
return n_events


rule femc_pi0_generate:
input:
script="benchmarks/femc_pi0/analysis/gen_particles.cxx",
params:
NEVENTS_GEN=1000,
N_EVENTS=get_n_events,
th_max=28,
th_min=2.0
output:
GEN_FILE="sim_output/femc_pi0/pi0_{P}GeV.hepmc"
shell:
"""
mkdir -p sim_output/femc_pi0
root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "pi0", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
root -l -b -q '{input.script}({params.N_EVENTS},"{output.GEN_FILE}", "pi0", {params.th_min}, {params.th_max}, 0., 360., {wildcards.P})'
"""

rule femc_pi0_simulate:
input:
GEN_FILE="sim_output/femc_pi0/pi0_{P}GeV.hepmc"
params:
N_EVENTS=get_n_events,
PHYSICS_LIST="FTFP_BERT"
output:
SIM_FILE="sim_output/femc_pi0/{DETECTOR_CONFIG}_sim_pi0_{P}GeV.edm4hep.root"
shell:
"""
NEVENTS_SIM=1000
# Running simulation
npsim \
--compactFile $DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml \
--numberOfEvents $NEVENTS_SIM \
--numberOfEvents {params.N_EVENTS} \
--physicsList {params.PHYSICS_LIST} \
--inputFiles {input.GEN_FILE} \
--outputFile {output.SIM_FILE}
Expand All @@ -37,10 +44,11 @@ rule femc_pi0_recon:
SIM_FILE="sim_output/femc_pi0/{DETECTOR_CONFIG}_sim_pi0_{P}GeV.edm4hep.root"
output:
REC_FILE="sim_output/femc_pi0/{DETECTOR_CONFIG}_rec_pi0_{P}GeV.edm4eic.root"
params:
N_EVENTS=get_n_events,
shell:
"""
NEVENTS_REC=1000
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents=$NEVENTS_REC
eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Pdd4hep:xml_files=$DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPInsertRecHits,EcalEndcapPInsertClusters,EcalEndcapPInsertRecHits,EcalEndcapPClusters -Pjana:nevents={params.N_EVENTS}
"""

rule femc_pi0_analysis:
Expand Down

0 comments on commit 6f48010

Please sign in to comment.