env.sh

#!/bin/bash

## =============================================================================
## Global configuration variables for the benchmark scripts
## The script defines the following environment variables that are meant to
## be overriden by the Gitlab continuous integration (CI)
##
##  - BEAMLINE_CONFIG:         compact detector files for the interaciton point beamline
##  - BEAMLINE_CONFIG_VERSION: compact detector files for the interaciton point beamline
##  - JUGGLER_DETECTOR:        detector package to be used for the benchmark
##  - JUGGLER_DETECTOR_VERISON:        detector package to be used for the benchmark
##  - JUGGLER_N_EVENTS:        events processed by simulation/reconstruction
##  - JUGGLER_INSTALL_PREFIX:  location where Juggler (digi/recon) is installed
##  - JUGGLER_N_THREADS:       Number of threads/processes to spawn in parallel
##  - JUGGLER_RNG_SEED:        Random seed for the RNG
##
## It also defines the following additional variables for internal usage
##  - LOCAL_PREFIX:           prefix for packages installed during the benchmark
##  - LOCAL_DATA_PATH:        local storage for pipeline jobs
##  - DETECTOR_PREFIX:        prefix for the detector definitions
##  - DETECTOR_PATH:          actual path with the detector definitions
##
## Finally, it makes sure LOCAL_PREFIX and JUGGLER_PREFIX are added to PATH
## and LD_LIBRARY_PATH
## =============================================================================

echo "Setting up the Physics Benchmarks environment"

## =============================================================================
## Default variable definitions, normally these should be set
## by the CI. In case of local development you may want to change these
## in case you would like to modify the detector package or
## number of events to be analyzed during the benchmark

if [ ! -n  "${BEAMLINE_CONFIG}" ] ; then 
  export BEAMLINE_CONFIG="ip6"
fi

if [ ! -n  "${BEAMLINE_CONFIG_VERSION}" ] ; then 
  export BEAMLINE_CONFIG_VERSION="master"
fi


## Detector package to be used during the benchmark process
if [ ! -n  "${JUGGLER_DETECTOR}" ] ; then 
  export JUGGLER_DETECTOR="athena"
fi

if [ ! -n  "${JUGGLER_DETECTOR_VERSION}" ] ; then 
  export JUGGLER_DETECTOR_VERSION="master"
fi


## Number of events that will be processed by the reconstruction
if [ ! -n  "${JUGGLER_N_EVENTS}" ] ; then 
  export JUGGLER_N_EVENTS=100
fi

## Maximum number of threads or processes a single pipeline should use
## (this is not enforced, but the different pipeline scripts should use
##  this to guide the number of parallel processes or threads they 
##  spawn).
if [ ! -n "${JUGGLER_N_THREADS}" ]; then
  export JUGGLER_N_THREADS=10
fi

## Random seed for event generation, should typically not be changed for
## reproductability.
if [ ! -n "${JUGGLER_RNG_SEED}" ]; then
  export JUGGLER_RNG_SEED=1
fi

## Install prefix for juggler, needed to locate the Juggler xenv files.
## Also used by the CI as install prefix for other packages where needed.
## You should not have to touch this. Note that for local usage a different 
## prefix structure is automatically used.
if [ ! -n  "${JUGGLER_INSTALL_PREFIX}" ] ; then 
  export JUGGLER_INSTALL_PREFIX="/usr/local"
fi
## Ensure the juggler prefix is an absolute path
export JUGGLER_INSTALL_PREFIX=`realpath ${JUGGLER_INSTALL_PREFIX}`


## Location of local data for pass data from job to job within pipeline.
## Not saved as artifacts.
## Local /scratch directory is presumed to be writable. 
if [ ! -n  "${LOCAL_DATA_PATH}" ] ; then 
  export LOCAL_DATA_PATH="/scratch/${CI_PROJECT_NAME}_${CI_PIPELINE_ID}"
