def getSampleDirectories(analysis, sampleList): source = getNtupleDirectory(analysis) directories = [] for s in sampleList: for d in glob.glob(os.path.join(source, s)): directories += [d] return directories
def getSampleDirectories(analysis,sampleList): source = getNtupleDirectory(analysis) directories = [] for s in sampleList: for d in glob.glob(os.path.join(source,s)): directories += [d] return directories
def __init__(self,analysis,sample,**kwargs): # default to access via sample/analysis self.analysis = analysis self.sample = sample self.shift = kwargs.pop('shift','') self.skipHists = kwargs.pop('skipHists',False) self.isData = isData(self.sample) self.intLumi = kwargs.get('intLumi',float(getLumi())) logging.debug('Initializing {0} {1} {2}'.format(self.analysis,self.sample,self.shift)) # backup passing custom parameters self.ntupleDirectory = kwargs.pop('ntupleDirectory','{0}/{1}'.format(getNtupleDirectory(self.analysis,shift=self.shift),self.sample)) self.inputFileList = kwargs.pop('inputFileList','') self.outputFile = kwargs.pop('outputFile',getNewFlatHistograms(self.analysis,self.sample,shift=self.shift)) if os.path.dirname(self.outputFile): python_mkdir(os.path.dirname(self.outputFile)) self.treeName = kwargs.pop('treeName',getTreeName(self.analysis)) if hasProgress: self.pbar = kwargs.pop('progressbar',ProgressBar(widgets=['{0}: '.format(sample),' ',SimpleProgress(),' ',Percentage(),' ',Bar(),' ',ETA()])) else: self.pbar = None # get stuff needed to flatten self.infile = 0 self.tchain = 0 self.initialized = False self.hists = {} self.datasets = {}
def __init__(self, analysis, sample, **kwargs): # default to access via sample/analysis self.analysis = analysis self.sample = sample self.shift = kwargs.pop('shift', '') logging.debug('Initializing {0} {1} {2}'.format( self.analysis, self.sample, self.shift)) # backup passing custom parameters self.ntupleDirectory = kwargs.pop( 'ntupleDirectory', '{0}/{1}'.format( getNtupleDirectory(self.analysis, shift=self.shift), self.sample)) self.inputFileList = kwargs.pop('inputFileList', '') self.outputFile = kwargs.pop('outputFile', '') self.json = kwargs.pop('json', getSkimJson(self.analysis, self.sample)) self.pickle = kwargs.pop('pickle', getSkimPickle(self.analysis, self.sample)) self.treeName = kwargs.pop('treeName', getTreeName(self.analysis)) if hasProgress: self.pbar = kwargs.pop( 'progressbar', ProgressBar(widgets=[ '{0}: '.format(sample), ' ', SimpleProgress(), ' events ', Percentage(), ' ', Bar(), ' ', ETA() ])) else: self.pbar = None # get stuff needed to flatten self.infile = 0 self.tchain = 0 self.initialized = False self.counts = {}
def __init__(self,analysis,sample,**kwargs): # default to access via sample/analysis self.analysis = analysis self.sample = sample self.shift = kwargs.pop('shift','') self.intLumi = kwargs.get('intLumi',float(getLumi())) logging.debug('Initializing {0} {1} {2}'.format(self.analysis,self.sample,self.shift)) # backup passing custom parameters self.ntupleDirectory = kwargs.pop('ntupleDirectory','{0}/{1}'.format(getNtupleDirectory(self.analysis,shift=self.shift),self.sample)) self.inputFileList = kwargs.pop('inputFileList','') self.outputFile = kwargs.pop('outputFile','') self.json = kwargs.pop('json',getSkimJson(self.analysis,self.sample)) self.pickle = kwargs.pop('pickle',getSkimPickle(self.analysis,self.sample)) self.treeName = kwargs.pop('treeName',getTreeName(self.analysis)) if hasProgress: self.pbar = kwargs.pop('progressbar',ProgressBar(widgets=['{0}: '.format(sample),' ',SimpleProgress(),' events ',Percentage(),' ',Bar(),' ',ETA()])) else: self.pbar = None # get stuff needed to flatten self.infile = 0 self.tchain = 0 self.initialized = False self.counts = {}