-
Notifications
You must be signed in to change notification settings - Fork 0
/
SubmitLHEGENSIM.py
102 lines (79 loc) · 3.3 KB
/
SubmitLHEGENSIM.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#!/usr/bin/python
#-----------------------------------------------
# Latest update: 2014.09.14
#-----------------------------------------------
import sys, os, pwd, commands
import optparse, shlex, re
import time
from time import gmtime, strftime
import math
#define function for parsing options
def parseOptions():
global observalbesTags, modelTags, runAllSteps
usage = ('usage: %prog [options]\n'
+ '%prog -h for help')
parser = optparse.OptionParser(usage)
# input options
parser.add_option('-t', '--tag', dest='TAG', type='string',default='', help='string to tag your job')
parser.add_option('-g', '--gridpack', dest='GRIDPACK', type='string',default='', help='gridpack you want to generate events from')
parser.add_option('-f', '--fragment', dest='FRAGMENT', type='string',default='', help='GEN-SIM fragment')
parser.add_option('-n', '--njobs', dest='NJOBS', type=int, default=1, help='number of jobs')
parser.add_option('-e', '--eventsperjob', dest='EVENTSPERJOB', type=int, default=1000, help='events per job')
# store options and arguments as global variables
global opt, args
(opt, args) = parser.parse_args()
# define function for processing the external os commands
def processCmd(cmd, quite = 0):
# print cmd
status, output = commands.getstatusoutput(cmd)
if (status !=0 and not quite):
print 'Error in processing command:\n ['+cmd+']'
print 'Output:\n ['+output+'] \n'
return output
def submitLHEGENSIM():
# parse the arguments and options
global opt, args
parseOptions()
gridpack = opt.GRIDPACK
if (gridpack==''):
print 'No gridpack specified!!!'
sys.exit()
gensimfragment = opt.FRAGMENT
if (gensimfragment==''):
print 'No GEN-SIM fragment specified!!!'
sys.exit()
njobs = int(opt.NJOBS)
eventsperjob = int(opt.EVENTSPERJOB)
# save working dir
currentDir = os.getcwd()
tag = opt.TAG+'_'+time.strftime("%Y%m%d")
outDir='resultsLHEGENSIM_'+tag
if (not os.path.isdir(outDir)):
cmd = 'mkdir '+outDir
processCmd(cmd)
outFilesDir = 'outFilesLHEGENSIM_'+tag
if not os.path.isdir(outFilesDir):
cmd = 'mkdir '+outFilesDir
processCmd(cmd)
errFilesDir = 'errFilesLHEGENSIM_'+tag
if not os.path.isdir(errFilesDir):
cmd = 'mkdir '+errFilesDir
processCmd(cmd)
# submit the jobs
print '[Submitting jobs]'
jobCount=0
for job in range(1,njobs+1):
jobName = 'jobLHEGENSIM_'+str(int(job))
cmd = 'qsub -o '+outFilesDir+'/'+jobName+'.out -e '+errFilesDir+'/'+jobName+'.err -v jobName='+jobName+',curDir='+currentDir+',outDir='+outDir+',gridpack='+gridpack+',job='+str(int(job))+',eventsperjob='+str(eventsperjob)+',gensimfragment='+gensimfragment+',tag='+opt.TAG+' -N \"'+jobName+'\" submitFileLHEGENSIM.pbs.sh'
#print cmd
output = processCmd(cmd)
while ('error' in output):
time.sleep(1.0);
output = processCmd(cmd)
if ('error' not in output):
print 'Submitted after retry - job for observable '+str(observable)
jobCount += 1
print '[Submitted '+str(jobCount)+' jobs]'
# run the submitLHEGENSIM() as main()
if __name__ == "__main__":
submitLHEGENSIM()