MatrixInjector.MatrixInjector Class Reference

Inheritance diagram for MatrixInjector.MatrixInjector:

Public Member Functions
def	__init__ (self, opt, mode='init', options='')
def	prepare (self, mReader, directories, mode='init')
def	submit (self)
def	upload (self)
def	uploadConf (self, filePath, label, where)
	the info are not in the task specific dict but in the general dict t_input.update(copy.deepcopy(self.defaultHarvest)) t_input['DQMConfigCacheID']=t_second['ConfigCacheID'] More...

Static Public Member Functions
def	get_wmsplit ()

Public Attributes
	batchName
	batchTime
	candidateWFName
	chainDicts
	couch
	couchCache
	count
	DbsUrl
	defaultChain
	defaultHarvest
	defaultInput
	defaultScratch
	defaultTask
	dqmgui
	group
	keep
	label
	longWFName
	memoryOffset
	memPerCore
	numberEventsInLuminosityBlock
	speciallabel
	testMode
	user
	version
	wmagent

Detailed Description

Definition at line 41 of file MatrixInjector.py.

Constructor & Destructor Documentation

__init__()

def MatrixInjector.MatrixInjector.__init__	(		self,
			opt,
			mode = 'init',
			options = ''
	)

Definition at line 43 of file MatrixInjector.py.

    def __init__(self,opt,mode='init',options=''):
        self.count=1040
 
        self.dqmgui=None
        self.wmagent=None
        for k in options.split(','):
            if k.startswith('dqm:'):
                self.dqmgui=k.split(':',1)[-1]
            elif k.startswith('wma:'):
                self.wmagent=k.split(':',1)[-1]
 
        self.testMode=((mode!='submit') and (mode!='force'))
        self.version =1
        self.keep = opt.keep
        self.memoryOffset = opt.memoryOffset
        self.memPerCore = opt.memPerCore
        self.numberEventsInLuminosityBlock = opt.numberEventsInLuminosityBlock
        self.batchName = ''
        self.batchTime = str(int(time.time()))
        if(opt.batchName):
            self.batchName = '__'+opt.batchName+'-'+self.batchTime
 
        #wagemt stuff
        if not self.wmagent:
            self.wmagent=os.getenv('WMAGENT_REQMGR')
        if not self.wmagent:
            if not opt.testbed :
                self.wmagent = 'cmsweb.cern.ch'
                self.DbsUrl = "https://"+self.wmagent+"/dbs/prod/global/DBSReader"
            else :
                self.wmagent = 'cmsweb-testbed.cern.ch'
                self.DbsUrl = "https://"+self.wmagent+"/dbs/int/global/DBSReader"
 
        if not self.dqmgui:
            self.dqmgui="https://cmsweb.cern.ch/dqm/relval"
        #couch stuff
        self.couch = 'https://'+self.wmagent+'/couchdb'
