helperFunctions Namespace Reference

Functions
def	addIndex (filename, njobs, index=None)
def	boolfromstring (string, name)
def	cache (function)
def	castorDirExists (path)
def	clean_name (s)
def	cppboolstring (string, name)
def	getCommandOutput2 (command)
def	getTagsMap (db)
def	parsecolor (color)
def	parsestyle (style)
def	pythonboolstring (string, name)
def	recursivesubclasses (cls)
def	replaceByMap (target, the_map)
	— Helpers —############################ More...
def	replacelast (string, old, new, count=1)

Variables
list	fileExtensions = ["_cfg.py", ".sh", ".root"]

Function Documentation

addIndex()

def helperFunctions.addIndex	(		filename,
			njobs,
			index = None
	)

Definition at line 93 of file helperFunctions.py.

References FastTimerService_cff.range, replacelast(), and str.

Referenced by genericValidation.GenericValidation.createFiles(), genericValidation.GenericValidationData.createScript(), and genericValidation.GenericValidationData.getRepMap().

def addIndex(filename, njobs, index = None):
    if index is None:
        return [addIndex(filename, njobs, i) for i in range(njobs)]
    if njobs == 1:
        return filename

    fileExtension = None
    for extension in fileExtensions:
        if filename.endswith(extension):
            fileExtension = extension
    if fileExtension is None:
        raise AllInOneError(fileName + " does not end with any of the extensions "
                                     + str(fileExtensions))
    return replacelast(filename, fileExtension, "_" + str(index) + fileExtension)

boolfromstring()

def helperFunctions.boolfromstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool

Definition at line 169 of file helperFunctions.py.

References nano_mu_local_reco_cff.bool, createfilelist.int, and str.

Referenced by pythonboolstring().

def boolfromstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool
    """
    #try as a string, not case sensitive
    if string.lower() == "true": return True
    if string.lower() == "false": return False
    #try as a number
    try:
        return str(bool(int(string)))
    except ValueError:
        pass
    #out of options
    raise ValueError("{} has to be true or false!".format(name))
    

cache()

def helperFunctions.cache

(

function

)

Definition at line 155 of file helperFunctions.py.

References HiBiasedCentrality_cfi.function, print(), and mkLumiAveragedPlots.tuple.

def cache(function):
    cache = {}
    def newfunction(*args, **kwargs):
        try:
            return cache[args, tuple(sorted(kwargs.items()))]
        except TypeError:
            print(args, tuple(sorted(kwargs.items())))
            raise
        except KeyError:
            cache[args, tuple(sorted(kwargs.items()))] = function(*args, **kwargs)
            return newfunction(*args, **kwargs)
    newfunction.__name__ = function.__name__
    return newfunction

castorDirExists()

def helperFunctions.castorDirExists

(

path

)

This function checks if the directory given by `path` exists.

Arguments:
- `path`: Path to castor directory

Definition at line 66 of file helperFunctions.py.

References getCommandOutput2().

def castorDirExists(path):
    """This function checks if the directory given by `path` exists.

    Arguments:
    - `path`: Path to castor directory
    """

    if path[-1] == "/":
        path = path[:-1]
    containingPath = os.path.join( *path.split("/")[:-1] )
    dirInQuestion = path.split("/")[-1]
    try:
        rawLines = getCommandOutput2("rfdir /"+containingPath).splitlines()
    except RuntimeError:
        return False
    for line in rawLines:
        if line.split()[0][0] == "d":
            if line.split()[8] == dirInQuestion:
                return True
    return False

clean_name()

def helperFunctions.clean_name

(

s

)

Transforms a string into a valid variable or method name.

Arguments:
- `s`: input string

Definition at line 212 of file helperFunctions.py.

Referenced by geometryComparison.GeometryComparison.getRepMap().

def clean_name(s):
    """Transforms a string into a valid variable or method name.

    Arguments:
    - `s`: input string
    """

    # Remove invalid characters
    s = re.sub(r"[^0-9a-zA-Z_]", "", s)

    # Remove leading characters until we find a letter or underscore
    s = re.sub(r"^[^a-zA-Z_]+", "", s)

    return s

cppboolstring()

def helperFunctions.cppboolstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool string for a C++ template

Definition at line 193 of file helperFunctions.py.

References pythonboolstring().

Referenced by plottingOptions.PlottingOptionsZMuMu.__init__(), plottingOptions.PlottingOptionsOffline.__init__(), and plottingOptions.PlottingOptionsPrimaryVertex.__init__().

def cppboolstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool string for a C++ template
    """
    return pythonboolstring(string, name).lower()

