HistControl.HistControl Class Reference

Inheritance diagram for HistControl.HistControl:

Collaboration diagram for HistControl.HistControl:

Public Member Functions
def	__init__ (self, name="HistControl", title="Controls the histograms", doOutputFile=False)
def	BookTProfile (self, name, title, numbinx, xmin, xmax)
def	BookHist1D (self, name, title, numbinx, xmin, xmax)
def	BookHist2D (self, name, title, numbinx, xmin, xmax, numbiny, ymin, ymax)
def	GetRandomString (self, N)
def	SaveTH1FToImage (self, histname, imagename, drawoptions='HIST', colors=[ROOT.kRed, ROOT.kBlue, ROOT.kGreen, ROOT.kOrange, ROOT.kCyan, ROOT.kMagenta], linewidth=2)
def	SaveTH2FToImage (self, histname, imagename, doth2f2graph=False, minY=1., maxY=-1.)
def	getRMSs (self, hist)
def	TwoDHist2Graph (self, name, doErros=False)
def	FillHist1D (self, name, value)
def	FillHist2D (self, name, valuex, valuey)
def	FillTProfile (self, name, valuex, valuey)
def	SaveAllToFile (self)

Private Attributes
	__OneDHists
	__TwoDHists
	__Graphs
	__TProfiles
	__OutputROOTFile
	__doOutputFile

Detailed Description

Definition at line 9 of file HistControl.py.

Constructor & Destructor Documentation

__init__()

def HistControl.HistControl.__init__	(		self,
			name = "HistControl",
			title = "Controls the histograms",
			doOutputFile = False
	)

Definition at line 11 of file HistControl.py.

    def __init__( self, name = "HistControl", title = "Controls the histograms", doOutputFile=False):
 
        # Initialise the base class:
        ROOT.TNamed.__init__( self, name, title )
        self.SetName( name )
        self.SetTitle( title )
 
        self.__OneDHists = {}
        self.__TwoDHists = {}
        self.__Graphs    = {}
        self.__TProfiles = {}
        self.__OutputROOTFile = None
 
        self.__doOutputFile = doOutputFile
            
        return
 

Member Function Documentation

BookHist1D()

def HistControl.HistControl.BookHist1D	(		self,
			name,
			title,
			numbinx,
			xmin,
			xmax
	)

Definition at line 36 of file HistControl.py.

    def BookHist1D(self, name, title,  numbinx, xmin, xmax):
 
        if name in self.__OneDHists.keys():
            print 'already stored a th1f object with name', name
            return False
        hist1d = ROOT.TH1F(name, title,  numbinx, xmin, xmax)
        self.__OneDHists[ name ] = hist1d
        return True
        
        

BookHist2D()

def HistControl.HistControl.BookHist2D	(		self,
			name,
			title,
			numbinx,
			xmin,
			xmax,
			numbiny,
			ymin,
			ymax
	)

Definition at line 46 of file HistControl.py.

    def BookHist2D(self, name, title,  numbinx, xmin, xmax, numbiny, ymin, ymax):
        if name in self.__TwoDHists.keys():
            print 'already stored a th2f object with name', name
            return False
        hist2d = ROOT.TH2F(name, title,  numbinx, xmin, xmax, numbiny, ymin, ymax)
        self.__TwoDHists[ name ] = hist2d
        return True
        
 

BookTProfile()

def HistControl.HistControl.BookTProfile	(		self,
			name,
			title,
			numbinx,
			xmin,
			xmax
	)

Definition at line 28 of file HistControl.py.

    def BookTProfile(self, name, title, numbinx, xmin, xmax):
        if name in self.__TProfiles.keys():
            print 'already stored a tprofile object with name', name
            return False
        tprof = ROOT.TProfile(name, title, numbinx, xmin, xmax)
        self.__TProfiles[ name ] = tprof
        return True
 

FillHist1D()

def HistControl.HistControl.FillHist1D	(		self,
			name,
			value
	)

Definition at line 214 of file HistControl.py.

    def FillHist1D(self, name, value):
        if name in self.__OneDHists:
            self.__OneDHists[ name ].Fill(value)
        else:
            print 'No such th1f with name', name
            pass
        return
 

FillHist2D()

def HistControl.HistControl.FillHist2D	(		self,
			name,
			valuex,
			valuey
	)

Definition at line 222 of file HistControl.py.

    def FillHist2D(self, name, valuex, valuey):
        if name in self.__TwoDHists:
            self.__TwoDHists[ name ].Fill(valuex, valuey)
        else:
            print 'No such th2f with name', name
            pass
        return
 

FillTProfile()