#        self.couchDB = 'reqmgr_config_cache'
        self.couchCache={} # so that we do not upload like crazy, and recyle cfgs
        self.user = os.getenv('USER')
        self.group = 'ppd'
        self.label = 'RelValSet_'+os.getenv('CMSSW_VERSION').replace('-','')+'_v'+str(self.version)
        self.speciallabel=''
        if opt.label:
            self.speciallabel= '_'+opt.label
        self.longWFName = []
 
        if not os.getenv('WMCORE_ROOT'):
            print('\n\twmclient is not setup properly. Will not be able to upload or submit requests.\n')
            if not self.testMode:
                print('\n\t QUIT\n')
                sys.exit(-18)
        else:
            print('\n\tFound wmclient\n')
            
        self.defaultChain={
            "RequestType" :    "TaskChain",                    #this is how we handle relvals
            "SubRequestType" : "RelVal",                       #this is how we handle relvals, now that TaskChain is also used for central MC production
            "RequestPriority": 500000,
            "Requestor": self.user,                           #Person responsible
            "Group": self.group,                              #group for the request
            "CMSSWVersion": os.getenv('CMSSW_VERSION'),       #CMSSW Version (used for all tasks in chain)
            "Campaign": os.getenv('CMSSW_VERSION'),           # = AcquisitionEra, will be reset later to the one of first task, will both be the CMSSW_VERSION
            "ScramArch": os.getenv('SCRAM_ARCH'),             #Scram Arch (used for all tasks in chain)
            "ProcessingVersion": self.version,                #Processing Version (used for all tasks in chain)
            "GlobalTag": None,                                #Global Tag (overridden per task)
            "ConfigCacheUrl": self.couch,                     #URL of CouchDB containing Config Cache
            "DbsUrl": self.DbsUrl,
            #- Will contain all configs for all Tasks
            #"SiteWhitelist" : ["T2_CH_CERN", "T1_US_FNAL"],   #Site whitelist
            "TaskChain" : None,                                  #Define number of tasks in chain.
            "nowmTasklist" : [],  #a list of tasks as we put them in
            "Multicore" : 1,   # do not set multicore for the whole chain
            "Memory" : 3000,
            "SizePerEvent" : 1234,
            "TimePerEvent" : 10,
            "PrepID": os.getenv('CMSSW_VERSION')
            }
 
        self.defaultHarvest={
            "EnableHarvesting" : "True",
            "DQMUploadUrl" : self.dqmgui,
            "DQMConfigCacheID" : None,
            "Multicore" : 1              # hardcode Multicore to be 1 for Harvest
            }
        
        self.defaultScratch={
            "TaskName" : None,                            #Task Name
            "ConfigCacheID" : None,                   #Generator Config id
            "GlobalTag": None,
            "SplittingAlgo"  : "EventBased",             #Splitting Algorithm
            "EventsPerJob" : None,                       #Size of jobs in terms of splitting algorithm
            "EventsPerLumi" : None,
            "RequestNumEvents" : None,                      #Total number of events to generate
            "Seeding" : "AutomaticSeeding",                          #Random seeding method
            "PrimaryDataset" : None,                          #Primary Dataset to be created
            "nowmIO": {},
            "Multicore" : opt.nThreads,                  # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified 
            "KeepOutput" : False
            }
        self.defaultInput={
            "TaskName" : "DigiHLT",                                      #Task Name
            "ConfigCacheID" : None,                                      #Processing Config id
            "GlobalTag": None,
            "InputDataset" : None,                                       #Input Dataset to be processed
            "SplittingAlgo"  : "LumiBased",                        #Splitting Algorithm
            "LumisPerJob" : 10,               #Size of jobs in terms of splitting algorithm
            "nowmIO": {},
            "Multicore" : opt.nThreads,                       # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified 
            "KeepOutput" : False
            }
        self.defaultTask={
            "TaskName" : None,                                 #Task Name
            "InputTask" : None,                                #Input Task Name (Task Name field of a previous Task entry)
            "InputFromOutputModule" : None,                    #OutputModule name in the input task that will provide files to process
            "ConfigCacheID" : None,                            #Processing Config id
            "GlobalTag": None,
            "SplittingAlgo"  : "LumiBased",                        #Splitting Algorithm
            "LumisPerJob" : 10,               #Size of jobs in terms of splitting algorithm
            "nowmIO": {},
            "Multicore" : opt.nThreads,                       # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified 
            "KeepOutput" : False
            }
 
        self.chainDicts={}
 
 

Member Function Documentation

get_wmsplit()

def MatrixInjector.MatrixInjector.get_wmsplit ( )

static

Return a "wmsplit" dictionary that contain non-default LumisPerJob values

