PrepareReferenceFile Namespace Reference

Functions
def	iterate (ROOT.TDirectory thisdir, ROOT.TDirectory targetdir, str prefix, typing.Pattern regex, bool excludeTrees)

Variables
	parser = argparse.ArgumentParser()
	default
	args = parser.parse_args()
	infile = ROOT.TFile.Open(args.infile, "READ")
	outfile = ROOT.TFile.Open(args.outfile, "RECREATE")
	regex = re.compile(args.excludeRegex)

Function Documentation

iterate()

def PrepareReferenceFile.iterate	(	ROOT.TDirectory	thisdir,
		ROOT.TDirectory	targetdir,
		str	prefix,
		typing.Pattern	regex,
		bool	excludeTrees
	)

Recurse to copy all objects except those of regex

Definition at line 10 of file PrepareReferenceFile.py.

def iterate(thisdir: ROOT.TDirectory, targetdir: ROOT.TDirectory, prefix: str, regex: typing.Pattern,
            excludeTrees: bool):
    """Recurse to copy all objects except those of regex"""
    for k in thisdir.GetListOfKeys():
        if k.GetClassName().startswith('TDirectory'):
            targetsubdirname = thisdir.GetPath()[len(prefix):] + '/' + k.GetName()
            if regex.fullmatch(targetsubdirname):
                print('Now skipping', targetsubdirname)
                continue
            targetsubdir = targetdir.mkdir(k.GetName())
            iterate(o := k.ReadObj(), targetsubdir, prefix, regex, excludeTrees)
            ROOT.SetOwnership(o, True)
        else:
            if k.GetClassName().startswith('TTree'):
                if not args.excludeTrees or k.GetName() == 'metadata':
                    targetdir.WriteObject(o := k.ReadObj().CloneTree(), k.GetName())
                    ROOT.SetOwnership(o, True)
            else:
                targetdir.WriteObject(o := k.ReadObj(), k.GetName())
                ROOT.SetOwnership(o, True)
 

Variable Documentation

args

PrepareReferenceFile.args = parser.parse_args()

Definition at line 39 of file PrepareReferenceFile.py.

default

PrepareReferenceFile.default

Definition at line 36 of file PrepareReferenceFile.py.

infile

PrepareReferenceFile.infile = ROOT.TFile.Open(args.infile, "READ")

Definition at line 41 of file PrepareReferenceFile.py.

outfile

PrepareReferenceFile.outfile = ROOT.TFile.Open(args.outfile, "RECREATE")

Definition at line 42 of file PrepareReferenceFile.py.

parser

PrepareReferenceFile.parser = argparse.ArgumentParser()

Definition at line 33 of file PrepareReferenceFile.py.

regex

PrepareReferenceFile.regex = re.compile(args.excludeRegex)

Definition at line 43 of file PrepareReferenceFile.py.