fi
mkdir -p "${LOCAL_DATA_PATH}"
if [ ! -d "${LOCAL_DATA_PATH}" ]; then 
  echo "LOCAL_DATA_PATH (${LOCAL_DATA_PATH}) does not exist!!"
  echo "Creating LOCAL_DATA_PATH=$(pwd)/local_data "
  export LOCAL_DATA_PATH="$(pwd)/local_data"
  mkdir -p "${LOCAL_DATA_PATH}"
fi

## =============================================================================
## Other utility variables that govern how some of the dependent packages
## are built and installed. You should not have to change these.

## local prefix to be used for local storage of packages
## downloaded/installed during the benchmark process
LOCAL_PREFIX=".local"
mkdir -p "${LOCAL_PREFIX}"
export LOCAL_PREFIX=`realpath ${LOCAL_PREFIX}`

## detector prefix: prefix for the detector definitions
export DETECTOR_PREFIX="${LOCAL_PREFIX}/detector"
mkdir -p ${DETECTOR_PREFIX}

## detector path: actual detector definition path
export DETECTOR_PATH="${DETECTOR_PREFIX}/${JUGGLER_DETECTOR}"

## build dir for ROOT to put its binaries etc.
export ROOT_BUILD_DIR=$LOCAL_PREFIX/root_build

export ROOT_INCLUDE_PATH=${LOCAL_PREFIX}/include:${ROOT_INCLUDE_PATH}

echo "JUGGLER_DETECTOR:           ${JUGGLER_DETECTOR}"
echo "JUGGLER_DETECTOR_VERSION:   ${JUGGLER_DETECTOR_VERSION}"
echo "JUGGLER_N_EVENTS:           ${JUGGLER_N_EVENTS}"
echo "JUGGLER_N_THREADS:          ${JUGGLER_N_THREADS}"
echo "JUGGLER_RNG_SEED:           ${JUGGLER_RNG_SEED}"
echo "JUGGLER_INSTALL_PREFIX:     ${JUGGLER_INSTALL_PREFIX}"
echo "LOCAL_PREFIX:               ${LOCAL_PREFIX}"
echo "DETECTOR_PREFIX:            ${DETECTOR_PREFIX}"
echo "DETECTOR_PATH:              ${DETECTOR_PATH}"
echo "ROOT_BUILD_DIR:             ${ROOT_BUILD_DIR}"
echo "ROOT_INCLUDE_PATH:          ${ROOT_INCLUDE_PATH}"
echo "BEAMLINE_CONFIG:            ${BEAMLINE_CONFIG}"
echo "BEAMLINE_CONFIG_VERSION:    ${BEAMLINE_CONFIG_VERSION}"
echo "LOCAL_DATA_PATH:            ${LOCAL_DATA_PATH}"

## =============================================================================
## Setup PATH and LD_LIBRARY_PATH to include our prefixes
echo "Adding JUGGLER_INSTALL_PREFIX and LOCAL_PREFIX to PATH and LD_LIBRARY_PATH"
export PATH=${LOCAL_PREFIX}/bin:${JUGGLER_INSTALL_PREFIX}/bin:${PATH}
export LD_LIBRARY_PATH=${LOCAL_PREFIX}/lib:${JUGGLER_INSTALL_PREFIX}/lib:${LD_LIBRARY_PATH}

mkdir -p ${LOCAL_DATA_PATH}/fieldmaps
pushd ${LOCAL_DATA_PATH}/fieldmaps
if [ ! -f "EIC_Magnetic_Field_Map_2021_05_07_radial_coords__cm___T_.120000.lines.Bmap " ] ; then
  echo "Downloading fieldmap: EIC_Magnetic_Field_Map_2021_05_07_radial_coords__cm___T_.120000.lines.Bmap"
  wget -q "https://eicweb.phy.anl.gov/EIC/detectors/athena/uploads/9179a98c5a80df1f2b98968c747c1c48/EIC_Magnetic_Field_Map_2021_05_07_radial_coords__cm___T_.120000.lines.Bmap" 
fi
popd
ln -sf ${LOCAL_DATA_PATH}/fieldmaps

## =============================================================================
## That's all!
echo "Environment setup complete."