QuickTriggerAnalysis/makeFriend.py at master · jmontejo/QuickTriggerAnalysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/env python
import sys,re, array
import ROOT

weights = {}
events = {}
#If you use a new EB run, you need to change this file! They're all located in
# the folder  /afs/cern.ch/user/a/attradm/public/
with open('/afs/cern.ch/user/a/attradm/public/EnhancedBiasWeights_360026.xml') as fp:
    for line in fp:
        #if "8732555" in line :
        #    print "the line is read "
        if "<weight " in line :
            sid = line.split(" ")[1]
            svalue = line.split(" ")[2]
            idd =  re.findall("\d+", sid)
#            value =  re.findall(r"[-+]?\d*\.\d+|\d+", svalue)
            value = re.findall("[-+]?\d+[\.]?\d*[eE]?[-+]?\d*", svalue )
            print idd[0], svalue, value[0]
            weights[idd[0]] =  value[0]
            #if idd[0] < minId :
            #    minId = idd[0]

        if "<e " in line :
            sid = line.split(" ")[1]
            svalue = line.split(" ")[2]
            iv =  int(re.findall("\-?\d+", sid)[0])
            if(iv & 0x80000000):
                ev = -0x100000000 + iv
            else :
                ev = iv
            value =  re.findall("\d+", svalue)[0]
            events[ev] =  float(weights[value])
            if "-1680982068" in line :
                print line, value, ev
            #if "8732555" in line : print line, value, ev
            #if ev == 8732555 :
            #    print ev,  value, weights[value],
            #    print "and ->  ",events[8732555]

print "Created events of length ", len(events)
fileName = sys.argv[1]

fTrig=ROOT.TFile(fileName)
tTrig=fTrig.Get("trig")
tTrig.SetBranchStatus("*",0)
tTrig.SetBranchStatus("eventNumber",1)

fOut =ROOT.TFile(fileName[:-5]+".EBweights.root", "UPDATE")
fOut.cd()
tOut = ROOT.TTree("trig","trig")
#tOut = tTrig.CloneTree()

EBweight = array.array("f",(0 for i in range(0,1)))
tOut.Branch("EBweight",EBweight ,"EBweight/F")


for entry in xrange( tTrig.GetEntries() ):
    if entry%10000 == 1 :
        print entry , " out of ",  tTrig.GetEntries()
    tTrig.GetEntry( entry )
    #if tTrig.eventNumber not in events.keys() :
    #    EBweight[0] = 0
    #    print "Can not find event ", tTrig.eventNumber
    #else :
    iv = tTrig.eventNumber
    if(iv & 0x80000000):
        iv = -0x100000000 + iv
    #print iv
    EBweight[0] = events[iv]
    tOut.Fill()

print "Write out EB ntuple ", fOut.GetName() , " with ", tOut.GetEntries(), " events"
fOut.Write()
fOut.Close()
exit(0)