validation Namespace Reference

Classes
class	Sample
class	SimpleSample
class	SimpleValidation
class	Validation

Functions
def	_copyDir (src, dst)
def	_copySubDir (oldfile, newfile, basenames, dirname)
def	_findDuplicates (lst)
def	_getGlobalTag (sample, release)
def	_getRelValUrl (release)
def	_processPlotsForSample (plotterFolder, sample)
def	_stripRelease (release)

Variables
	_doConversionSamples
	_doElectronSamples
	_globalTags
	_releasePostfixes
	_relvalUrls
	IgnoreCommandLineOptions

Function Documentation

def validation._copyDir (   src,   dst  )

private

Copy non-TTree objects from src TDirectory to dst TDirectory.

Definition at line 959 of file validation.py.

Referenced by _copySubDir().

def _copyDir(src, dst):
    """Copy non-TTree objects from src TDirectory to dst TDirectory."""
    keys = src.GetListOfKeys()
    for key in keys:
        classname = key.GetClassName()
        cl = ROOT.TClass.GetClass(classname)
        if not cl:
            continue
        if not (cl.InheritsFrom("TTree") and cl.InheritsFrom("TDirectory")):
            dst.cd()
            obj = key.ReadObj()
            obj.Write()
            obj.Delete()

def validation._copySubDir (   oldfile,   newfile,   basenames,   dirname  )

private

Copy a subdirectory from oldfile to newfile.

Arguments:
oldfile   -- String for source TFile
newfile   -- String for destination TFile
basenames -- List of strings for base directories, first existing one is picked
dirname   -- String for directory name under the base directory

Definition at line 924 of file validation.py.

References _copyDir(), join(), and split.

Referenced by validation.Validation._doPlots().

def _copySubDir(oldfile, newfile, basenames, dirname):
    """Copy a subdirectory from oldfile to newfile.

    Arguments:
    oldfile   -- String for source TFile
    newfile   -- String for destination TFile
    basenames -- List of strings for base directories, first existing one is picked
    dirname   -- String for directory name under the base directory
    """
    oldf = ROOT.TFile.Open(oldfile)

    dirold = None
    for basename in basenames:
        dirold = oldf.GetDirectory(basename)
        if dirold:
            break
    if not dirold:
        raise Exception("Did not find any of %s directories from file %s" % (",".join(basenames), oldfile))
    if dirname:
        d = dirold.Get(dirname)
        if not d:
            raise Exception("Did not find directory %s under %s" % (dirname, dirold.GetPath()))
        dirold = d

    newf = ROOT.TFile.Open(newfile, "RECREATE")
    dirnew = newf
    for d in basenames[0].split("/"):
        dirnew = dirnew.mkdir(d)
    if dirname:
        dirnew = dirnew.mkdir(dirname)
    _copyDir(dirold, dirnew)

    oldf.Close()
    return newf

def validation._findDuplicates (   lst )

private

Definition at line 973 of file validation.py.

References list().

Referenced by validation.Validation._doPlots(), validation.SimpleValidation._doPlots(), validation.Validation._doPlotsFastFull(), and validation.Validation._doPlotsPileup().

def _findDuplicates(lst):
    found = set()
    found2 = set()
    for x in lst:
        if x in found:
            found2.add(x)
        else:
            found.add(x)
    return list(found2)

def validation._getGlobalTag (   sample,   release  )

private

Get a GlobalTag.

Arguments:
sample  -- Sample object
release -- CMSSW release string

Definition at line 197 of file validation.py.

Referenced by validation.Validation._doPlots(), validation.Validation._doPlotsFastFull(), validation.Validation._doPlotsPileup(), validation.Validation._getRefFileAndSelection(), validation.Sample.datasetpattern(), and validation.Sample.filename().

def _getGlobalTag(sample, release):
    """Get a GlobalTag.

    Arguments:
    sample  -- Sample object
    release -- CMSSW release string
    """
    if not release in _globalTags:
        print "Release %s not found from globaltag map in validation.py" % release
        sys.exit(1)
    gtmap = _globalTags[release]
    if sample.hasOverrideGlobalTag():
        ogt = sample.overrideGlobalTag()
        if release in ogt:
            gtmap = _globalTags[ogt[release]]
    if sample.fullsim():
        if sample.hasScenario():
            return gtmap[sample.scenario()]
        if sample.pileupEnabled():
            puType = sample.pileupType()
            if "50ns" in puType:
                return gtmap.get("fullsim_50ns", gtmap["default"])
            if "25ns" in puType:
                return gtmap.get("fullsim_25ns", gtmap["default"])
    if sample.fastsim():
        fsgt = gtmap.get("fastsim", gtmap["default"])
        if sample.pileupEnabled():
            puType = sample.pileupType()
            if "25ns" in puType:
                return gtmap.get("fastsim_25ns", fsgt)
        return fsgt
    return gtmap["default"]

# Mapping from release series to RelVal download URLs

def validation._getRelValUrl (   release )

private

Get RelVal download URL for a given release.

Definition at line 254 of file validation.py.

Referenced by validation.Validation.download().

def _getRelValUrl(release):
    """Get RelVal download URL for a given release."""
    version_re = re.compile("CMSSW_(?P<X>\d+)_(?P<Y>\d+)")
    m = version_re.search(release)
    if not m:
        raise Exception("Regex %s does not match to release version %s" % (version_re.pattern, release))
    version = "%s_%s_X" % (m.group("X"), m.group("Y"))
    if not version in _relvalUrls:
        print "No RelVal URL for version %s, please update _relvalUrls" % version
        sys.exit(1)
    return _relvalUrls[version]

def validation._processPlotsForSample (   plotterFolder,   sample  )

private

Definition at line 266 of file validation.py.

Referenced by validation.Validation._doFastsimFastVsFullPlots(), validation.Validation._doPhase2PileupPlots(), and validation.SimpleValidation._doPlotsForPlotter().

def _processPlotsForSample(plotterFolder, sample):
    if plotterFolder.onlyForPileup() and not sample.pileupEnabled():
        return False
    if plotterFolder.onlyForElectron() and not sample.doElectron():
        return False
    if plotterFolder.onlyForConversion() and not sample.doConversion():
        return False
    return True

def validation._stripRelease (   release )

private

Definition at line 190 of file validation.py.

Referenced by validation.Validation._doPlots(), validation.Validation._doPlotsFastFull(), validation.Validation._doPlotsPileup(), and validation.Sample.filename().

def _stripRelease(release):
    for pf in _releasePostfixes:
        if pf in release:
            return release.replace(pf, "")
    return release

Variable Documentation

validation._doConversionSamples

private

Definition at line 249 of file validation.py.

validation._doElectronSamples

private

Definition at line 244 of file validation.py.

validation._globalTags

private

Definition at line 15 of file validation.py.

validation._releasePostfixes

private

Definition at line 188 of file validation.py.

validation._relvalUrls

private

Definition at line 231 of file validation.py.

validation.IgnoreCommandLineOptions

Definition at line 9 of file validation.py.