15def ReadPileupHistograms(dirName):
16 if not dirName.endswith("/"):
17 dirName = dirName + "/"
18
19
20
21 rootFileList = sorted(glob.glob(dirName+"*.root"))
22 if len(rootFileList) != 1:
23 print "Directory contains multiple or no root files - was expecting one:",dirName
24 return None
25 rootFile = TFile(rootFileList[0],"READ")
26
27 histos = {}
28 for aJetDef in jetDefs:
29 histos[aJetDef] = {}
30
31
32 for aSystName in SystematicNames:
33 systematicName = aSystName + "_" + aJetDef
34 histo = rootFile.Get(systematicName)
35 if histo is None:
36 print "Failed to get histogram:",systematicName
37 return None
38 histo.SetDirectory(0)
39
40 histos[aJetDef][aSystName] = histo
41
42
43 rootFile.Close()
44
45 return histos
46
47
48