Definition at line 171 of file MatrixInjector.py.

    def get_wmsplit():
        """
        Return a "wmsplit" dictionary that contain non-default LumisPerJob values
        """
        wmsplit = {}
        try:
            #from Configuration.PyReleaseValidation.relval_steps import wmsplit
            wmsplit = {}
            wmsplit['DIGIHI']=5
            wmsplit['RECOHI']=5
            wmsplit['HLTD']=5
            wmsplit['RECODreHLT']=2
            wmsplit['DIGIPU']=4
            wmsplit['DIGIPU1']=4
            wmsplit['RECOPU1']=1
            wmsplit['DIGIUP15_PU50']=1
            wmsplit['RECOUP15_PU50']=1
            wmsplit['DIGIUP15_PU25']=1
            wmsplit['RECOUP15_PU25']=1
            wmsplit['DIGIUP15_PU25HS']=1
            wmsplit['RECOUP15_PU25HS']=1
            wmsplit['DIGIHIMIX']=5
            wmsplit['RECOHIMIX']=5
            wmsplit['RECODSplit']=1
            wmsplit['SingleMuPt10_UP15_ID']=1
            wmsplit['DIGIUP15_ID']=1
            wmsplit['RECOUP15_ID']=1
            wmsplit['TTbar_13_ID']=1
            wmsplit['SingleMuPt10FS_ID']=1
            wmsplit['TTbarFS_ID']=1
            wmsplit['RECODR2_50nsreHLT']=5
            wmsplit['RECODR2_25nsreHLT']=5
            wmsplit['RECODR2_2016reHLT']=5
            wmsplit['RECODR2_50nsreHLT_HIPM']=5
            wmsplit['RECODR2_25nsreHLT_HIPM']=5
            wmsplit['RECODR2_2016reHLT_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimSingleMu']=1
            wmsplit['RECODR2_2016reHLT_skimDoubleEG']=1
            wmsplit['RECODR2_2016reHLT_skimMuonEG']=1
            wmsplit['RECODR2_2016reHLT_skimJetHT']=1
            wmsplit['RECODR2_2016reHLT_skimMET']=1
            wmsplit['RECODR2_2016reHLT_skimSinglePh']=1
            wmsplit['RECODR2_2016reHLT_skimMuOnia']=1
            wmsplit['RECODR2_2016reHLT_skimSingleMu_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimDoubleEG_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimMuonEG_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimJetHT_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimMET_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimSinglePh_HIPM']=1
            wmsplit['RECODR2_2016reHLT_skimMuOnia_HIPM']=1
            wmsplit['RECODR2_2017reHLT_Prompt']=1
            wmsplit['RECODR2_2017reHLT_skimSingleMu_Prompt_Lumi']=1
            wmsplit['RECODR2_2017reHLT_skimDoubleEG_Prompt']=1
            wmsplit['RECODR2_2017reHLT_skimMET_Prompt']=1
            wmsplit['RECODR2_2017reHLT_skimMuOnia_Prompt']=1
            wmsplit['RECODR2_2017reHLT_Prompt_L1TEgDQM']=1
            wmsplit['RECODR2_2018reHLT_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimSingleMu_Prompt_Lumi']=1
            wmsplit['RECODR2_2018reHLT_skimDoubleEG_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimMET_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimMuOnia_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimEGamma_Prompt_L1TEgDQM']=1
            wmsplit['RECODR2_2018reHLT_skimMuonEG_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimCharmonium_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt_HEfail']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt_BadHcalMitig']=1
            wmsplit['RECODR2_2018reHLTAlCaTkCosmics_Prompt']=1
            wmsplit['RECODR2_2018reHLT_skimDisplacedJet_Prompt']=1
            wmsplit['RECODR2_2018reHLT_ZBPrompt']=1
            wmsplit['RECODR2_2018reHLT_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimSingleMu_Offline_Lumi']=1
            wmsplit['RECODR2_2018reHLT_skimDoubleEG_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimMET_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimMuOnia_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimEGamma_Offline_L1TEgDQM']=1
            wmsplit['RECODR2_2018reHLT_skimMuonEG_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimCharmonium_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Offline_HEfail']=1
            wmsplit['RECODR2_2018reHLT_skimJetHT_Offline_BadHcalMitig']=1
            wmsplit['RECODR2_2018reHLTAlCaTkCosmics_Offline']=1
            wmsplit['RECODR2_2018reHLT_skimDisplacedJet_Offline']=1
            wmsplit['RECODR2_2018reHLT_ZBOffline']=1
            wmsplit['HLTDR2_50ns']=1
            wmsplit['HLTDR2_25ns']=1
            wmsplit['HLTDR2_2016']=1
            wmsplit['HLTDR2_2017']=1
            wmsplit['HLTDR2_2018']=1
            wmsplit['HLTDR2_2018_BadHcalMitig']=1
            wmsplit['Hadronizer']=1
            wmsplit['DIGIUP15']=1
            wmsplit['RECOUP15']=1
            wmsplit['RECOAODUP15']=5
            wmsplit['DBLMINIAODMCUP15NODQM']=5
            wmsplit['DigiFull']=5
            wmsplit['RecoFull']=5
            wmsplit['DigiFullPU']=1
            wmsplit['RecoFullPU']=1
            wmsplit['RECOHID11']=1
            wmsplit['DigiFullTriggerPU_2026D17PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D17PU']=1
            wmsplit['DIGIUP17']=1
            wmsplit['RECOUP17']=1
            wmsplit['DIGIUP17_PU25']=1
            wmsplit['RECOUP17_PU25']=1
            wmsplit['DIGICOS_UP16']=1
            wmsplit['RECOCOS_UP16']=1
            wmsplit['DIGICOS_UP17']=1
            wmsplit['RECOCOS_UP17']=1
            wmsplit['DIGICOS_UP18']=1
            wmsplit['RECOCOS_UP18']=1
            wmsplit['DIGICOS_UP21']=1
            wmsplit['RECOCOS_UP21']=1
            wmsplit['HYBRIDRepackHI2015VR']=1
            wmsplit['HYBRIDZSHI2015']=1
            wmsplit['RECOHID15']=1
            wmsplit['RECOHID18']=1
            wmsplit['DigiFullTriggerPU_2026D35PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D35PU']=1
            wmsplit['DigiFullTriggerPU_2026D41PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D41PU']=1
            wmsplit['DigiFullTriggerPU_2026D43PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D43PU']=1
            wmsplit['DigiFullTriggerPU_2026D44PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D44PU']=1
            wmsplit['DigiFullTriggerPU_2026D45PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D45PU']=1
            wmsplit['DigiFullTriggerPU_2026D46PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D46PU']=1
            wmsplit['DigiFullTriggerPU_2026D47PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D47PU']=1
            wmsplit['DigiFullTriggerPU_2026D48PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D48PU']=1
            wmsplit['DigiFullTriggerPU_2026D49PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D49PU']=1
            wmsplit['DigiFullTriggerPU_2026D50PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D50PU']=1
            wmsplit['DigiFullTriggerPU_2026D51PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D51PU']=1
            wmsplit['DigiFullTriggerPU_2026D52PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D52PU']=1
            wmsplit['DigiFullTriggerPU_2026D57PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D57PU']=1
            wmsplit['DigiFullTriggerPU_2026D58PU'] = 1
            wmsplit['RecoFullGlobalPU_2026D58PU']=1
        except Exception as ex:
            print('Exception while building a wmsplit dictionary: %s' % (str(ex)))
            return {}
 
        return wmsplit
 

