generateEDF Namespace Reference

Classes
class	LumiInfo
	LumiInfo Class More...
class	LumiInfoCont
	LumiInfoCont Class More...

Functions
def	loadEvents
	General Functions More...
def	makeEDFplot

Variables
string	action = 'store_true'
list	allowedEDF = ['time', 'instLum', 'instIntLum']
	## Main More...
tuple	cont = LumiInfoCont(args[0], **options.__dict__)
	load Luminosity info ## More...
string	default = 'Empirical Distribution Function'
	done = False
string	help = 'title of plot (default %default)'
tuple	inputGroup = optparse.OptionGroup(parser, "Input Options")
tuple	modeGroup = optparse.OptionGroup(parser, "Mode Options")
tuple	nonSpaceRE = re.compile(r'\S')
tuple	parser = optparse.OptionParser("Usage: %prog [options] lumi.csv events.txt output.png", description='Script for generating EDF curves. See https://twiki.cern.ch/twiki/bin/viewauth/CMS/SWGuideGenerateEDF for more details.')
tuple	pieces = sepRE.split(line)
tuple	plotGroup = optparse.OptionGroup(parser, "Plot Options")
int	prevRecLumi = 0
tuple	rangeGroup = optparse.OptionGroup(parser, "Range Options")
int	recLumIndex = 0
list	recLumis = []
	look for which runs correspond to what total ## recorded integrated luminosity ## More...
tuple	recLumValue = float(piece)
tuple	sepRE = re.compile(r'[\s,;:]+')
string	type = 'string'

Function Documentation

def generateEDF.loadEvents	(		filename,
			cont,
			options
	)

General Functions

Definition at line 240 of file generateEDF.py.

References bitset_utilities.append(), and print().

def loadEvents (filename, cont, options):
    eventsDict = {}
    print("loading events from '%s'" % filename)
    events = open (filename, 'r')
    runIndex, lumiIndex, eventIndex, weightIndex = 0, 1, 2, 3
    if options.relOrder:
        lumiIndex, eventIndex = 2,1
    minPieces = 3
    totalWeight = 0.
    if options.weights:
        minPieces = 4
    for line in events:
        pieces = sepRE.split (line.strip())
        if len (pieces) < minPieces:
            if nonSpaceRE.search (line):
                print("skipping", line)
            continue
        try:
            run, lumi, event = int( pieces[runIndex]   ), \
                               int( pieces[lumiIndex]  ), \
                               int( pieces[eventIndex] )
        except:
            continue
        key = (run, lumi)
        if key not in cont:
            if options.ignore:
                print("Warning, %s is not found in the lumi information" \
                      % key.__str__())
                continue
            else:
                raise RuntimeError("%s is not found in lumi information.  Use '--ignoreNoLumiEvents' option to ignore these events and continue." \
                      % key.__str__())
        if options.edfMode != 'time' and not cont[key].xingInfo:
            if options.ignore:
                print("Warning, %s does not have Xing information" \
                      % key.__str__())
                continue
            else:
                raise RuntimeError("%s  does not have Xing information.  Use '--ignoreNoLumiEvents' option to ignore these events and continue." \
                      % key.__str__())            
        if options.weights:
            weight = float (pieces[weightIndex])
        else:
            weight = 1
        eventsDict.setdefault( key, []).append( (event, weight) )
        totalWeight += weight
    events.close()
    return eventsDict, totalWeight