handle zero-events nanoAOD files

cms-sw · Dec 9, 2023 · 31bbfd0 · 31bbfd0
1 parent ce34e99
commit 31bbfd0
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/PhysicsTools/NanoAOD/scripts/haddnano.py b/PhysicsTools/NanoAOD/scripts/haddnano.py
@@ -32,7 +32,7 @@ def zeroFill(tree, brName, brObj, allowNonBool=False):
 fileHandles = []
 goFast = True
 for fn in files:
-    print("Adding file" + str(fn))
+    print("Adding file", str(fn))
     fileHandles.append(ROOT.TFile.Open(fn))
     if fileHandles[-1].GetCompressionSettings() != fileHandles[0].GetCompressionSettings():
         goFast = False
@@ -44,7 +44,7 @@ def zeroFill(tree, brName, brObj, allowNonBool=False):
 
 for e in fileHandles[0].GetListOfKeys():
     name = e.GetName()
-    print("Merging" + str(name))
+    print("Merging", str(name))
     obj = e.ReadObj()
     cl = ROOT.TClass.GetClass(e.GetClassName())
     inputs = ROOT.TList()
@@ -53,15 +53,26 @@ def zeroFill(tree, brName, brObj, allowNonBool=False):
         obj = obj.CloneTree(-1, "fast" if goFast else "")
         branchNames = set([x.GetName() for x in obj.GetListOfBranches()])
     for fh in fileHandles[1:]:
+        if isTree and obj.GetName() == 'Events' and obj.GetEntries() == 0 :
+            # Zero-events first file. Skip to avoid messing up branches.
+            print(" 'Events' tree contsins no events; skipping")
+            obj = fh.GetListOfKeys().FindObject(name).ReadObj()
+            obj = obj.CloneTree(-1, "fast" if goFast else "")
+            branchNames = set([x.GetName() for x in obj.GetListOfBranches()])
+            continue
         otherObj = fh.GetListOfKeys().FindObject(name).ReadObj()
+        if isTree and obj.GetName() == 'Events' and otherObj.GetEntries() == 0 :
+            # Zero-events file; skip
+            print(" 'Events' tree contains no events; skipping")
+            continue
         inputs.Add(otherObj)
         if isTree and obj.GetName() == 'Events':
             otherObj.SetAutoFlush(0)
             otherBranches = set([x.GetName()
                                  for x in otherObj.GetListOfBranches()])
             missingBranches = list(branchNames - otherBranches)
             additionalBranches = list(otherBranches - branchNames)
-            print("missing: " + str(missingBranches) + "\n Additional:" + str(additionalBranches))
+            print("missing: " + str(missingBranches) + "\n Additional: " + str(additionalBranches))
             for br in missingBranches:
                 # fill "Other"
                 zeroFill(otherObj, br, obj.GetListOfBranches().FindObject(br))
@@ -76,7 +87,7 @@ def zeroFill(tree, brName, brObj, allowNonBool=False):
                                  for x in otherObj.GetListOfBranches()])
             missingBranches = list(branchNames - otherBranches)
             additionalBranches = list(otherBranches - branchNames)
-            print("missing: " + str(missingBranches) + "\n Additional:" + str(additionalBranches))
+            print("missing: " + str(missingBranches) + "\n Additional: " + str(additionalBranches))
             for br in missingBranches:
                 # fill "Other"
                 zeroFill(otherObj, br, obj.GetListOfBranches(