References edm.print(), and str.

prepare()

def MatrixInjector.MatrixInjector.prepare	(		self,
			mReader,
			directories,
			mode = 'init'
	)

Definition at line 323 of file MatrixInjector.py.

    def prepare(self,mReader, directories, mode='init'):
        wmsplit = MatrixInjector.get_wmsplit()
        acqEra=False
        for (n,dir) in directories.items():
            chainDict=copy.deepcopy(self.defaultChain)
            print("inspecting",dir)
            nextHasDSInput=None
            for (x,s) in mReader.workFlowSteps.items():
                #x has the format (num, prefix)
                #s has the format (num, name, commands, stepList)
                if x[0]==n:
                    #print "found",n,s[3]
                    #chainDict['RequestString']='RV'+chainDict['CMSSWVersion']+s[1].split('+')[0]
                    index=0
                    splitForThisWf=None
                    thisLabel=self.speciallabel
                    #if 'HARVESTGEN' in s[3]:
                    if len( [step for step in s[3] if "HARVESTGEN" in step] )>0:
                        chainDict['TimePerEvent']=0.01
                        thisLabel=thisLabel+"_gen"
                    # for double miniAOD test
                    if len( [step for step in s[3] if "DBLMINIAODMCUP15NODQM" in step] )>0:
                        thisLabel=thisLabel+"_dblMiniAOD"
                    processStrPrefix=''
                    setPrimaryDs=None
                    nanoedmGT=''
                    for step in s[3]:
                        
                        if 'INPUT' in step or (not isinstance(s[2][index],str)):
                            nextHasDSInput=s[2][index]
 
                        else:
 
                            if (index==0):
                                #first step and not input -> gen part
                                chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultScratch))
                                try:
                                    chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
                                except:
                                    print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
                                    return -15
 
                                chainDict['nowmTasklist'][-1]['PrimaryDataset']='RelVal'+s[1].split('+')[0]
                                if not '--relval' in s[2][index]:
                                    print('Impossible to create task from scratch without splitting information with --relval')
                                    return -12
                                else:
                                    arg=s[2][index].split()
                                    ns=list(map(int,arg[len(arg) - arg[-1::-1].index('--relval')].split(',')))
                                    chainDict['nowmTasklist'][-1]['RequestNumEvents'] = ns[0]
                                    chainDict['nowmTasklist'][-1]['EventsPerJob'] = ns[1]
                                    chainDict['nowmTasklist'][-1]['EventsPerLumi'] = ns[1]
                                    #overwrite EventsPerLumi if numberEventsInLuminosityBlock is set in cmsDriver
                                    if 'numberEventsInLuminosityBlock' in s[2][index]:
                                        nEventsInLuminosityBlock = re.findall('process.source.numberEventsInLuminosityBlock=cms.untracked.uint32\(([ 0-9 ]*)\)', s[2][index],re.DOTALL)
                                        if nEventsInLuminosityBlock[-1].isdigit() and int(nEventsInLuminosityBlock[-1]) < ns[1]:
                                            chainDict['nowmTasklist'][-1]['EventsPerLumi'] = int(nEventsInLuminosityBlock[-1])
                                    if(self.numberEventsInLuminosityBlock > 0 and self.numberEventsInLuminosityBlock <= ns[1]):
                                        chainDict['nowmTasklist'][-1]['EventsPerLumi'] = self.numberEventsInLuminosityBlock
                                if 'FASTSIM' in s[2][index] or '--fast' in s[2][index]:
                                    thisLabel+='_FastSim'
                                if 'lhe' in s[2][index] in s[2][index]:
                                    chainDict['nowmTasklist'][-1]['LheInputFiles'] =True
 
                            elif nextHasDSInput:
                                chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultInput))
                                try:
                                    chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
                                except:
                                    print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
                                    return -15
                                chainDict['nowmTasklist'][-1]['InputDataset']=nextHasDSInput.dataSet
                                if ('DQMHLTonRAWAOD' in step) :
                                    chainDict['nowmTasklist'][-1]['IncludeParents']=True
                                splitForThisWf=nextHasDSInput.split
                                chainDict['nowmTasklist'][-1]['LumisPerJob']=splitForThisWf
                                if step in wmsplit:
                                    chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit[step]
                                # get the run numbers or #events
                                if len(nextHasDSInput.run):
                                    chainDict['nowmTasklist'][-1]['RunWhitelist']=nextHasDSInput.run
                                if len(nextHasDSInput.ls):
                                    chainDict['nowmTasklist'][-1]['LumiList']=nextHasDSInput.ls
                                #print "what is s",s[2][index]
                                if '--data' in s[2][index] and nextHasDSInput.label:
                                    thisLabel+='_RelVal_%s'%nextHasDSInput.label
                                if 'filter' in chainDict['nowmTasklist'][-1]['nowmIO']:
                                    print("This has an input DS and a filter sequence: very likely to be the PyQuen sample")
                                    processStrPrefix='PU_'
                                    setPrimaryDs = 'RelVal'+s[1].split('+')[0]
                                    if setPrimaryDs:
                                        chainDict['nowmTasklist'][-1]['PrimaryDataset']=setPrimaryDs
                                nextHasDSInput=None
                            else:
                                #not first step and no inputDS
                                chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultTask))
                                try:
                                    chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
                                except:
                                    print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
                                    return -15
                                if splitForThisWf:
                                    chainDict['nowmTasklist'][-1]['LumisPerJob']=splitForThisWf
                                if step in wmsplit:
                                    chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit[step]
 
                            # change LumisPerJob for Hadronizer steps. 
                            if 'Hadronizer' in step: 
                                chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit['Hadronizer']
 
                            #print step
                            chainDict['nowmTasklist'][-1]['TaskName']=step
                            if setPrimaryDs:
                                chainDict['nowmTasklist'][-1]['PrimaryDataset']=setPrimaryDs
                            chainDict['nowmTasklist'][-1]['ConfigCacheID']='%s/%s.py'%(dir,step)
                            chainDict['nowmTasklist'][-1]['GlobalTag']=chainDict['nowmTasklist'][-1]['nowmIO']['GT'] # copy to the proper parameter name
                            chainDict['GlobalTag']=chainDict['nowmTasklist'][-1]['nowmIO']['GT'] #set in general to the last one of the chain
                            if 'NANOEDM' in step :
                                nanoedmGT = chainDict['nowmTasklist'][-1]['nowmIO']['GT']
                            if 'NANOMERGE' in step :
                                chainDict['GlobalTag'] = nanoedmGT
                            if 'pileup' in chainDict['nowmTasklist'][-1]['nowmIO']:
                                chainDict['nowmTasklist'][-1]['MCPileup']=chainDict['nowmTasklist'][-1]['nowmIO']['pileup']
                            if '--pileup ' in s[2][index]:      # catch --pileup (scenarion) and not --pileup_ (dataset to be mixed) => works also making PRE-MIXed dataset
                                processStrPrefix='PU_'          # take care of pu overlay done with GEN-SIM mixing
                                if (  s[2][index].split()[  s[2][index].split().index('--pileup')+1 ]  ).find('25ns')  > 0 :
                                    processStrPrefix='PU25ns_'
                                elif   (  s[2][index].split()[  s[2][index].split().index('--pileup')+1 ]  ).find('50ns')  > 0 :
                                    processStrPrefix='PU50ns_'
                            if 'premix_stage2' in s[2][index] and '--pileup_input' in s[2][index]: # take care of pu overlay done with DIGI mixing of premixed events
                                if s[2][index].split()[ s[2][index].split().index('--pileup_input')+1  ].find('25ns')  > 0 :
                                    processStrPrefix='PUpmx25ns_'
                                elif s[2][index].split()[ s[2][index].split().index('--pileup_input')+1  ].find('50ns')  > 0 :
                                    processStrPrefix='PUpmx50ns_'
 
                            if acqEra:
                                #chainDict['AcquisitionEra'][step]=(chainDict['CMSSWVersion']+'-PU_'+chainDict['nowmTasklist'][-1]['GlobalTag']).replace('::All','')+thisLabel
                                chainDict['AcquisitionEra'][step]=chainDict['CMSSWVersion']
                                chainDict['ProcessingString'][step]=processStrPrefix+chainDict['nowmTasklist'][-1]['GlobalTag'].replace('::All','').replace('-','_')+thisLabel
                                if 'NANOMERGE' in step :
                                    chainDict['ProcessingString'][step]=processStrPrefix+nanoedmGT.replace('::All','').replace('-','_')+thisLabel
                            else:
                                #chainDict['nowmTasklist'][-1]['AcquisitionEra']=(chainDict['CMSSWVersion']+'-PU_'+chainDict['nowmTasklist'][-1]['GlobalTag']).replace('::All','')+thisLabel
                                chainDict['nowmTasklist'][-1]['AcquisitionEra']=chainDict['CMSSWVersion']
                                chainDict['nowmTasklist'][-1]['ProcessingString']=processStrPrefix+chainDict['nowmTasklist'][-1]['GlobalTag'].replace('::All','').replace('-','_')+thisLabel
                                if 'NANOMERGE' in step :
                                    chainDict['nowmTasklist'][-1]['ProcessingString']=processStrPrefix+nanoedmGT.replace('::All','').replace('-','_')+thisLabel
 
                            if (self.batchName):
                                chainDict['nowmTasklist'][-1]['Campaign'] = chainDict['nowmTasklist'][-1]['AcquisitionEra']+self.batchName
 
                            # specify different ProcessingString for double miniAOD dataset
                            if ('DBLMINIAODMCUP15NODQM' in step): 
                                chainDict['nowmTasklist'][-1]['ProcessingString']=chainDict['nowmTasklist'][-1]['ProcessingString']+'_miniAOD' 
 
                            if( chainDict['nowmTasklist'][-1]['Multicore'] ):
                                # the scaling factor of 1.2GB / thread is empirical and measured on a SECOND round of tests with PU samples
                                # the number of threads is NO LONGER assumed to be the same for all tasks
                                # https://hypernews.cern.ch/HyperNews/CMS/get/edmFramework/3509/1/1/1.html
                                # now change to 1.5GB / additional thread according to discussion:
                                # https://hypernews.cern.ch/HyperNews/CMS/get/relval/4817/1/1.html
