#!/bin/bash

#
# variables from arguments string in jdl
#
# format:
#
# 1: any unique identifier
# 2: condor process number
# 3: output directory (to store all files created here)
# 4: running directory (CMSSW_X_Y_Z/subdir)
# 5: the filename of the parameter set (inside run dir)
# 6: starting seed, where the seed in the config will be
#    the starting seed + 100 * the process number,
#    if job doesn't require seed, just provide a dummy
#    input number
# 7: number of events per job

UNIQUE_ID=$1
CONDOR_PROCESS=$2
OUT_DIR=$3
RUN_DIR=$4
PARAMETER_SET=$5
RANDOM_NUMBER=$6
NUM_EVENTS_PER_JOB=$7

#
# header 
#

echo ""
echo "CMSSW on Condor"
echo ""

START_TIME=`/bin/date`
echo "started at $START_TIME"

echo ""
echo "parameter set:"
echo "UNIQUE_ID: $UNIQUE_ID"
echo "CONDOR_PROCESS: $CONDOR_PROCESS"
echo "OUT_DIR: $OUT_DIR"
echo "RUN_DIR: $RUN_DIR"
echo "PARAMETER_SET: $PARAMETER_SET"
echo "RANDOM_NUMBER: $RANDOM_NUMBER"
echo "NUM_EVENTS_PER_JOB: $NUM_EVENTS_PER_JOB"

#
# setup software environment at FNAL for the given CMSSW release
#
export VO_CMS_SW_DIR=/sharesoft/cmssw
. $VO_CMS_SW_DIR/cmsset_default.sh
cd $RUN_DIR
eval `scramv1 runtime -sh`

#
# modify parameter-set
#

FINAL_PREFIX_NAME=`echo ${OUT_DIR}/${UNIQUE_ID}_${CONDOR_PROCESS}`
FINAL_PARAMETER_SET=`echo $FINAL_PREFIX_NAME.py`
RANDOM_LOG=`echo ${UNIQUE_ID}_${CONDOR_PROCESS}.random`
FINAL_LOG=`echo $FINAL_PREFIX_NAME.log`
FINAL_FILENAME=`echo $FINAL_PREFIX_NAME.output.root`
FINAL_HISTONAME=`echo $FINAL_PREFIX_NAME.histo.root`
let "FINAL_RANDOMNUMBER = $CONDOR_PROCESS*100 + $RANDOM_NUMBER"
let "SKIP_EVENTS = $CONDOR_PROCESS*$NUM_EVENTS_PER_JOB"
let "PROCESS_PLUS_ONE = $CONDOR_PROCESS+1"
echo ""
echo "Writing final parameter-set: $FINAL_PARAMETER_SET to OUT_DIR: $OUT_DIR"
echo ""

cat $PARAMETER_SET \
| sed -e s~CONDOR_RANDOMLOG~$RANDOM_LOG~ \
| sed -e s/CONDOR_RANDOMNUMBER/$FINAL_RANDOMNUMBER/ \
| sed -e s/CONDOR_MAXEVENTS/$NUM_EVENTS_PER_JOB/ \
| sed -e s/CONDOR_SKIPEVENTS/$SKIP_EVENTS/ \
| sed -e s~CONDOR_OUTPUTFILENAME~$FINAL_FILENAME~ \
| sed -e s~CONDOR_HISTOFILENAME~$FINAL_HISTONAME~ \
> $FINAL_PARAMETER_SET

#
# run cmssw
#

echo "run: time cmsRun $FINAL_PARAMETER_SET > $FINAL_LOG 2>&1"
cmsRun $FINAL_PARAMETER_SET >> $FINAL_LOG 2>&1
exitcode=$?

# The RandomNumberServiceHelper wants the log placed in the running dir.
# Transfer the random # log to the out_dir once job completes.
echo "Moving random log, if it exists.  Ignore error if file does not exist, as not all jobs have random seed logs."
mv $RUN_DIR/$RANDOM_LOG $OUT_DIR/$RANDOM_LOG

#
# end run
#

echo ""
END_TIME=`/bin/date`
echo "finished at $END_TIME"
exit $exitcode