cmsswPreprocessor.py

Go to the documentation of this file.

import os                                                               
import re 
import imp
from PhysicsTools.HeppyCore.framework.config import CFG
class CmsswPreprocessor :
    def __init__(self,configFile,command="cmsRun") :
        self.configFile=configFile
        self.command=command
    
    def run(self,component,wd,firstEvent,nEvents):
        print wd,firstEvent,nEvents
        if nEvents is None:
            nEvents = -1
        cmsswConfig = imp.load_source("cmsRunProcess",self.configFile)
        inputfiles= []
        for fn in component.files :
            if not re.match("file:.*",fn) and not re.match("root:.*",fn) :
                fn="file:"+fn
            inputfiles.append(fn)
        cmsswConfig.process.source.fileNames = inputfiles
        cmsswConfig.process.maxEvents.input=nEvents
        #fixme: implement skipEvent / firstevent

        outfilename=wd+"/cmsswPreProcessing.root"
        for outName in cmsswConfig.process.endpath.moduleNames() :
            out = getattr(cmsswConfig.process,outName)
            out.fileName = outfilename
        if not hasattr(component,"options"):
            component.options = CFG(name="postCmsrunOptions")
                #use original as primary and new as secondary 
                #component.options.inputFiles= component.files
        #component.options.secondaryInputFiles=[outfilename]

                #use new as primary and original as secondary
                component.options.secondaryInputFiles= component.files
        component.options.inputFiles=[outfilename]
                component.files=[outfilename]

        configfile=wd+"/cmsRun_config.py"
        f = open(configfile, 'w')
        f.write(cmsswConfig.process.dumpPython())
        f.close()
        runstring="%s %s >& %s/cmsRun.log" % (self.command,configfile,wd)
        print "Running pre-processor: %s " %runstring
        os.system(runstring)
        return component