getCommandOutput2()

def helperFunctions.getCommandOutput2

(

command

)

This function executes `command` and returns it output.

Arguments:
- `command`: Shell command to be invoked by this function.

Definition at line 51 of file helperFunctions.py.

Referenced by castorDirExists(), and geometryComparison.GeometryComparison.createScript().

def getCommandOutput2(command):
    """This function executes `command` and returns it output.

    Arguments:
    - `command`: Shell command to be invoked by this function.
    """

    child = os.popen(command)
    data = child.read()
    err = child.close()
    if err:
        raise RuntimeError('%s failed w/ exit code %d' % (command, err))
    return data

getTagsMap()

def helperFunctions.getTagsMap

(

db

)

Definition at line 200 of file helperFunctions.py.

References python.cmstools.all(), conddblib.connect(), conddblib.make_url(), and FastTimerService_cff.range.

Referenced by alignment.Alignment.__getConditions(), and alignment.Alignment.__testDbExist().

def getTagsMap(db):
    con = conddblib.connect(url = conddblib.make_url(db))
    session = con.session()
    TAG = session.get_dbtype(conddblib.Tag)
    dictionary = {}
    for i in range(0,len(session.query(TAG.object_type).order_by(TAG.name).all())):
        q1 = session.query(TAG.object_type).order_by(TAG.name).all()[i][0]
        q2 = session.query(TAG.name).order_by(TAG.name).all()[i][0]
        dictionary[q1]=q2

    return dictionary

parsecolor()

def helperFunctions.parsecolor

(

color

)

Definition at line 108 of file helperFunctions.py.

References createfilelist.int, and str.

Referenced by preexistingValidation.PreexistingValidation.getRepMap().

def parsecolor(color):
    try: #simplest case: it's an int
        return int(color)
    except ValueError:
        pass

    try:   #kRed, kBlue, ...
        color = str(getattr(ROOT, color))
        return int(color)
    except (AttributeError, ValueError):
        pass

    if color.count("+") + color.count("-") == 1:  #kRed+5, kGreen-2
        if "+" in color:                          #don't want to deal with nonassociativity of -
            split = color.split("+")
            color1 = parsecolor(split[0])
            color2 = parsecolor(split[1])
            return color1 + color2

        if "-" in color:
            split = color.split("-")
            color1 = parsecolor(split[0])
            color2 = parsecolor(split[1])
            return color1 - color2

    raise AllInOneError("color has to be an integer, a ROOT constant (kRed, kBlue, ...), or a two-term sum or difference (kGreen-5)!")

parsestyle()

def helperFunctions.parsestyle

(

style

)

Definition at line 135 of file helperFunctions.py.

References createfilelist.int, and str.

Referenced by preexistingValidation.PreexistingValidation.getRepMap().

def parsestyle(style):
    try: #simplest case: it's an int
        return int(style)
    except ValueError:
        pass

    try: #kStar, kDot, ...
        style = str(getattr(ROOT,style))
        return int(style)
    except (AttributeError, ValueError):
        pass

    raise AllInOneError("style has to be an integer or a ROOT constant (kDashed, kStar, ...)!")

pythonboolstring()

def helperFunctions.pythonboolstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool string for a python template

Definition at line 186 of file helperFunctions.py.

References boolfromstring(), and str.

