Merge pull request #84 from blinkseb/fatjet_config

adding tau1, tau2 and tau3 as config file string
cp3-llbb · Oct 15, 2015 · ffe65c6 · ffe65c6
2 parents cc5b051 + e7bac28
commit ffe65c6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/interface/FatJetsProducer.h b/interface/FatJetsProducer.h
@@ -33,6 +33,10 @@ class FatJetsProducer: public CandidatesProducer<pat::Jet>, public BTaggingScale
                             &CandidatesProducer<pat::Jet>::tree["toptag_" + btag].write<std::vector<std::vector<float>>>());
                 }
             }
+            Njettinesstau1 = config.getUntrackedParameter<std::string>("Njettinesstau1","NjettinessAK8:tau1");
+            Njettinesstau2 = config.getUntrackedParameter<std::string>("Njettinesstau2","NjettinessAK8:tau2");
+            Njettinesstau3 = config.getUntrackedParameter<std::string>("Njettinesstau3","NjettinessAK8:tau3");
+
         }
 
         virtual ~FatJetsProducer() {}
@@ -57,6 +61,12 @@ class FatJetsProducer: public CandidatesProducer<pat::Jet>, public BTaggingScale
         std::vector<std::string> m_subjets_btag_discriminators;
         std::map<std::string, std::vector<std::vector<float>>*> m_softdrop_btag_discriminators_branches;
         std::map<std::string, std::vector<std::vector<float>>*> m_toptag_btag_discriminators_branches;
+
+        std::string Njettinesstau1;
+        std::string Njettinesstau2;
+        std::string Njettinesstau3;
+
+
     public:
         // Tree members
         std::vector<float>& area = tree["area"].write<std::vector<float>>();

diff --git a/src/FatJetsProducer.cc b/src/FatJetsProducer.cc
@@ -22,9 +22,9 @@ void FatJetsProducer::produce(edm::Event& event, const edm::EventSetup& eventSet
         passTightID.push_back(Tools::Jets::passTightId(jet));
         passTightLeptonVetoID.push_back(Tools::Jets::passTightLeptonVetoId(jet));
 
-        tau1.push_back(jet.userFloat("NjettinessAK8:tau1"));
-        tau2.push_back(jet.userFloat("NjettinessAK8:tau2"));
-        tau3.push_back(jet.userFloat("NjettinessAK8:tau3"));
+        tau1.push_back(jet.userFloat(Njettinesstau1));
+        tau2.push_back(jet.userFloat(Njettinesstau2));
+        tau3.push_back(jet.userFloat(Njettinesstau3));
 
         softdrop_mass.push_back(jet.userFloat("ak8PFJetsCHSSoftDropMass"));
         trimmed_mass.push_back(jet.userFloat("ak8PFJetsCHSTrimmedMass"));