#!/usr/bin/env bash
#------------------------------------------------------------------------------
# usage: cdfopr/scripts/submit_stnmaker_prod dataset[:book] \
#                                            sec1 sec2 [caf[:mode]] [queue:nevents]
#
#  edit the script to put in 
#  - right location of the tarball 
#  - queue
#  - logic defining output directory 
#  - user email address for CAF
#
# to run 1 job on local machine: 
# ------------------------------
#         cdfopr/scripts/submit_stnmaker_prod zewk0d 1 10 local:test
#
# to run 300 jobs on Fermilab CAF:
# --------------------------------
#         cdfopr/scripts/submit_stnmaker_prod wewk9t 501 800 caf
#------------------------------------------------------------------------------
 dataset=`echo $1 | awk -F : '{print $1}'`
    book=`echo $1 | awk -F : '{print $2}'`
    caf="local"; 
    job1=$2
    job2=$3
    mode="submit";
  prefix=""
  extra_parameters="none"
  dh_access=DFC

    if [ .$4 != "." ] ; then 
      caf=`echo $4 | awk -F : '{print $1}'` 
        x=`echo $4 | awk -F : '{print $2}'`; 
       if [ .$x != "." ] ; then mode=$x ; fi
    fi

    if [ $mode == "debug" ] ; then prefix="echo" ; fi

if [ $caf == "cafcondor" ] ; then queue=medium ; else queue=medium ; fi

if [ .$5  != "."         ] ; then 
  queue=`echo $5 | awk -F : '{print $1}'` 
      x=`echo $5 | awk -F : '{print $2}'`; 

  if [ .$x != "." ] ; then nevents=$x ; fi
fi

echo book=$book dataset=$dataset mode=$mode caf=$caf

input_tcl_file=
    host=`hostname -f`
     tar=stnmaker_prod_6.1.4_dev_243_07.tar.gz
    user=`klist | grep Default | awk '{print $3}' | awk -F @ '{print $1}'`

if [ ".$CDF_TARBALL_DIR" != "." ] ; then 
  tarball_dir=$CDF_TARBALL_DIR
else
  case $host in 
    "ncdf131.fnal.gov"  ) 
       tarball_dir=/cdf/data01/tarballs ;;
    "fcdflnx3.fnal.gov" ) 
       tarball_dir=/cdf/opr/cdfopr/tarballs ;;
    * )
       tarball_dir= ;;
  esac
fi

if [ $caf == "caf" ] ; then 
  if [ $user == "cdfopr" ] ; then
    queue=cdfopr
  else
    queue=medium
  fi
elif [ $caf == ".cafcondor" ] ; then 
  queue=medium ; 
else 
  queue=medium ; 
fi

tarball=$tarball_dir/$tar

output_node=cdfopr@fcdfdata053.fnal.gov
 output_dir=/cdf/scratch/cdfopr/datasets/$book/$dataset/tmp

if [ .$book == "." ] ; then book="cdfpewk" ; fi

input_dataset=$dataset

case $dataset in
    "bewk0d" )                                     # incl high-Pt electrons
        job=stnmaker.prod
        input_dataset=bhel0d
        input_tcl_file=cdfopr/tcl/stnmaker/filter_hpte.tcl
        output_node=murat@fcdfdata122
        output_dir=/export/data1/ewk/murat/datasets/cdfpewk/$dataset/tmp
        queue=short
	;;
    "bewk2d" )                                     # incl high-Pt muons
        job=stnmaker.prod
        input_dataset=bhmu0d
        input_tcl_file=cdfopr/tcl/stnmaker/filter_hptm.tcl
        output_node=cdfopr@fcdflnx6
        output_dir=/cdf/data58/s0/production/datasets/cdfpewk/$dataset/tmp
        queue=short
	;;
    "eewk0d" )                                     # MET25_PS100 and such
        job=stnmaker.prod
	input_dataset="emet0d"
	input_tcl_file=cdfopr/tcl/stnmaker/filter_met25_ps100.tcl
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
        queue=short
	;;
    "gcrs0d" )                                     # zero-bias (0d-0e) ntupling
        job=stnmaker.prod
	;;
    "sewk3d" )                                     # SUM ET dataset (0D)
	input_dataset="gset0d"
        job=stnmaker.prod
        output_node=ewk@fcdfdata054
        output_dir=/cdf/scratch/ewk/datasets/$book/$dataset/tmp
        extra_parameters=" -x USE_WWW=1"
	;;
    "mewkpp" )
        job=stnmaker.prod:mc
	;;
    "mewktp" )
        job=stnmaker.prod:mc
	;;
    "tmet0d" )                                     # W->tau nu strip+ntupling
        job=stnmaker.prod
        input_dataset=etau0d
        input_tcl_file=cdfopr/tcl/stnmaker/filter_taumet.tcl
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
        queue=short
	;;
    "wewk0d" )                                     # W->enu (calib)
        job=stnmaker.prod
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
	;;
    "wewk10" )                                     # W_NOTRACK, 314 files
        job=stnmaker.prod
        input_dataset=wewk10
        output_node=cdfopr@fcdflnx6
        output_dir=/cdf/data58/s0/production/datasets/cdfpewk/$dataset/tmp
        queue=short
	;;
    "wewk2d" )                                     # W->munu (calib)
        job=stnmaker.prod
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
	;;
    "wewk8t" )                                     # 5.3.3 W->taunu MC
        job=stnmaker.prod:mc
	;;
    "wewk9t" )                                     # 5.3.3 W->taunu MC v2
        job=stnmaker.prod:mc
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/$book/$dataset/tmp
	;;
    "wewk9t_test" )                                     # 5.3.3 W->taunu MC v2
        job=stnmaker.prod:mc
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/$book/$dataset/tmp
        extra_parameters=" -x USE_WWW=1"
	;;
    "wewkee" )
        job=stnmaker.prod:mc
	;;
    "zewk0d" )                                     # Z->ee
        job=stnmaker.prod
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
	;;
    "zewk2d" )                                     # Z->mumu
        job=stnmaker.prod
        output_node=ewk@fcdfdata030
        output_dir=/cdf/scratch/ewk/datasets/cdfpewk/$dataset/tmp
	;;
    "zewk3m" )                                     # Z->mumu 5.3.3 MC
        job=stnmaker.prod:mc
	;;
    "zewkae" )                                     # Z->ee 5.3.3 MC
        job=stnmaker.prod:mc
	;;
    * )