#                                chainDict['nowmTasklist'][-1]['Memory'] = 3000 + int( chainDict['nowmTasklist'][-1]['Multicore']  -1 )*1500
                                chainDict['nowmTasklist'][-1]['Memory'] = self.memoryOffset + int( chainDict['nowmTasklist'][-1]['Multicore']  -1 ) * self.memPerCore
 
                        index+=1
                    #end of loop through steps
                    chainDict['RequestString']='RV'+chainDict['CMSSWVersion']+s[1].split('+')[0]
                    if processStrPrefix or thisLabel:
                        chainDict['RequestString']+='_'+processStrPrefix+thisLabel
                    #check candidate WF name
                    self.candidateWFName = self.user+'_'+chainDict['RequestString']
                    if (len(self.candidateWFName)>MAXWORKFLOWLENGTH):
                        self.longWFName.append(self.candidateWFName)
 

References mps_setup.append, MatrixInjector.MatrixInjector.batchName, MatrixInjector.MatrixInjector.defaultChain, MatrixInjector.MatrixInjector.defaultInput, MatrixInjector.MatrixInjector.defaultScratch, MatrixInjector.MatrixInjector.defaultTask, spr.find(), createfilelist.int, list(), genParticles_cff.map, MatrixInjector.MatrixInjector.memoryOffset, MatrixInjector.MatrixInjector.memPerCore, MatrixInjector.MatrixInjector.numberEventsInLuminosityBlock, edm.print(), readEcalDQMStatus.read, python.rootplot.root2matplotlib.replace(), MatrixInjector.MatrixInjector.speciallabel, and cms::dd.split().