Referenced by primaryVertexValidation.PrimaryVertexValidation.__init__(), offlineValidation.OfflineValidation.__init__(), cppboolstring(), and geometryComparison.GeometryComparison.createConfiguration().

def pythonboolstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool string for a python template
    """
    return str(boolfromstring(string, name))

recursivesubclasses()

def helperFunctions.recursivesubclasses

(

cls

)

Definition at line 149 of file helperFunctions.py.

Referenced by presentation.ValidationPlots.validationclass().

def recursivesubclasses(cls):
    result = [cls]
    for subcls in cls.__subclasses__():
        result += recursivesubclasses(subcls)
    return result

replaceByMap()

def helperFunctions.replaceByMap	(		target,
			the_map
	)

— Helpers —############################

This function replaces `.oO[key]Oo.` by `the_map[key]` in target.

Arguments:
- `target`: String which contains symbolic tags of the form `.oO[key]Oo.`
- `the_map`: Dictionary which has to contain the `key`s in `target` as keys

Definition at line 12 of file helperFunctions.py.

References mps_monitormerge.items, and str.

Referenced by zMuMuValidation.ZMuMuValidation.appendToPlots(), geometryComparison.GeometryComparison.createConfiguration(), genericValidation.GenericValidation.createFiles(), genericValidation.ValidationWithPlots.createPlottingScript(), geometryComparison.GeometryComparison.createScript(), genericValidation.ValidationWithComparison.doComparison(), genericValidation.ParallelValidation.doInitMerge(), genericValidation.ParallelValidation.doMerge(), genericValidation.ValidationWithPlots.doRunPlots(), zMuMuValidation.ZMuMuValidation.filesToCompare(), alignment.Alignment.getConditions(), genericValidation.GenericValidationData.getRepMap(), offlineValidation.OfflineValidation.initMerge(), and plottingOptions.PlottingOptionsTrackSplitting.validsubdets().

def replaceByMap(target, the_map):
    """This function replaces `.oO[key]Oo.` by `the_map[key]` in target.

    Arguments:
    - `target`: String which contains symbolic tags of the form `.oO[key]Oo.`
    - `the_map`: Dictionary which has to contain the `key`s in `target` as keys
    """

    result = target
    for key in the_map:
        lifeSaver = 10e3
        iteration = 0
        while ".oO[" in result and "]Oo." in result:
            for key in the_map:
                try:
                    result = result.replace(".oO["+key+"]Oo.",the_map[key])
                except TypeError:   #try a dict
                    try:
                        for keykey, value in the_map[key].items():
                           result = result.replace(".oO[" + key + "['" + keykey + "']]Oo.", value)
                           result = result.replace(".oO[" + key + '["' + keykey + '"]]Oo.', value)
                    except AttributeError:   #try a list
                        try:
                            for index, value in enumerate(the_map[key]):
                                result = result.replace(".oO[" + key + "[" + str(index) + "]]Oo.", value)
                        except TypeError:
                            raise TypeError("Something is wrong in replaceByMap!  Need a string, dict, or list, but the_map(%s)=%s!"%(repr(key), repr(the_map[key])))
                iteration += 1
            if iteration > lifeSaver:
                problematicLines = ""
                for line in result.splitlines():
                    if  ".oO[" in result and "]Oo." in line:
                        problematicLines += "%s\n"%line
                msg = ("Oh Dear, there seems to be an endless loop in "
                       "replaceByMap!!\n%s\n%s"%(problematicLines, the_map))
                raise AllInOneError(msg)
    return result

replacelast()

def helperFunctions.replacelast	(		string,
			old,
			new,
			count = 1
	)

Replace the last occurances of a string

Definition at line 87 of file helperFunctions.py.

Referenced by addIndex().

def replacelast(string, old, new, count = 1):
    """Replace the last occurances of a string"""
    return new.join(string.rsplit(old,count))

Variable Documentation

fileExtensions

list helperFunctions.fileExtensions = ["_cfg.py", ".sh", ".root"]

Definition at line 91 of file helperFunctions.py.

---

Generated for CMSSW Reference Manual by  1.8.14