def HistControl.HistControl.FillTProfile	(		self,
			name,
			valuex,
			valuey
	)

Definition at line 230 of file HistControl.py.

    def FillTProfile(self, name, valuex, valuey):
        if name in self.__TProfiles:
            self.__TProfiles[ name ].Fill(valuex, valuey)
        else:
            print 'No such tprofile with name', name
            pass
        return
 
 

GetRandomString()

def HistControl.HistControl.GetRandomString	(		self,
			N
	)

Definition at line 57 of file HistControl.py.

    def GetRandomString(self, N):
        str = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(N))
        print 'randomstring', str
        return str
 

getRMSs()

def HistControl.HistControl.getRMSs	(		self,
			hist
	)

Definition at line 134 of file HistControl.py.

    def getRMSs(self, hist):
 
        mean = hist.GetMean()
        nentries = hist.GetEntries()
        rmsDown = 0.
        rmsUp   = 0.
        nDown = 0
        nUp   = 0
        print 'mean', mean
        for bin in range(1, hist.GetNbinsX()+1):
            if hist.GetBinCenter(bin) < mean:
                print 'low', hist.GetBinContent(bin)
                nDown = nDown + hist.GetBinContent(bin)
                rmsDown += float(hist.GetBinContent(bin))*(mean - float(hist.GetBinCenter(bin)))**2
            else:
                nUp = nUp + hist.GetBinContent(bin)
                rmsUp   += float(hist.GetBinContent(bin))*(mean - float(hist.GetBinCenter(bin)))**2
                pass
            pass
        if nDown > 0:
            rmsDown = ROOT.TMath.Sqrt(rmsDown/float(nentries))
        else:
            rmsDown = 0.
            pass
        if nUp > 0:
            rmsUp   = ROOT.TMath.Sqrt(rmsUp/float(nentries))
        else:
            rmsUp   = 0.
 
        return (rmsDown,rmsUp)
 
 

SaveAllToFile()

def HistControl.HistControl.SaveAllToFile

(

self

)

Definition at line 239 of file HistControl.py.

    def SaveAllToFile(self):
        if not self.__doOutputFile:
            print 'Cannot save hists to file. No output file set.'
            return
 
        outputFile = ROOT.TFile(self.GetName()+'.root', 'RECREATE')
        for name1D, hist1D in self.__OneDHists.iteritems():
            if hist1D.GetEntries() != 0:
                hist1D.Write()
        for name2D, hist2D in self.__TwoDHists.iteritems():
            if hist2D.GetEntries() != 0:
                hist2D.Write()
        for nametprof, histtprof in self.__TProfiles.iteritems():
            histtprof.Write()
        #for graph in self.__Graphs:
        #    graph.Write()
        return

SaveTH1FToImage()

def HistControl.HistControl.SaveTH1FToImage	(		self,
			histname,
			imagename,
			drawoptions = 'HIST',
			colors = [ROOT.kRed, ROOT.kBlue, ROOT.kGreen, ROOT.kOrange, ROOT.kCyan, ROOT.kMagenta],
			linewidth = 2
	)

Definition at line 62 of file HistControl.py.

    def SaveTH1FToImage(self, histname, imagename, drawoptions = 'HIST',
                        colors = [ROOT.kRed, ROOT.kBlue, ROOT.kGreen, ROOT.kOrange, ROOT.kCyan, ROOT.kMagenta],
                        linewidth = 2):
        canvas = ROOT.TCanvas('canvas'+self.GetRandomString(6))
        canvas.cd(0)
        if isinstance(histname, list):
            maxy = -999.
            for (index,hname) in enumerate(histname):
                if self.__OneDHists[ hname ].GetMaximum() > maxy:
                    maxy = self.__OneDHists[ hname ].GetMaximum()
                    pass
                pass
            for (index,hname) in enumerate(histname):
                self.__OneDHists[ hname ].SetMaximum(maxy*1.2)
                pass
            for (index,hname) in enumerate(histname):
                self.__OneDHists[ hname ].SetLineWidth(linewidth)
                if index >= len(colors):
                    print 'add more colors'
                    assert false
                    pass
                self.__OneDHists[ hname ].SetLineColor(colors[index])
                if index == 0:
                    self.__OneDHists[ hname ].Draw(drawoptions)
                else:
                    self.__OneDHists[ hname ].Draw(drawoptions+'SAME')
                    pass
                pass
            pass
        else:
            self.__OneDHists[ hname ].SetLineWidth(linewidth)
            self.__OneDHists[ hname ].SetLineColor(colors[0])
            self.__OneDHists[ hname ].Draw(drawoptions)
            pass
        canvas.SaveAs('output/'+imagename+'.png')
        pass
 

