Inheritance diagram for tree.Tree:

Public Member Functions
def	__init__ (self, name, title, defaultFloatType="D", defaultIntType="I")

def	branch_ (self, selfmap, varName, type, len, postfix="", storageType="default", title=None)

def	copyStructure (self, tree)

def	fill (self, varName, value)

def	reset (self)

def	setDefaultFloatType (self, defaultFloatType)

def	setDefaultIntType (self, defaultIntType)

def	var (self, varName, type=float, default=-99, title=None, storageType="default", filler=None)

def	vector (self, varName, lenvar, maxlen=None, type=float, default=-99, title=None, storageType="default", filler=None)

def	vfill (self, varName, values)

Public Attributes
	defaultFloatType

	defaultIntType

	defaults

	fillers

	tree

	vars

	vecdefaults

	vecvars

Detailed Description

Definition at line 6 of file tree.py.

Constructor & Destructor Documentation

def tree.Tree.__init__	(	self,
		name,
		title,
		defaultFloatType = `"D"`,
		defaultIntType = `"I"`
	)

Definition at line 8 of file tree.py.

     def __init__(self, name, title, defaultFloatType="D", defaultIntType="I"):
         self.vars = {}
         self.vecvars = {}
         self.tree = TTree(name, title)
         self.defaults = {}
         self.vecdefaults = {}
         self.defaultFloatType = defaultFloatType
         self.defaultIntType = defaultIntType
         self.fillers = {}
 

Member Function Documentation

def tree.Tree.branch_	(	self,
		selfmap,
		varName,
		type,
		len,
		postfix = `""`,
		storageType = `"default"`,
		title = `None`
	)

Backend function used to create scalar and vector branches. 
   Users should call "var" and "vector", not this function directly.

Definition at line 33 of file tree.py.

References tree.Tree.defaultFloatType, and tree.Tree.defaultIntType.

Referenced by tree.Tree.var(), and tree.Tree.vector().

     def branch_(self, selfmap, varName, type, len, postfix="", storageType="default", title=None):
         """Backend function used to create scalar and vector branches. 
            Users should call "var" and "vector", not this function directly."""
         if storageType == "default": 
             storageType = self.defaultIntType if type is int else self.defaultFloatType
         if type is float  :
             if storageType == "F": 
                 selfmap[varName]=numpy.zeros(len,numpy.float32)
                 self.tree.Branch(varName,selfmap[varName],varName+postfix+'/F')
             elif storageType == "D":
                 selfmap[varName]=numpy.zeros(len,numpy.float64)
                 self.tree.Branch(varName,selfmap[varName],varName+postfix+'/D')
             else:
                 raise RuntimeError('Unknown storage type %s for branch %s' % (storageType, varName))
         elif type is int: 
             dtypes = {
                 "i" : numpy.uint32,
                 "s" : numpy.uint16,
                 "b" : numpy.uint8,
                 "l" : numpy.uint64,
                 "I" : numpy.int32,
                 "S" : numpy.int16,
                 "B" : numpy.int8,
                 "L" : numpy.int64,
             }
             if storageType not in dtypes: 
                 raise RuntimeError('Unknown storage type %s for branch %s' % (storageType, varName))
             selfmap[varName]=numpy.zeros(len,dtypes[storageType])
             self.tree.Branch(varName,selfmap[varName],varName+postfix+'/'+storageType)
         else:
             raise RuntimeError('Unknown type %s for branch %s' % (type, varName))
         if title:
             self.tree.GetBranch(varName).SetTitle(title)
 

def tree.Tree.copyStructure	(	self,
		tree
	)

Definition at line 24 of file tree.py.

References CentralityTable::BinValues.var, TaggingVariablePlotter::VariableConfig.var, PhysicsTools::SourceVariableSet::PosVar.var, tree.Tree.var(), reco::parser::Grammar::definition< ScannerT >.var, and PhysicsTools::MVAComputer::InputVar.var.

     def copyStructure(self, tree):
         for branch in tree.GetListOfBranches():
             name = branch.GetName() 
             typeName = branch.GetListOfLeaves()[0].GetTypeName()
             type = float
             if typeName == 'Int_t':
                 type = int
             self.var(name, type)            
     

def tree.Tree.fill	(	self,
		varName,
		value
	)

Definition at line 117 of file tree.py.

References tree.Tree.fillers, tree.Tree.vars, PlotAlignmentValidation::DMRPlotInfo.vars, and JsonOutputProducer.vars.

     def fill(self, varName, value ):
         if isinstance(self.vars[varName], numpy.ndarray):
             self.vars[varName][0]=value
         else:
             self.fillers[varName](self.vars[varName],value)
 

