lumiregperbunch Namespace Reference

Classes
class	bsmeas

Functions
def	readroot ()
def	timeof (run, lumisection)

Variables
	allmeas
	bxmeas
	file
	filename
	FILL
	line
	lines
	lumimax
	lumimid
	lumimin
	lumirange
	meastime
	OUTDIR
	rfbucket
	runlstime
	runno
	sortedtimes
	thismeas
	WORKDIR

Function Documentation

readroot()

def lumiregperbunch.readroot

(

)

Definition at line 107 of file lumiregperbunch.py.

def readroot():
    rls=[]
    bxlist=[]
    allmeas={}
    
    DIRES=['X0','Y0','Z0','width_X0','Width_Y0','Sigma_Z0','dxdz','dydz']
    # DIRES=['X0']
    rootfile="BxAnalysis_Fill_"+FILL+".root"
    filein=TFile(rootfile)
    for dire in DIRES:
        filein.cd(dire)
        # get list of histograms
        histolist=gDirectory.GetListOfKeys()
        iter = histolist.MakeIterator()
        key = iter.Next()
        while key:
            if key.GetClassName() == 'TH1F':
                td = key.ReadObj()
                histoname = td.GetName()
                if "bx" in histoname:
#                    print histoname
                    bx=histoname.split('_')[-1]
                    if bx not in bxlist:
# this is to be removed                        
#                        if len(bxlist)>=2:
#                            key = iter.Next()
#                            continue
# end to be removed                        
                        bxlist.append(bx)
                        allmeas[bx]={}
#                    print bx,histoname
                    histo=gDirectory.Get(histoname)
                    nbin=histo.GetNbinsX()
 
                    thisbx=allmeas[bx]
                    
                    for bin in range(1,nbin+1):
                        label=histo.GetXaxis().GetBinLabel(bin)
                        label=label.strip()
                        if ":" not in label:
                            # not a valid label of type run:lumi-lumi, skip it
                            continue
                        
                        cont=histo.GetBinContent(bin)
                        if cont!=cont:
                            # it's a nan
                            cont=-999.0
                        err=histo.GetBinError(bin)
                        if err!=err:
                            err=-999.0
                        #                        if len(bxlist)==1:
                        #                            rls.append(label)
                        #                            print label
                        #                        else:
                        if label not in rls:
                            print("New range:",label," found in ",histoname)
                            rls.append(label)
                                
                        if label in thisbx.keys():
                            thismeas=thisbx[label]
                        else:
                            thisbx[label]=bsmeas()
                            thismeas=thisbx[label]
                        #  now filling up
                        if dire=='X0':
                            thismeas.x=cont
                            thismeas.ex=err
                        if dire=='Y0':
                            thismeas.y=cont
                            thismeas.ey=cont
                        if dire=='Z0':
                            thismeas.z=cont
                            thismeas.ez=err
                        if dire=='width_X0':
                            thismeas.wx=cont
                            thismeas.ewx=err
                        if dire=='Width_Y0':
                            thismeas.wy=cont
                            thismeas.ewy=err
                        if dire=='Sigma_Z0':
                            thismeas.wz=cont
                            thismeas.ewz=err
                        if dire=='dxdz':
                            thismeas.dxdz=cont
                            thismeas.edxdz=err
                        if dire=='dydz':
                            thismeas.dydz=cont
                            thismeas.edydz=err
 
    
            key = iter.Next()
        
        
    #    for name in pippo:
    #        print name
    
    filein.Close()
 
    # let's try to show it
#    for bx in allmeas.keys():
#        print "bx=",bx
#        bxmeas=allmeas[bx]
#        for meas in bxmeas.keys():
#            print "meas time=",meas
#            thismeas=bxmeas[meas]
#            print thismeas.x,thismeas.ex,thismeas.y,thismeas.ey,thismeas.z,thismeas.ez
#            print thismeas.wx,thismeas.ewx,thismeas.wy,thismeas.ewy,thismeas.wz,thismeas.ewz
#            print thismeas.dxdz,thismeas.edxdz,thismeas.dydz,thismeas.edydz
    return allmeas
 

References print(), and FastTimerService_cff.range.

timeof()

def lumiregperbunch.timeof	(		run,
			lumisection
	)

