plotmaker Namespace Reference

Functions
def	make2DPdf (name, hist, prefix)
def	makePdf (name, hists, prefix)

Variables
int	ScalarColor = ROOT.kPink+4
int	DarkPhotonColor = ROOT.kAzure-3
int	ElectronColor = ROOT.kRed-3
int	MuonColor = ROOT.kBlue-3
int	PionColor = ROOT.kGreen-3
	parser = optparse.OptionParser()
	dest
	help
	default
	options
	args
	hist_file = ROOT.TFile.Open(options.hist_file_in, 'r')
dictionary	hist1DDict
dictionary	hist2DDict = { 'decayZR2D': None, 'decayEtaR2D':None }
	keys = hist_file.GetListOfKeys()
	keyName = key.GetName()
	hist = ROOT.TH1F()

Function Documentation

make2DPdf()

def plotmaker.make2DPdf	(		name,
			hist,
			prefix
	)

Definition at line 17 of file plotmaker.py.

def make2DPdf(name, hist, prefix):
    if not hist:
        return
    canvas = ROOT.TCanvas('whoCares'+name , 'name', 1600,1600)
    canvas.cd(0)
    hist.Draw('COLZ')
    if prefix != '':
        prefix += '_'
        pass
    canvas.SaveAs(prefix+name+'.pdf')
 

makePdf()

def plotmaker.makePdf	(		name,
			hists,
			prefix
	)

Definition at line 28 of file plotmaker.py.

def makePdf(name, hists, prefix):
    if len(hists) == 0:
        return
    if prefix != '' and prefix[-1] != '_':
        prefix += '_'
        pass
 
    if isinstance(hists, list):
        maxY = -999
        maxEntries = -999
        for hist in hists:
            hist.Scale( 1./float(hist.GetEntries()) )
            hist.SetYTitle('Normalized to Unity')
            if hist.GetMaximum() > maxY and ('Electron' in hist.GetName() or 'Pion' in hist.GetName() or 'Muon' in hist.GetName()):
                maxY = hist.GetMaximum()
                maxEntries = hist.GetEntries()
                pass
            hist.SetLineWidth(2)
            pass
    
        legend = ROOT.TLegend(.6, .65, .85, .89)
        legend.SetFillColor(0)
        legend.SetBorderSize(0)
        legend.SetTextFont(102)
        legend.SetTextSize(0.04)
        
        canvas = ROOT.TCanvas('whoCares'+name , 'name', 1600,1200)
        canvas.cd(0)
        scalarhist = None
        darkphotonhist = None
        for index,hist in enumerate(hists):
            #print hist.GetName()
            drawOptions = 'HIST'
            if index!=0:
                drawOptions += 'SAME'
                pass
 
            if 'Electron' in hist.GetName():
                legend.AddEntry(hist, '#splitline{Electrons}{'+str(int(hist.GetEntries()))+' events}', 'f')
                hist.SetLineColor(ElectronColor)
                hist.SetMaximum(1.2*maxY )
                pass
            elif 'Muon' in hist.GetName():
                legend.AddEntry(hist, '#splitline{Muons}{'+str(int(hist.GetEntries()))+' events}', 'f')
                hist.SetLineColor(MuonColor)
                hist.SetMaximum(1.2*maxY )
                pass
            elif 'Pion' in hist.GetName():
                legend.AddEntry(hist, '#splitline{Pions}{'+str(int(hist.GetEntries()))+' events}', 'f')
                hist.SetLineColor(PionColor)
                hist.SetMaximum(1.2*maxY )
                pass
            elif 'DarkPhoton' in hist.GetName() or 'darkPhoton' in hist.GetName():
                hist.SetLineColor(DarkPhotonColor)
                darkphotonhist = hist
                continue
                pass
            elif 'Scalar' in hist.GetName() or 'scalar' in hist.GetName():
                hist.SetLineColor(ScalarColor)
                scalarhist = hist
                continue
                pass
            hist.Draw(drawOptions)
            pass
        if legend.GetNRows():
            legend.Draw()
            pass
        canvas.SaveAs(prefix+name+'.pdf')
        if darkphotonhist:
            if not 'darkPhoton' in name:
                name = 'darkPhoton'+name
                pass
            makePdf(name, darkphotonhist, prefix)
            pass
        if scalarhist:
            if not 'scalar' in name:
                name = 'scalar'+name
                pass
            makePdf(name, scalarhist, prefix)
            pass
    else:
        canvas = ROOT.TCanvas('whoCares2'+name , 'name', 1600,1200)
        canvas.cd(0)
        hists.Draw('HIST')
        canvas.SaveAs(prefix+name+'.pdf')
 
 

Variable Documentation

args

plotmaker.args

Definition at line 122 of file plotmaker.py.

DarkPhotonColor

int plotmaker.DarkPhotonColor = ROOT.kAzure-3

Definition at line 12 of file plotmaker.py.

default

plotmaker.default

Definition at line 119 of file plotmaker.py.

dest

plotmaker.dest

Definition at line 118 of file plotmaker.py.

ElectronColor

int plotmaker.ElectronColor = ROOT.kRed-3

Definition at line 13 of file plotmaker.py.

help

plotmaker.help

Definition at line 118 of file plotmaker.py.

hist

plotmaker.hist = ROOT.TH1F()

Definition at line 148 of file plotmaker.py.

hist1DDict

dictionary plotmaker.hist1DDict

Initial value:

=  { 'darkPhotonPt': [], 'darkPhotonEta':[], 'darkPhotonPhi': [], 'Polarization': [],
                   'scalarPt': [], 'scalarEta':[], 'scalarPhi': [],
                   'decayR1D': [], 'decayZ1D':[],
                   'OpeningR':[], 'OpeningPhi':[], 'OpeningEta':[],
                   'ptBalance':[], 'subLeadingPt':[], 'leadingPt':[]}

Definition at line 135 of file plotmaker.py.

hist2DDict

dictionary plotmaker.hist2DDict = { 'decayZR2D': None, 'decayEtaR2D':None }

Definition at line 141 of file plotmaker.py.

hist_file

plotmaker.hist_file = ROOT.TFile.Open(options.hist_file_in, 'r')

Definition at line 129 of file plotmaker.py.

keyName

plotmaker.keyName = key.GetName()

Definition at line 145 of file plotmaker.py.

keys

plotmaker.keys = hist_file.GetListOfKeys()

Definition at line 143 of file plotmaker.py.

MuonColor

int plotmaker.MuonColor = ROOT.kBlue-3

Definition at line 14 of file plotmaker.py.

options

plotmaker.options

Definition at line 122 of file plotmaker.py.

parser

plotmaker.parser = optparse.OptionParser()

Definition at line 117 of file plotmaker.py.

PionColor

int plotmaker.PionColor = ROOT.kGreen-3

Definition at line 15 of file plotmaker.py.

ScalarColor

int plotmaker.ScalarColor = ROOT.kPink+4

Definition at line 11 of file plotmaker.py.