| 1 |
|
# |
| 2 |
< |
# |
| 3 |
< |
# Written in python because ROOT's c-interpreter (CINT) |
| 4 |
< |
# chokes and dies on code that tries to do this. |
| 2 |
> |
# Run by typing: |
| 3 |
> |
# python plotFromMultipleFiles.py |
| 4 |
|
# |
| 5 |
|
# Creates histograms from TTree in several files |
| 6 |
|
# that have different scales. |
| 7 |
|
# |
| 8 |
+ |
# Written in python for its power, and also because |
| 9 |
+ |
# ROOT's C-interpreter (CINT) chokes and dies on |
| 10 |
+ |
# code that tries to do things here. |
| 11 |
+ |
# |
| 12 |
|
# Michael Anderson |
| 13 |
|
# Nov 4, 2009 |
| 14 |
|
|
| 15 |
< |
from ROOT import TFile, TH1F, TTree |
| 15 |
> |
from ROOT import TFile, TH1F, TH2F, TH3F, TTree # Import any ROOT class you want |
| 16 |
|
from math import pi |
| 17 |
< |
from array import array # used to make Float_t's that ROOT wants |
| 18 |
< |
from datetime import datetime # used in output filename |
| 19 |
< |
import sys # used for exiting program |
| 17 |
> |
from array import array # used to make Float_t array ROOT wants |
| 18 |
> |
from datetime import datetime # used in output filename |
| 19 |
> |
import sys # used for exiting program |
| 20 |
|
|
| 21 |
|
######################################## |
| 22 |
|
# Class for root files to store |
| 23 |
|
# file's name, scale, cuts, etc... |
| 24 |
|
class rootFile: |
| 25 |
< |
def __init__(self, fileName, theScale, theCuts): |
| 25 |
> |
def __init__(self, fileName, scale=1.0, cuts=""): |
| 26 |
|
self.name = fileName |
| 27 |
< |
self.scale = theScale |
| 28 |
< |
self.cuts = theCuts |
| 29 |
< |
self.file = TFile(fileName, "read") # Open TFile |
| 27 |
> |
self.scale = scale |
| 28 |
> |
self.cuts = cuts |
| 29 |
> |
self.file = TFile(fileName, "read") # Open TFile |
| 30 |
|
if self.file.IsZombie(): |
| 31 |
|
print "Error opening %s, exiting..." % self.name |
| 32 |
|
sys.exit(0) |
| 33 |
< |
self.ttree = TTree() |
| 34 |
< |
self.file.GetObject(ttreeName, self.ttree) # ttreeName is set in variables below |
| 33 |
> |
print "Opened %s, scale=%.2e, cuts='%s'" % (fileName, scale, cuts) |
| 34 |
> |
self.ttree = TTree() # Create empty TTree, and |
| 35 |
> |
try: # try to get TTree from file. |
| 36 |
> |
self.file.GetObject(ttreeName, self.ttree) # ttreeName set in variables below |
| 37 |
> |
except: |
| 38 |
> |
print "Error: %s not found in %s, exiting..." % (ttreeName, fileName) |
| 39 |
> |
sys.exit(0) |
| 40 |
|
######################################## |
| 41 |
|
|
| 42 |
|
|
| 43 |
|
######################################## |
| 44 |
|
# Variables |
| 45 |
< |
invLum = 50.0 # Set inverse lum to scale plots to |
| 45 |
> |
invLum = 50.0 # Set inverse lum to scale plots to |
| 46 |
|
|
| 47 |
|
ttreeName = "TreePhotonJet" # TTree name in all files |
| 48 |
|
|
| 51 |
|
rootFile("QCD_Pt30_Summer09.root",invLum*(1.091E7-1.93E6)/1.779232e+06, "event_genEventScale>30&&event_genEventScale<80"), |
| 52 |
|
rootFile("QCD_Pt80_Summer09.root",invLum*(1.93E6-6.2E4)/2.14780e+06, "event_genEventScale>80&&event_genEventScale<170")] |
| 53 |
|
|
| 54 |
< |
# Cut applied to all files |
| 54 |
> |
# Cut(s) applied to all files |
| 55 |
|
cutForAllFiles = "photon_et>15.0&&abs(photon_eta)<2.5" |
| 56 |
|
|
| 57 |
+ |
outputFilename = "QCD_combined_%s.root" % datetime.now().strftime("%Y_%m_%d_%H_%M") |
| 58 |
+ |
|
| 59 |
|
# Histogram bins |
| 60 |
|
# Converting with "array" is done because ROOT wants an array of Float_t |
| 61 |
|
bins_et = array('f', [15.0, 20.0, 27.0, 35.0, 45.0, 57.0, 72.0, 90.0, 120.0, 150.0, 200.0, 300.0, 400.0, 550.0]) |
| 62 |
|
bins_eta = array('f', [-2.5, -1.55, -1.45, -0.9, 0.0, 0.9, 1.45, 1.55, 2.5]) |
| 63 |
+ |
bins_fisher = array('f', [i/20.0 for i in range(-40,21)]) # runs from -2.0 to 1.0 in steps of 0.05 |
| 64 |
|
|
| 65 |
< |
listOfPlots = {'photon_et' : TH1F("photonEt", "Photon E_{T} ;E_{T} (GeV);entries/15 GeV bin", len(bins_et)-1 , bins_et ), |
| 66 |
< |
'photon_eta': TH1F("photonEta", "Photon #eta ;#eta;entries/0.1 bin" , len(bins_eta)-1, bins_eta ), |
| 67 |
< |
'photon_phi': TH1F("photonPhi", "Photon #phi ;#phi;entries/bin" , 62, (-1.-1./30.)*pi, (1.+1./30.)*pi)} |
| 68 |
< |
|
| 69 |
< |
outputFilename = "QCD_combined_%s.root" % datetime.now().strftime("%Y_%m_%d_%H_%M") |
| 65 |
> |
# Variable name (in TTree) : Histogram to plot into |
| 66 |
> |
listOfPlots = {'photon_et' : TH1F("photonEt", "Photon E_{T} ;E_{T} (GeV);entries/bin" , len(bins_et)-1 , bins_et ), |
| 67 |
> |
'photon_eta': TH1F("photonEta", "Photon #eta ;#eta;entries/bin" , len(bins_eta)-1, bins_eta ), |
| 68 |
> |
'photon_phi': TH1F("photonPhi", "Photon #phi ;#phi;entries/bin" , 32, (-1.-1./15.)*pi, (1.+1./15.)*pi), |
| 69 |
> |
'fishValue' : TH1F("photonFisherValue","Photon Fisher Value;fisher Value", 40, -2.0, 1.0), |
| 70 |
> |
'fishValue:abs(photon_eta):photon_et' : TH3F("photonEtEtaFishValue", "Photon E_{T} vs |#eta| vs FishValue;E_{T};|#eta|;Fisher Value", len(bins_et)-1, bins_et, len(bins_eta)-1, bins_eta, len(bins_fisher)-1, bins_fisher)} |
| 71 |
|
# END of Varibles |
| 72 |
|
######################################## |
| 73 |
|
|
| 74 |
|
|
| 75 |
|
######################################## |
| 76 |
< |
# Print names of opened files |
| 65 |
< |
for aFile in listOfFiles: |
| 66 |
< |
print "Opened %s, scale=%.2e, cuts='%s'" % (aFile.name, aFile.scale, aFile.cuts) |
| 67 |
< |
|
| 76 |
> |
# Open output file |
| 77 |
|
outputFile = TFile(outputFilename, "recreate") |
| 78 |
|
if not outputFile.IsZombie(): |
| 79 |
|
print "Opened %s for output." % outputFilename |
| 80 |
|
else: |
| 81 |
< |
print "Error opening "+self.name+" for output exiting..." |
| 81 |
> |
print "Error opening %s for output exiting..." % outputFilename |
| 82 |
|
sys.exit(0) |
| 83 |
|
|
| 84 |
< |
print "Creating plots...", |
| 84 |
> |
print "\nCuts applied to all files:\n %s" % cutForAllFiles |
| 85 |
> |
print "Creating plots..." |
| 86 |
|
# Loop over all things to plot |
| 87 |
|
for plot in listOfPlots: |
| 88 |
+ |
print " %s >> %s" % (plot, listOfPlots[plot].GetName()) # Say what plot is being made |
| 89 |
|
# Loop over all TTrees (from the different files) |
| 90 |
|
for aFile in listOfFiles: |
| 91 |
|
tempHist = listOfPlots[plot].Clone("temp") # Create temp histogram |
| 92 |
|
cuts = "%s&&%s" % (cutForAllFiles, aFile.cuts) # Set cuts |
| 93 |
< |
aFile.ttree.Draw( "%s >> temp" % plot, cuts, "goff" ) # Draw into it (with cuts) |
| 93 |
> |
aFile.ttree.Draw( "%s >> temp" % plot, cuts, "goff" ) # Draw into it (with cuts) graphics off |
| 94 |
|
tempHist.Scale(aFile.scale) # Scale it |
| 95 |
|
listOfPlots[plot].Add(tempHist) # Add it to total histogram |
| 96 |
|
print "done." |