SaveTH2FToImage()

def HistControl.HistControl.SaveTH2FToImage	(		self,
			histname,
			imagename,
			doth2f2graph = False,
			minY = 1.,
			maxY = -1.
	)

Definition at line 99 of file HistControl.py.

    def SaveTH2FToImage(self, histname, imagename, doth2f2graph=False, minY=1., maxY=-1.):
        canvas = ROOT.TCanvas('canvas'+self.GetRandomString(6))
        canvas.cd(0)
        hist_frame = None
        if (minY<maxY) and doth2f2graph:
            hist_frame = ROOT.TH1F('hist_frame', self.__TwoDHists[ histname ].GetTitle(),
                                   100, self.__TwoDHists[ histname ].GetXaxis().GetXmin(), self.__TwoDHists[ histname ].GetXaxis().GetXmax())
            hist_frame.SetXTitle( self.__TwoDHists[ histname ].GetXaxis().GetTitle() )
            hist_frame.SetYTitle( self.__TwoDHists[ histname ].GetYaxis().GetTitle() )
            hist_frame.SetMinimum(minY)
            hist_frame.SetMaximum(maxY)
        if isinstance(histname, list):
            print 'still need to figure this out'
            pass
        else:
            if doth2f2graph:
                graph_err = self.TwoDHist2Graph(histname, doErros=True)
                graph     = self.TwoDHist2Graph(histname, doErros=False)
                if hist_frame is not None:
                    hist_frame.Draw()
                    graph.SetLineWidth(3)
                    graph_err.Draw('3SAME')
                    #graph.SetMarkerSize(0)
                    graph.Draw('CSAME')
                else:
                    graph.Draw('ALP')
                    pass
            else:
                self.__TwoDHists[ histname ].Draw()
                pass
            pass
        canvas.SaveAs('output/'+imagename+'.png')
        pass
 
 

TwoDHist2Graph()

def HistControl.HistControl.TwoDHist2Graph	(		self,
			name,
			doErros = False
	)

Definition at line 166 of file HistControl.py.

    def TwoDHist2Graph(self, name, doErros=False):
        if not name in self.__TwoDHists.keys():
            print 'No such TH2F as', name
            return None
 
        hist = self.__TwoDHists[ name ]
        
        list_mean   = []
        list_x      = []
        list_xel    = []
        list_yel    = []
        list_xeh    = []
        list_yeh    = []
 
        for bin in range(1, hist.GetNbinsX()+1 ):
            hist_projY = hist.ProjectionY('pY'+str(bin), bin, bin)
            if hist_projY.GetEntries() != 0:
                mean = hist_projY.GetMean()
                list_mean.append( mean )
                list_x.append( hist.GetXaxis().GetBinCenter(bin) )
                list_xel.append(0.)
                list_xeh.append(0.)
                if doErros:
                    (yel,yeh) = self.getRMSs(hist_projY)
                else:
                    (yel,yeh) = (0., 0.)
                    pass
                list_yel.append(yel)
                list_yeh.append(yeh)
                pass
            hist_projY.Delete()
            pass
 
        array_mean      = array.array('d', list_mean)
        array_x         = array.array('d', list_x)
        array_xel       = array.array('d', list_xel)
        array_yel       = array.array('d', list_yel)
        array_xeh       = array.array('d', list_xeh)
        array_yeh       = array.array('d', list_yeh)
 
        if doErros:
            graph_mean    = ROOT.TGraphAsymmErrors(len(list_mean), array_x, array_mean, array_xel,array_xeh,array_yel,array_yeh)
            graph_mean.SetFillColor(ROOT.kRed)
        else:
            graph_mean    = ROOT.TGraph(len(list_mean), array_x, array_mean)
        return graph_mean
 
 

Member Data Documentation

__doOutputFile

HistControl.HistControl.__doOutputFile

private

Definition at line 24 of file HistControl.py.

__Graphs

HistControl.HistControl.__Graphs

private

Definition at line 20 of file HistControl.py.

__OneDHists

HistControl.HistControl.__OneDHists

private

Definition at line 18 of file HistControl.py.

__OutputROOTFile

HistControl.HistControl.__OutputROOTFile

private

Definition at line 22 of file HistControl.py.

__TProfiles

HistControl.HistControl.__TProfiles

private

Definition at line 21 of file HistControl.py.

__TwoDHists

HistControl.HistControl.__TwoDHists

private

Definition at line 19 of file HistControl.py.

---

The documentation for this class was generated from the following file:

• HistControl.py