Definition at line 41 of file lumiregperbunch.py.

def timeof(run,lumisection):
    # first check if this run is already in the list, otherwise read it
    if run not in runlstime.keys():
        print("Reading lumi time from lumireg localcopy files")
        filename="localcopy/BeamFitResults_Run"+run+".txt"
        if not os.path.exists(filename):
            print("WARNING: file ",filename," does not exist. Returning null.")
            return -1
 
        # reading file
        lstime={}
        in_file = open(filename)
        pieces = paragraphs(in_file,start_of_new_beamspot_measurement,True)
        for piece in pieces:
            if len(piece) < 1:
                continue
            try:
                tmp = BeamspotMeasurement(piece)
            except Exception as err:
                print("    ERROR Found corrupt " \
                      "beamspot measurement entry!", file=sys.stderr)
                print("    !!! %s !!!" % str(err), file=sys.stderr)
                continue
            # Argh!
            runfromfile=tmp.run_number
            (lumimin,lumimax)=tmp.lumi_range
            time_begin=tmp.time_begin
            time_end=tmp.time_end
            time_begin=calendar.timegm(time_begin.timetuple())
            time_end=calendar.timegm(time_end.timetuple())-23 # assume end of lumisection
            lstime[lumimin]=time_begin
            lstime[lumimax]=time_end
            
        # order lumisections and make a list
        lslist=sorted(lstime.keys())
        lstimesorted=[]
        for ls in lslist:
            lstimesorted.append((ls,lstime[ls]))
        runlstime[run]=lstimesorted
 
        #            print runfromfile
        #            print lumirange
        #            print time_begin, calendar.timegm(time_begin.timetuple())
        #            print time_end, calendar.timegm(time_end.timetuple())
        
        in_file.close()
    # now give a time
    dcloselumi=999999
    closelumi=-1
    closetime=-1
    lstimesorted=runlstime[run]
    
    for pair in lstimesorted:
        (lumi,time)=pair
        if abs(lumisection-lumi)<dcloselumi:
            dcloselumi=abs(lumisection-lumi)
            closelumi=lumi
            closetime=time
    if closelumi!=-1:
        finaltime=closetime+(lumisection-closelumi)*23
    else:
        finaltime=-1
        
    return finaltime
 
 

References funct.abs(), print(), and str.

Variable Documentation

allmeas

lumiregperbunch.allmeas

Definition at line 223 of file lumiregperbunch.py.

bxmeas

lumiregperbunch.bxmeas

Definition at line 228 of file lumiregperbunch.py.

file

lumiregperbunch.file

Definition at line 270 of file lumiregperbunch.py.

filename

lumiregperbunch.filename

Definition at line 269 of file lumiregperbunch.py.

FILL

lumiregperbunch.FILL

Definition at line 11 of file lumiregperbunch.py.

line

lumiregperbunch.line

Definition at line 247 of file lumiregperbunch.py.

lines

lumiregperbunch.lines

Definition at line 229 of file lumiregperbunch.py.

lumimax

lumiregperbunch.lumimax

Definition at line 237 of file lumiregperbunch.py.

lumimid

lumiregperbunch.lumimid

Definition at line 239 of file lumiregperbunch.py.

lumimin

lumiregperbunch.lumimin

Definition at line 235 of file lumiregperbunch.py.

lumirange

lumiregperbunch.lumirange

Definition at line 234 of file lumiregperbunch.py.

meastime

lumiregperbunch.meastime

Definition at line 240 of file lumiregperbunch.py.

OUTDIR

lumiregperbunch.OUTDIR

Definition at line 12 of file lumiregperbunch.py.

rfbucket

lumiregperbunch.rfbucket

Definition at line 268 of file lumiregperbunch.py.

runlstime

lumiregperbunch.runlstime

Definition at line 14 of file lumiregperbunch.py.

runno

lumiregperbunch.runno

Definition at line 232 of file lumiregperbunch.py.

sortedtimes

lumiregperbunch.sortedtimes

Definition at line 271 of file lumiregperbunch.py.

thismeas

lumiregperbunch.thismeas

Definition at line 243 of file lumiregperbunch.py.

WORKDIR

lumiregperbunch.WORKDIR

Definition at line 266 of file lumiregperbunch.py.