HLTProducers/src/L2TauJetsMerger.cc

0001 #include "RecoTauTag/HLTProducers/interface/L2TauJetsMerger.h"
0002 #include "Math/GenVector/VectorUtil.h"
0003 #include "DataFormats/HLTReco/interface/TriggerTypeDefs.h"
0004 #include "FWCore/Utilities/interface/EDMException.h"
0005 //
0006 // class decleration
0007 //
0008 using namespace reco;
0009 using namespace std;
0010 using namespace edm;
0011
0012 L2TauJetsMerger::L2TauJetsMerger(const edm::ParameterSet& iConfig)
0013     : jetSrc(iConfig.getParameter<vtag>("JetSrc")), mEt_Min(iConfig.getParameter<double>("EtMin")) {
0014   for (vtag::const_iterator it = jetSrc.begin(); it != jetSrc.end(); ++it) {
0015     edm::EDGetTokenT<CaloJetCollection> aToken = consumes<CaloJetCollection>(*it);
0016     jetSrc_token.push_back(aToken);
0017   }
0018
0019   produces<CaloJetCollection>();
0020 }
0021
0022 L2TauJetsMerger::~L2TauJetsMerger() {}
0023
0024 void L2TauJetsMerger::produce(edm::StreamID iSId, edm::Event& iEvent, const edm::EventSetup& iES) const {
0025   using namespace edm;
0026   using namespace std;
0027   using namespace reco;
0028
0029   //Getting the Collections of L2ReconstructedJets from L1Seeds
0030   //and removing the collinear jets
0031   CaloJetCollection myTmpJets;
0032
0033   int iL1Jet = 0;
0034   for (vtoken_cjets::const_iterator s = jetSrc_token.begin(); s != jetSrc_token.end(); ++s) {
0035     edm::Handle<CaloJetCollection> tauJets;
0036     iEvent.getByToken(*s, tauJets);
0037     for (CaloJetCollection::const_iterator iTau = tauJets->begin(); iTau != tauJets->end(); ++iTau) {
0038       if (iTau->et() > mEt_Min) {
0039         //Add the Pdg Id here
0040         CaloJet myJet = *iTau;
0041         myJet.setPdgId(15);
0042         myTmpJets.push_back(myJet);
0043       }
0044     }
0045     iL1Jet++;
0046   }
0047
0048   std::unique_ptr<CaloJetCollection> tauL2jets(new CaloJetCollection);
0049
0050   //Removing the collinear jets correctly!
0051
0052   //First sort the jets you have merged
0053   SorterByPt sorter;
0054   std::sort(myTmpJets.begin(), myTmpJets.end(), sorter);
0055
0056   //Remove Collinear Jets by prefering the highest ones!
0057   while (!myTmpJets.empty()) {
0058     tauL2jets->push_back(myTmpJets[0]);
0059     CaloJetCollection tmp;
0060     for (unsigned int i = 1; i < myTmpJets.size(); ++i) {
0061       double DR = ROOT::Math::VectorUtil::DeltaR(myTmpJets[0].p4(), myTmpJets[i].p4());
0062       if (DR > 0.1)
0063         tmp.push_back(myTmpJets[i]);
0064     }
0065     myTmpJets.swap(tmp);
0066   }
0067
0068   iEvent.put(std::move(tauL2jets));
0069 }
0070
0071 void L2TauJetsMerger::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
0072   edm::ParameterSetDescription desc;
0073   std::vector<edm::InputTag> inputTags;
0074   inputTags.push_back(edm::InputTag("hltAkIsoTau1Regional"));
0075   inputTags.push_back(edm::InputTag("hltAkIsoTau2Regional"));
0076   inputTags.push_back(edm::InputTag("hltAkIsoTau3Regional"));
0077   inputTags.push_back(edm::InputTag("hltAkIsoTau4Regional"));
0078   desc.add<std::vector<edm::InputTag> >("JetSrc", inputTags)->setComment("CaloJet collections to merge");
0079   desc.add<double>("EtMin", 20.0)->setComment("Minimal ET of jet to merge");
0080   descriptions.setComment("Merges CaloJet collections removing duplicates");
0081   descriptions.add("L2TauJetsMerger", desc);
0082 }