ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/COMP/CRAB/python/cms_cmssw.py

Comparing COMP/CRAB/python/cms_cmssw.py (file contents):
Revision 1.116.2.6 by fanzago, Fri Oct 12 09:46:06 2007 UTC vs.
Revision 1.178 by spiga, Sun Apr 20 09:34:40 2008 UTC

#	Line 5 | Line 5 | from crab_util import *
5		from BlackWhiteListParser import BlackWhiteListParser
6		import common
7		import Scram
8	+	from LFNBaseName import *
9
10		import os, string, glob
11
#	Line 13 | Line 14 | class Cmssw(JobType):
14		JobType.__init__(self, 'CMSSW')
15		common.logger.debug(3,'CMSSW::__init__')
16
17	+	self.argsList = []
18	+
19		self._params = {}
20		self.cfg_params = cfg_params
18	–
21		# init BlackWhiteListParser
22		self.blackWhiteListParser = BlackWhiteListParser(cfg_params)
23
24	<	try:
23	<	self.MaxTarBallSize = float(self.cfg_params['EDG.maxtarballsize'])
24	<	except KeyError:
25	<	self.MaxTarBallSize = 9.5
24	>	self.MaxTarBallSize = float(self.cfg_params.get('EDG.maxtarballsize',9.5))
25
26		# number of jobs requested to be created, limit obj splitting
27		self.ncjobs = ncjobs
28
29		log = common.logger
30	<
30	>
31		self.scram = Scram.Scram(cfg_params)
32		self.additional_inbox_files = []
33		self.scriptExe = ''
#	Line 37 | Line 36 | class Cmssw(JobType):
36		self.tgz_name = 'default.tgz'
37		self.additional_tgz_name = 'additional.tgz'
38		self.scriptName = 'CMSSW.sh'
39	<	self.pset = ''      #scrip use case Da
39	>	self.pset = ''      #scrip use case Da
40		self.datasetPath = '' #scrip use case Da
41
42		# set FJR file name
43		self.fjrFileName = 'crab_fjr.xml'
44
45		self.version = self.scram.getSWVersion()
46	<
46	>
47		#
48		# Try to block creation in case of arch/version mismatch
49		#
50
51	<	a = string.split(self.version, "_")
51	>	#        a = string.split(self.version, "_")
52	>	#
53	>	#        if int(a[1]) == 1 and (int(a[2]) < 5 and self.executable_arch.find('slc4') == 0):
54	>	#            msg = "Warning: You are using %s version of CMSSW  with %s architecture. \n--> Did you compile your libraries with SLC3? Otherwise you can find some problems running on SLC4 Grid nodes.\n"%(self.version, self.executable_arch)
55	>	#            common.logger.message(msg)
56	>	#        if int(a[1]) == 1 and (int(a[2]) >= 5 and self.executable_arch.find('slc3') == 0):
57	>	#            msg = "Error: CMS does not support %s with %s architecture"%(self.version, self.executable_arch)
58	>	#            raise CrabException(msg)
59	>	#
60
54	–	if int(a[1]) == 1 and (int(a[2]) < 5 and self.executable_arch.find('slc4') == 0):
55	–	msg = "Error: CMS does not support %s with %s architecture"%(self.version, self.executable_arch)
56	–	raise CrabException(msg)
57	–	if int(a[1]) == 1 and (int(a[2]) >= 5 and self.executable_arch.find('slc3') == 0):
58	–	msg = "Error: CMS does not support %s with %s architecture"%(self.version, self.executable_arch)
59	–	raise CrabException(msg)
60	–
61	–	common.taskDB.setDict('codeVersion',self.version)
62	–	self.setParam_('application', self.version)
61
62		### collect Data cards
63
64	<	## get DBS mode
65	<	try:
68	<	self.use_dbs_1 = int(self.cfg_params['CMSSW.use_dbs_1'])
69	<	except KeyError:
70	<	self.use_dbs_1 = 0
71	<
72	<	try:
73	<	tmp =  cfg_params['CMSSW.datasetpath']
74	<	log.debug(6, "CMSSW::CMSSW(): datasetPath = "+tmp)
75	<	if string.lower(tmp)=='none':
76	<	self.datasetPath = None
77	<	self.selectNoInput = 1
78	<	else:
79	<	self.datasetPath = tmp
80	<	self.selectNoInput = 0
81	<	except KeyError:
82	<	msg = "Error: datasetpath not defined "
64	>	if not cfg_params.has_key('CMSSW.datasetpath'):
65	>	msg = "Error: datasetpath not defined "
66		raise CrabException(msg)
67	<
68	<	# ML monitoring
69	<	# split dataset path style: /PreProdR3Minbias/SIM/GEN-SIM
70	<	if not self.datasetPath:
71	<	self.setParam_('dataset', 'None')
89	<	self.setParam_('owner', 'None')
67	>	tmp =  cfg_params['CMSSW.datasetpath']
68	>	log.debug(6, "CMSSW::CMSSW(): datasetPath = "+tmp)
69	>	if string.lower(tmp)=='none':
70	>	self.datasetPath = None
71	>	self.selectNoInput = 1
72		else:
73	<	try:
74	<	datasetpath_split = self.datasetPath.split("/")
93	<	# standard style
94	<	self.setParam_('datasetFull', self.datasetPath)
95	<	if self.use_dbs_1 == 1 :
96	<	self.setParam_('dataset', datasetpath_split[1])
97	<	self.setParam_('owner', datasetpath_split[-1])
98	<	else:
99	<	self.setParam_('dataset', datasetpath_split[1])
100	<	self.setParam_('owner', datasetpath_split[2])
101	<	except:
102	<	self.setParam_('dataset', self.datasetPath)
103	<	self.setParam_('owner', self.datasetPath)
104	<
105	<	self.setTaskid_()
106	<	self.setParam_('taskId', self.cfg_params['taskId'])
73	>	self.datasetPath = tmp
74	>	self.selectNoInput = 0
75
76		self.dataTiers = []
77
78		## now the application
79	<	try:
80	<	self.executable = cfg_params['CMSSW.executable']
113	<	self.setParam_('exe', self.executable)
114	<	log.debug(6, "CMSSW::CMSSW(): executable = "+self.executable)
115	<	msg = "Default executable cmsRun overridden. Switch to " + self.executable
116	<	log.debug(3,msg)
117	<	except KeyError:
118	<	self.executable = 'cmsRun'
119	<	self.setParam_('exe', self.executable)
120	<	msg = "User executable not defined. Use cmsRun"
121	<	log.debug(3,msg)
122	<	pass
79	>	self.executable = cfg_params.get('CMSSW.executable','cmsRun')
80	>	log.debug(6, "CMSSW::CMSSW(): executable = "+self.executable)
81
82	<	try:
125	<	self.pset = cfg_params['CMSSW.pset']
126	<	log.debug(6, "Cmssw::Cmssw(): PSet file = "+self.pset)
127	<	if self.pset.lower() != 'none' :
128	<	if (not os.path.exists(self.pset)):
129	<	raise CrabException("User defined PSet file "+self.pset+" does not exist")
130	<	else:
131	<	self.pset = None
132	<	except KeyError:
82	>	if not cfg_params.has_key('CMSSW.pset'):
83		raise CrabException("PSet file missing. Cannot run cmsRun ")
84	+	self.pset = cfg_params['CMSSW.pset']
85	+	log.debug(6, "Cmssw::Cmssw(): PSet file = "+self.pset)
86	+	if self.pset.lower() != 'none' :
87	+	if (not os.path.exists(self.pset)):
88	+	raise CrabException("User defined PSet file "+self.pset+" does not exist")
89	+	else:
90	+	self.pset = None
91
92		# output files
93		## stuff which must be returned always via sandbox
#	Line 140 | Line 97 | class Cmssw(JobType):
97		self.output_file_sandbox.append(self.fjrFileName)
98
99		# other output files to be returned via sandbox or copied to SE
100	<	try:
101	<	self.output_file = []
102	<	tmp = cfg_params['CMSSW.output_file']
103	<	if tmp != '':
104	<	tmpOutFiles = string.split(cfg_params['CMSSW.output_file'],',')
105	<	log.debug(7, 'cmssw::cmssw(): output files '+str(tmpOutFiles))
106	<	for tmp in tmpOutFiles:
107	<	tmp=string.strip(tmp)
151	<	self.output_file.append(tmp)
152	<	pass
153	<	else:
154	<	log.message("No output file defined: only stdout/err and the CRAB Framework Job Report will be available\n")
100	>	self.output_file = []
101	>	tmp = cfg_params.get('CMSSW.output_file',None)
102	>	if tmp :
103	>	tmpOutFiles = string.split(tmp,',')
104	>	log.debug(7, 'cmssw::cmssw(): output files '+str(tmpOutFiles))
105	>	for tmp in tmpOutFiles:
106	>	tmp=string.strip(tmp)
107	>	self.output_file.append(tmp)
108		pass
109	<	pass
157	<	except KeyError:
109	>	else:
110		log.message("No output file defined: only stdout/err and the CRAB Framework Job Report will be available\n")
111	<	pass
111	>	pass
112
113		# script_exe file as additional file in inputSandbox
114	<	try:
115	<	self.scriptExe = cfg_params['USER.script_exe']
116	<	if self.scriptExe != '':
117	<	if not os.path.isfile(self.scriptExe):
118	<	msg ="ERROR. file "+self.scriptExe+" not found"
119	<	raise CrabException(msg)
168	<	self.additional_inbox_files.append(string.strip(self.scriptExe))
169	<	except KeyError:
170	<	self.scriptExe = ''
114	>	self.scriptExe = cfg_params.get('USER.script_exe',None)
115	>	if self.scriptExe :
116	>	if not os.path.isfile(self.scriptExe):
117	>	msg ="ERROR. file "+self.scriptExe+" not found"
118	>	raise CrabException(msg)
119	>	self.additional_inbox_files.append(string.strip(self.scriptExe))
120
121		#CarlosDaniele
122		if self.datasetPath == None and self.pset == None and self.scriptExe == '' :
123	<	msg ="Error. script_exe  not defined"
124	<	raise CrabException(msg)
123	>	msg ="Error. script_exe  not defined"
124	>	raise CrabException(msg)
125
126		## additional input files
127	<	try:
127	>	if cfg_params.has_key('USER.additional_input_files'):
128		tmpAddFiles = string.split(cfg_params['USER.additional_input_files'],',')
129		for tmp in tmpAddFiles:
130		tmp = string.strip(tmp)
#	Line 199 | Line 148 | class Cmssw(JobType):
148		pass
149		pass
150		common.logger.debug(5,"Additional input files: "+str(self.additional_inbox_files))
151	<	except KeyError:
203	<	pass
204	<
205	<	# files per job
206	<	try:
207	<	if (cfg_params['CMSSW.files_per_jobs']):
208	<	raise CrabException("files_per_jobs no longer supported.  Quitting.")
209	<	except KeyError:
210	<	pass
151	>	pass
152
153		## Events per job
154	<	try:
154	>	if cfg_params.has_key('CMSSW.events_per_job'):
155		self.eventsPerJob =int( cfg_params['CMSSW.events_per_job'])
156		self.selectEventsPerJob = 1
157	<	except KeyError:
157	>	else:
158		self.eventsPerJob = -1
159		self.selectEventsPerJob = 0
160	<
160	>
161		## number of jobs
162	<	try:
162	>	if cfg_params.has_key('CMSSW.number_of_jobs'):
163		self.theNumberOfJobs =int( cfg_params['CMSSW.number_of_jobs'])
164		self.selectNumberOfJobs = 1
165	<	except KeyError:
165	>	else:
166		self.theNumberOfJobs = 0
167		self.selectNumberOfJobs = 0
168
169	<	try:
169	>	if cfg_params.has_key('CMSSW.total_number_of_events'):
170		self.total_number_of_events = int(cfg_params['CMSSW.total_number_of_events'])
171		self.selectTotalNumberEvents = 1
172	<	except KeyError:
172	>	else:
173		self.total_number_of_events = 0
174		self.selectTotalNumberEvents = 0
175
176	<	if self.pset != None: #CarlosDaniele
176	>	if self.pset != None: #CarlosDaniele
177		if ( (self.selectTotalNumberEvents + self.selectEventsPerJob + self.selectNumberOfJobs) != 2 ):
178		msg = 'Must define exactly two of total_number_of_events, events_per_job, or number_of_jobs.'
179		raise CrabException(msg)
#	Line 241 | Line 182 | class Cmssw(JobType):
182		msg = 'Must specify  number_of_jobs.'
183		raise CrabException(msg)
184
185	<	## source seed for pythia
186	<	try:
187	<	self.sourceSeed = int(cfg_params['CMSSW.pythia_seed'])
188	<	except KeyError:
189	<	self.sourceSeed = None
190	<	common.logger.debug(5,"No seed given")
191	<
192	<	try:
193	<	self.sourceSeedVtx = int(cfg_params['CMSSW.vtx_seed'])
194	<	except KeyError:
195	<	self.sourceSeedVtx = None
196	<	common.logger.debug(5,"No vertex seed given")
185	>	## New method of dealing with seeds
186	>	self.incrementSeeds = []
187	>	self.preserveSeeds = []
188	>	if cfg_params.has_key('CMSSW.preserve_seeds'):
189	>	tmpList = cfg_params['CMSSW.preserve_seeds'].split(',')
190	>	for tmp in tmpList:
191	>	tmp.strip()
192	>	self.preserveSeeds.append(tmp)
193	>	if cfg_params.has_key('CMSSW.increment_seeds'):
194	>	tmpList = cfg_params['CMSSW.increment_seeds'].split(',')
195	>	for tmp in tmpList:
196	>	tmp.strip()
197	>	self.incrementSeeds.append(tmp)
198	>
199	>	## Old method of dealing with seeds
200	>	## FUTURE: This is for old CMSSW and old CRAB. Can throw exceptions after a couple of CRAB releases and then
201	>	## remove
202	>	self.sourceSeed = cfg_params.get('CMSSW.pythia_seed',None)
203	>	if self.sourceSeed:
204	>	print "pythia_seed is a deprecated parameter. Use preserve_seeds or increment_seeds in the future.\n","Added to increment_seeds."
205	>	self.incrementSeeds.append('sourceSeed')
206	>
207	>	self.sourceSeedVtx = cfg_params.get('CMSSW.vtx_seed',None)
208	>	if self.sourceSeedVtx:
209	>	print "vtx_seed is a deprecated parameter. Use preserve_seeds or increment_seeds in the future.\n","Added to increment_seeds."
210	>	self.incrementSeeds.append('VtxSmeared')
211	>
212	>	self.sourceSeedG4 = cfg_params.get('CMSSW.g4_seed',None)
213	>	if self.sourceSeedG4:
214	>	print "g4_seed is a deprecated parameter. Use preserve_seeds or increment_seeds in the future.\n","Added to increment_seeds."
215	>	self.incrementSeeds.append('g4SimHits')
216	>
217	>	self.sourceSeedMix = cfg_params.get('CMSSW.mix_seed',None)
218	>	if self.sourceSeedMix:
219	>	print "mix_seed is a deprecated parameter. Use preserve_seeds or increment_seeds in the future.\n","Added to increment_seeds."
220	>	self.incrementSeeds.append('mix')
221
222	<	try:
258	<	self.sourceSeedG4 = int(cfg_params['CMSSW.g4_seed'])
259	<	except KeyError:
260	<	self.sourceSeedG4 = None
261	<	common.logger.debug(5,"No g4 sim hits seed given")
222	>	self.firstRun = cfg_params.get('CMSSW.first_run',None)
223
263	–	try:
264	–	self.sourceSeedMix = int(cfg_params['CMSSW.mix_seed'])
265	–	except KeyError:
266	–	self.sourceSeedMix = None
267	–	common.logger.debug(5,"No mix seed given")
268	–
269	–	try:
270	–	self.firstRun = int(cfg_params['CMSSW.first_run'])
271	–	except KeyError:
272	–	self.firstRun = None
273	–	common.logger.debug(5,"No first run given")
224		if self.pset != None: #CarlosDaniele
225	<	ver = string.split(self.version,"_")
276	<	if (int(ver[1])>=1 and int(ver[2])>=5):
277	<	import PsetManipulator150 as pp
278	<	else:
279	<	import PsetManipulator as pp
225	>	import PsetManipulator as pp
226		PsetEdit = pp.PsetManipulator(self.pset) #Daniele Pset
227
228	+	# Copy/return
229	+
230	+	self.copy_data = int(cfg_params.get('USER.copy_data',0))
231	+	self.return_data = int(cfg_params.get('USER.return_data',0))
232	+
233		#DBSDLS-start
234	<	## Initialize the variables that are extracted from DBS/DLS and needed in other places of the code
234	>	## Initialize the variables that are extracted from DBS/DLS and needed in other places of the code
235		self.maxEvents=0  # max events available   ( --> check the requested nb. of evts in Creator.py)
236		self.DBSPaths={}  # all dbs paths requested ( --> input to the site local discovery script)
237		self.jobDestination=[]  # Site destination(s) for each job (list of lists)
#	Line 289 | Line 240 | class Cmssw(JobType):
240		blockSites = {}
241		if self.datasetPath:
242		blockSites = self.DataDiscoveryAndLocation(cfg_params)
243	<	#DBSDLS-end
243	>	#DBSDLS-end
244
245		self.tgzNameWithPath = self.getTarBall(self.executable)
246	<
246	>
247		## Select Splitting
248	<	if self.selectNoInput:
248	>	if self.selectNoInput:
249		if self.pset == None: #CarlosDaniele
250		self.jobSplittingForScript()
251		else:
#	Line 305 | Line 256 | class Cmssw(JobType):
256		# modify Pset
257		if self.pset != None: #CarlosDaniele
258		try:
259	<	if (self.datasetPath): # standard job
260	<	# allow to processa a fraction of events in a file
310	<	PsetEdit.inputModule("INPUT")
311	<	PsetEdit.maxEvent("INPUTMAXEVENTS")
312	<	PsetEdit.skipEvent("INPUTSKIPEVENTS")
313	<	else:  # pythia like job
314	<	PsetEdit.maxEvent(self.eventsPerJob)
315	<	if (self.firstRun):
316	<	PsetEdit.pythiaFirstRun("INPUTFIRSTRUN")  #First Run
317	<	if (self.sourceSeed) :
318	<	PsetEdit.pythiaSeed("INPUT")
319	<	if (self.sourceSeedVtx) :
320	<	PsetEdit.vtxSeed("INPUTVTX")
321	<	if (self.sourceSeedG4) :
322	<	PsetEdit.g4Seed("INPUTG4")
323	<	if (self.sourceSeedMix) :
324	<	PsetEdit.mixSeed("INPUTMIX")
325	<	# add FrameworkJobReport to parameter-set
259	>	# Add FrameworkJobReport to parameter-set, set max events.
260	>	# Reset later for data jobs by writeCFG which does all modifications
261		PsetEdit.addCrabFJR(self.fjrFileName)
262	+	PsetEdit.maxEvent(self.eventsPerJob)
263		PsetEdit.psetWriter(self.configFilename())
264		except:
265		msg='Error while manipuliating ParameterSet: exiting...'
#	Line 332 | Line 268 | class Cmssw(JobType):
268		def DataDiscoveryAndLocation(self, cfg_params):
269
270		import DataDiscovery
335	–	import DataDiscovery_DBS2
271		import DataLocation
272		common.logger.debug(10,"CMSSW::DataDiscoveryAndLocation()")
273
#	Line 341 | Line 276 | class Cmssw(JobType):
276		## Contact the DBS
277		common.logger.message("Contacting Data Discovery Services ...")
278		try:
279	<
345	<	if self.use_dbs_1 == 1 :
346	<	self.pubdata=DataDiscovery.DataDiscovery(datasetPath, cfg_params)
347	<	else :
348	<	self.pubdata=DataDiscovery_DBS2.DataDiscovery_DBS2(datasetPath, cfg_params)
279	>	self.pubdata=DataDiscovery.DataDiscovery(datasetPath, cfg_params)
280		self.pubdata.fetchDBSInfo()
281
282		except DataDiscovery.NotExistingDatasetError, ex :
#	Line 357 | Line 288 | class Cmssw(JobType):
288		except DataDiscovery.DataDiscoveryError, ex:
289		msg = 'ERROR ***: failed Data Discovery in DBS :  %s'%ex.getErrorMessage()
290		raise CrabException(msg)
360	–	except DataDiscovery_DBS2.NotExistingDatasetError_DBS2, ex :
361	–	msg = 'ERROR ***: failed Data Discovery in DBS : %s'%ex.getErrorMessage()
362	–	raise CrabException(msg)
363	–	except DataDiscovery_DBS2.NoDataTierinProvenanceError_DBS2, ex :
364	–	msg = 'ERROR ***: failed Data Discovery in DBS : %s'%ex.getErrorMessage()
365	–	raise CrabException(msg)
366	–	except DataDiscovery_DBS2.DataDiscoveryError_DBS2, ex:
367	–	msg = 'ERROR ***: failed Data Discovery in DBS :  %s'%ex.getErrorMessage()
368	–	raise CrabException(msg)
291
292		self.filesbyblock=self.pubdata.getFiles()
293		self.eventsbyblock=self.pubdata.getEventsPerBlock()
294		self.eventsbyfile=self.pubdata.getEventsPerFile()
295
296		## get max number of events
297	<	self.maxEvents=self.pubdata.getMaxEvents() ##  self.maxEvents used in Creator.py
297	>	self.maxEvents=self.pubdata.getMaxEvents() ##  self.maxEvents used in Creator.py
298
299		## Contact the DLS and build a list of sites hosting the fileblocks
300		try:
#	Line 381 | Line 303 | class Cmssw(JobType):
303		except DataLocation.DataLocationError , ex:
304		msg = 'ERROR ***: failed Data Location in DLS \n %s '%ex.getErrorMessage()
305		raise CrabException(msg)
306	<
306	>
307
308		sites = dataloc.getSites()
309		allSites = []
#	Line 395 | Line 317 | class Cmssw(JobType):
317		common.logger.message("Requested dataset: " + datasetPath + " has " + str(self.maxEvents) + " events in " + str(len(self.filesbyblock.keys())) + " blocks.\n")
318
319		return sites
320	<
320	>
321	>	# to Be Removed  DS -- BL
322	>	#  def setArgsList(self, argsList):
323	>	#      self.argsList = argsList
324	>
325		def jobSplittingByBlocks(self, blockSites):
326		"""
327		Perform job splitting. Jobs run over an integer number of files
#	Line 445 | Line 371 | class Cmssw(JobType):
371		totalNumberOfJobs = 999999999
372		else :
373		totalNumberOfJobs = self.ncjobs
448	–
374
375		blocks = blockSites.keys()
376		blockCount = 0
#	Line 465 | Line 390 | class Cmssw(JobType):
390		blockCount += 1
391		if block not in jobsOfBlock.keys() :
392		jobsOfBlock[block] = []
393	<
393	>
394		if self.eventsbyblock.has_key(block) :
395		numEventsInBlock = self.eventsbyblock[block]
396		common.logger.debug(5,'Events in Block File '+str(numEventsInBlock))
397	<
397	>
398		files = self.filesbyblock[block]
399		numFilesInBlock = len(files)
400		if (numFilesInBlock <= 0):
#	Line 477 | Line 402 | class Cmssw(JobType):
402		fileCount = 0
403
404		# ---- New block => New job ---- #
405	<	parString = "\\{"
405	>	parString = ""
406		# counter for number of events in files currently worked on
407		filesEventCount = 0
408		# flag if next while loop should touch new file
409		newFile = 1
410		# job event counter
411		jobSkipEventCount = 0
412	<
412	>
413		# ---- Iterate over the files in the block until we've met the requested ---- #
414		# ---- total # of events or we've gone over all the files in this block  ---- #
415		while ( (eventsRemaining > 0) and (fileCount < numFilesInBlock) and (jobCount < totalNumberOfJobs) ):
#	Line 500 | Line 425 | class Cmssw(JobType):
425		newFile = 0
426		except KeyError:
427		common.logger.message("File "+str(file)+" has unknown number of events: skipping")
503	–
428
429	+	eventsPerJobRequested = min(eventsPerJobRequested, eventsRemaining)
430		# if less events in file remain than eventsPerJobRequested
431	<	if ( filesEventCount - jobSkipEventCount < eventsPerJobRequested ) :
431	>	if ( filesEventCount - jobSkipEventCount < eventsPerJobRequested):
432		# if last file in block
433		if ( fileCount == numFilesInBlock-1 ) :
434		# end job using last file, use remaining events in block
435		# close job and touch new file
436		fullString = parString[:-2]
512	–	fullString += '\\}'
437		list_of_lists.append([fullString,str(-1),str(jobSkipEventCount)])
438		common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(filesEventCount - jobSkipEventCount)+" events (last file in block).")
439		self.jobDestination.append(blockSites[block])
#	Line 522 | Line 446 | class Cmssw(JobType):
446		eventsRemaining = eventsRemaining - filesEventCount + jobSkipEventCount
447		jobSkipEventCount = 0
448		# reset file
449	<	parString = "\\{"
449	>	parString = ""
450		filesEventCount = 0
451		newFile = 1
452		fileCount += 1
#	Line 534 | Line 458 | class Cmssw(JobType):
458		elif ( filesEventCount - jobSkipEventCount == eventsPerJobRequested ) :
459		# close job and touch new file
460		fullString = parString[:-2]
537	–	fullString += '\\}'
461		list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
462		common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(eventsPerJobRequested)+" events.")
463		self.jobDestination.append(blockSites[block])
#	Line 546 | Line 469 | class Cmssw(JobType):
469		eventsRemaining = eventsRemaining - eventsPerJobRequested
470		jobSkipEventCount = 0
471		# reset file
472	<	parString = "\\{"
472	>	parString = ""
473		filesEventCount = 0
474		newFile = 1
475		fileCount += 1
476	<
476	>
477		# if more events in file remain than eventsPerJobRequested
478		else :
479		# close job but don't touch new file
480		fullString = parString[:-2]
558	–	fullString += '\\}'
481		list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
482		common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(eventsPerJobRequested)+" events.")
483		self.jobDestination.append(blockSites[block])
#	Line 570 | Line 492 | class Cmssw(JobType):
492		jobSkipEventCount = eventsPerJobRequested - (filesEventCount - jobSkipEventCount - self.eventsbyfile[file])
493		# remove all but the last file
494		filesEventCount = self.eventsbyfile[file]
495	<	parString = "\\{"
574	<	parString += '\\\"' + file + '\\\"\,'
495	>	parString = '\\\"' + file + '\\\"\,'
496		pass # END if
497		pass # END while (iterate over files in the block)
498		pass # END while (iterate over blocks in the dataset)
#	Line 579 | Line 500 | class Cmssw(JobType):
500		if (eventsRemaining > 0 and jobCount < totalNumberOfJobs ):
501		common.logger.message("Could not run on all requested events because some blocks not hosted at allowed sites.")
502		common.logger.message(str(jobCount)+" job(s) can run on "+str(totalEventCount)+" events.\n")
503	<
503	>
504		# screen output
505		screenOutput = "List of jobs and available destination sites:\n\n"
506
#	Line 591 | Line 512 | class Cmssw(JobType):
512		for block in blocks:
513		if block in jobsOfBlock.keys() :
514		blockCounter += 1
515	<	screenOutput += "Block %5i: jobs %20s: sites: %s\n" % (blockCounter,spanRanges(jobsOfBlock[block]),','.join(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)))
515	>	screenOutput += "Block %5i: jobs %20s: sites: %s\n" % (blockCounter,spanRanges(jobsOfBlock[block]),
516	>	','.join(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)))
517		if len(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)) == 0:
518	<	noSiteBlock.append( spanRanges(jobsOfBlock[block]) )
518	>	noSiteBlock.append( spanRanges(jobsOfBlock[block]) )
519		bloskNoSite.append( blockCounter )
520	<
520	>
521		common.logger.message(screenOutput)
522		if len(noSiteBlock) > 0 and len(bloskNoSite) > 0:
523		msg = 'WARNING: No sites are hosting any part of data for block:\n                '
#	Line 611 | Line 533 | class Cmssw(JobType):
533		for range_jobs in noSiteBlock:
534		msg += str(range_jobs) + virgola
535		msg += '\n               will not be submitted and this block of data can not be analyzed!\n'
536	+	if self.cfg_params.has_key('EDG.se_white_list'):
537	+	msg += 'WARNING: SE White List: '+self.cfg_params['EDG.se_white_list']+'\n'
538	+	msg += '(Hint: By whitelisting you force the job to run at this particular site(s).\n'
539	+	msg += 'Please check if the dataset is available at this site!)\n'
540	+	if self.cfg_params.has_key('EDG.ce_white_list'):
541	+	msg += 'WARNING: CE White List: '+self.cfg_params['EDG.ce_white_list']+'\n'
542	+	msg += '(Hint: By whitelisting you force the job to run at this particular site(s).\n'
543	+	msg += 'Please check if the dataset is available at this site!)\n'
544	+
545		common.logger.message(msg)
546
547		self.list_of_args = list_of_lists
#	Line 621 | Line 552 | class Cmssw(JobType):
552		Perform job splitting based on number of event per job
553		"""
554		common.logger.debug(5,'Splitting per events')
555	<	common.logger.message('Required '+str(self.eventsPerJob)+' events per job ')
556	<	common.logger.message('Required '+str(self.theNumberOfJobs)+' jobs in total ')
557	<	common.logger.message('Required '+str(self.total_number_of_events)+' events in total ')
555	>
556	>	if (self.selectEventsPerJob):
557	>	common.logger.message('Required '+str(self.eventsPerJob)+' events per job ')
558	>	if (self.selectNumberOfJobs):
559	>	common.logger.message('Required '+str(self.theNumberOfJobs)+' jobs in total ')
560	>	if (self.selectTotalNumberEvents):
561	>	common.logger.message('Required '+str(self.total_number_of_events)+' events in total ')
562
563		if (self.total_number_of_events < 0):
564		msg='Cannot split jobs per Events with "-1" as total number of events'
#	Line 632 | Line 567 | class Cmssw(JobType):
567		if (self.selectEventsPerJob):
568		if (self.selectTotalNumberEvents):
569		self.total_number_of_jobs = int(self.total_number_of_events/self.eventsPerJob)
570	<	elif(self.selectNumberOfJobs) :
570	>	elif(self.selectNumberOfJobs) :
571		self.total_number_of_jobs =self.theNumberOfJobs
572	<	self.total_number_of_events =int(self.theNumberOfJobs*self.eventsPerJob)
572	>	self.total_number_of_events =int(self.theNumberOfJobs*self.eventsPerJob)
573
574		elif (self.selectNumberOfJobs) :
575		self.total_number_of_jobs = self.theNumberOfJobs
576		self.eventsPerJob = int(self.total_number_of_events/self.total_number_of_jobs)
577	<
577	>
578		common.logger.debug(5,'N jobs  '+str(self.total_number_of_jobs))
579
580		# is there any remainder?
#	Line 655 | Line 590 | class Cmssw(JobType):
590		self.list_of_args = []
591		for i in range(self.total_number_of_jobs):
592		## Since there is no input, any site is good
593	<	# self.jobDestination.append(["Any"])
659	<	self.jobDestination.append([""]) #must be empty to write correctly the xml
593	>	self.jobDestination.append([""]) #must be empty to write correctly the xml
594		args=[]
595		if (self.firstRun):
596	<	## pythia first run
663	<	#self.list_of_args.append([(str(self.firstRun)+str(i))])
596	>	## pythia first run
597		args.append(str(self.firstRun)+str(i))
665	–	else:
666	–	## no first run
667	–	#self.list_of_args.append([str(i)])
668	–	args.append(str(i))
669	–	if (self.sourceSeed):
670	–	args.append(str(self.sourceSeed)+str(i))
671	–	if (self.sourceSeedVtx):
672	–	## + vtx random seed
673	–	args.append(str(self.sourceSeedVtx)+str(i))
674	–	if (self.sourceSeedG4):
675	–	## + G4 random seed
676	–	args.append(str(self.sourceSeedG4)+str(i))
677	–	if (self.sourceSeedMix):
678	–	## + Mix random seed
679	–	args.append(str(self.sourceSeedMix)+str(i))
680	–	pass
681	–	pass
598		self.list_of_args.append(args)
683	–	pass
684	–
685	–	# print self.list_of_args
599
600		return
601
#	Line 711 | Line 624 | class Cmssw(JobType):
624		return
625
626		def split(self, jobParams):
627	<
715	<	common.jobDB.load()
627	>
628		#### Fabio
629		njobs = self.total_number_of_jobs
630		arglist = self.list_of_args
631		# create the empty structure
632		for i in range(njobs):
633		jobParams.append("")
634	<
634	>
635	>	listID=[]
636	>	listField=[]
637		for job in range(njobs):
638		jobParams[job] = arglist[job]
639	<	# print str(arglist[job])
640	<	# print jobParams[job]
641	<	common.jobDB.setArguments(job, jobParams[job])
642	<	common.logger.debug(5,"Job "+str(job)+" Destination: "+str(self.jobDestination[job]))
643	<	common.jobDB.setDestination(job, self.jobDestination[job])
639	>	listID.append(job+1)
640	>	job_ToSave ={}
641	>	concString = ' '
642	>	argu=''
643	>	if len(jobParams[job]):
644	>	argu +=   concString.join(jobParams[job] )
645	>	job_ToSave['arguments']= str(job+1)+' '+argu## new BL--DS
646	>	job_ToSave['dlsDestination']= self.jobDestination[job]## new BL--DS
647	>	#common._db.updateJob_(job,job_ToSave)## new BL--DS
648	>	listField.append(job_ToSave)
649	>	msg="Job "+str(job)+" Arguments:   "+str(job+1)+" "+argu+"\n"  \
650	>	+"                     Destination: "+str(self.jobDestination[job])
651	>	common.logger.debug(5,msg)
652	>	#common.logger.debug(5,"Job "+str(job)+" Destination: "+str(self.jobDestination[job]))
653	>	common._db.updateJob_(listID,listField)## new BL--DS
654	>	## Pay Attention Here....DS--BL
655	>	self.argsList = (len(jobParams[1])+1)
656
731	–	common.jobDB.save()
657		return
658	<
734	<	def getJobTypeArguments(self, nj, sched):
735	<	result = ''
736	<	for i in common.jobDB.arguments(nj):
737	<	result=result+str(i)+" "
738	<	return result
739	<
658	>
659		def numberOfJobs(self):
660		# Fabio
661		return self.total_number_of_jobs
#	Line 745 | Line 664 | class Cmssw(JobType):
664		"""
665		Return the TarBall with lib and exe
666		"""
667	<
667	>
668		# if it exist, just return it
669		#
670		# Marco. Let's start to use relative path for Boss XML files
#	Line 768 | Line 687 | class Cmssw(JobType):
687		# print "swVersion = ", swVersion
688		swReleaseTop = self.scram.getReleaseTop_()
689		#print "swReleaseTop = ", swReleaseTop
690	<
690	>
691		## check if working area is release top
692		if swReleaseTop == '' or swArea == swReleaseTop:
693	+	common.logger.debug(3,"swArea = "+swArea+" swReleaseTop ="+swReleaseTop)
694		return
695
696		import tarfile
#	Line 781 | Line 701 | class Cmssw(JobType):
701		exeWithPath = self.scram.findFile_(executable)
702		if ( not exeWithPath ):
703		raise CrabException('User executable '+executable+' not found')
704	<
704	>
705		## then check if it's private or not
706		if exeWithPath.find(swReleaseTop) == -1:
707		# the exe is private, so we must ship
#	Line 790 | Line 710 | class Cmssw(JobType):
710		# distinguish case when script is in user project area or given by full path somewhere else
711		if exeWithPath.find(path) >= 0 :
712		exe = string.replace(exeWithPath, path,'')
713	<	tar.add(path+exe,os.path.basename(executable))
713	>	tar.add(path+exe,exe)
714		else :
715		tar.add(exeWithPath,os.path.basename(executable))
716		pass
717		else:
718		# the exe is from release, we'll find it on WN
719		pass
720	<
720	>
721		## Now get the libraries: only those in local working area
722		libDir = 'lib'
723		lib = swArea+'/' +libDir
724		common.logger.debug(5,"lib "+lib+" to be tarred")
725		if os.path.exists(lib):
726		tar.add(lib,libDir)
727	<
727	>
728		## Now check if module dir is present
729		moduleDir = 'module'
730		module = swArea + '/' + moduleDir
#	Line 818 | Line 738 | class Cmssw(JobType):
738		common.logger.debug(5,"data "+root+"/data"+" to be tarred")
739		tar.add(root+"/data",root[swAreaLen:]+"/data")
740
821	–	## Add ProdAgent dir to tar
822	–	paDir = 'ProdAgentApi'
823	–	pa = os.environ['CRABDIR'] + '/' + 'ProdAgentApi'
824	–	if os.path.isdir(pa):
825	–	tar.add(pa,paDir)
741
742	<	### FEDE FOR DBS PUBLICATION
828	<	## Add PRODCOMMON dir to tar
742	>	## Add ProdCommon dir to tar
743		prodcommonDir = 'ProdCommon'
744		prodcommonPath = os.environ['CRABDIR'] + '/' + 'ProdCommon'
745		if os.path.isdir(prodcommonPath):
746		tar.add(prodcommonPath,prodcommonDir)
747	<	#############################
834	<
747	>
748		common.logger.debug(5,"Files added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
749		tar.close()
750		except :
#	Line 843 | Line 756 | class Cmssw(JobType):
756		raise CrabException('Input sandbox size of ' + str(float(tarballinfo.st_size)/1024.0/1024.0) + ' MB is larger than the allowed ' + str(self.MaxTarBallSize) + ' MB input sandbox limit and not supported by the used GRID submission system. Please make sure that no unnecessary files are in all data directories in your local CMSSW project area as they are automatically packed into the input sandbox.')
757
758		## create tar-ball with ML stuff
759	<	self.MLtgzfile =  common.work_space.pathForTgz()+'share/MLfiles.tgz'
759	>	self.MLtgzfile =  common.work_space.pathForTgz()+'share/MLfiles.tgz'
760		try:
761		tar = tarfile.open(self.MLtgzfile, "w:gz")
762		path=os.environ['CRABDIR'] + '/python/'
763	<	for file in ['report.py', 'DashboardAPI.py', 'Logger.py', 'ProcInfo.py', 'apmon.py', 'parseCrabFjr.py']:
763	>	for file in ['report.py', 'DashboardAPI.py', 'Logger.py', 'ProcInfo.py', 'apmon.py', 'parseCrabFjr.py','writeCfg.py', 'JobReportErrorCode.py']:
764		tar.add(path+file,file)
765		common.logger.debug(5,"Files added to "+self.MLtgzfile+" : "+str(tar.getnames()))
766		tar.close()
767		except :
768		raise CrabException('Could not create ML files tar-ball')
769	<
769	>
770		return
771	<
771	>
772		def additionalInputFileTgz(self):
773		"""
774		Put all additional files into a tar ball and return its name
#	Line 869 | Line 782 | class Cmssw(JobType):
782		tar.close()
783		return tarName
784
785	<	def wsSetupEnvironment(self, nj):
785	>	def wsSetupEnvironment(self, nj=0):
786		"""
787		Returns part of a job script which prepares
788		the execution environment for the job 'nj'.
789		"""
790		# Prepare JobType-independent part
791	<	txt = ''
792	<
793	<	## OLI_Daniele at this level  middleware already known
881	<
882	<	txt += 'if [ $middleware == LCG ]; then \n'
883	<	txt += '    echo "### First set SCRAM ARCH and BUILD_ARCH to ' + self.executable_arch + ' ###"\n'
884	<	txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
885	<	txt += '    export BUILD_ARCH='+self.executable_arch+'\n'
791	>	txt = '\n#Written by cms_cmssw::wsSetupEnvironment\n'
792	>	txt += 'echo ">>> setup environment"\n'
793	>	txt += 'if [ $middleware == LCG ]; then \n'
794		txt += self.wsSetupCMSLCGEnvironment_()
795		txt += 'elif [ $middleware == OSG ]; then\n'
796		txt += '    WORKING_DIR=`/bin/mktemp  -d $OSG_WN_TMP/cms_XXXXXXXXXXXX`\n'
797	<	txt += '    echo "Created working directory: $WORKING_DIR"\n'
798	<	txt += '    if [ ! -d $WORKING_DIR ] ;then\n'
799	<	txt += '        echo "SET_CMS_ENV 10016 ==> OSG $WORKING_DIR could not be created on WN `hostname`"\n'
800	<	txt += '    echo "JOB_EXIT_STATUS = 10016"\n'
893	<	txt += '    echo "JobExitCode=10016" | tee -a $RUNTIME_AREA/$repo\n'
894	<	txt += '    dumpStatus $RUNTIME_AREA/$repo\n'
895	<	txt += '        rm -f $RUNTIME_AREA/$repo \n'
896	<	txt += '        echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
897	<	txt += '        echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
898	<	txt += '        exit 1\n'
797	>	txt += '    if [ ! $? == 0 ] ;then\n'
798	>	txt += '        echo "ERROR ==> OSG $WORKING_DIR could not be created on WN `hostname`"\n'
799	>	txt += '        job_exit_code=10016\n'
800	>	txt += '        func_exit\n'
801		txt += '    fi\n'
802	+	txt += '    echo ">>> Created working directory: $WORKING_DIR"\n'
803		txt += '\n'
804		txt += '    echo "Change to working directory: $WORKING_DIR"\n'
805		txt += '    cd $WORKING_DIR\n'
806	<	txt += self.wsSetupCMSOSGEnvironment_()
807	<	txt += '    echo "### Set SCRAM ARCH to ' + self.executable_arch + ' ###"\n'
808	<	txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
806	>	txt += '    echo ">>> current directory (WORKING_DIR): $WORKING_DIR"\n'
807	>	txt += self.wsSetupCMSOSGEnvironment_()
808	>	#txt += '    echo "### Set SCRAM ARCH to ' + self.executable_arch + ' ###"\n'
809	>	#txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
810		txt += 'fi\n'
811
812		# Prepare JobType-specific part
813		scram = self.scram.commandName()
814		txt += '\n\n'
815	<	txt += 'echo "### SPECIFIC JOB SETUP ENVIRONMENT ###"\n'
815	>	txt += 'echo ">>> specific cmssw setup environment:"\n'
816	>	txt += 'echo "CMSSW_VERSION =  '+self.version+'"\n'
817		txt += scram+' project CMSSW '+self.version+'\n'
818		txt += 'status=$?\n'
819		txt += 'if [ $status != 0 ] ; then\n'
820	<	txt += '   echo "SET_EXE_ENV 10034 ==>ERROR CMSSW '+self.version+' not found on `hostname`" \n'
821	<	txt += '   echo "JOB_EXIT_STATUS = 10034"\n'
822	<	txt += '   echo "JobExitCode=10034" | tee -a $RUNTIME_AREA/$repo\n'
918	<	txt += '   dumpStatus $RUNTIME_AREA/$repo\n'
919	<	txt += '   rm -f $RUNTIME_AREA/$repo \n'
920	<	txt += '   echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
921	<	txt += '   echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
922	<	## OLI_Daniele
923	<	txt += '    if [ $middleware == OSG ]; then \n'
924	<	txt += '        echo "Remove working directory: $WORKING_DIR"\n'
925	<	txt += '        cd $RUNTIME_AREA\n'
926	<	txt += '        /bin/rm -rf $WORKING_DIR\n'
927	<	txt += '        if [ -d $WORKING_DIR ] ;then\n'
928	<	txt += '            echo "SET_CMS_ENV 10018 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after CMSSW CMSSW_0_6_1 not found on `hostname`"\n'
929	<	txt += '            echo "JOB_EXIT_STATUS = 10018"\n'
930	<	txt += '            echo "JobExitCode=10018" | tee -a $RUNTIME_AREA/$repo\n'
931	<	txt += '            dumpStatus $RUNTIME_AREA/$repo\n'
932	<	txt += '            rm -f $RUNTIME_AREA/$repo \n'
933	<	txt += '            echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
934	<	txt += '            echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
935	<	txt += '        fi\n'
936	<	txt += '    fi \n'
937	<	txt += '   exit 1 \n'
820	>	txt += '    echo "ERROR ==> CMSSW '+self.version+' not found on `hostname`" \n'
821	>	txt += '    job_exit_code=10034\n'
822	>	txt += '    func_exit\n'
823		txt += 'fi \n'
939	–	txt += 'echo "CMSSW_VERSION =  '+self.version+'"\n'
824		txt += 'cd '+self.version+'\n'
825		########## FEDE FOR DBS2 ######################
826		txt += 'SOFTWARE_DIR=`pwd`\n'
827	<	txt += 'echo SOFTWARE_DIR=$SOFTWARE_DIR \n'
827	>	txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
828		###############################################
829		### needed grep for bug in scramv1 ###
946	–	txt += scram+' runtime -sh\n'
830		txt += 'eval `'+scram+' runtime -sh | grep -v SCRAMRT_LSB_JOBNAME`\n'
948	–	txt += 'echo $PATH\n'
949	–
831		# Handle the arguments:
832		txt += "\n"
833		txt += "## number of arguments (first argument always jobnumber)\n"
834		txt += "\n"
835	<	#        txt += "narg=$#\n"
955	<	txt += "if [ $nargs -lt 2 ]\n"
835	>	txt += "if [ $nargs -lt "+str(self.argsList)+" ]\n"
836		txt += "then\n"
837	<	txt += "    echo 'SET_EXE_ENV 1 ==> ERROR Too few arguments' +$nargs+ \n"
838	<	txt += '    echo "JOB_EXIT_STATUS = 50113"\n'
839	<	txt += '    echo "JobExitCode=50113" | tee -a $RUNTIME_AREA/$repo\n'
960	<	txt += '    dumpStatus $RUNTIME_AREA/$repo\n'
961	<	txt += '    rm -f $RUNTIME_AREA/$repo \n'
962	<	txt += '    echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
963	<	txt += '    echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
964	<	## OLI_Daniele
965	<	txt += '    if [ $middleware == OSG ]; then \n'
966	<	txt += '        echo "Remove working directory: $WORKING_DIR"\n'
967	<	txt += '        cd $RUNTIME_AREA\n'
968	<	txt += '        /bin/rm -rf $WORKING_DIR\n'
969	<	txt += '        if [ -d $WORKING_DIR ] ;then\n'
970	<	txt += '            echo "SET_EXE_ENV 50114 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after Too few arguments for CRAB job wrapper"\n'
971	<	txt += '            echo "JOB_EXIT_STATUS = 50114"\n'
972	<	txt += '            echo "JobExitCode=50114" | tee -a $RUNTIME_AREA/$repo\n'
973	<	txt += '            dumpStatus $RUNTIME_AREA/$repo\n'
974	<	txt += '            rm -f $RUNTIME_AREA/$repo \n'
975	<	txt += '            echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
976	<	txt += '            echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
977	<	txt += '        fi\n'
978	<	txt += '    fi \n'
979	<	txt += "    exit 1\n"
837	>	txt += "    echo 'ERROR ==> Too few arguments' +$nargs+ \n"
838	>	txt += '    job_exit_code=50113\n'
839	>	txt += "    func_exit\n"
840		txt += "fi\n"
841		txt += "\n"
842
843		# Prepare job-specific part
844		job = common.job_list[nj]
845		### FEDE FOR DBS OUTPUT PUBLICATION
846	<	if (self.datasetPath):
846	>	if (self.datasetPath):
847		txt += '\n'
848		txt += 'DatasetPath='+self.datasetPath+'\n'
849
850		datasetpath_split = self.datasetPath.split("/")
851	<
851	>
852		txt += 'PrimaryDataset='+datasetpath_split[1]+'\n'
853		txt += 'DataTier='+datasetpath_split[2]+'\n'
994	–	#txt += 'ProcessedDataset='+datasetpath_split[3]+'\n'
854		txt += 'ApplicationFamily=cmsRun\n'
855
856		else:
857		txt += 'DatasetPath=MCDataTier\n'
858		txt += 'PrimaryDataset=null\n'
859		txt += 'DataTier=null\n'
1001	–	#txt += 'ProcessedDataset=null\n'
860		txt += 'ApplicationFamily=MCDataTier\n'
861	<	if self.pset != None: #CarlosDaniele
861	>	if self.pset != None:
862		pset = os.path.basename(job.configFilename())
863		txt += '\n'
864		txt += 'cp  $RUNTIME_AREA/'+pset+' .\n'
865		if (self.datasetPath): # standard job
866	<	#txt += 'InputFiles=$2\n'
867	<	txt += 'InputFiles=${args[1]}\n'
868	<	txt += 'MaxEvents=${args[2]}\n'
1011	<	txt += 'SkipEvents=${args[3]}\n'
866	>	txt += 'InputFiles=${args[1]}; export InputFiles\n'
867	>	txt += 'MaxEvents=${args[2]}; export MaxEvents\n'
868	>	txt += 'SkipEvents=${args[3]}; export SkipEvents\n'
869		txt += 'echo "Inputfiles:<$InputFiles>"\n'
1013	–	txt += 'sed "s#{\'INPUT\'}#$InputFiles#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
870		txt += 'echo "MaxEvents:<$MaxEvents>"\n'
1015	–	txt += 'sed "s#INPUTMAXEVENTS#$MaxEvents#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
871		txt += 'echo "SkipEvents:<$SkipEvents>"\n'
1017	–	txt += 'sed "s#INPUTSKIPEVENTS#$SkipEvents#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
872		else:  # pythia like job
873	<	seedIndex=1
873	>	txt += 'PreserveSeeds='  + ','.join(self.preserveSeeds)  + '; export PreserveSeeds\n'
874	>	txt += 'IncrementSeeds=' + ','.join(self.incrementSeeds) + '; export IncrementSeeds\n'
875	>	txt += 'echo "PreserveSeeds: <$PreserveSeeds>"\n'
876	>	txt += 'echo "IncrementSeeds:<$IncrementSeeds>"\n'
877		if (self.firstRun):
878	<	txt += 'FirstRun=${args['+str(seedIndex)+']}\n'
878	>	txt += 'FirstRun=${args[1]}; export FirstRun\n'
879		txt += 'echo "FirstRun: <$FirstRun>"\n'
1023	–	txt += 'sed "s#\<INPUTFIRSTRUN\>#$FirstRun#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1024	–	seedIndex=seedIndex+1
880
1026	–	if (self.sourceSeed):
1027	–	txt += 'Seed=${args['+str(seedIndex)+']}\n'
1028	–	txt += 'sed "s#\<INPUT\>#$Seed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1029	–	seedIndex=seedIndex+1
1030	–	## the following seeds are not always present
1031	–	if (self.sourceSeedVtx):
1032	–	txt += 'VtxSeed=${args['+str(seedIndex)+']}\n'
1033	–	txt += 'echo "VtxSeed: <$VtxSeed>"\n'
1034	–	txt += 'sed "s#\<INPUTVTX\>#$VtxSeed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1035	–	seedIndex += 1
1036	–	if (self.sourceSeedG4):
1037	–	txt += 'G4Seed=${args['+str(seedIndex)+']}\n'
1038	–	txt += 'echo "G4Seed: <$G4Seed>"\n'
1039	–	txt += 'sed "s#\<INPUTG4\>#$G4Seed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1040	–	seedIndex += 1
1041	–	if (self.sourceSeedMix):
1042	–	txt += 'mixSeed=${args['+str(seedIndex)+']}\n'
1043	–	txt += 'echo "MixSeed: <$mixSeed>"\n'
1044	–	txt += 'sed "s#\<INPUTMIX\>#$mixSeed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1045	–	seedIndex += 1
1046	–	pass
1047	–	pass
881		txt += 'mv -f '+pset+' pset.cfg\n'
882
883		if len(self.additional_inbox_files) > 0:
884		txt += 'if [ -e $RUNTIME_AREA/'+self.additional_tgz_name+' ] ; then\n'
885		txt += '  tar xzvf $RUNTIME_AREA/'+self.additional_tgz_name+'\n'
886		txt += 'fi\n'
887	<	pass
887	>	pass
888
889	<	if self.pset != None: #CarlosDaniele
1057	<	txt += 'echo "### END JOB SETUP ENVIRONMENT ###"\n\n'
1058	<
889	>	if self.pset != None:
890		txt += '\n'
891		txt += 'echo "***** cat pset.cfg *********"\n'
892		txt += 'cat pset.cfg\n'
893		txt += 'echo "****** end pset.cfg ********"\n'
894		txt += '\n'
1064	–	### FEDE FOR DBS OUTPUT PUBLICATION
895		txt += 'PSETHASH=`EdmConfigHash < pset.cfg` \n'
896		txt += 'echo "PSETHASH = $PSETHASH" \n'
1067	–	##############
897		txt += '\n'
1069	–	# txt += 'echo "***** cat pset1.cfg *********"\n'
1070	–	# txt += 'cat pset1.cfg\n'
1071	–	# txt += 'echo "****** end pset1.cfg ********"\n'
898		return txt
899
900	<	def wsBuildExe(self, nj=0):
900	>	def wsUntarSoftware(self, nj=0):
901		"""
902		Put in the script the commands to build an executable
903		or a library.
904		"""
905
906	<	txt = ""
906	>	txt = '\n#Written by cms_cmssw::wsUntarSoftware\n'
907
908		if os.path.isfile(self.tgzNameWithPath):
909	<	txt += 'echo "tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+'"\n'
909	>	txt += 'echo ">>> tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+' :" \n'
910		txt += 'tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+'\n'
911		txt += 'untar_status=$? \n'
912		txt += 'if [ $untar_status -ne 0 ]; then \n'
913	<	txt += '   echo "SET_EXE 1 ==> ERROR Untarring .tgz file failed"\n'
914	<	txt += '   echo "JOB_EXIT_STATUS = $untar_status" \n'
915	<	txt += '   echo "JobExitCode=$untar_status" | tee -a $RUNTIME_AREA/$repo\n'
1090	<	txt += '   if [ $middleware == OSG ]; then \n'
1091	<	txt += '       echo "Remove working directory: $WORKING_DIR"\n'
1092	<	txt += '       cd $RUNTIME_AREA\n'
1093	<	txt += '       /bin/rm -rf $WORKING_DIR\n'
1094	<	txt += '       if [ -d $WORKING_DIR ] ;then\n'
1095	<	txt += '           echo "SET_EXE 50999 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after Untarring .tgz file failed"\n'
1096	<	txt += '           echo "JOB_EXIT_STATUS = 50999"\n'
1097	<	txt += '           echo "JobExitCode=50999" | tee -a $RUNTIME_AREA/$repo\n'
1098	<	txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1099	<	txt += '           rm -f $RUNTIME_AREA/$repo \n'
1100	<	txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1101	<	txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1102	<	txt += '       fi\n'
1103	<	txt += '   fi \n'
1104	<	txt += '   \n'
1105	<	txt += '   exit 1 \n'
913	>	txt += '   echo "ERROR ==> Untarring .tgz file failed"\n'
914	>	txt += '   job_exit_code=$untar_status\n'
915	>	txt += '   func_exit\n'
916		txt += 'else \n'
917		txt += '   echo "Successful untar" \n'
918		txt += 'fi \n'
919		txt += '\n'
920	<	txt += 'echo "Include ProdAgentApi and PRODCOMMON in PYTHONPATH"\n'
920	>	txt += 'echo ">>> Include ProdCommon in PYTHONPATH:"\n'
921		txt += 'if [ -z "$PYTHONPATH" ]; then\n'
922	<	#### FEDE FOR DBS OUTPUT PUBLICATION
1113	<	txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdAgentApi:$SOFTWARE_DIR/ProdCommon\n'
1114	<	#txt += '   export PYTHONPATH=`pwd`/ProdAgentApi:`pwd`/ProdCommon\n'
1115	<	#txt += '   export PYTHONPATH=ProdAgentApi\n'
922	>	txt += '   export PYTHONPATH=$RUNTIME_AREA/ProdCommon\n'
923		txt += 'else\n'
924	<	txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdAgentApi:$SOFTWARE_DIR/ProdCommon:${PYTHONPATH}\n'
1118	<	#txt += '   export PYTHONPATH=`pwd`/ProdAgentApi:`pwd`/ProdCommon:${PYTHONPATH}\n'
1119	<	#txt += '   export PYTHONPATH=ProdAgentApi:${PYTHONPATH}\n'
924	>	txt += '   export PYTHONPATH=$RUNTIME_AREA/ProdCommon:${PYTHONPATH}\n'
925		txt += 'echo "PYTHONPATH=$PYTHONPATH"\n'
1121	–	###################
926		txt += 'fi\n'
927		txt += '\n'
928
929		pass
930	<
930	>
931	>	return txt
932	>
933	>	def wsBuildExe(self, nj=0):
934	>	"""
935	>	Put in the script the commands to build an executable
936	>	or a library.
937	>	"""
938	>
939	>	txt = '\n#Written by cms_cmssw::wsBuildExe\n'
940	>	txt += 'echo ">>> moving CMSSW software directories in `pwd`" \n'
941	>
942	>	txt += 'rm -r lib/ module/ \n'
943	>	txt += 'mv $RUNTIME_AREA/lib/ . \n'
944	>	txt += 'mv $RUNTIME_AREA/module/ . \n'
945	>	txt += 'mv $RUNTIME_AREA/ProdCommon/ . \n'
946	>
947	>	txt += 'if [ -z "$PYTHONPATH" ]; then\n'
948	>	txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdCommon\n'
949	>	txt += 'else\n'
950	>	txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdCommon:${PYTHONPATH}\n'
951	>	txt += 'echo "PYTHONPATH=$PYTHONPATH"\n'
952	>	txt += 'fi\n'
953	>	txt += '\n'
954	>
955		return txt
956
957		def modifySteeringCards(self, nj):
958		"""
959	<	modify the card provided by the user,
959	>	modify the card provided by the user,
960		writing a new card into share dir
961		"""
962	<
962	>
963		def executableName(self):
964		if self.scriptExe: #CarlosDaniele
965		return "sh "
#	Line 1139 | Line 967 | class Cmssw(JobType):
967		return self.executable
968
969		def executableArgs(self):
970	+	# FUTURE: This function tests the CMSSW version. Can be simplified as we drop support for old versions
971		if self.scriptExe:#CarlosDaniele
972		return   self.scriptExe + " $NJob"
973		else:
1145	–	# if >= CMSSW_1_5_X, add -e
974		version_array = self.scram.getSWVersion().split('_')
975		major = 0
976		minor = 0
#	Line 1150 | Line 978 | class Cmssw(JobType):
978		major = int(version_array[1])
979		minor = int(version_array[2])
980		except:
981	<	msg = "Cannot parse CMSSW version string: " + "_".join(version_array) + " for major and minor release number!"
981	>	msg = "Cannot parse CMSSW version string: " + "_".join(version_array) + " for major and minor release number!"
982		raise CrabException(msg)
983	+
984	+	ex_args = ""
985	+	# FUTURE: This tests the CMSSW version. Can remove code as versions deprecated
986	+	# Framework job report
987		if major >= 1 and minor >= 5 :
988	<	return " -e -p pset.cfg"
988	>	ex_args += " -j $RUNTIME_AREA/crab_fjr_$NJob.xml"
989	>	# Type of cfg file
990	>	if major >= 2 :
991	>	ex_args += " -p pset.py"
992		else:
993	<	return " -p pset.cfg"
993	>	ex_args += " -p pset.cfg"
994	>	return ex_args
995
996		def inputSandbox(self, nj):
997		"""
#	Line 1175 | Line 1011 | class Cmssw(JobType):
1011		## additional input files
1012		tgz = self.additionalInputFileTgz()
1013		inp_box.append(tgz)
1014	+	## executable
1015	+	wrapper = os.path.basename(str(common._db.queryTask('scriptName')))
1016	+	inp_box.append(common.work_space.pathForTgz() +'job/'+ wrapper)
1017		return inp_box
1018
1019		def outputSandbox(self, nj):
#	Line 1185 | Line 1024 | class Cmssw(JobType):
1024
1025		## User Declared output files
1026		for out in (self.output_file+self.output_file_sandbox):
1027	<	n_out = nj + 1
1027	>	n_out = nj + 1
1028		out_box.append(self.numberFile_(out,str(n_out)))
1029		return out_box
1030
#	Line 1200 | Line 1039 | class Cmssw(JobType):
1039		Returns part of a job script which renames the produced files.
1040		"""
1041
1042	<	txt = '\n'
1043	<	txt += '# directory content\n'
1042	>	txt = '\n#Written by cms_cmssw::wsRenameOutput\n'
1043	>	txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
1044	>	txt += 'echo ">>> current directory content:"\n'
1045		txt += 'ls \n'
1046	+	txt += '\n'
1047
1207	–	txt += 'output_exit_status=0\n'
1208	–
1209	–	for fileWithSuffix in (self.output_file_sandbox):
1210	–	output_file_num = self.numberFile_(fileWithSuffix, '$NJob')
1211	–	txt += '\n'
1212	–	txt += '# check output file\n'
1213	–	txt += 'if [ -e ./'+fileWithSuffix+' ] ; then\n'
1214	–	txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA\n'
1215	–	txt += '    cp $RUNTIME_AREA/'+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1216	–	txt += 'else\n'
1217	–	txt += '    exit_status=60302\n'
1218	–	txt += '    echo "ERROR: Problem with output file '+fileWithSuffix+'"\n'
1219	–	if common.scheduler.boss_scheduler_name == 'condor_g':
1220	–	txt += '    if [ $middleware == OSG ]; then \n'
1221	–	txt += '        echo "prepare dummy output file"\n'
1222	–	txt += '        echo "Processing of job output failed" > $RUNTIME_AREA/'+output_file_num+'\n'
1223	–	txt += '    fi \n'
1224	–	txt += 'fi\n'
1225	–
1048		for fileWithSuffix in (self.output_file):
1049		output_file_num = self.numberFile_(fileWithSuffix, '$NJob')
1050		txt += '\n'
1051		txt += '# check output file\n'
1052		txt += 'if [ -e ./'+fileWithSuffix+' ] ; then\n'
1053	<	txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA\n'
1054	<	txt += '    cp $RUNTIME_AREA/'+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1053	>	if (self.copy_data == 1):  # For OSG nodes, file is in $WORKING_DIR, should not be moved to $RUNTIME_AREA
1054	>	txt += '    mv '+fileWithSuffix+' '+output_file_num+'\n'
1055	>	txt += '    ln -s `pwd`/'+output_file_num+' $RUNTIME_AREA/'+fileWithSuffix+'\n'
1056	>	else:
1057	>	txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1058	>	txt += '    ln -s $RUNTIME_AREA/'+output_file_num+' $RUNTIME_AREA/'+fileWithSuffix+'\n'
1059		txt += 'else\n'
1060	<	txt += '    exit_status=60302\n'
1061	<	txt += '    echo "ERROR: Problem with output file '+fileWithSuffix+'"\n'
1062	<	txt += '    echo "JOB_EXIT_STATUS = $exit_status"\n'
1237	<	txt += '    output_exit_status=$exit_status\n'
1238	<	if common.scheduler.boss_scheduler_name == 'condor_g':
1060	>	txt += '    job_exit_code=60302\n'
1061	>	txt += '    echo "WARNING: Output file '+fileWithSuffix+' not found"\n'
1062	>	if common.scheduler.name().upper() == 'CONDOR_G':
1063		txt += '    if [ $middleware == OSG ]; then \n'
1064		txt += '        echo "prepare dummy output file"\n'
1065		txt += '        echo "Processing of job output failed" > $RUNTIME_AREA/'+output_file_num+'\n'
#	Line 1244 | Line 1068 | class Cmssw(JobType):
1068		file_list = []
1069		for fileWithSuffix in (self.output_file):
1070		file_list.append(self.numberFile_(fileWithSuffix, '$NJob'))
1071	<
1071	>
1072		txt += 'file_list="'+string.join(file_list,' ')+'"\n'
1073	+	txt += '\n'
1074	+	txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
1075	+	txt += 'echo ">>> current directory content:"\n'
1076	+	txt += 'ls \n'
1077	+	txt += '\n'
1078		txt += 'cd $RUNTIME_AREA\n'
1079	+	txt += 'echo ">>> current directory (RUNTIME_AREA):  $RUNTIME_AREA"\n'
1080		return txt
1081
1082		def numberFile_(self, file, txt):
#	Line 1264 | Line 1094 | class Cmssw(JobType):
1094		result = name + '_' + txt + "." + ext
1095		else:
1096		result = name + '_' + txt
1097	<
1097	>
1098		return result
1099
1100		def getRequirements(self, nj=[]):
1101		"""
1102	<	return job requirements to add to jdl files
1102	>	return job requirements to add to jdl files
1103		"""
1104		req = ''
1105		if self.version:
#	Line 1284 | Line 1114 | class Cmssw(JobType):
1114		'", other.GlueHostApplicationSoftwareRunTimeEnvironment)'
1115
1116		req = req + ' && (other.GlueHostNetworkAdapterOutboundIP)'
1117	+	if common.scheduler.name() == "glitecoll":
1118	+	req += ' && other.GlueCEStateStatus == "Production" '
1119
1120		return req
1121
#	Line 1291 | Line 1123 | class Cmssw(JobType):
1123		""" return the config filename """
1124		return self.name()+'.cfg'
1125
1294	–	### OLI_DANIELE
1126		def wsSetupCMSOSGEnvironment_(self):
1127		"""
1128		Returns part of a job script which is prepares
1129		the execution environment and which is common for all CMS jobs.
1130		"""
1131	<	txt = '\n'
1132	<	txt += '   echo "### SETUP CMS OSG  ENVIRONMENT ###"\n'
1133	<	txt += '   if [ -f $GRID3_APP_DIR/cmssoft/cmsset_default.sh ] ;then\n'
1134	<	txt += '      # Use $GRID3_APP_DIR/cmssoft/cmsset_default.sh to setup cms software\n'
1135	<	txt += '       export SCRAM_ARCH='+self.executable_arch+'\n'
1136	<	txt += '       source $GRID3_APP_DIR/cmssoft/cmsset_default.sh '+self.version+'\n'
1306	<	txt += '   elif [ -f $OSG_APP/cmssoft/cms/cmsset_default.sh ] ;then\n'
1131	>	txt = '\n#Written by cms_cmssw::wsSetupCMSOSGEnvironment_\n'
1132	>	txt += '    echo ">>> setup CMS OSG environment:"\n'
1133	>	txt += '    echo "set SCRAM ARCH to ' + self.executable_arch + '"\n'
1134	>	txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
1135	>	txt += '    echo "SCRAM_ARCH = $SCRAM_ARCH"\n'
1136	>	txt += '    if [ -f $OSG_APP/cmssoft/cms/cmsset_default.sh ] ;then\n'
1137		txt += '      # Use $OSG_APP/cmssoft/cms/cmsset_default.sh to setup cms software\n'
1138	<	txt += '       export SCRAM_ARCH='+self.executable_arch+'\n'
1139	<	txt += '       source $OSG_APP/cmssoft/cms/cmsset_default.sh '+self.version+'\n'
1140	<	txt += '   else\n'
1141	<	txt += '       echo "SET_CMS_ENV 10020 ==> ERROR $GRID3_APP_DIR/cmssoft/cmsset_default.sh and $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1142	<	txt += '       echo "JOB_EXIT_STATUS = 10020"\n'
1143	<	txt += '       echo "JobExitCode=10020" | tee -a $RUNTIME_AREA/$repo\n'
1314	<	txt += '       dumpStatus $RUNTIME_AREA/$repo\n'
1315	<	txt += '       rm -f $RUNTIME_AREA/$repo \n'
1316	<	txt += '       echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1317	<	txt += '       echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1318	<	txt += '       exit 1\n'
1319	<	txt += '\n'
1320	<	txt += '       echo "Remove working directory: $WORKING_DIR"\n'
1321	<	txt += '       cd $RUNTIME_AREA\n'
1322	<	txt += '       /bin/rm -rf $WORKING_DIR\n'
1323	<	txt += '       if [ -d $WORKING_DIR ] ;then\n'
1324	<	txt += '           echo "SET_CMS_ENV 10017 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after $GRID3_APP_DIR/cmssoft/cmsset_default.sh and $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1325	<	txt += '           echo "JOB_EXIT_STATUS = 10017"\n'
1326	<	txt += '           echo "JobExitCode=10017" | tee -a $RUNTIME_AREA/$repo\n'
1327	<	txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1328	<	txt += '           rm -f $RUNTIME_AREA/$repo \n'
1329	<	txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1330	<	txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1331	<	txt += '       fi\n'
1332	<	txt += '\n'
1333	<	txt += '       exit 1\n'
1334	<	txt += '   fi\n'
1138	>	txt += '        source $OSG_APP/cmssoft/cms/cmsset_default.sh '+self.version+'\n'
1139	>	txt += '    else\n'
1140	>	txt += '        echo "ERROR ==> $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1141	>	txt += '        job_exit_code=10020\n'
1142	>	txt += '        func_exit\n'
1143	>	txt += '    fi\n'
1144		txt += '\n'
1145	<	txt += '   echo "SET_CMS_ENV 0 ==> setup cms environment ok"\n'
1146	<	txt += '   echo " END SETUP CMS OSG  ENVIRONMENT "\n'
1145	>	txt += '    echo "==> setup cms environment ok"\n'
1146	>	txt += '    echo "SCRAM_ARCH = $SCRAM_ARCH"\n'
1147
1148		return txt
1149	<
1149	>
1150		### OLI_DANIELE
1151		def wsSetupCMSLCGEnvironment_(self):
1152		"""
1153		Returns part of a job script which is prepares
1154		the execution environment and which is common for all CMS jobs.
1155		"""
1156	<	txt  = '   \n'
1157	<	txt += '   echo " ### SETUP CMS LCG  ENVIRONMENT ### "\n'
1158	<	txt += '   if [ ! $VO_CMS_SW_DIR ] ;then\n'
1159	<	txt += '       echo "SET_CMS_ENV 10031 ==> ERROR CMS software dir not found on WN `hostname`"\n'
1160	<	txt += '       echo "JOB_EXIT_STATUS = 10031" \n'
1161	<	txt += '       echo "JobExitCode=10031" | tee -a $RUNTIME_AREA/$repo\n'
1162	<	txt += '       dumpStatus $RUNTIME_AREA/$repo\n'
1163	<	txt += '       rm -f $RUNTIME_AREA/$repo \n'
1164	<	txt += '       echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1165	<	txt += '       echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1166	<	txt += '       exit 1\n'
1167	<	txt += '   else\n'
1168	<	txt += '       echo "Sourcing environment... "\n'
1169	<	txt += '       if [ ! -s $VO_CMS_SW_DIR/cmsset_default.sh ] ;then\n'
1170	<	txt += '           echo "SET_CMS_ENV 10020 ==> ERROR cmsset_default.sh file not found into dir $VO_CMS_SW_DIR"\n'
1171	<	txt += '           echo "JOB_EXIT_STATUS = 10020"\n'
1172	<	txt += '           echo "JobExitCode=10020" | tee -a $RUNTIME_AREA/$repo\n'
1173	<	txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1174	<	txt += '           rm -f $RUNTIME_AREA/$repo \n'
1175	<	txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1176	<	txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1177	<	txt += '           exit 1\n'
1178	<	txt += '       fi\n'
1179	<	txt += '       echo "sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1180	<	txt += '       source $VO_CMS_SW_DIR/cmsset_default.sh\n'
1181	<	txt += '       result=$?\n'
1182	<	txt += '       if [ $result -ne 0 ]; then\n'
1374	<	txt += '           echo "SET_CMS_ENV 10032 ==> ERROR problem sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1375	<	txt += '           echo "JOB_EXIT_STATUS = 10032"\n'
1376	<	txt += '           echo "JobExitCode=10032" | tee -a $RUNTIME_AREA/$repo\n'
1377	<	txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1378	<	txt += '           rm -f $RUNTIME_AREA/$repo \n'
1379	<	txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1380	<	txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1381	<	txt += '           exit 1\n'
1382	<	txt += '       fi\n'
1383	<	txt += '   fi\n'
1384	<	txt += '   \n'
1385	<	txt += '   echo "SET_CMS_ENV 0 ==> setup cms environment ok"\n'
1386	<	txt += '   echo "### END SETUP CMS LCG ENVIRONMENT ###"\n'
1156	>	txt = '\n#Written by cms_cmssw::wsSetupCMSLCGEnvironment_\n'
1157	>	txt += '    echo ">>> setup CMS LCG environment:"\n'
1158	>	txt += '    echo "set SCRAM ARCH and BUILD_ARCH to ' + self.executable_arch + ' ###"\n'
1159	>	txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
1160	>	txt += '    export BUILD_ARCH='+self.executable_arch+'\n'
1161	>	txt += '    if [ ! $VO_CMS_SW_DIR ] ;then\n'
1162	>	txt += '        echo "ERROR ==> CMS software dir not found on WN `hostname`"\n'
1163	>	txt += '        job_exit_code=10031\n'
1164	>	txt += '        func_exit\n'
1165	>	txt += '    else\n'
1166	>	txt += '        echo "Sourcing environment... "\n'
1167	>	txt += '        if [ ! -s $VO_CMS_SW_DIR/cmsset_default.sh ] ;then\n'
1168	>	txt += '            echo "ERROR ==> cmsset_default.sh file not found into dir $VO_CMS_SW_DIR"\n'
1169	>	txt += '            job_exit_code=10020\n'
1170	>	txt += '            func_exit\n'
1171	>	txt += '        fi\n'
1172	>	txt += '        echo "sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1173	>	txt += '        source $VO_CMS_SW_DIR/cmsset_default.sh\n'
1174	>	txt += '        result=$?\n'
1175	>	txt += '        if [ $result -ne 0 ]; then\n'
1176	>	txt += '            echo "ERROR ==> problem sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1177	>	txt += '            job_exit_code=10032\n'
1178	>	txt += '            func_exit\n'
1179	>	txt += '        fi\n'
1180	>	txt += '    fi\n'
1181	>	txt += '    \n'
1182	>	txt += '    echo "==> setup cms environment ok"\n'
1183		return txt
1184
1185	<	### FEDE FOR DBS OUTPUT PUBLICATION
1185	>	### FEDE FOR DBS OUTPUT PUBLICATION
1186		def modifyReport(self, nj):
1187		"""
1188	<	insert the part of the script that modifies the FrameworkJob Report
1188	>	insert the part of the script that modifies the FrameworkJob Report
1189		"""
1190
1191	<	txt = ''
1192	<	try:
1193	<	publish_data = int(self.cfg_params['USER.publish_data'])
1398	<	except KeyError:
1399	<	publish_data = 0
1400	<	if (publish_data == 1):
1401	<	txt += 'echo "Modify Job Report" \n'
1402	<	#txt += 'chmod a+x $RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py\n'
1403	<	################ FEDE FOR DBS2 #############################################
1404	<	txt += 'chmod a+x $SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py\n'
1405	<	#############################################################################
1406	<	#try:
1407	<	#    publish_data = int(self.cfg_params['USER.publish_data'])
1408	<	#except KeyError:
1409	<	#    publish_data = 0
1410	<
1411	<	txt += 'if [ -z "$SE" ]; then\n'
1412	<	txt += '    SE="" \n'
1413	<	txt += 'fi \n'
1414	<	txt += 'if [ -z "$SE_PATH" ]; then\n'
1415	<	txt += '    SE_PATH="" \n'
1416	<	txt += 'fi \n'
1417	<	txt += 'echo "SE = $SE"\n'
1418	<	txt += 'echo "SE_PATH = $SE_PATH"\n'
1419	<
1420	<	#if (publish_data == 1):
1421	<	#processedDataset = self.cfg_params['USER.processed_datasetname']
1191	>	txt = '\n#Written by cms_cmssw::modifyReport\n'
1192	>	publish_data = int(self.cfg_params.get('USER.publish_data',0))
1193	>	if (publish_data == 1):
1194		processedDataset = self.cfg_params['USER.publish_data_name']
1195	<	txt += 'ProcessedDataset='+processedDataset+'\n'
1196	<	#### LFN=/store/user/<user>/processedDataset_PSETHASH
1197	<	txt += 'if [ "$SE_PATH" == "" ]; then\n'
1198	<	#### FEDE: added slash in LFN ##############
1195	>	LFNBaseName = LFNBase(processedDataset)
1196	>
1197	>	txt += 'if [ $copy_exit_status -eq 0 ]; then\n'
1198	>	txt += '    FOR_LFN=%s_${PSETHASH}/\n'%(LFNBaseName)
1199	>	txt += 'else\n'
1200		txt += '    FOR_LFN=/copy_problems/ \n'
1201	<	txt += 'else \n'
1202	<	txt += '    tmp=`echo $SE_PATH | awk -F \'store\' \'{print$2}\'` \n'
1203	<	#####  FEDE TO BE CHANGED, BECAUSE STORE IS HARDCODED!!!! ########
1204	<	txt += '    FOR_LFN=/store$tmp \n'
1205	<	txt += 'fi \n'
1201	>	txt += '    SE=""\n'
1202	>	txt += '    SE_PATH=""\n'
1203	>	txt += 'fi\n'
1204	>
1205	>	txt += 'echo ">>> Modify Job Report:" \n'
1206	>	txt += 'chmod a+x $SOFTWARE_DIR/ProdCommon/ProdCommon/FwkJobRep/ModifyJobReport.py\n'
1207	>	txt += 'ProcessedDataset='+processedDataset+'\n'
1208		txt += 'echo "ProcessedDataset = $ProcessedDataset"\n'
1209	+	txt += 'echo "SE = $SE"\n'
1210	+	txt += 'echo "SE_PATH = $SE_PATH"\n'
1211		txt += 'echo "FOR_LFN = $FOR_LFN" \n'
1212		txt += 'echo "CMSSW_VERSION = $CMSSW_VERSION"\n\n'
1213	<	#txt += 'echo "$RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1214	<	txt += 'echo "$SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1438	<	txt += '$SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1439	<	#txt += '$RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1440	<
1213	>	txt += 'echo "$SOFTWARE_DIR/ProdCommon/ProdCommon/FwkJobRep/ModifyJobReport.py $RUNTIME_AREA/crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1214	>	txt += '$SOFTWARE_DIR/ProdCommon/ProdCommon/FwkJobRep/ModifyJobReport.py $RUNTIME_AREA/crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1215		txt += 'modifyReport_result=$?\n'
1442	–	txt += 'echo modifyReport_result = $modifyReport_result\n'
1216		txt += 'if [ $modifyReport_result -ne 0 ]; then\n'
1217	<	txt += '    exit_status=1\n'
1218	<	txt += '    echo "ERROR: Problem with ModifyJobReport"\n'
1217	>	txt += '    modifyReport_result=70500\n'
1218	>	txt += '    job_exit_code=$modifyReport_result\n'
1219	>	txt += '    echo "ModifyReportResult=$modifyReport_result" | tee -a $RUNTIME_AREA/$repo\n'
1220	>	txt += '    echo "WARNING: Problem with ModifyJobReport"\n'
1221		txt += 'else\n'
1222	<	txt += '    mv NewFrameworkJobReport.xml crab_fjr_$NJob.xml\n'
1222	>	txt += '    mv NewFrameworkJobReport.xml $RUNTIME_AREA/crab_fjr_$NJob.xml\n'
1223		txt += 'fi\n'
1449	–	else:
1450	–	txt += 'echo "no data publication required"\n'
1451	–	#txt += 'ProcessedDataset=no_data_to_publish \n'
1452	–	#### FEDE: added slash in LFN ##############
1453	–	#txt += 'FOR_LFN=/local/ \n'
1454	–	#txt += 'echo "ProcessedDataset = $ProcessedDataset"\n'
1455	–	#txt += 'echo "FOR_LFN = $FOR_LFN" \n'
1456	–	return txt
1457	–
1458	–	def cleanEnv(self):
1459	–	### OLI_DANIELE
1460	–	txt = ''
1461	–	txt += 'if [ $middleware == OSG ]; then\n'
1462	–	txt += '    cd $RUNTIME_AREA\n'
1463	–	txt += '    echo "Remove working directory: $WORKING_DIR"\n'
1464	–	txt += '    /bin/rm -rf $WORKING_DIR\n'
1465	–	txt += '    if [ -d $WORKING_DIR ] ;then\n'
1466	–	txt += '              echo "SET_EXE 60999 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after cleanup of WN"\n'
1467	–	txt += '              echo "JOB_EXIT_STATUS = 60999"\n'
1468	–	txt += '              echo "JobExitCode=60999" | tee -a $RUNTIME_AREA/$repo\n'
1469	–	txt += '              dumpStatus $RUNTIME_AREA/$repo\n'
1470	–	txt += '        rm -f $RUNTIME_AREA/$repo \n'
1471	–	txt += '        echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1472	–	txt += '        echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1473	–	txt += '    fi\n'
1474	–	txt += 'fi\n'
1475	–	txt += '\n'
1224		return txt
1225
1226		def setParam_(self, param, value):
#	Line 1481 | Line 1229 | class Cmssw(JobType):
1229		def getParams(self):
1230		return self._params
1231
1484	–	def setTaskid_(self):
1485	–	self._taskId = self.cfg_params['taskId']
1486	–
1487	–	def getTaskid(self):
1488	–	return self._taskId
1489	–
1232		def uniquelist(self, old):
1233		"""
1234		remove duplicates from a list
#	Line 1496 | Line 1238 | class Cmssw(JobType):
1238		nd[e]=0
1239		return nd.keys()
1240
1241	<
1500	<	def checkOut(self, limit):
1241	>	def outList(self):
1242		"""
1243		check the dimension of the output files
1244		"""
1245	<	txt = 'echo "*****************************************"\n'
1246	<	txt += 'echo "** Starting output sandbox limit check **"\n'
1506	<	txt += 'echo "*****************************************"\n'
1507	<	allOutFiles = ""
1245	>	txt = ''
1246	>	txt += 'echo ">>> list of expected files on output sandbox"\n'
1247		listOutFiles = []
1248	<	for fileOut in (self.output_file+self.output_file_sandbox):
1249	<	if fileOut.find('crab_fjr') == -1:
1250	<	allOutFiles = allOutFiles + " " + self.numberFile_(fileOut, '$NJob')
1251	<	listOutFiles.append(self.numberFile_(fileOut, '$NJob'))
1252	<	txt += 'echo "OUTPUT files: '+str(allOutFiles)+'";\n'
1253	<	txt += 'ls -gGhrta;\n'
1254	<	txt += 'sum=0;\n'
1255	<	txt += 'for file in '+str(allOutFiles)+' ; do\n'
1256	<	txt += '    if [ -e $file ]; then\n'
1257	<	txt += '        tt=`ls -gGrta $file | awk \'{ print $3 }\'`\n'
1258	<	txt += '        sum=`expr $sum + $tt`\n'
1259	<	txt += '    else\n'
1260	<	txt += '        echo "WARNING: output file $file not found!"\n'
1261	<	txt += '    fi\n'
1262	<	txt += 'done\n'
1524	<	txt += 'echo "Total Output dimension: $sum";\n'
1525	<	txt += 'limit='+str(limit)+';\n'
1526	<	txt += 'echo "OUTPUT FILES LIMIT SET TO: $limit";\n'
1527	<	txt += 'if [ $limit -lt $sum ]; then\n'
1528	<	txt += '    echo "WARNING: output files have to big size - something will be lost;"\n'
1529	<	txt += '    echo "         checking the output file sizes..."\n'
1530	<	"""
1531	<	txt += '    dim=0;\n'
1532	<	txt += '    exclude=0;\n'
1533	<	txt += '    for files in '+str(allOutFiles)+' ; do\n'
1534	<	txt += '        sumTemp=0;\n'
1535	<	txt += '        for file2 in '+str(allOutFiles)+' ; do\n'
1536	<	txt += '            if [ $file != $file2 ]; then\n'
1537	<	txt += '                tt=`ls -gGrta $file2 | awk \'{ print $3 }\';`\n'
1538	<	txt += '                sumTemp=`expr $sumTemp + $tt`;\n'
1539	<	txt += '            fi\n'
1540	<	txt += '        done\n'
1541	<	txt += '        if [ $sumTemp -lt $limit ]; then\n'
1542	<	txt += '            if [ $dim -lt $sumTemp ]; then\n'
1543	<	txt += '                dim=$sumTemp;\n'
1544	<	txt += '                exclude=$file;\n'
1545	<	txt += '            fi\n'
1546	<	txt += '        fi\n'
1547	<	txt += '    done\n'
1548	<	txt += '    echo "Dimension calculated: $dim"; echo "File to exclude: $exclude";\n'
1549	<	"""
1550	<	txt += '    tot=0;\n'
1551	<	txt += '    for file2 in '+str(allOutFiles)+' ; do\n'
1552	<	txt += '        tt=`ls -gGrta $file2 | awk \'{ print $3 }\';`\n'
1553	<	txt += '        tot=`expr $tot + $tt`;\n'
1554	<	txt += '        if [ $limit -lt $tot ]; then\n'
1555	<	txt += '            tot=`expr $tot - $tt`;\n'
1556	<	txt += '            fileLast=$file;\n'
1557	<	txt += '            break;\n'
1558	<	txt += '        fi\n'
1559	<	txt += '    done\n'
1560	<	txt += '    echo "Dimension calculated: $tot"; echo "First file to exclude: $file";\n'
1561	<	txt += '    flag=0;\n'
1562	<	txt += '    for filess in '+str(allOutFiles)+' ; do\n'
1563	<	txt += '        if [ $fileLast = $filess ]; then\n'
1564	<	txt += '            flag=1;\n'
1565	<	txt += '        fi\n'
1566	<	txt += '        if [ $flag -eq 1 ]; then\n'
1567	<	txt += '            rm -f $filess;\n'
1568	<	txt += '        fi\n'
1569	<	txt += '    done\n'
1570	<	txt += '    ls -agGhrt;\n'
1571	<	txt += '    echo "WARNING: output files are too big in dimension: can not put in the output_sandbox.";\n'
1572	<	txt += '    echo "JOB_EXIT_STATUS = 70000";\n'
1573	<	txt += '    exit_status=70000;\n'
1574	<	txt += 'else'
1575	<	txt += '    echo "Total Output dimension $sum is fine.";\n'
1576	<	txt += 'fi\n'
1577	<	txt += 'echo "*****************************************"\n'
1578	<	txt += 'echo "*** Ending output sandbox limit check ***"\n'
1579	<	txt += 'echo "*****************************************"\n'
1248	>	stdout = 'CMSSW_$NJob.stdout'
1249	>	stderr = 'CMSSW_$NJob.stderr'
1250	>	if (self.return_data == 1):
1251	>	for file in (self.output_file+self.output_file_sandbox):
1252	>	listOutFiles.append(self.numberFile_(file, '$NJob'))
1253	>	listOutFiles.append(stdout)
1254	>	listOutFiles.append(stderr)
1255	>	else:
1256	>	for file in (self.output_file_sandbox):
1257	>	listOutFiles.append(self.numberFile_(file, '$NJob'))
1258	>	listOutFiles.append(stdout)
1259	>	listOutFiles.append(stderr)
1260	>	txt += 'echo "output files: '+string.join(listOutFiles,' ')+'"\n'
1261	>	txt += 'filesToCheck="'+string.join(listOutFiles,' ')+'"\n'
1262	>	txt += 'export filesToCheck\n'
1263		return txt

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines