mergePhysValFiles Namespace Reference

Functions
	mergeFolder (path)

Variables
	start = time.clock()
	parser = argparse.ArgumentParser(description='Merge specific folder(s) in root files.')
	help
	default
	nargs
	args = parser.parse_args()
	folder = os.path.abspath(args.input)
	mergeDirs = args.dir
	origDir = os.getcwd()
	out = args.output
	pattern = args.pattern
	output_file = os.path.abspath(args.output)
	files = glob.glob(folder + "/" + pattern)
	f = ROOT.TFile(output_file, "recreate")
	f2 = ROOT.TFile(files[1])
list	errors = []
	newfolder = f.mkdir(mergeDir,mergeDir)
	end = time.clock()
list	categories
str	restCategory = 'other'
list	categories_with_subcategories_type_1 = ['SV', 'tracks']
list	sub_categories_type_1
list	categories_with_subcategories_type_2 = ['tagger_GN2v01','tagger_GN3XPV01']
list	sub_categories_type_2
list	categories_with_subcategories_type_3 = ['old_taggers']
list	sub_categories_type_3 = [ ]
list	categories_with_subcategories_type_4 = ['jet']
list	sub_categories_type_4 = [ 'jet']
list	jetcontainers
dict	tagfolders = {}

Function Documentation

mergeFolder()

mergePhysValFiles.mergeFolder

(

path

)

Definition at line 44 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

def mergeFolder(path) :
    print("Merging folder " + path)
    d = f2.Get(path)
    if not d:
        error = "ERROR: Cannot find directory " + path + ". Omitting."
        print(error)
        errors.append(error)
        return
    dirlist = d.GetListOfKeys()
    for subdir in dirlist:
        obj = subdir.ReadObj()
        if obj.IsA().InheritsFrom(ROOT.TH1.Class()):
            print("Now merging "+obj.GetName())
            h1 = obj
            hpath = d.GetPath()
            hname = hpath[hpath.find(":")+2:]+"/"+obj.GetName()
            print("Path: "+hname)
            for tup in files:
                if tup==files[1]: continue
                nextfile = ROOT.TFile(tup)
                h2 = nextfile.Get(hname)
                if not h2:
                    error = "ERROR: Cannot find " + hname + " in file " + tup + ". Omitting."
                    print(error)
                    errors.append(error)
                    continue
                h1.Add(h2)
            subfolder = f.Get(hpath[hpath.find(":")+2:])
            subfolder.cd()
            h1.Write()
        if obj.IsA().InheritsFrom(ROOT.TDirectory.Class()):
            print("Found subdirectory "+obj.GetName())
            hpath = obj.GetPath()
            subfolder = f.mkdir(hpath[hpath.find(":")+2:],obj.GetTitle())
            subfolder.cd()
            mergeFolder(hpath[hpath.find(":")+2:])
 

Variable Documentation

args

mergePhysValFiles.args = parser.parse_args()

Definition at line 20 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

categories

list mergePhysValFiles.categories

Initial value:

=  ['jet',
              'tracks',
              'SV',
              'tagger_GN2v01',
              'tagger_GN3XPV01',
              'old_taggers',
             ]

Definition at line 17 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

categories_with_subcategories_type_1

list mergePhysValFiles.categories_with_subcategories_type_1 = ['SV', 'tracks']

Definition at line 29 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

categories_with_subcategories_type_2

list mergePhysValFiles.categories_with_subcategories_type_2 = ['tagger_GN2v01','tagger_GN3XPV01']

Definition at line 38 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

categories_with_subcategories_type_3

list mergePhysValFiles.categories_with_subcategories_type_3 = ['old_taggers']

Definition at line 45 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

categories_with_subcategories_type_4

list mergePhysValFiles.categories_with_subcategories_type_4 = ['jet']

Definition at line 49 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

default

mergePhysValFiles.default

Definition at line 16 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

end

mergePhysValFiles.end = time.clock()

Definition at line 92 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

errors

list mergePhysValFiles.errors = []

Definition at line 42 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

f

mergePhysValFiles.f = ROOT.TFile(output_file, "recreate")

Definition at line 31 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

f2

mergePhysValFiles.f2 = ROOT.TFile(files[1])

Definition at line 33 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

files

mergePhysValFiles.files = glob.glob(folder + "/" + pattern)

Definition at line 29 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

folder

mergePhysValFiles.folder = os.path.abspath(args.input)

Definition at line 21 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

help

mergePhysValFiles.help

Definition at line 16 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

jetcontainers

list mergePhysValFiles.jetcontainers

Initial value:

=  ['AntiKt4EMTopoJets',
                 'AntiKt4EMPFlowJets',
                 'AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets',
                 ]

Definition at line 54 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

mergeDirs

mergePhysValFiles.mergeDirs = args.dir

Definition at line 22 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

nargs

mergePhysValFiles.nargs

Definition at line 18 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

newfolder

mergePhysValFiles.newfolder = f.mkdir(mergeDir,mergeDir)

Definition at line 82 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

origDir

mergePhysValFiles.origDir = os.getcwd()

Definition at line 23 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

out

mergePhysValFiles.out = args.output

Definition at line 24 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

output_file

mergePhysValFiles.output_file = os.path.abspath(args.output)

Definition at line 26 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

parser

mergePhysValFiles.parser = argparse.ArgumentParser(description='Merge specific folder(s) in root files.')

Definition at line 15 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

pattern

mergePhysValFiles.pattern = args.pattern

Definition at line 25 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

restCategory

str mergePhysValFiles.restCategory = 'other'

Definition at line 26 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

start

mergePhysValFiles.start = time.clock()

Definition at line 13 of file DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py.

sub_categories_type_1

list mergePhysValFiles.sub_categories_type_1

Initial value:

=  [ '_incl',
                   '_b',
                   '_c',
                   '_l',
                   '_muon',
                 ]

Definition at line 31 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

sub_categories_type_2

list mergePhysValFiles.sub_categories_type_2

Initial value:

=  [ '_pt_ttbar',
                   '_pt_Zprime',
                   '_Lxy'
                 ]

Definition at line 40 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

sub_categories_type_3

list mergePhysValFiles.sub_categories_type_3 = [ ]

Definition at line 47 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

sub_categories_type_4

list mergePhysValFiles.sub_categories_type_4 = [ 'jet']

Definition at line 51 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.

tagfolders

dict mergePhysValFiles.tagfolders = {}

Definition at line 93 of file PhysicsAnalysis/JetTagging/JetTagValidation/JetTagDQA/scripts/mergePhysValFiles.py.