finalise btagging SF and rename base module

Author: Oguz-Guzel

python/basePlotter.py renamed to python/baseAnalysis.py

@@ -5,6 +5,7 @@
 from itertools import chain
 
 import utils
+import definitions as defs
 
 from bamboo import treefunctions as op
 from bamboo import treedecorators as td
@@ -116,9 +117,11 @@ def prepare_ondemand(self, tree, sample=None, sampleCfg=None, backend=None):
                 "v12", year=era[:4], isMC=isMC, systVariations=systVars),
             backend=self.args.backend or backend)
 
+        # MC weight
         if isMC:
             noSel = noSel.refine('genWeight', weight=tree.genWeight)
 
+        # Triggers
         self.triggersPerPrimaryDataset = {}
 
         def addHLTPath(PD, HLT):
@@ -130,8 +133,6 @@ def addHLTPath(PD, HLT):
             except AttributeError:
                 print("Couldn't find branch tree.HLT.%s, cross check!" % HLT)
 
-        # Triggers
-
         # Muon
         addHLTPath('Muon', 'IsoMu24')
         addHLTPath('Muon', 'IsoMu27')
@@ -181,12 +182,16 @@ def addHLTPath(PD, HLT):
         cmJMEArgs.update({"jsonFileSubjet": JSONFiles[era]["AK4"], })
         configureJets(tree._FatJet, jetType="AK8PFPuppi", **cmJMEArgs)
 
+        # define objects
+        defs.defineObjects(self, tree)
+
         # btagging SF
-        from bamboo.scalefactors import get_bTagSF_itFit, makeBtagWeightItFit
-        def btvSF(flav): return get_bTagSF_itFit(
-            BTV_SF_JSONFiles[era], "particleNet", "btagDeepFlavB", flav, noSel)
-        btvWeight = makeBtagWeightItFit(tree._Jet, btvSF)
-        noSel = noSel.refine("btag", weight=btvWeight)
+        if isMC:
+            from bamboo.scalefactors import get_bTagSF_itFit, makeBtagWeightItFit
+            def btvSF(flav): return get_bTagSF_itFit(
+                BTV_SF_JSONFiles[era], "particleNet", "btagDeepFlavB", flav, noSel)
+            btvWeight = makeBtagWeightItFit(self.ak4Jets, btvSF)
+            noSel = noSel.refine("btag", weight=btvWeight)
 
         return tree, noSel, be, lumiArgs
 

python/controlPlotter.py

@@ -3,13 +3,13 @@
 from bamboo.plots import EquidistantBinning as EqBin
 from bamboo import treefunctions as op
 
-from basePlotter import NanoBaseHHWWbb
+from baseAnalysis import NanoBaseHHWWbb
 from selections import makeDLSelection, makeSLSelection
 import definitions as defs
 
 
 class controlPlotter(NanoBaseHHWWbb):
-    """ Class to create control plots and skims"""
+    """ Class to create control plots, cutflow reports and skims"""
 
     def __init__(self, args):
         super(controlPlotter, self).__init__(args)
@@ -19,9 +19,6 @@ def __init__(self, args):
     def definePlots(self, tree, noSel, sample=None, sampleCfg=None):
         plots = []
 
-        # call object definitions
-        defs.defineObjects(self, tree)
-
         # cutflow report
         yields = CutFlowReport("yields", recursive=True)
         plots.append(yields)

python/selections.py

@@ -1,5 +1,4 @@
 from bamboo import treefunctions as op
-import definitions as defs
 
 # common variables for DL and SL channels