CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Public Attributes
tree.Tree Class Reference
Inheritance diagram for tree.Tree:

Public Member Functions

def __init__
 
def branch_
 
def copyStructure
 
def fill
 
def reset
 
def setDefaultFloatType
 
def setDefaultIntType
 
def var
 
def vector
 
def vfill
 

Public Attributes

 defaultFloatType
 
 defaultIntType
 
 defaults
 
 fillers
 
 tree
 
 vars
 
 vecdefaults
 
 vecvars
 

Detailed Description

Definition at line 5 of file tree.py.

Constructor & Destructor Documentation

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

Definition at line 7 of file tree.py.

7 
8  def __init__(self, name, title, defaultFloatType="D", defaultIntType="I"):
9  self.vars = {}
10  self.vecvars = {}
11  self.tree = TTree(name, title)
12  self.defaults = {}
13  self.vecdefaults = {}
14  self.defaultFloatType = defaultFloatType
15  self.defaultIntType = defaultIntType
16  self.fillers = {}
tree.Tree.__init__
def __init__
Definition: tree.py:7
tree.Tree.vecvars
vecvars
Definition: tree.py:9
tree.Tree.vars
vars
Definition: tree.py:8
tree.Tree.tree
tree
Definition: tree.py:10
tree.Tree.vecdefaults
vecdefaults
Definition: tree.py:12
tree.Tree.defaults
defaults
Definition: tree.py:11
tree.Tree.defaultIntType
defaultIntType
Definition: tree.py:14
tree.Tree.fillers
fillers
Definition: tree.py:15
tree.Tree.defaultFloatType
defaultFloatType
Definition: tree.py:13

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 32 of file tree.py.

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

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

32 
33  def branch_(self, selfmap, varName, type, len, postfix="", storageType="default", title=None):
34  """Backend function used to create scalar and vector branches.
35  Users should call "var" and "vector", not this function directly."""
36  if storageType == "default":
37  storageType = self.defaultIntType if type is int else self.defaultFloatType
38  if type is float :
39  if storageType == "F":
40  selfmap[varName]=numpy.zeros(len,numpy.float32)
41  self.tree.Branch(varName,selfmap[varName],varName+postfix+'/F')
42  elif storageType == "D":
43  selfmap[varName]=numpy.zeros(len,numpy.float64)
44  self.tree.Branch(varName,selfmap[varName],varName+postfix+'/D')
45  else:
46  raise RuntimeError('Unknown storage type %s for branch %s' % (storageType, varName))
47  elif type is int:
48  dtypes = {
49  "i" : numpy.uint32,
50  "s" : numpy.uint16,
51  "b" : numpy.uint8,
52  "l" : numpy.uint64,
53  "I" : numpy.int32,
54  "S" : numpy.int16,
55  "B" : numpy.int8,
56  "L" : numpy.int64,
57  }
58  if storageType not in dtypes:
59  raise RuntimeError('Unknown storage type %s for branch %s' % (storageType, varName))
60  selfmap[varName]=numpy.zeros(len,dtypes[storageType])
61  self.tree.Branch(varName,selfmap[varName],varName+postfix+'/'+storageType)
62  else:
63  raise RuntimeError('Unknown type %s for branch %s' % (type, varName))
64  if title:
65  self.tree.GetBranch(varName).SetTitle(title)
tree.Tree.branch_
def branch_
Definition: tree.py:32
tree.Tree.defaultIntType
defaultIntType
Definition: tree.py:14
tree.Tree.defaultFloatType
defaultFloatType
Definition: tree.py:13
def tree.Tree.copyStructure (   self,
  tree 
)

Definition at line 23 of file tree.py.

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

23 
24  def copyStructure(self, tree):
25  for branch in tree.GetListOfBranches():
26  name = branch.GetName()
27  typeName = branch.GetListOfLeaves()[0].GetTypeName()
28  type = float
29  if typeName == 'Int_t':
30  type = int
31  self.var(name, type)
tree.Tree.var
def var
Definition: tree.py:66
tree.Tree.copyStructure
def copyStructure
Definition: tree.py:23
def tree.Tree.fill (   self,
  varName,
  value 
)