def tree.Tree.reset ( self )

Definition at line 104 of file tree.py.

References tree.Tree.defaults, geometryComparison.GeometryComparison.defaults, primaryVertexValidation.PrimaryVertexValidation.defaults, offlineValidation.OfflineValidation.defaults, zMuMuValidation.ZMuMuValidation.defaults, preexistingValidation.PreexistingValidation.defaults, plottingOptions.BasePlottingOptions.defaults, genericValidation.GenericValidation.defaults, preexistingValidation.PreexistingOfflineValidation.defaults, plottingOptions.PlottingOptionsTrackSplitting.defaults, plottingOptions.PlottingOptionsZMuMu.defaults, plottingOptions.PlottingOptionsOffline.defaults, plottingOptions.PlottingOptionsPrimaryVertex.defaults, genericValidation.GenericValidationData.defaults, genericValidation.GenericValidationData_CTSR.defaults, svgfig.Path.defaults, svgfig.Curve.defaults, svgfig.Poly.defaults, svgfig.Text.defaults, svgfig.TextGlobal.defaults, svgfig.Dots.defaults, svgfig.Line.defaults, svgfig.LineGlobal.defaults, svgfig.VLine.defaults, svgfig.HLine.defaults, svgfig.Rect.defaults, svgfig.Ellipse.defaults, svgfig.Ticks.defaults, svgfig.CurveAxis.defaults, svgfig.LineAxis.defaults, svgfig.XAxis.defaults, svgfig.YAxis.defaults, svgfig.Axes.defaults, svgfig.HGrid.defaults, svgfig.VGrid.defaults, svgfig.Grid.defaults, svgfig.XErrorBars.defaults, svgfig.YErrorBars.defaults, tree.Tree.fillers, tree.Tree.vars, PlotAlignmentValidation::DMRPlotInfo.vars, JsonOutputProducer.vars, tree.Tree.vecdefaults, and tree.Tree.vecvars.

     def reset(self):
         for name,value in six.iteritems(self.vars):
             if name in self.fillers:
                 self.fillers[name](value, self.defaults[name])
             else:
                 value[0]=self.defaults[name]
         for name,value in six.iteritems(self.vecvars):
             if isinstance(value, numpy.ndarray):
                 value.fill(self.vecdefaults[name])
             else:
                 if isinstance(value, ROOT.TObject) and value.ClassName() == "TClonesArray":
                     value.ExpandCreateFast(0)
             

def tree.Tree.setDefaultFloatType	(	self,
		defaultFloatType
	)

Definition at line 18 of file tree.py.

References tree.Tree.defaultFloatType.

     def setDefaultFloatType(self, defaultFloatType):
         self.defaultFloatType = defaultFloatType
 

def tree.Tree.setDefaultIntType	(	self,
		defaultIntType
	)

Definition at line 21 of file tree.py.

References tree.Tree.defaultIntType.

     def setDefaultIntType(self, defaultIntType):
         self.defaultIntType = defaultIntType
         

def tree.Tree.var	(	self,
		varName,
		type = `float`,
		default = `-99`,
		title = `None`,
		storageType = `"default"`,
		filler = `None`
	)

Definition at line 67 of file tree.py.

References tree.Tree.branch_(), fwlite::RecordWriter::DataBuffer.branch_, edm::TypeInBranchType.branch_, fwlite::internal::Data.branch_, BareRootProductGetter::Buffer.branch_, edm::PoolOutputModule::SpecialSplitLevelForBranch.branch_, tree.Tree.defaults, geometryComparison.GeometryComparison.defaults, primaryVertexValidation.PrimaryVertexValidation.defaults, offlineValidation.OfflineValidation.defaults, zMuMuValidation.ZMuMuValidation.defaults, preexistingValidation.PreexistingValidation.defaults, plottingOptions.BasePlottingOptions.defaults, genericValidation.GenericValidation.defaults, preexistingValidation.PreexistingOfflineValidation.defaults, plottingOptions.PlottingOptionsTrackSplitting.defaults, plottingOptions.PlottingOptionsZMuMu.defaults, plottingOptions.PlottingOptionsOffline.defaults, plottingOptions.PlottingOptionsPrimaryVertex.defaults, genericValidation.GenericValidationData.defaults, genericValidation.GenericValidationData_CTSR.defaults, svgfig.Path.defaults, svgfig.Curve.defaults, svgfig.Poly.defaults, svgfig.Text.defaults, svgfig.TextGlobal.defaults, svgfig.Dots.defaults, svgfig.Line.defaults, svgfig.LineGlobal.defaults, svgfig.VLine.defaults, svgfig.HLine.defaults, svgfig.Rect.defaults, svgfig.Ellipse.defaults, svgfig.Ticks.defaults, svgfig.CurveAxis.defaults, svgfig.LineAxis.defaults, svgfig.XAxis.defaults, svgfig.YAxis.defaults, svgfig.Axes.defaults, svgfig.HGrid.defaults, svgfig.VGrid.defaults, svgfig.Grid.defaults, svgfig.XErrorBars.defaults, svgfig.YErrorBars.defaults, tree.Tree.fillers, tree.Tree.vars, PlotAlignmentValidation::DMRPlotInfo.vars, and JsonOutputProducer.vars.