#-----------------------------------------------------------------------
#  default: get definition of the dataset from tikiwiki
#  fewk01, wewkfe
#-----------------------------------------------------------------------
        tiki_host=www-cdf.fnal.gov
        delimitor=stnmaker_prod.parameters
        parameter_file=$dataset.$delimitor
	. cdfopr/scripts/tiki_get_data $tiki_host wiki_page . $book.$dataset $delimitor > $parameter_file
	cat $parameter_file

        script=cdfopr/scripts/tiki_get_parameter

              mc_flag=`$script $parameter_file mc_flag        $job1`
        input_dataset=`$script $parameter_file input_dataset  $job1`
       output_dataset=`$script $parameter_file output_dataset $job1`
                queue=`$script $parameter_file queue          $job1`
          output_node=`$script $parameter_file output_node    $job1`
           output_dir=`$script $parameter_file output_dir     $job1`
            dh_access=`$script $parameter_file dh_access      $job1`
       input_tcl_file=`$script $parameter_file input_tcl_file $job1`

        rm $parameter_file
        
        job=stnmaker.prod:$mc_flag

        echo [submit_stnmaker_prod]: mc_flag=$mc_flag
        echo [submit_stnmaker_prod]: input_dataset=$input_dataset
        echo [submit_stnmaker_prod]: input_tcl_file=$input_tcl_file
        echo [submit_stnmaker_prod]: output_dataset=$output_dataset
        echo [submit_stnmaker_prod]: output_node=$output_node
        echo [submit_stnmaker_prod]: output_dir=$output_dir
        echo [submit_stnmaker_prod]: queue=$queue
        echo [submit_stnmaker_prod]: dh_access=$dh_access
        echo [submit_stnmaker_prod]: input_tcl_file=$input_tcl_file
#-----------------------------------------------------------------------
#  get script if anything
#-----------------------------------------------------------------------
        delimitor=stnmaker_prod.script
        script_file=$dataset.$delimitor
	. cdfopr/scripts/tiki_get_data $tiki_host wiki_page . $book.$dataset $delimitor > $script_file
        cat    $script_file
        source $script_file

        extra_parameters=" -x USE_TIKI=$tiki_host -x DATA_CATALOG=$dh_access"
        ;;
esac

parameters="-e bin/$BFARCH/stnmaker_prod.exe -j $job"
parameters=${parameters}" -b $book -D $input_dataset -o ${output_node}:$output_dir"
parameters=${parameters}" -x CAF=$caf"

if [ .$input_tcl_file == ."dfc" ] ; then
  parameters=${parameters}" -x USE_DFC=1"
fi

output_tarball=${output_node}:$output_dir/stnmaker_prod_$dataset.$.tgz

if [ .$input_tcl_file != "." ] ; then
  parameters=${parameters}" -i $input_tcl_file "
fi

if [ ."$nevents" != "." ] ; then
  parameters=${parameters}" -n $nevents"
fi

if [ "$extra_parameters" != "none" ] ; then
  parameters=${parameters}" $extra_parameters"
fi


if   [ $mode == "test"     ] ; then 
  if [ $caf == "local" ] ; then
#-----------------------------------------------------------------------
#  local test job
#-----------------------------------------------------------------------
    cmd="$prefix ./cdfopr/scripts/run.sh -V debug -J $job1 $parameters -n 1000 &"
  else 
#-----------------------------------------------------------------------
#  test remote job
#-----------------------------------------------------------------------
    source ~cdfsoft/cdf2.shrc
    setup cdfsoft2 development

    if [ $caf == "caf"       ] ; then queue=test ; fi
    if [ $caf == "cafcondor" ] ; then queue=test ; fi

    cmd="$prefix CafSubmit --tarFile=$tarball        \
                      --outLocation=$output_tarball  \
                      --dhaccess=$dh_access          \
                      --dataset=$input_dataset       \
                      --procType=$queue              \
                      --email=${user}@fnal.gov       \
                      --farm=$caf                    \
                      --start=$job1 --end=$job2      \
         ./cdfopr/scripts/run.sh -V debug -J $ $parameters -n 10"
  fi
else
#-----------------------------------------------------------------------
#  remote job
#-----------------------------------------------------------------------
  if [ $caf == "local" ] ; then
#-----------------------------------------------------------------------
#  local job
#-----------------------------------------------------------------------
    cmd="$prefix ./cdfopr/scripts/run.sh -V debug -J $job1 $parameters &"
  else 
    source ~cdfsoft/cdf2.shrc
    setup cdfsoft2 development

    cmd="$prefix CafSubmit --tarFile=$tarball \
                      --outLocation=$output_tarball  \
                      --dhaccess=$dh_access          \
                      --dataset=$input_dataset       \
                      --procType=$queue              \
                      --email=${user}@fnal.gov       \
                      --farm=$caf                    \
                      --start=$job1 --end=$job2      \
       ./cdfopr/scripts/run.sh -V debug -J $ $parameters"
  fi
fi

echo $cmd
$cmd