Definition at line 116 of file tree.py.

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

116 
117  def fill(self, varName, value ):
118  if isinstance(self.vars[varName], numpy.ndarray):
119  self.vars[varName][0]=value
120  else:
121  self.fillers[varName](self.vars[varName],value)
tree.Tree.fill
def fill
Definition: tree.py:116
tree.Tree.vars
vars
Definition: tree.py:8
tree.Tree.fillers
fillers
Definition: tree.py:15
def tree.Tree.reset (   self)

Definition at line 103 of file tree.py.

References tree.Tree.defaults, geometryComparison.GeometryComparison.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, and tree.Tree.vecdefaults.

103 
104  def reset(self):
105  for name,value in self.vars.iteritems():
106  if name in self.fillers:
107  self.fillers[name](value, self.defaults[name])
108  else:
109  value[0]=self.defaults[name]
110  for name,value in self.vecvars.iteritems():
111  if isinstance(value, numpy.ndarray):
112  value.fill(self.vecdefaults[name])
113  else:
114  if isinstance(value, ROOT.TObject) and value.ClassName() == "TClonesArray":
115  value.ExpandCreateFast(0)
tree.Tree.vecdefaults
vecdefaults
Definition: tree.py:12
tree.Tree.defaults
defaults
Definition: tree.py:11
tree.Tree.reset
def reset
Definition: tree.py:103
tree.Tree.fillers
fillers
Definition: tree.py:15
def tree.Tree.setDefaultFloatType (   self,
  defaultFloatType 
)

Definition at line 17 of file tree.py.

References tree.Tree.defaultFloatType.

17 
18  def setDefaultFloatType(self, defaultFloatType):
19  self.defaultFloatType = defaultFloatType
tree.Tree.setDefaultFloatType
def setDefaultFloatType
Definition: tree.py:17
tree.Tree.defaultFloatType
defaultFloatType
Definition: tree.py:13
def tree.Tree.setDefaultIntType (   self,
  defaultIntType 
)

Definition at line 20 of file tree.py.

References tree.Tree.defaultIntType.

20 
21  def setDefaultIntType(self, defaultIntType):
22  self.defaultIntType = defaultIntType
tree.Tree.setDefaultIntType
def setDefaultIntType
Definition: tree.py:20
tree.Tree.defaultIntType
defaultIntType
Definition: tree.py:14
def tree.Tree.var (   self,
  varName,
  type = float,
  default = -99,
  title = None,
  storageType = "default",
  filler = None 
)

Definition at line 66 of file tree.py.

References tree.Tree.branch_(), fwlite::RecordWriter::DataBuffer.branch_, edm::TypeInBranchType.branch_, fwlite::internal::Data.branch_, BareRootProductGetter::Buffer.branch_, tree.Tree.defaults, geometryComparison.GeometryComparison.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().

66 
67  def var(self, varName,type=float, default=-99, title=None, storageType="default", filler=None ):
68  if type in [int, float]:
69  self.branch_(self.vars, varName, type, 1, title=title, storageType=storageType)
70  self.defaults[varName] = default
71  elif __builtins__['type'](type) == str:
72  # create a value, looking up the type from ROOT and calling the default constructor
73  self.vars[varName] = getattr(ROOT,type)()
74  if type in [ "TLorentzVector" ]: # custom streamer classes
75  self.tree.Branch(varName+".", type, self.vars[varName], 8000,-1)
76  else:
77  self.tree.Branch(varName+".", type, self.vars[varName])
78  if filler is None:
79  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)")
80  self.fillers[varName] = filler
81  else:
82  raise RuntimeError('Unknown type %s for branch %s: it is not int, float or a string' % (type, varName))
83  self.defaults[varName] = default
tree.Tree.branch_
def branch_
Definition: tree.py:32
tree.Tree.var
def var
Definition: tree.py:66
tree.Tree.vars
vars
Definition: tree.py:8
tree.Tree.defaults
defaults
Definition: tree.py:11
tree.Tree.fillers
fillers
Definition: tree.py:15
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 84 of file tree.py.

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

