Back to home page

EIC code displayed by LXR

 
 

    


Warning, /detector_benchmarks/benchmarks/insert_neutron/Snakefile is written in an unsupported language. File is not indexed.

0001 rule insert_neutron_generate:
0002     input:
0003         script="benchmarks/insert_neutron/analysis/gen_particles.cxx",
0004     output:
0005         GEN_FILE="sim_output/insert_neutron/neutron_{P}GeV.hepmc",
0006     params:
0007         NEVENTS_GEN=1000,
0008         th_max=5.7,
0009         th_min=2.0,
0010         P=lambda wildcards: wildcards.P,
0011     shell:
0012         """
0013 mkdir -p sim_output/insert_neutron
0014 root -l -b -q '{input.script}({params.NEVENTS_GEN},"{output.GEN_FILE}", "neutron", {params.th_min}, {params.th_max}, 0., 360., {params.P})'
0015 """
0016 
0017 rule insert_neutron_simulate:
0018     input:
0019         GEN_FILE="sim_output/insert_neutron/neutron_{P}GeV.hepmc",
0020         warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
0021         geometry_lib=find_epic_libraries(),
0022     output:
0023         SIM_FILE="sim_output/insert_neutron/{DETECTOR_CONFIG}_sim_neutron_{P}GeV_{INDEX}.edm4hep.root",
0024     params:
0025         INDEX=lambda wildcards: wildcards.INDEX,
0026         SEED=lambda wildcards: "1" + wildcards.INDEX,
0027         PHYSICS_LIST="FTFP_BERT",
0028         DETECTOR_PATH=os.environ["DETECTOR_PATH"],
0029         DETECTOR_CONFIG=lambda wildcards: wildcards.DETECTOR_CONFIG,
0030         DD4HEP_HASH=get_spack_package_hash("dd4hep"),
0031         NPSIM_HASH=get_spack_package_hash("npsim"),
0032     cache: True
0033     shell:
0034         """
0035 NEVENTS_SIM=200
0036 exec npsim \
0037    --compactFile {params.DETECTOR_PATH}/{params.DETECTOR_CONFIG}.xml \
0038    --random.seed {params.SEED} \
0039    --skipNEvents $(( $NEVENTS_SIM * {params.INDEX} )) \
0040    --numberOfEvents $NEVENTS_SIM \
0041    --physicsList {params.PHYSICS_LIST} \
0042    --inputFiles {input.GEN_FILE} \
0043    --outputFile {output.SIM_FILE}
0044 """
0045 
0046 rule insert_neutron_recon:
0047     input:
0048         SIM_FILE="sim_output/insert_neutron/{DETECTOR_CONFIG}_sim_neutron_{P}GeV_{INDEX}.edm4hep.root",
0049         warmup="warmup/{DETECTOR_CONFIG}.edm4hep.root",
0050     output:
0051         REC_FILE="sim_output/insert_neutron/{DETECTOR_CONFIG}_rec_neutron_{P}GeV_{INDEX}.edm4eic.root",
0052     params:
0053         DETECTOR_CONFIG=lambda wildcards: wildcards.DETECTOR_CONFIG,
0054         EICRECON_HASH=get_spack_package_hash("eicrecon"),
0055     cache: True
0056     shell:
0057         """
0058 NEVENTS_REC=200
0059 env DETECTOR_CONFIG={params.DETECTOR_CONFIG} \
0060   eicrecon {input.SIM_FILE} -Ppodio:output_file={output.REC_FILE} -Ppodio:output_collections=MCParticles,HcalEndcapPInsertRecHits,HcalEndcapPInsertClusters,HcalEndcapPInsertSubcellHits,EcalEndcapPClusters  -Pjana:nevents=$NEVENTS_REC
0061 """
0062 
0063 rule insert_neutron_analysis:
0064     input:
0065         expand("sim_output/insert_neutron/{DETECTOR_CONFIG}_rec_neutron_{P}GeV_{INDEX}.edm4eic.root",
0066             P=[20, 30, 40, 50, 60, 70, 80],
0067             DETECTOR_CONFIG=["{DETECTOR_CONFIG}"],
0068             INDEX=range(5),
0069         ),
0070         script="benchmarks/insert_neutron/analysis/neutron_plots.py",
0071     output:
0072         results_dir=directory("results/{DETECTOR_CONFIG}/insert_neutron"),
0073     shell:
0074         """
0075 mkdir -p {output.results_dir}
0076 python {input.script} {output.results_dir}
0077 """