From a6f560af1677e56744c9ee1c2407a02a50d3c653 Mon Sep 17 00:00:00 2001
From: Sylvester Joosten <sjoosten@anl.gov>
Date: Sat, 7 Nov 2020 23:29:11 +0000
Subject: [PATCH] Generalized event generation step
---
dvmp/scripts/generate.sh | 189 ++++++++++++++++++++++++++++++++++++
dvmp/scripts/print_fname.sh | 60 ++++++++++++
2 files changed, 249 insertions(+)
create mode 100755 dvmp/scripts/generate.sh
create mode 100755 dvmp/scripts/print_fname.sh
diff --git a/dvmp/scripts/generate.sh b/dvmp/scripts/generate.sh
new file mode 100755
index 00000000..76a62a52
--- /dev/null
+++ b/dvmp/scripts/generate.sh
@@ -0,0 +1,189 @@
+#!/bin/bash
+
+## Init the environment
+source config/env.sh
+
+## Generator configuration
+## We use generator-level n-events, which needs to be larger or equal to
+## the number of juggler events
+export NEVENTS=1000
+if [ ${JUGGLER_N_EVENTS} \> ${NEVENTS} ]; then
+ NEVENTS=${JUGGLER_N_EVENTS}
+fi
+
+## Our random seed
+export RNG_SEED=1
+
+## Setup local environment
+export DATA_PATH=$RESULTS_PATH/data/dvmp
+
+## Maximum number of generators to run in parallel
+export MT=10
+
+run_generator() {
+ EBEAM=
+ PBEAM=
+ DECAY=
+ CONFIG=
+
+ while [ $# -gt 0 ]; do
+ key=$1
+ case $key in
+ --ebeam)
+ EBEAM="$2"
+ shift
+ shift
+ ;;
+ --pbeam)
+ PBEAM="$2"
+ shift
+ shift
+ ;;
+ --decay)
+ DECAY="$2"
+ shift
+ shift
+ ;;
+ --config)
+ CONFIG="$2"
+ shift
+ shift
+ ;;
+ *)
+ echo "Unknown option $key to run_generator(), aborting..."
+ exit 1
+ esac
+ done
+
+ if [ -z $EBEAM ]; then
+ echo "EBEAM not defined in run_generator, aborting..."
+ exit 1
+ elif [ -z $PBEAM ]; then
+ echo "PBEAM not defined in run_generator, aborting..."
+ exit 1
+ elif [ -z $DECAY ]; then
+ echo "DECAY not defined in run_generator, aborting..."
+ exit 1
+ elif [ $DECAY != "electron" ] && [ $DECAY != "muon" ] ; then
+ echo "Unknown decay channel $DECAY, aborting..."
+ exit 1
+ elif [ -z $CONFIG ]; then
+ echo "CONFIG not defined in run_generator, aborting..."
+ fi
+
+ pushd dvmp
+ FNAME=`scripts/print_fname.sh \
+ --ebeam $EBEAM \
+ --pbeam $PBEAM \
+ --decay $DECAY \
+ --config $CONFIG \
+ --type gen`
+ if [ -f "${DATA_PATH}/${FNAME}.hepmc" ]; then
+ echo "Found cached generator output for $FNAME, no need to rerun"
+ return
+ fi
+
+ echo "Generator output for $FNAME not found in cache, need to run generator"
+
+ ## process decay info
+ BRANCHING=
+ DECAY_PID=
+ if [ $DECAY = "electron"]; then
+ BRANCHING="0.05971"
+ DECAY_PID="11"
+ elif [ $DECAY = "muon"]; then
+ BRANCHING="0.05961"
+ DECAY_PID="13"
+ fi
+
+ ## generate the config file for this generator setup
+ CONFIG_IN="generator/${CONFIG}.json.in"
+ echo "Creating generator configuration file ${FNAME}.json"
+ if [ ! -f ${CONFIG_IN} ]; then
+ echo "ERROR: cannot find master config file ${CONFIG_IN}"
+ exit 1
+ fi
+ sed "s/@TAG@/${FNAME}/" $CONFIG_IN | \
+ sed "s/@EBEAM@/${EBEAM}/" | \
+ sed "s/@PBEAM@/${PBEAM}/" | \
+ sed "s/@DECAY_LEPTON@/${DECAY_PID}/" | \
+ sed "s/@BRANCHING@/${BRANCHING}/" > ${FNAME}.json
+
+ ## New we can run the generator
+ echo "Running the generator"
+ lager -r ${RNG_SEED} -c ${FNAME}.json -e ${NEVENTS} -o .
+
+ ## Finally, move relevant output into the artifacts directory
+ echo "Moving generator output into ${DATA_PATH}"
+ mkdir -p ${DATA_PATH}
+ for ext in hepmc json log root; do
+ mv *.${FNAME}.*.${ext} ${DATA_PATH}/${FNAME}.${ext}
+ done
+}
+
+
+
+## Generates different configurations from the master configuration
+## for both electron and muon decay channels
+
+echo "Generating generator configuration files for J/psi -> e+e- and mu+mu-"
+
+EBEAM=
+PBEAM=
+DECAYS=()
+CONFS=()
+
+while [ $# -gt 0 ]
+do
+ key="$1"
+ case $key in
+ --config)
+ CONFS+=("$2")
+ shift # past argument
+ shift # past value
+ ;;
+ --decay)
+ DECAYS+=("$2")
+ shift # past argument
+ shift # past value
+ ;;
+ --ebeam)
+ EBEAM="$2"
+ shift # past argument
+ shift # past value
+ ;;
+ --pbeam)
+ PBEAM="$2"
+ shift # past argument
+ shift # past value
+ ;;
+ *) # unknown option
+ echo "unknown option"
+ exit 1
+ ;;
+ esac
+done
+
+if [ ${#CONFS[@]} -eq 0 ]; then
+ echo "ERROR: need one or more config names: --config <config name> "
+ exit 1
+elif [ ${#DECAYS[@]} -eq 0 ]; then
+ echo "ERROR: need one or more decay types: --decay muon/electron"
+ exit 1
+elif [ -z $EBEAM ]; then
+ echo "ERROR: EBEAM not defined: --EBEAM <energy>"
+ exit 1
+elif [ -z $PBEAM ]; then
+ echo "ERROR: PBEAM not defined: --PBEAM <energy>"
+ exit 1
+fi
+
+## loop over all our configurations and run the generator in parallel
+
+parallel -j ${MT} \
+ run_generator --ebeam ${EBEAM} --pbeam ${PBEAM} --config {1} --decay {2} \
+ ::: "${CONFS[@]}" ::: "${DECAYS[@]}"
+
+CONFIG_BASE=`basename ${CONFIG} .json.in`
+
+echo "Event generation finished"
diff --git a/dvmp/scripts/print_fname.sh b/dvmp/scripts/print_fname.sh
new file mode 100755
index 00000000..3f298666
--- /dev/null
+++ b/dvmp/scripts/print_fname.sh
@@ -0,0 +1,60 @@
+#!/bin/bash
+
+## Simple script to output a unified file name based on a set of data options
+
+EBEAM=
+PBEAM=
+DECAY=
+CONFIG=
+TYPE=
+
+while [ $# -gt 0 ]; do
+ key=$1
+ case $key in
+ --ebeam)
+ EBEAM="$2"
+ shift
+ shift
+ ;;
+ --pbeam)
+ PBEAM="$2"
+ shift
+ shift
+ ;;
+ --decay)
+ DECAY="$2"
+ shift
+ shift
+ ;;
+ --config)
+ CONFIG="$2"
+ shift
+ shift
+ ;;
+ --stage)
+ TYPE="$2"
+ shift
+ shift
+ ;;
+ *)
+ echo "Unknown option $key to print_fname, aborting..."
+ exit 1
+ esac
+done
+
+if [ -z $EBEAM ]; then
+ echo "EBEAM not defined in print_fname, aborting..."
+ exit 1
+elif [ -z $PBEAM ]; then
+ echo "PBEAM not defined in print_fname, aborting..."
+ exit 1
+elif [ -z $DECAY ]; then
+ echo "DECAY not defined in print_fname, aborting..."
+ exit 1
+elif [ -z $CONFIG ]; then
+ echo "CONFIG not defined in print_fname, aborting..."
+elif [ -z $TYPE ]; then
+ echo "TYPE not defined in print_fname, aborting..."
+fi
+
+echo "${CONFIG}_${DECAY}-${EBEAM}on${PBEAM}-${TYPE}"
--
GitLab