submit()

def MatrixInjector.MatrixInjector.submit

(

self

)

Definition at line 642 of file MatrixInjector.py.

    def submit(self):
        try:
            from modules.wma import makeRequest,approveRequest
            from wmcontrol import random_sleep
            print('\n\tFound wmcontrol\n')
        except:
            print('\n\tUnable to find wmcontrol modules. Please include it in your python path\n')
            if not self.testMode:
                print('\n\t QUIT\n')
                sys.exit(-17)
 
        import pprint
        for (n,d) in self.chainDicts.items():
            if self.testMode:
                print("Only viewing request",n)
                print(pprint.pprint(d))
            else:
                #submit to wmagent each dict
                print("For eyes before submitting",n)
                print(pprint.pprint(d))
                print("Submitting",n,"...........")
                workFlow=makeRequest(self.wmagent,d,encodeDict=True)
                print("...........",n,"submitted")
                random_sleep()
        if self.testMode and len(self.longWFName)>0:
            print("\n*** WARNING: "+str(len(self.longWFName))+" workflows have too long names for submission (>"+str(MAXWORKFLOWLENGTH)+ "characters) ***")
            print('\n'.join(self.longWFName))

References MatrixInjector.MatrixInjector.chainDicts, mps_monitormerge.items, join(), MatrixInjector.MatrixInjector.longWFName, edm.print(), str, MatrixInjector.MatrixInjector.testMode, and MatrixInjector.MatrixInjector.wmagent.