84 
85  def vector(self, varName, lenvar, maxlen=None, type=float, default=-99, title=None, storageType="default", filler=None ):
86  """either lenvar is a string, and maxlen an int (variable size array), or lenvar is an int and maxlen is not specified (fixed array)"""
87  if type in [int, float]:
88  if __builtins__['type'](lenvar) == int: # need the __builtins__ since 'type' is a variable here :-/
89  self.branch_(self.vecvars, varName, type, lenvar, postfix="[%d]" % lenvar, title=title, storageType=storageType)
90  else:
91  if maxlen == None: RuntimeError, 'You must specify a maxlen if making a dynamic array';
92  self.branch_(self.vecvars, varName, type, maxlen, postfix="[%s]" % lenvar, title=title, storageType=storageType)
93  elif __builtins__['type'](type) == str:
94  self.vecvars[varName] = ROOT.TClonesArray(type,(lenvar if __builtins__['type'](lenvar) == int else maxlen))
95  if type in [ "TLorentzVector" ]: # custom streamer classes
96  self.tree.Branch(varName+".", self.vecvars[varName], 32000, -1)
97  else:
98  self.tree.Branch(varName+".", self.vecvars[varName])
99  if filler is None:
100  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)")
101  self.fillers[varName] = filler
102  self.vecdefaults[varName] = default
tree.Tree.vector
def vector
Definition: tree.py:84
tree.Tree.vecvars
vecvars
Definition: tree.py:9
tree.Tree.branch_
def branch_
Definition: tree.py:32
tree.Tree.vecdefaults
vecdefaults
Definition: tree.py:12
tree.Tree.fillers
fillers
Definition: tree.py:15
def tree.Tree.vfill (   self,
  varName,
  values 
)

Definition at line 122 of file tree.py.

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

122 
123  def vfill(self, varName, values ):
124  a = self.vecvars[varName]
125  if isinstance(a, numpy.ndarray):
126  for (i,v) in enumerate(values):
127  a[i]=v
128  else:
129  if isinstance(a, ROOT.TObject) and a.ClassName() == "TClonesArray":
130  a.ExpandCreateFast(len(values))
131  fillit = self.fillers[varName]
132  for (i,v) in enumerate(values):
133  fillit(a[i],v)
tree.Tree.vecvars
vecvars
Definition: tree.py:9
tree.Tree.vfill
def vfill
Definition: tree.py:122
tree.Tree.fillers
fillers
Definition: tree.py:15

Member Data Documentation

tree.Tree.defaultFloatType

Definition at line 13 of file tree.py.

Referenced by tree.Tree.branch_(), and tree.Tree.setDefaultFloatType().

tree.Tree.defaultIntType

Definition at line 14 of file tree.py.

Referenced by tree.Tree.branch_(), and tree.Tree.setDefaultIntType().

tree.Tree.defaults

Definition at line 11 of file tree.py.

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

tree.Tree.fillers

Definition at line 15 of file tree.py.

Referenced by tree.Tree.fill(), tree.Tree.reset(), tree.Tree.var(), tree.Tree.vector(), and tree.Tree.vfill().

tree.Tree.tree

Definition at line 10 of file tree.py.

Referenced by svgfig.SVG.__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer.beginLoop(), MetTreeProducer.MetTreeProducer.declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer.declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer.fillTree(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer.process(), MetTreeProducer.MetTreeProducer.process(), and python.cmstools.EventTree.SetAlias().

tree.Tree.vars

Definition at line 8 of file tree.py.

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

tree.Tree.vecdefaults

Definition at line 12 of file tree.py.

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

tree.Tree.vecvars

Definition at line 9 of file tree.py.

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

Generated for CMSSW Reference Manual by doxygen 1.8.5