Define classes. More...

Inheritance diagram for python.rootplot.utilities.Hist2D:

Public Member Functions
def	__getitem__ (self, index)

def	__init__ (self, hist, label="__nolabel__", title=None, xlabel=None, ylabel=None)

def	__iter__ (self)

def	__len__ (self)

def	TH2F (self, name="")

Public Attributes
	binlabelsx

	binlabelsy

	content

	entries

	label

	name

	nbinsx

	nbinsy

	rootclass

	title

	x

	xedges

	xlabel

	y

	yedges

	ylabel

Private Member Functions
def	_flat_content (self)

Detailed Description

Define classes.

A container to hold the parameters from a 2D ROOT histogram.

Definition at line 43 of file utilities.py.

Constructor & Destructor Documentation

def python.rootplot.utilities.Hist2D.__init__	(	self,
		hist,
		label = `"__nolabel__"`,
		title = `None`,
		xlabel = `None`,
		ylabel = `None`
	)

Definition at line 46 of file utilities.py.

                  xlabel=None, ylabel=None):
         try:
             if not hist.InheritsFrom("TH2"):
                 raise TypeError("%s does not inherit from TH2" % hist)
         except:
             raise TypeError("%s is not a ROOT object" % hist)
         self.rootclass = hist.ClassName()
         self.name = hist.GetName()
         self.nbinsx = nx = hist.GetNbinsX()
         self.nbinsy = ny = hist.GetNbinsY()
         self.binlabelsx = process_bin_labels([hist.GetXaxis().GetBinLabel(i)
                                                for i in range(1, nx + 1)])
         if self.binlabelsx:
             self.nbinsx = nx = self.binlabelsx.index('')
             self.binlabelsx = self.binlabelsx[:ny]
         self.binlabelsy = process_bin_labels([hist.GetYaxis().GetBinLabel(i)
                                                for i in range(1, ny + 1)])
         if self.binlabelsy:
             self.nbinsy = ny = self.binlabelsy.index('')
             self.binlabelsy = self.binlabelsy[:ny]
         self.entries = hist.GetEntries()
         self.content = [[hist.GetBinContent(i, j) for i in range(1, nx + 1)]
                         for j in range(1, ny + 1)]
         self.xedges = [hist.GetXaxis().GetBinLowEdge(i)
                              for i in range(1, nx + 2)]
         self.yedges = [hist.GetYaxis().GetBinLowEdge(i)
                              for i in range(1, ny + 2)]
         self.x      = [(self.xedges[i+1] + self.xedges[i])/2
                              for i in range(nx)]
         self.y      = [(self.yedges[i+1] + self.yedges[i])/2
                              for i in range(ny)]
         self.title  = title or hist.GetTitle()
         self.xlabel = xlabel or hist.GetXaxis().GetTitle()
         self.ylabel = ylabel or hist.GetYaxis().GetTitle()
         self.label  = label

Member Function Documentation

def python.rootplot.utilities.Hist2D.__getitem__	(	self,
		index
	)

Return contents of indexth bin: x.__getitem__(y) <==> x[y]

Definition at line 86 of file utilities.py.

References python.rootplot.utilities.Hist2D._flat_content().

     def __getitem__(self, index):
         """Return contents of indexth bin: x.__getitem__(y) <==> x[y]"""
         return self._flat_content()[index]

def python.rootplot.utilities.Hist2D.__iter__ ( self )

Iterate through bins: x.__iter__() <==> iter(x)

Definition at line 92 of file utilities.py.

References python.rootplot.utilities.Hist2D._flat_content().

     def __iter__(self):
         """Iterate through bins: x.__iter__() <==> iter(x)"""
         return iter(self._flat_content())

def python.rootplot.utilities.Hist2D.__len__ ( self )

Return the number of bins: x.__len__() <==> len(x)

Definition at line 89 of file utilities.py.

References python.rootplot.utilities.Hist2D._flat_content().

     def __len__(self):
         """Return the number of bins: x.__len__() <==> len(x)"""
         return len(self._flat_content())

def python.rootplot.utilities.Hist2D._flat_content ( self )

private

Definition at line 81 of file utilities.py.

References DQMChannel.content, fixedArray< T, S >.content, BlobComplexObjects.content, python.rootplot.utilities.Hist2D.content, and PhysicsTools::MVATrainerLooper::TrainerContainer.content.

Referenced by python.rootplot.utilities.Hist2D.__getitem__(), python.rootplot.utilities.Hist2D.__iter__(), and python.rootplot.utilities.Hist2D.__len__().

     def _flat_content(self):
         flatcontent = []
         for row in self.content:
             flatcontent += row
         return flatcontent

def python.rootplot.utilities.Hist2D.TH2F	(	self,
		name = `""`
	)

Return a ROOT.TH2F object with contents of this Hist2D.

Definition at line 95 of file utilities.py.

     def TH2F(self, name=""):
         """Return a ROOT.TH2F object with contents of this Hist2D."""
         th2f = ROOT.TH2F(name, "",
                          self.nbinsx, array.array('f', self.xedges),
                          self.nbinsy, array.array('f', self.yedges))
         th2f.SetTitle("%s;%s;%s" % (self.title, self.xlabel, self.ylabel))
         for ix in range(self.nbinsx):
             for iy in range(self.nbinsy):
                 th2f.SetBinContent(ix + 1, iy + 1, self.content[iy][ix])
         return th2f
 