upload()

def MatrixInjector.MatrixInjector.upload

(

self

)

Definition at line 622 of file MatrixInjector.py.

    def upload(self):
        for (n,d) in self.chainDicts.items():
            for it in d:
                if it.startswith("Task") and it!='TaskChain':
                    #upload
                    couchID=self.uploadConf(d[it]['ConfigCacheID'],
                                            str(n)+d[it]['TaskName'],
                                            d['ConfigCacheUrl']
                                            )
                    print(d[it]['ConfigCacheID']," uploaded to couchDB for",str(n),"with ID",couchID)
                    d[it]['ConfigCacheID']=couchID
                if it =='DQMConfigCacheID':
                    couchID=self.uploadConf(d['DQMConfigCacheID'],
                                            str(n)+'harvesting',
                                            d['ConfigCacheUrl']
                                            )
                    print(d['DQMConfigCacheID'],"uploaded to couchDB for",str(n),"with ID",couchID)
                    d['DQMConfigCacheID']=couchID
                        
            

References MatrixInjector.MatrixInjector.chainDicts, mps_monitormerge.items, edm.print(), str, and MatrixInjector.MatrixInjector.uploadConf().

uploadConf()

def MatrixInjector.MatrixInjector.uploadConf	(		self,
			filePath,
			label,
			where
	)

the info are not in the task specific dict but in the general dict t_input.update(copy.deepcopy(self.defaultHarvest)) t_input['DQMConfigCacheID']=t_second['ConfigCacheID']

batch name appended to Campaign name chainDict['Campaign'] = chainDict['AcquisitionEra'] clean things up now provide the number of tasks

Definition at line 595 of file MatrixInjector.py.

    def uploadConf(self,filePath,label,where):
        labelInCouch=self.label+'_'+label
        cacheName=filePath.split('/')[-1]
        if self.testMode:
            self.count+=1
            print('\tFake upload of',filePath,'to couch with label',labelInCouch)
            return self.count
        else:
            try:
                from modules.wma import upload_to_couch,DATABASE_NAME
            except:
                print('\n\tUnable to find wmcontrol modules. Please include it in your python path\n')
                print('\n\t QUIT\n')
                sys.exit(-16)
 
            if cacheName in self.couchCache:
                print("Not re-uploading",filePath,"to",where,"for",label)
                cacheId=self.couchCache[cacheName]
            else:
                print("Loading",filePath,"to",where,"for",label)
                
                pool = multiprocessing.Pool(1)
                cacheIds = pool.map( upload_to_couch_oneArg, [(filePath,labelInCouch,self.user,self.group,where)] )
                cacheId = cacheIds[0]
                self.couchCache[cacheName]=cacheId
            return cacheId
    

References MatrixInjector.MatrixInjector.couchCache, TmCcu.count, TmModule.count, TmApvPair.count, TmPsu.count, MatrixInjector.MatrixInjector.count, ValidationMisalignedTracker.count, SiStripDetSummary::Values.count, MatrixInjector.MatrixInjector.group, ElectronLikelihoodCategoryData.label, entry< T >.label, classes.PlotData.label, SiPixelFedFillerWordEventNumber.label, L1GtBoardTypeStringToEnum.label, HcalLutSet.label, TtEvent::HypoClassKeyStringToEnum.label, L1GtPsbQuadStringToEnum.label, MatrixInjector.MatrixInjector.label, ValidationMisalignedTracker.label, HcalDbProducer::ServiceToken< ProductType, LABEL, EffectiveType >.label, L1GtConditionTypeStringToEnum.label, L1GtConditionCategoryStringToEnum.label, cond::payloadInspector::ModuleVersion.label, edm.print(), MatrixInjector.MatrixInjector.testMode, EcalTPGParamReaderFromDB.user, popcon::RpcObGasData.user, popcon::RPCObPVSSmapData.user, popcon::RpcDataT.user, popcon::RpcDataV.user, popcon::RpcDataGasMix.user, popcon::RpcDataUXC.user, popcon::RpcDataI.user, popcon::RpcDataS.user, popcon::RpcDataFebmap.user, and MatrixInjector.MatrixInjector.user.