Referenced by tree.Tree.copyStructure().

     def var(self, varName,type=float, default=-99, title=None, storageType="default", filler=None ):
         if type in [int, float]:
             self.branch_(self.vars, varName, type, 1, title=title, storageType=storageType)
             self.defaults[varName] = default
         elif __builtins__['type'](type) == str:
             # create a value, looking up the type from ROOT and calling the default constructor
             self.vars[varName] = getattr(ROOT,type)()
             if type in [ "TLorentzVector" ]: # custom streamer classes
                 self.tree.Branch(varName+".", type, self.vars[varName], 8000,-1)
             else:
                 self.tree.Branch(varName+".", type, self.vars[varName])
             if filler is None:
                 raise RuntimeError("Error: when brancing with an object, filler should be set to a function that takes as argument an object instance and a value, and set the instance to the value (as otherwise python assignment of objects changes the address as well)")
             self.fillers[varName] = filler
         else:
             raise RuntimeError('Unknown type %s for branch %s: it is not int, float or a string' % (type, varName))
         self.defaults[varName] = default
 

def tree.Tree.vector	(	self,
		varName,
		lenvar,
		maxlen = `None`,
		type = `float`,
		default = `-99`,
		title = `None`,
		storageType = `"default"`,
		filler = `None`
	)

either lenvar is a string, and maxlen an int (variable size array), or lenvar is an int and maxlen is not specified (fixed array)

Definition at line 85 of file tree.py.

References tree.Tree.branch_(), fwlite::RecordWriter::DataBuffer.branch_, edm::TypeInBranchType.branch_, fwlite::internal::Data.branch_, BareRootProductGetter::Buffer.branch_, edm::PoolOutputModule::SpecialSplitLevelForBranch.branch_, tree.Tree.fillers, tree.Tree.vecdefaults, and tree.Tree.vecvars.

     def vector(self, varName, lenvar, maxlen=None, type=float, default=-99, title=None, storageType="default", filler=None ):
         """either lenvar is a string, and maxlen an int (variable size array), or lenvar is an int and maxlen is not specified (fixed array)"""
         if type in [int, float]:
             if __builtins__['type'](lenvar) == int:  # need the __builtins__ since 'type' is a variable here :-/
                 self.branch_(self.vecvars, varName, type, lenvar, postfix="[%d]" % lenvar, title=title, storageType=storageType)
             else:
                 if maxlen == None: RuntimeError, 'You must specify a maxlen if making a dynamic array';
                 self.branch_(self.vecvars, varName, type, maxlen, postfix="[%s]" % lenvar, title=title, storageType=storageType)
         elif __builtins__['type'](type) == str:
             self.vecvars[varName] = ROOT.TClonesArray(type,(lenvar if __builtins__['type'](lenvar) == int else maxlen))
             if type in [ "TLorentzVector" ]: # custom streamer classes
                 self.tree.Branch(varName+".", self.vecvars[varName], 32000, -1)
             else:
                 self.tree.Branch(varName+".", self.vecvars[varName])
             if filler is None:
                 raise RuntimeError("Error: when brancing with an object, filler should be set to a function that takes as argument an object instance and a value, and set the instance to the value (as otherwise python assignment of objects changes the address as well)")
             self.fillers[varName] = filler
         self.vecdefaults[varName] = default
 

def tree.Tree.vfill	(	self,
		varName,
		values
	)

Definition at line 123 of file tree.py.

References tree.Tree.fillers, and tree.Tree.vecvars.

     def vfill(self, varName, values ):
         a = self.vecvars[varName]
         if isinstance(a, numpy.ndarray):
             for (i,v) in enumerate(values):
                 a[i]=v
         else:
             if isinstance(a, ROOT.TObject) and a.ClassName() == "TClonesArray":
                 a.ExpandCreateFast(len(values))
             fillit = self.fillers[varName]
             for (i,v) in enumerate(values):
                 fillit(a[i],v)