From 5695f1c003f723909253f777bdb982eefe2f1d2c Mon Sep 17 00:00:00 2001
From: Christian Weber <cweber@bnl.gov>
Date: Wed, 30 Nov 2022 16:56:32 -0500
Subject: [PATCH] Add 2l2j, 2e2j and 2l2j to the plottable flavors

---
 plotPostProcess.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/plotPostProcess.py b/plotPostProcess.py
index d3fbb34..f82ef5f 100644
--- a/plotPostProcess.py
+++ b/plotPostProcess.py
@@ -1130,7 +1130,9 @@ if __name__ == '__main__':
     help = "If we only want to list a subset of the kninematic variables, list them here.\
             Use like --kinematicsToPlot m4l m34. If none are specified, or argument is not used, we plot all kinematic variables" ) 
 
-    parser.add_argument( "--flavorsToPlot", nargs='*', default=["2e2mu", "2mu2e", "4e", "4mu", "All"], 
+    flavorsToPlotDefaults = ["2e2mu", "2mu2e", "4e", "4mu", "2l2j", "2e2j", "2mu2j","All"]
+    parser.add_argument( "--flavorsToPlot", nargs='*', 
+    default=flavorsToPlotDefaults, 
     help = "If we only want to list a subset of the kninematic variables, list them here.\
             Use like --kinematicsToPlot m4l m34. If none are specified, or argument is not used, we plot all kinematic variables" ) 
 
@@ -1472,8 +1474,7 @@ if __name__ == '__main__':
 
             backgroundMergedTH1ForRatioHist = backgroundMergedTH1.Clone( backgroundMergedTH1.GetName() + "_ratioHist")
 
-
-            inferredFlavor  = re.search("(All)|(2e2mu)|(4mu)|(4e)|(2mu2e)|(2l2e)|(2l2mu)", histEnding).group()
+            inferredFlavor  = re.search("(%s)" %")|(".join(flavorsToPlotDefaults), histEnding).group()
 
             ################# add in systematic uncertainties #################
             if addSystematicUncertaintyToNominal  and systematicChannel == "Nominal" and "ZXVR1a" not in histEnding : # and "ZXSR" in histEnding
-- 
GitLab