Referenced by MatrixInjector.MatrixInjector.upload().

Member Data Documentation

batchName

MatrixInjector.MatrixInjector.batchName

Definition at line 60 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

batchTime

MatrixInjector.MatrixInjector.batchTime

Definition at line 61 of file MatrixInjector.py.

candidateWFName

MatrixInjector.MatrixInjector.candidateWFName

Definition at line 493 of file MatrixInjector.py.

chainDicts

MatrixInjector.MatrixInjector.chainDicts

Definition at line 167 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.submit(), and MatrixInjector.MatrixInjector.upload().

couch

MatrixInjector.MatrixInjector.couch

Definition at line 79 of file MatrixInjector.py.

couchCache

MatrixInjector.MatrixInjector.couchCache

Definition at line 81 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.uploadConf().

count

MatrixInjector.MatrixInjector.count

Definition at line 44 of file MatrixInjector.py.

Referenced by core.TriggerBitFilter.TriggerBitFilter.process(), core.JSONAnalyzer.JSONAnalyzer.process(), core.SkimAnalyzerCount.SkimAnalyzerCount.process(), objects.VertexAnalyzer.VertexAnalyzer.process(), and MatrixInjector.MatrixInjector.uploadConf().

DbsUrl

MatrixInjector.MatrixInjector.DbsUrl

Definition at line 71 of file MatrixInjector.py.

defaultChain

MatrixInjector.MatrixInjector.defaultChain

Definition at line 98 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

defaultHarvest

MatrixInjector.MatrixInjector.defaultHarvest

Definition at line 122 of file MatrixInjector.py.

defaultInput

MatrixInjector.MatrixInjector.defaultInput

Definition at line 143 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

defaultScratch

MatrixInjector.MatrixInjector.defaultScratch

Definition at line 129 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

defaultTask

MatrixInjector.MatrixInjector.defaultTask

Definition at line 154 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

dqmgui

MatrixInjector.MatrixInjector.dqmgui

Definition at line 46 of file MatrixInjector.py.

group

MatrixInjector.MatrixInjector.group

Definition at line 83 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.uploadConf().

keep

MatrixInjector.MatrixInjector.keep

Definition at line 56 of file MatrixInjector.py.

label

MatrixInjector.MatrixInjector.label

Definition at line 84 of file MatrixInjector.py.

Referenced by python.rootplot.root2matplotlib.Hist.bar(), python.rootplot.root2matplotlib.Hist.barh(), python.rootplot.root2matplotlib.Hist.errorbar(), python.rootplot.root2matplotlib.Hist.errorbarh(), core.TriggerMatchAnalyzer.TriggerMatchAnalyzer.process(), core.AutoHandle.AutoHandle.ReallyLoad(), and MatrixInjector.MatrixInjector.uploadConf().

longWFName

MatrixInjector.MatrixInjector.longWFName

Definition at line 88 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.submit().

memoryOffset

MatrixInjector.MatrixInjector.memoryOffset

Definition at line 57 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

memPerCore

MatrixInjector.MatrixInjector.memPerCore

Definition at line 58 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

numberEventsInLuminosityBlock

MatrixInjector.MatrixInjector.numberEventsInLuminosityBlock

Definition at line 59 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

speciallabel

MatrixInjector.MatrixInjector.speciallabel

Definition at line 85 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.prepare().

testMode

MatrixInjector.MatrixInjector.testMode

Definition at line 54 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.submit(), and MatrixInjector.MatrixInjector.uploadConf().

user

MatrixInjector.MatrixInjector.user

Definition at line 82 of file MatrixInjector.py.

Referenced by dataset.BaseDataset.printInfo(), DTWorkflow.DTWorkflow.remote_out_path(), production_tasks.CheckDatasetExists.run(), production_tasks.GenerateMask.run(), production_tasks.SourceCFG.run(), production_tasks.FullCFG.run(), production_tasks.MonitorJobs.run(), production_tasks.CleanJobFiles.run(), and MatrixInjector.MatrixInjector.uploadConf().

version

MatrixInjector.MatrixInjector.version

Definition at line 55 of file MatrixInjector.py.

Referenced by validation.Sample.datasetpattern(), and validation.Sample.filename().

wmagent

MatrixInjector.MatrixInjector.wmagent

Definition at line 47 of file MatrixInjector.py.

Referenced by MatrixInjector.MatrixInjector.submit().