haddnano Namespace Reference

Functions
def	zeroFill

Variables
tuple	additionalBranches = list(otherBranches - branchNames)
tuple	branchNames = set([x.GetName() for x in obj.GetListOfBranches()])
tuple	cl = ROOT.TClass.GetClass(e.GetClassName())
list	fileHandles = []
list	files = sys.argv[2:]
	goFast = True
tuple	inputs = ROOT.TList()
tuple	isTree = obj.IsA()
tuple	missingBranches = list(branchNames - otherBranches)
tuple	name = e.GetName()
tuple	obj = e.ReadObj()
tuple	of = ROOT.TFile(ofname, "recreate")
list	ofname = sys.argv[1]
tuple	otherBranches
tuple	otherObj = fh.GetListOfKeys()

Function Documentation

def haddnano.zeroFill	(		tree,
			brName,
			brObj,
			allowNonBool = False
	)

Definition at line 12 of file haddnano.py.

References print(), and sistrip::SpyUtilities.range().

def zeroFill(tree, brName, brObj, allowNonBool=False):
    # typename: (numpy type code, root type code)
    branch_type_dict = {'Bool_t': ('?', 'O'), 'Float_t': ('f4', 'F'), 'UInt_t': (
        'u4', 'i'), 'Long64_t': ('i8', 'L'), 'Double_t': ('f8', 'D')}
    brType = brObj.GetLeaf(brName).GetTypeName()
    if (not allowNonBool) and (brType != "Bool_t"):
        print(("Did not expect to back fill non-boolean branches", tree, brName, brObj.GetLeaf(br).GetTypeName()))
    else:
        if brType not in branch_type_dict:
            raise RuntimeError('Impossible to backfill branch of type %s' % brType)
        buff = numpy.zeros(1, dtype=numpy.dtype(branch_type_dict[brType][0]))
        b = tree.Branch(brName, buff, brName + "/" +
                        branch_type_dict[brType][1])
        # be sure we do not trigger flushing
        b.SetBasketSize(tree.GetEntries() * 2)
        for x in range(0, tree.GetEntries()):
            b.Fill()
        b.ResetAddress()