Public Member Functions
def	__init__ (self, name, objectType, maxlen, help="", mcOnly=False, sortAscendingBy=None, sortDescendingBy=None, filter=None)

def	__repr__ (self)

def	fillBranchesScalar (self, treeNumpy, collection, isMC)

def	fillBranchesVector (self, treeNumpy, collection, isMC)

def	get_cpp_declaration (self, isMC)

def	get_cpp_wrapper_class (self, isMC)

def	get_py_wrapper_class (self, isMC)

def	makeBranchesScalar (self, treeNumpy, isMC)

def	makeBranchesVector (self, treeNumpy, isMC)

Public Attributes
	filter

	help

	maxlen

	mcOnly

	name

	objectType

	sortAscendingBy

	sortDescendingBy

Detailed Description

Type defining a set of branches associated to a list of objects (i.e. an instance of NTupleObjectType)

Definition at line 157 of file autovars.py.

Constructor & Destructor Documentation

def core.autovars.NTupleCollection.__init__	(	self,
		name,
		objectType,
		maxlen,
		help = `""`,
		mcOnly = `False`,
		sortAscendingBy = `None`,
		sortDescendingBy = `None`,
		filter = `None`
	)

Definition at line 159 of file autovars.py.

     def __init__(self, name, objectType, maxlen, help="", mcOnly=False, sortAscendingBy=None, sortDescendingBy=None, filter=None):
         self.name = name
         self.objectType = objectType
         self.maxlen = maxlen
         self.help = help
         if objectType.mcOnly and mcOnly == False: 
             #print "collection %s is set to mcOnly since the type %s is mcOnly" % (name, objectType.name)
             mcOnly = True
         self.mcOnly = mcOnly
         if sortAscendingBy != None and sortDescendingBy != None:
             raise RuntimeError("Cannot specify two sort conditions")
         self.filter = filter
         self.sortAscendingBy  = sortAscendingBy
         self.sortDescendingBy = sortDescendingBy

Member Function Documentation

def core.autovars.NTupleCollection.__repr__ ( self )

Definition at line 214 of file autovars.py.

     def __repr__(self):
         return "<NTupleCollection[%s]>" % self.name
 

def core.autovars.NTupleCollection.fillBranchesScalar	(	self,
		treeNumpy,
		collection,
		isMC
	)

Definition at line 191 of file autovars.py.

     def fillBranchesScalar(self,treeNumpy,collection,isMC):
         if not isMC and self.objectType.mcOnly: return
         if self.filter != None: collection = [ o for o in collection if self.filter(o) ]
         if self.sortAscendingBy != None: collection  = sorted(collection, key=self.sortAscendingBy)
         if self.sortDescendingBy != None: collection = sorted(collection, key=self.sortDescendingBy, reverse=True)
         num = min(self.maxlen,len(collection))
         treeNumpy.fill("n"+self.name, num)
         allvars = self.objectType.allVars(isMC)
         for i in range(num): 
             o = collection[i]
             for v in allvars:
                 treeNumpy.fill("%s%d_%s" % (self.name, i+1, v.name), v(o))

def core.autovars.NTupleCollection.fillBranchesVector	(	self,
		treeNumpy,
		collection,
		isMC
	)

Definition at line 203 of file autovars.py.

     def fillBranchesVector(self,treeNumpy,collection,isMC):
         if not isMC and self.objectType.mcOnly: return
         if self.filter != None: collection = [ o for o in collection if self.filter(o) ]
         if self.sortAscendingBy != None: collection  = sorted(collection, key=self.sortAscendingBy)
         if self.sortDescendingBy != None: collection = sorted(collection, key=self.sortDescendingBy, reverse=True)
         num = min(self.maxlen,len(collection))
         treeNumpy.fill("n"+self.name, num)
         allvars = self.objectType.allVars(isMC)
         for v in allvars:
             name="%s_%s" % (self.name, v.name) if v.name != "" else self.name
             treeNumpy.vfill(name, [ v(collection[i]) for i in range(num) ])

def core.autovars.NTupleCollection.get_cpp_declaration	(	self,
		isMC
	)

Definition at line 217 of file autovars.py.

     def get_cpp_declaration(self, isMC):
         s = []
         for v in self.objectType.allVars(isMC):
             s += ["{0} {1}__{2}[{3}];".format(v.type.__name__, self.name, v.name, self.maxlen)]
         return "\n".join(s)
 

def core.autovars.NTupleCollection.get_cpp_wrapper_class	(	self,
		isMC
	)

Definition at line 223 of file autovars.py.

     def get_cpp_wrapper_class(self, isMC):
         s = "class %s {\n" % self.name
         s += "public:\n"
         for v in self.objectType.allVars(isMC):
             s += "    {0} {1};\n".format(v.type.__name__, v.name)
         s += "};\n"
         return s
 

def core.autovars.NTupleCollection.get_py_wrapper_class	(	self,
		isMC
	)

Definition at line 231 of file autovars.py.

     def get_py_wrapper_class(self, isMC):
         s = "class %s:\n" % self.name
         s += "    def __init__(self, tree, n):\n"
         for v in self.objectType.allVars(isMC):
             if len(v.name)>0:
                 s += "        self.{0} = tree.{1}_{2}[n];\n".format(v.name, self.name, v.name)
             else:
                 s += "        self.{0} = tree.{0}[n];\n".format(self.name)
 
         s += "    @staticmethod\n"
         s += "    def make_array(event):\n"
         s += "        return [{0}(event.input, i) for i in range(event.input.n{0})]\n".format(self.name)
         return s
 
 

def core.autovars.NTupleCollection.makeBranchesScalar	(	self,
		treeNumpy,
		isMC
	)

Definition at line 173 of file autovars.py.

     def makeBranchesScalar(self,treeNumpy,isMC):
         if not isMC and self.objectType.mcOnly: return
         treeNumpy.var("n"+self.name, int)
         allvars = self.objectType.allVars(isMC)
         for v in allvars:
             for i in range(1,self.maxlen+1):
                 h = v.help
                 if self.help: h = "%s for %s [%d]" % ( h if h else v.name, self.help, i-1 )
                 treeNumpy.var("%s%d_%s" % (self.name, i, v.name), type=v.type, default=v.default, title=h, filler=v.filler)

def core.autovars.NTupleCollection.makeBranchesVector	(	self,
		treeNumpy,
		isMC
	)

Definition at line 182 of file autovars.py.

     def makeBranchesVector(self,treeNumpy,isMC):
         if not isMC and self.objectType.mcOnly: return
         treeNumpy.var("n"+self.name, int)
         allvars = self.objectType.allVars(isMC)
         for v in allvars:
             h = v.help
             if self.help: h = "%s for %s" % ( h if h else v.name, self.help )
             name="%s_%s" % (self.name, v.name) if v.name != "" else self.name
             treeNumpy.vector(name, "n"+self.name, self.maxlen, type=v.type, default=v.default, title=h, filler=v.filler)