| 1 | #!/bin/sh | 
|---|
| 2 | # | 
|---|
| 3 | # ======================================================================== | 
|---|
| 4 | # | 
|---|
| 5 | # * | 
|---|
| 6 | # * This file is part of MARS, the MAGIC Analysis and Reconstruction | 
|---|
| 7 | # * Software. It is distributed to you in the hope that it can be a useful | 
|---|
| 8 | # * and timesaving tool in analysing Data of imaging Cerenkov telescopes. | 
|---|
| 9 | # * It is distributed WITHOUT ANY WARRANTY. | 
|---|
| 10 | # * | 
|---|
| 11 | # * Permission to use, copy, modify and distribute this software and its | 
|---|
| 12 | # * documentation for any purpose is hereby granted without fee, | 
|---|
| 13 | # * provided that the above copyright notice appear in all copies and | 
|---|
| 14 | # * that both that copyright notice and this permission notice appear | 
|---|
| 15 | # * in supporting documentation. It is provided "as is" without express | 
|---|
| 16 | # * or implied warranty. | 
|---|
| 17 | # * | 
|---|
| 18 | # | 
|---|
| 19 | # | 
|---|
| 20 | #   Author(s): Daniela Dorner  05/2006 <mailto:dorner@astro.uni-wuerzburg.de> | 
|---|
| 21 | # | 
|---|
| 22 | #   Copyright: MAGIC Software Development, 2000-2010 | 
|---|
| 23 | # | 
|---|
| 24 | # | 
|---|
| 25 | # ======================================================================== | 
|---|
| 26 | # | 
|---|
| 27 | # This a script, which launches other scripts (all scripts, that are run | 
|---|
| 28 | # on primary basis) | 
|---|
| 29 | # | 
|---|
| 30 |  | 
|---|
| 31 | source `dirname $0`/sourcefile | 
|---|
| 32 | printprocesslog "INFO starting $0" | 
|---|
| 33 |  | 
|---|
| 34 | set -C | 
|---|
| 35 |  | 
|---|
| 36 | # function to continue in loop and go to next script | 
|---|
| 37 | function sleepawhile() | 
|---|
| 38 | { | 
|---|
| 39 | usedsleeptime=$sleeptime | 
|---|
| 40 | case $1 in | 
|---|
| 41 | "error") if ! [ "$errorsleeptime" = "" ] | 
|---|
| 42 | then | 
|---|
| 43 | if [ $errorsleeptime -lt $sleeptimelimit ] | 
|---|
| 44 | then | 
|---|
| 45 | errorsleeptime=`echo " $errorsleeptime + $errorsleeptimedefault " | bc` | 
|---|
| 46 | fi | 
|---|
| 47 | usedsleeptime=$errorsleeptime | 
|---|
| 48 | fi | 
|---|
| 49 | ;; | 
|---|
| 50 | "ok") errorsleeptime=$errorsleeptimedefault | 
|---|
| 51 | ;; | 
|---|
| 52 | esac | 
|---|
| 53 | echo `date +%F\ %T`" sleeping "$usedsleeptime" seconds... (status: "$1")" >> $jmscriptlog 2>&1 | 
|---|
| 54 | echo "" >> $jmscriptlog 2>&1 | 
|---|
| 55 | sleep $usedsleeptime | 
|---|
| 56 | continue | 
|---|
| 57 | } | 
|---|
| 58 |  | 
|---|
| 59 | echo "" >> $jmscriptlog 2>&1 | 
|---|
| 60 | echo "" >> $jmscriptlog 2>&1 | 
|---|
| 61 | echo -n `date +%F\ %T`" starting jobmanager for setup "$AUTOMATIONSETUP >> $jmscriptlog 2>&1 | 
|---|
| 62 |  | 
|---|
| 63 | user=`whoami` | 
|---|
| 64 |  | 
|---|
| 65 | # choose commands according to queueing system (defined in setup) | 
|---|
| 66 | case $queuesys in | 
|---|
| 67 | sge)  echo " on queuing system 'sun grid engine'" >> $jmscriptlog 2>&1 | 
|---|
| 68 | # (-hard) -l hostname=compute-* | 
|---|
| 69 | #   for qstat this returns the jobs running on that node + all jobs in the queue | 
|---|
| 70 | alias 'queuesubmit'='/opt/gridengine/bin/lx26-amd64/qsub -b y -t 1-`echo $tosubmit` -v AUTOMATIONSETUP=$AUTOMATIONSETUP -e `echo $runlogpath`/error-`echo $date`.log -o `echo $runlogpath`/log-`echo $date`.log `echo $noderequirementsub` `echo $scriptspath`/`echo ${scripts[$i]}` ' | 
|---|
| 71 | #            alias 'queuesubmit'='/opt/gridengine/bin/lx26-amd64/qsub -b y -v AUTOMATIONSETUP=$AUTOMATIONSETUP -e `echo $runlogpath`/error-`echo $date`.log -o `echo $runlogpath`/log-`echo $date`.log `echo $scriptspath`/`echo ${scripts[$i]}` ' | 
|---|
| 72 | #            alias 'queuesubmit'='/opt/gridengine/bin/lx26-amd64/qsub -sc runlogpath=`echo $runlogpath` -sc date=`echo $date` -sc scriptspath=`echo $scriptspath` -sc script=`echo ${scripts[$i]}` `echo $scriptspath`/job.sge ' | 
|---|
| 73 | # FIXME: get complete scriptname (including command line option), needed for runstereo | 
|---|
| 74 | alias 'checkqueue'="/opt/gridengine/bin/lx26-amd64/qstat \`echo \$noderequirementstat\`  | awk ' { print \"Owner\"\$4\" \" \$3\"Jobstatus\"\$5 } '" | 
|---|
| 75 | break | 
|---|
| 76 | ;; | 
|---|
| 77 | pbs)  echo " on queuing system 'pbs'" >> $jmscriptlog 2>&1 | 
|---|
| 78 | alias 'queuesubmit'='$pbspath/qsub -t 1-`echo $tosubmit` -l walltime=`echo $walltime` -l pmem=`echo $pmem` -v AUTOMATIONSETUP=$AUTOMATIONSETUP,SOURCEFILEPATH=$SOURCEFILEPATH,SCRIPTNAME=`echo ${scripts[$i]}` -e `echo $runlogpath`/error-`echo $date`.log -o `echo $runlogpath`/log-`echo $date`.log `echo $noderequirementsub` `echo $scriptspath`/`echo ${scripts[$i]}` ' | 
|---|
| 79 | # check queue (restricted to current user only) | 
|---|
| 80 | alias 'checkqueue'="$pbspath/qstat -a -u $user | awk ' { print \"Owner\"\$2\" \" \$4\"Jobstatus\"\$10 } '" | 
|---|
| 81 | break | 
|---|
| 82 | ;; | 
|---|
| 83 | condor)  echo " on queuing system 'condor'" >> $jmscriptlog 2>&1 | 
|---|
| 84 | alias 'queuesubmit'='$condorpath/condor_submit -a path=`echo $scriptspath` -a prog=`echo ${scripts[$i]}` -a date=`echo $date` -a dir=`echo $runlogpath` -a num=`echo $tosubmit` -a automationsetup=$AUTOMATIONSETUP `echo $scriptspath`/run.condor' | 
|---|
| 85 | alias 'checkqueue'='$condorpath/condor_q -global -format "Owner%s " Owner -format "%s" CMD -format "Jobstatus%s\n" Jobstatus ' | 
|---|
| 86 | break | 
|---|
| 87 | ;; | 
|---|
| 88 | *)  echo "" >> $jmscriptlog 2>&1 | 
|---|
| 89 | finish >> $jmscriptlog 2>&1 | 
|---|
| 90 | ;; | 
|---|
| 91 | esac | 
|---|
| 92 |  | 
|---|
| 93 | echo "" >> $jmscriptlog 2>&1 | 
|---|
| 94 |  | 
|---|
| 95 | # for processing with local storage on different nodes | 
|---|
| 96 | currentnode=$minnode | 
|---|
| 97 | numevaluated=0 | 
|---|
| 98 |  | 
|---|
| 99 | # endless loop | 
|---|
| 100 | notcount=0 | 
|---|
| 101 | errorsleeptime=$errorsleeptimedefault | 
|---|
| 102 | while (( $notcount < 100 )) | 
|---|
| 103 | do | 
|---|
| 104 | # get and set some information for the processing | 
|---|
| 105 | source `dirname $0`/sourcefile | 
|---|
| 106 | # reset some values | 
|---|
| 107 | tosubmit=0 | 
|---|
| 108 | idleratio=0 | 
|---|
| 109 | addtoscript= | 
|---|
| 110 |  | 
|---|
| 111 | # get processes in queue | 
|---|
| 112 | q=(`checkqueue 2>&1 `) | 
|---|
| 113 | if echo $q | egrep \(Error\|failed\) | 
|---|
| 114 | then | 
|---|
| 115 | echo `date +%F\ %T`" WARN checking query ($queuesys) failed" >> $jmscriptlog 2>&1 | 
|---|
| 116 | printprocesslog "WARN checking query ($queuesys) failed" | 
|---|
| 117 | echo `date +%F\ %T`" WARN checking query ($queuesys) failed" >> $jmerrorlog | 
|---|
| 118 | sleepawhile "error" | 
|---|
| 119 | fi | 
|---|
| 120 |  | 
|---|
| 121 | # general check whether one should submit something depending on chosen algorithm | 
|---|
| 122 | # algorithm 1: | 
|---|
| 123 | #   submit new jobs in case there are less than $limitidle idle jobs in the queue | 
|---|
| 124 | # algorithm 2: | 
|---|
| 125 | #   submit new jobs in case the total number of jobs in the queue has fallen below $totalpno | 
|---|
| 126 | case $algorithm in | 
|---|
| 127 | 1) # algorithm 1 | 
|---|
| 128 | # get number of idle jobs in the queue | 
|---|
| 129 | q5=( `echo ${q[@]} | egrep -o \(Jobstatus1\|Jobstatusq\|JobstatusQ\)` ) | 
|---|
| 130 | idle=${#q5[@]} | 
|---|
| 131 | if [ $idle -gt $limitidle ] | 
|---|
| 132 | then | 
|---|
| 133 | echo `date +%F\ %T`" more than "$limitidle" jobs waiting ("$idle")" >> $jmscriptlog 2>&1 | 
|---|
| 134 | sleepawhile "ok" | 
|---|
| 135 | fi | 
|---|
| 136 | ;; | 
|---|
| 137 | 2) # algorithm 2 | 
|---|
| 138 | # get processes of user in queue | 
|---|
| 139 | q1=( `echo ${q[@]} | egrep -o "Owner$user"`) | 
|---|
| 140 | queued=${#q1[@]} | 
|---|
| 141 | hour=`date +%k` | 
|---|
| 142 | # choose array of total number of jobs to be done | 
|---|
| 143 | #   according to the day of the week | 
|---|
| 144 | dayofweek=`date +%u` | 
|---|
| 145 | case $dayofweek in | 
|---|
| 146 | 0 | 6)  totalpno=${pnototal[$hour]} ;; | 
|---|
| 147 | *)  totalpno=${pnototalwe[$hour]} ;; | 
|---|
| 148 | esac | 
|---|
| 149 | # get total number of jobs to be submitted | 
|---|
| 150 | if [ $queued -gt $totalpno ] | 
|---|
| 151 | then | 
|---|
| 152 | echo `date +%F\ %T`" more than "$totalpno" jobs waiting ("$queued")" >> $jmscriptlog 2>&1 | 
|---|
| 153 | sleepawhile "ok" | 
|---|
| 154 | else | 
|---|
| 155 | tosubmittotal=`echo "$totalpno - $queued" | bc -l` | 
|---|
| 156 | fi | 
|---|
| 157 | ;; | 
|---|
| 158 | *) echo "Please give an algorithm to calculate the number of allowed jobs." | 
|---|
| 159 | exit | 
|---|
| 160 | ;; | 
|---|
| 161 | esac | 
|---|
| 162 | echo `date +%F\ %T`" Total number of jobs to be submitted: "$tosubmittotal >> $jmscriptlog 2>&1 | 
|---|
| 163 |  | 
|---|
| 164 |  | 
|---|
| 165 | # first loop to determine | 
|---|
| 166 | # a) how many jobs of this script have to be done | 
|---|
| 167 | # b) how many jobs of this script are running or queued | 
|---|
| 168 | for (( i=0 ; i < ${#scripts[@]} ; i++ )) | 
|---|
| 169 | do | 
|---|
| 170 | # set the step to be evaluated | 
|---|
| 171 | step=${scriptscolname[$i]} | 
|---|
| 172 | getstepinfo | 
|---|
| 173 |  | 
|---|
| 174 | # check if the script is restricted to one node | 
|---|
| 175 | #   (i.e. where output of previous step(s) is stored) | 
|---|
| 176 | #   this information is taken from the steps.rc file | 
|---|
| 177 | #   currently this is implemented for sge only | 
|---|
| 178 | # then get number of jobs to be done | 
|---|
| 179 | if [ "$noderestricted" = "yes" ] | 
|---|
| 180 | then | 
|---|
| 181 | # get number of next node | 
|---|
| 182 | if [ $numevaluated -ge $numrestrictedscripts ] | 
|---|
| 183 | then | 
|---|
| 184 | currentnode=`echo " $currentnode + 1 " | bc -l` | 
|---|
| 185 | numevaluated=1 | 
|---|
| 186 | else | 
|---|
| 187 | numevaluated=`echo " $numevaluated + 1 " | bc -l` | 
|---|
| 188 | fi | 
|---|
| 189 | if [ $currentnode -gt $maxnode ] | 
|---|
| 190 | then | 
|---|
| 191 | currentnode=$minnode | 
|---|
| 192 | fi | 
|---|
| 193 | # check if node is excluded | 
|---|
| 194 | for excludednode in ${excludednodes[@]} | 
|---|
| 195 | do | 
|---|
| 196 | if [ $currentnode -eq $excludednode ] | 
|---|
| 197 | then | 
|---|
| 198 | echo `date +%F\ %T`" Node compute-0-$currentnode is currently excluded." >> $jmscriptlog 2>&1 | 
|---|
| 199 | continue 2 | 
|---|
| 200 | fi | 
|---|
| 201 | done | 
|---|
| 202 | # define requirement for submission | 
|---|
| 203 | # FIXME: currently only for sge at isdc | 
|---|
| 204 | echo `date +%F\ %T`" Checking for node $currentnode. " >> $jmscriptlog 2>&1 | 
|---|
| 205 | noderequirementsub=" -hard -l hostname=compute-0-${currentnode}" | 
|---|
| 206 | noderequirementstat=" -l hostname=compute-0-${currentnode}" | 
|---|
| 207 | # get number of jobs to be done from the DB | 
|---|
| 208 | getstatus $currentnode >> $jmscriptlog 2>&1 | 
|---|
| 209 | else | 
|---|
| 210 | noderequirementsub="" | 
|---|
| 211 | noderequirementstat="" | 
|---|
| 212 | # get number of jobs to be done from the DB | 
|---|
| 213 | getstatus >> $jmscriptlog 2>&1 | 
|---|
| 214 | fi | 
|---|
| 215 | # store the number of processes to be done for this script | 
|---|
| 216 | todo[$i]=$numproc | 
|---|
| 217 | tododb[$i]=$numproc | 
|---|
| 218 |  | 
|---|
| 219 | # FIXME: sge cuts scriptname to 8 digits in qstat | 
|---|
| 220 | # number of idle jobs, i.e. jobs waiting in the queue to run | 
|---|
| 221 | #   condor: 1 | 
|---|
| 222 | #   sge: q | 
|---|
| 223 | #   pbs: Q | 
|---|
| 224 | q4=( `echo ${q[@]} | egrep -o \("${scripts[$i]}"Jobstatus1\|"${scripts[$i]}"Jobstatusq\|"${scripts[$i]}"JobstatusQ\)` ) | 
|---|
| 225 | idlescript[$i]=${#q4[@]} | 
|---|
| 226 |  | 
|---|
| 227 | q2=( `echo ${q[@]} | egrep -o "${scripts[$i]}"`) | 
|---|
| 228 | queuedscript[$i]=${#q2[@]} | 
|---|
| 229 |  | 
|---|
| 230 | stillfree[$i]=`echo "${maxjobs[$i]} - ${queuedscript[$i]} " | bc -l` | 
|---|
| 231 |  | 
|---|
| 232 | if [ $numproc -eq 0 ] || [ ${todo[$i]} -le ${idlescript[$i]} ] || [ ${maxjobs[$i]} -le ${queuedscript[$i]} ] | 
|---|
| 233 | then | 
|---|
| 234 | # store the fraction of cpus to add it to another process | 
|---|
| 235 | idleratio=`echo " ${ratio[$i]} + $idleratio " | bc -l` | 
|---|
| 236 | ratio[$i]=0 | 
|---|
| 237 | todo[$i]=0 | 
|---|
| 238 | idlenum=$i | 
|---|
| 239 | continue | 
|---|
| 240 | fi | 
|---|
| 241 | done | 
|---|
| 242 | echo `date +%F\ %T`" Evaluated scripts: "${scripts[@]} >> $jmscriptlog 2>&1 | 
|---|
| 243 | echo `date +%F\ %T`" Running scripts: "${queuedscript[@]}" (max: "${maxjobs[@]}")" >> $jmscriptlog 2>&1 | 
|---|
| 244 | echo `date +%F\ %T`" Number of jobs to be done (from DB): "${tododb[@]} >> $jmscriptlog 2>&1 | 
|---|
| 245 | echo `date +%F\ %T`" Number of jobs to be done (updated): "${todo[@]} >> $jmscriptlog 2>&1 | 
|---|
| 246 | echo `date +%F\ %T`" Ratio: "${ratio[@]}" (idle: "$idleratio")" >> $jmscriptlog 2>&1 | 
|---|
| 247 |  | 
|---|
| 248 | # loop to update the ratio taking into account the ratio of | 
|---|
| 249 | #   a) steps where nothing has to done | 
|---|
| 250 | #   b) steps where already enough jobs are in the queue | 
|---|
| 251 | # sum up this idle ratio | 
|---|
| 252 | # determine for which step still most jobs have to be done | 
|---|
| 253 | if ! [ "$idleratio" = "0" ] | 
|---|
| 254 | then | 
|---|
| 255 | addtoscript= | 
|---|
| 256 | for (( i=0 ; i < ${#scripts[@]} ; i++ )) | 
|---|
| 257 | do | 
|---|
| 258 | if [ ${todo[$i]} -gt ${todo[$idlenum]} ] && [ ${todo[$i]} -gt 0 ] | 
|---|
| 259 | then | 
|---|
| 260 | if ! [ "$addtoscript" = "" ] | 
|---|
| 261 | then | 
|---|
| 262 | if [ ${todo[$i]} -lt ${todo[$addtoscript]} ] | 
|---|
| 263 | then | 
|---|
| 264 | continue | 
|---|
| 265 | fi | 
|---|
| 266 | fi | 
|---|
| 267 | addtoscript=$i | 
|---|
| 268 | fi | 
|---|
| 269 | done | 
|---|
| 270 |  | 
|---|
| 271 | # continue in case nothing has to be done for all steps | 
|---|
| 272 | # else: update the ratio for the step where most jobs have to be done | 
|---|
| 273 | #   by adding the idle ratio | 
|---|
| 274 | if [ "$addtoscript" = "" ] | 
|---|
| 275 | then | 
|---|
| 276 | echo `date +%F\ %T`" No jobs to be done for any step." >> $jmscriptlog 2>&1 | 
|---|
| 277 | sleepawhile "ok" | 
|---|
| 278 | else | 
|---|
| 279 | ratio[$addtoscript]=`echo " ${ratio[$addtoscript]} + $idleratio " | bc -l` | 
|---|
| 280 | fi | 
|---|
| 281 | fi | 
|---|
| 282 | echo `date +%F\ %T`" Updated ratio: "${ratio[@]} >> $jmscriptlog 2>&1 | 
|---|
| 283 |  | 
|---|
| 284 |  | 
|---|
| 285 | # loop to submit jobs to queueing system | 
|---|
| 286 | for (( i=0 ; i < ${#scripts[@]} ; i++ )) | 
|---|
| 287 | do | 
|---|
| 288 | # calculate number of jobs to be submitted | 
|---|
| 289 | tosubmit=`echo "scale=0; $tosubmittotal * ${ratio[$i]} / 1 " | bc -l` | 
|---|
| 290 | if [ ${todo[$i]} -lt $tosubmit ] | 
|---|
| 291 | then | 
|---|
| 292 | echo `date +%F\ %T`" Updating tosubmit for "${scripts[$i]}" from "$tosubmit" to "${todo[$i]} >> $jmscriptlog 2>&1 | 
|---|
| 293 | tosubmit=${todo[$i]} | 
|---|
| 294 | fi | 
|---|
| 295 | if [ $tosubmit -eq 0 ] | 
|---|
| 296 | then | 
|---|
| 297 | echo `date +%F\ %T`" No jobs to be submitted for script '"${scripts[$i]}"'" >> $jmscriptlog 2>&1 | 
|---|
| 298 | continue | 
|---|
| 299 | fi | 
|---|
| 300 | if [ $tosubmit -gt ${stillfree[$i]} ] | 
|---|
| 301 | then | 
|---|
| 302 | echo `date +%F\ %T`" Updating tosubmit for "${scripts[$i]}" from "$tosubmit" to "${stillfree[$i]} >> $jmscriptlog 2>&1 | 
|---|
| 303 | tosubmit=${stillfree[$i]} | 
|---|
| 304 | fi | 
|---|
| 305 |  | 
|---|
| 306 | # set the step to be evaluated | 
|---|
| 307 | step=${scriptscolname[$i]} | 
|---|
| 308 | # check if walltime has to be set | 
|---|
| 309 | if [ "$setwalltime" = "yes" ] | 
|---|
| 310 | then | 
|---|
| 311 | walltime=${walltimes[$i]} | 
|---|
| 312 | fi | 
|---|
| 313 | # check if memory has to be set | 
|---|
| 314 | if [ "$setpmem" = "yes" ] | 
|---|
| 315 | then | 
|---|
| 316 | pmem=${pmems[$i]} | 
|---|
| 317 | fi | 
|---|
| 318 |  | 
|---|
| 319 | # submit $tosubmit scripts to queuing system | 
|---|
| 320 | echo `date +%F\ %T`" Submitting "$tosubmit" jobs for script '"${scripts[$i]}"' to "$queuesys >> $jmscriptlog 2>&1 | 
|---|
| 321 | date=`date +%Y-%m-%d` | 
|---|
| 322 | if ! queuesubmit 2>> $jmerrorlog | 
|---|
| 323 | then | 
|---|
| 324 | echo `date`" WARN submitting job ($queuesys) failed" >> $jmerrorlog | 
|---|
| 325 | echo `date +%F\ %T`" WARN $queuesys is not working -> sleeping $errorsleeptime [\$errorsleeptime]" >> $jmscriptlog 2>&1 | 
|---|
| 326 | printprocesslog "WARN submitting ${scripts[$i]} ($queuesys) failed" | 
|---|
| 327 | sleepawhile "error" | 
|---|
| 328 | fi | 
|---|
| 329 | done | 
|---|
| 330 | sleepawhile "ok" | 
|---|
| 331 | done | 
|---|
| 332 |  | 
|---|