ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/Creator.py
Revision: 1.35
Committed: Thu Apr 6 16:15:45 2006 UTC (19 years, 1 month ago) by slacapra
Content type: text/x-python
Branch: MAIN
Changes since 1.34: +3 -1 lines
Log Message:
More explicit print out

File Contents

# User Rev Content
1 nsmirnov 1.1 from Actor import Actor
2     from WorkSpace import WorkSpace
3     from JobList import JobList
4     from JobDB import JobDB
5     from ScriptWriter import ScriptWriter
6     from Scheduler import Scheduler
7     from crab_logger import Logger
8     from crab_exceptions import *
9     from crab_util import *
10     import common
11    
12 slacapra 1.8 import os, string, math
13 corvo 1.26 # marco
14 nsmirnov 1.1 class Creator(Actor):
15 corvo 1.30 def __init__(self, job_type_name, cfg_params, ncjobs):
16 nsmirnov 1.1 self.job_type_name = job_type_name
17 corvo 1.30 self.job_type = None
18 nsmirnov 1.1 self.cfg_params = cfg_params
19 nsmirnov 1.2 self.total_njobs = 0
20     self.total_number_of_events = 0
21     self.job_number_of_events = 0
22 slacapra 1.7 self.first_event = 0
23 slacapra 1.14
24 corvo 1.30 self.createJobTypeObject()
25 nsmirnov 1.3 common.logger.debug(5, __name__+": JobType "+self.job_type.name()+" created")
26 nsmirnov 1.1
27     self.job_type.prepareSteeringCards()
28 nsmirnov 1.3 common.logger.debug(5, __name__+": Steering cards prepared")
29 nsmirnov 1.1
30 nsmirnov 1.2 self.defineTotalNumberOfJobs_()
31 nsmirnov 1.3 common.logger.debug(5, __name__+": total # of jobs = "+`self.total_njobs`)
32 nsmirnov 1.1
33     # Set number of jobs to be created
34    
35     self.ncjobs = ncjobs
36     if ncjobs == 'all' : self.ncjobs = self.total_njobs
37     if ncjobs > self.total_njobs : self.ncjobs = self.total_njobs
38 spiga 1.11
39 slacapra 1.35 # This is code for proto-monitoring
40 spiga 1.20 fileCODE1 = open(common.work_space.shareDir()+"/.code","a")
41     ### commented for FAMOS
42 afanfani 1.34 if self.job_type.name() == 'ORCA' or self.job_type.name() == 'ORCA_DBSDLS' or self.job_type.name() == 'ORCA_COMMON' or self.job_type.name() == 'ORCA_COMMON_DBSDLS':
43 spiga 1.20 self.owner = cfg_params['ORCA.owner']
44     self.dataset = cfg_params['ORCA.dataset']
45     fileCODE1.write('::'+str(self.job_type.name())+'::'+str(self.ncjobs)+'::'+str(self.dataset)+'::'+str(self.owner))
46     pass
47     elif self.job_type.name() == 'FAMOS':
48 spiga 1.27 self.inputFile = cfg_params['FAMOS.input_lfn']
49     self.executable = cfg_params['FAMOS.executable']
50     fileCODE1.write('::'+str(self.job_type.name())+'::'+str(self.ncjobs)+'::'+str(self.inputFile)+'::'+str(self.executable))
51 spiga 1.20 pass
52 spiga 1.11 fileCODE1.close()
53    
54 slacapra 1.35 # This is code for dashboard
55 corvo 1.33 try:
56     fl = open(common.work_space.shareDir() + '/' + self.cfg_params['apmon'].fName, 'w')
57     self.cfg_params['GridName'] = runCommand("voms-proxy-info -identity")
58     common.logger.debug(5, "GRIDNAME: "+self.cfg_params['GridName'])
59     taskType = 'analysis'
60    
61     params = {'tool': common.prog_name, 'tool_ui': os.environ['HOSTNAME'], \
62     'scheduler': self.cfg_params['CRAB.scheduler'], 'GridName': self.cfg_params['GridName'].strip(), \
63     'taskType': taskType, 'vo': self.cfg_params['EDG.virtual_organization'], 'user': self.cfg_params['user']}
64     jtParam = self.job_type.getParams()
65     for i in jtParam.iterkeys():
66     params[i] = string.strip(jtParam[i])
67     for j, k in params.iteritems():
68     # print "Values: %s %s"%(j, k)
69     fl.write(j + ':' + k + '\n')
70     fl.close()
71     except:
72     exctype, value = sys.exc_info()[:2]
73 slacapra 1.35 common.logger.message("Creator::run Exception raised in collection params for dashboard: %s %s"%(exctype, value))
74 corvo 1.33 pass
75 nsmirnov 1.1
76     #TODO: deprecated code, not needed,
77     # will be eliminated when WorkSpace.saveConfiguration()
78     # will be improved.
79     #
80     # Set/Save Job Type name
81    
82     jt_fname = common.work_space.shareDir() + 'jobtype'
83     if os.path.exists(jt_fname):
84     # Read stored job type name
85     jt_file = open(jt_fname, 'r')
86     jt = jt_file.read()
87     if self.job_type_name:
88     if ( jt != self.job_type_name+'\n' ):
89     msg = 'Job Type mismatch: requested <' + self.job_type_name
90     msg += '>, found <' + jt[:-1] + '>.'
91     raise CrabException(msg)
92     pass
93     else:
94     self.job_type_name = jt[:-1]
95     pass
96     jt_file.close()
97     pass
98     else:
99     # Save job type name
100     jt_file = open(jt_fname, 'w')
101     jt_file.write(self.job_type_name+'\n')
102     jt_file.close()
103     pass
104     #end of deprecated code
105    
106 nsmirnov 1.3 common.logger.debug(5, "Creator constructor finished")
107 nsmirnov 1.1 return
108    
109 nsmirnov 1.2 def defineTotalNumberOfJobs_(self):
110     """
111     Calculates the total number of jobs to be created.
112     """
113    
114     try:
115 slacapra 1.8 self.first_event = int(self.cfg_params['USER.first_event'])
116 slacapra 1.7 except KeyError:
117     self.first_event = 0
118 slacapra 1.8 common.logger.debug(1,"First event ot be analyzed: "+str(self.first_event))
119 slacapra 1.7
120 slacapra 1.10 # TODO Could we find a better way to get this number?
121 slacapra 1.8 maxAvailableEvents = int(self.job_type.maxEvents)
122     common.logger.debug(1,"Available events: "+str(maxAvailableEvents))
123 slacapra 1.7
124 slacapra 1.8 # some sanity check
125     if self.first_event>=maxAvailableEvents:
126 slacapra 1.7 raise CrabException('First event is bigger than maximum number of available events!')
127    
128 slacapra 1.8 # the total number of events to be analyzed
129 slacapra 1.7 try:
130 nsmirnov 1.2 n = self.cfg_params['USER.total_number_of_events']
131     if n == 'all': n = '-1'
132 slacapra 1.8 if n == '-1':
133     self.total_number_of_events = (maxAvailableEvents - self.first_event)
134     common.logger.debug(1,"Analysing all available events "+str(self.total_number_of_events))
135     else:
136     if maxAvailableEvents<(int(n)+self.first_event):
137 slacapra 1.17 raise CrabException('(First event + total events)='+str(int(n)+self.first_event)+' is bigger than maximum number of available events '+str(maxAvailableEvents)+' !! Use "total_number_of_events=-1" to analyze to whole dataset')
138 slacapra 1.8 self.total_number_of_events = int(n)
139 slacapra 1.7 except KeyError:
140     common.logger.message("total_number_of_events not defined, set it to maximum available")
141 slacapra 1.8 self.total_number_of_events = (maxAvailableEvents - self.first_event)
142 slacapra 1.7 pass
143 slacapra 1.8 common.logger.message("Total number of events to be analyzed: "+str(self.total_number_of_events))
144    
145 slacapra 1.7
146 slacapra 1.8 # read user directives
147 slacapra 1.7 eventPerJob=0
148     try:
149     eventPerJob = self.cfg_params['USER.job_number_of_events']
150 nsmirnov 1.2 except KeyError:
151     pass
152    
153 slacapra 1.7 jobsPerTask=0
154 nsmirnov 1.2 try:
155 slacapra 1.7 jobsPerTask = int(self.cfg_params['USER.total_number_of_jobs'])
156 nsmirnov 1.2 except KeyError:
157 slacapra 1.7 pass
158    
159     # If both the above set, complain and use event per jobs
160     if eventPerJob>0 and jobsPerTask>0:
161 nsmirnov 1.2 msg = 'Warning. '
162 slacapra 1.7 msg += 'job_number_of_events and total_number_of_jobs are both defined '
163     msg += 'Using job_number_of_events.'
164 nsmirnov 1.2 common.logger.message(msg)
165 slacapra 1.7 jobsPerTask = 0
166     if eventPerJob==0 and jobsPerTask==0:
167     msg = 'Warning. '
168     msg += 'job_number_of_events and total_number_of_jobs are not defined '
169     msg += 'Creating just one job for all events.'
170     common.logger.message(msg)
171     jobsPerTask = 1
172    
173     # first case: events per job defined
174     if eventPerJob>0:
175     n=eventPerJob
176     if n == 'all' or n == '-1' or (int(n)>self.total_number_of_events and self.total_number_of_events>0):
177     common.logger.message("Asking more events than available: set it to maximum available")
178     self.job_number_of_events = self.total_number_of_events
179 slacapra 1.8 self.total_njobs = 1
180 slacapra 1.7 else:
181     self.job_number_of_events = int(n)
182 slacapra 1.8 self.total_njobs = int((self.total_number_of_events-1)/self.job_number_of_events)+1
183 slacapra 1.7 # second case: jobs per task defined
184     elif jobsPerTask>0:
185 slacapra 1.8 common.logger.debug(2,"total number of events: "+str(self.total_number_of_events)+" JobPerTask "+str(jobsPerTask))
186     self.job_number_of_events = int(math.floor((self.total_number_of_events)/jobsPerTask))
187     self.total_njobs = jobsPerTask
188 slacapra 1.7 # should not happen...
189     else:
190     raise CrabException('Somthing wrong with splitting')
191    
192 slacapra 1.8 common.logger.debug(2,"total number of events: "+str(self.total_number_of_events)+
193 slacapra 1.7 " events per job: "+str(self.job_number_of_events))
194 slacapra 1.8
195 slacapra 1.7 return
196    
197     def writeJobsSpecsToDB(self):
198     """
199     Write firstEvent and maxEvents in the DB for future use
200     """
201    
202     common.jobDB.load()
203     # case one: write first and max events
204     nJobs=self.nJobs()
205    
206     firstEvent=self.first_event
207 fanzago 1.21 lastJobsNumberOfEvents = self.job_number_of_events
208 fanzago 1.28
209     try:
210     self.events_management = common.analisys_common_info['events_management']
211 fanzago 1.31 #print "self.events_management = ", self.events_management
212 fanzago 1.28 except KeyError: self.events_management = 0
213    
214 slacapra 1.7 # last jobs is different...
215     for job in range(nJobs-1):
216     common.jobDB.setFirstEvent(job, firstEvent)
217     common.jobDB.setMaxEvents(job, self.job_number_of_events)
218 fanzago 1.28 ### fede checks parameter for famos and orca
219     #try:
220     # self.events_management = common.analisys_common_info['events_management']
221     # print "self.events_management = ", self.events_management
222     #except KeyError: self.events_management = 0
223 spiga 1.27 firstEvent=firstEvent+self.job_number_of_events
224 slacapra 1.7
225     # this is the last job
226     common.jobDB.setFirstEvent(nJobs-1, firstEvent)
227 fanzago 1.21 if int(self.events_management) == 1:
228     lastJobsNumberOfEvents= (self.total_number_of_events+self.first_event)-firstEvent
229 slacapra 1.8 common.jobDB.setMaxEvents(nJobs-1, lastJobsNumberOfEvents)
230 slacapra 1.7
231 slacapra 1.15 if (lastJobsNumberOfEvents!=self.job_number_of_events):
232     common.logger.message(str(self.total_njobs-1)+' jobs will be created for '+str(self.job_number_of_events)+' events each plus 1 for '+str(lastJobsNumberOfEvents)+' events for a total of '+str(self.job_number_of_events*(self.total_njobs-1)+lastJobsNumberOfEvents)+' events')
233     else:
234     common.logger.message(str(self.total_njobs)+' jobs will be created for '+str(self.job_number_of_events)+' events each for a total of '+str(self.job_number_of_events*(self.total_njobs-1)+lastJobsNumberOfEvents)+' events')
235 slacapra 1.7
236     # case two (to be implemented) write eventCollections for each jobs
237 nsmirnov 1.2
238 slacapra 1.9 # save the DB
239     common.jobDB.save()
240 nsmirnov 1.2 return
241    
242 nsmirnov 1.1 def nJobs(self):
243     return self.total_njobs
244    
245     def createJobTypeObject(self):
246     file_name = 'cms_'+ string.lower(self.job_type_name)
247     klass_name = string.capitalize(self.job_type_name)
248    
249     try:
250     klass = importName(file_name, klass_name)
251     except KeyError:
252     msg = 'No `class '+klass_name+'` found in file `'+file_name+'.py`'
253     raise CrabException(msg)
254     except ImportError, e:
255     msg = 'Cannot create job type '+self.job_type_name
256     msg += ' (file: '+file_name+', class '+klass_name+'):\n'
257     msg += str(e)
258     raise CrabException(msg)
259    
260     self.job_type = klass(self.cfg_params)
261     return
262 nsmirnov 1.5
263     def jobType(self):
264     return self.job_type
265 nsmirnov 1.1
266     def run(self):
267 nsmirnov 1.4 """
268     The main method of the class.
269     """
270 nsmirnov 1.1
271 nsmirnov 1.3 common.logger.debug(5, "Creator::run() called")
272    
273 nsmirnov 1.1 # Instantiate ScriptWriter
274    
275     script_writer = ScriptWriter('crab_template.sh')
276    
277     # Loop over jobs
278    
279     njc = 0
280 nsmirnov 1.4 for nj in range(self.total_njobs):
281 nsmirnov 1.1 if njc == self.ncjobs : break
282 nsmirnov 1.4 st = common.jobDB.status(nj)
283 nsmirnov 1.1 if st != 'X': continue
284    
285 nsmirnov 1.6 common.logger.message("Creating job # "+`(nj+1)`)
286 nsmirnov 1.3
287 nsmirnov 1.1 # Prepare configuration file
288    
289 nsmirnov 1.4 self.job_type.modifySteeringCards(nj)
290 fanzago 1.12 # Create script (sh)
291 fanzago 1.13 #print "nel for prima del modifyTemplateScript, nj vale", nj
292 nsmirnov 1.4 script_writer.modifyTemplateScript(nj)
293     os.chmod(common.job_list[nj].scriptFilename(), 0744)
294 nsmirnov 1.1
295 fanzago 1.12 # Create scheduler scripts (jdl)
296     common.scheduler.createSchScript(nj)
297    
298 nsmirnov 1.4 common.jobDB.setStatus(nj, 'C')
299 slacapra 1.9 # common: write input and output sandbox
300     common.jobDB.setInputSandbox(nj, self.job_type.inputSandbox(nj))
301    
302     outputSandbox=self.job_type.outputSandbox(nj)
303 gutsche 1.32 stdout=common.job_list[nj].stdout()
304 slacapra 1.9 stderr=common.job_list[nj].stderr()
305     outputSandbox.append(common.job_list[nj].stdout())
306     # check if out!=err
307     if stdout != stderr:
308     outputSandbox.append(common.job_list[nj].stderr())
309     common.jobDB.setOutputSandbox(nj, outputSandbox)
310 corvo 1.18 common.jobDB.setTaskId(nj, self.cfg_params['taskId'])
311 slacapra 1.9
312 nsmirnov 1.1 njc = njc + 1
313     pass
314    
315     ####
316 fanzago 1.12
317 nsmirnov 1.1 common.jobDB.save()
318    
319     msg = '\nTotal of %d jobs created'%njc
320     if njc != self.ncjobs: msg = msg + ' from %d requested'%self.ncjobs
321     msg = msg + '.\n'
322     common.logger.message(msg)
323     return