HLTJetCollectionsForBoostedLeptonPlusJets.cc

Go to the documentation of this file.

#include <string>
#include <vector>

#include "TVector3.h"

#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/RecoCandidate/interface/RecoEcalCandidate.h"
#include "DataFormats/EgammaCandidates/interface/Electron.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "HLTrigger/JetMET/interface/HLTJetCollectionsForBoostedLeptonPlusJets.h"
#include "HLTrigger/HLTcore/interface/defaultModuleLabel.h"
#include "CommonTools/Utils/interface/PtComparator.h"

template <typename jetType>
HLTJetCollectionsForBoostedLeptonPlusJets<jetType>::HLTJetCollectionsForBoostedLeptonPlusJets(
    const edm::ParameterSet& iConfig)
    : hltLeptonTag(iConfig.getParameter<edm::InputTag>("HltLeptonTag")),
      sourceJetTag(iConfig.getParameter<edm::InputTag>("SourceJetTag")),
      minDeltaR_(iConfig.getParameter<double>("minDeltaR")) {
  using namespace edm;
  using namespace std;

  typedef vector<RefVector<vector<jetType>, jetType, refhelper::FindUsingAdvance<vector<jetType>, jetType>>>
      JetCollectionVector;
  typedef vector<jetType> JetCollection;

  m_theLeptonToken = consumes<trigger::TriggerFilterObjectWithRefs>(hltLeptonTag);
  m_theJetToken = consumes<std::vector<jetType>>(sourceJetTag);
  produces<JetCollectionVector>();
  produces<JetCollection>();
}

template <typename jetType>
HLTJetCollectionsForBoostedLeptonPlusJets<jetType>::~HLTJetCollectionsForBoostedLeptonPlusJets() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

template <typename jetType>
void HLTJetCollectionsForBoostedLeptonPlusJets<jetType>::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("HltLeptonTag", edm::InputTag("triggerFilterObjectWithRefs"));
  //(2)
  desc.add<edm::InputTag>("SourceJetTag", edm::InputTag("jetCollection"));
  //(2)
  desc.add<double>("minDeltaR", 0.5);
  descriptions.add(defaultModuleLabel<HLTJetCollectionsForBoostedLeptonPlusJets<jetType>>(), desc);
}

//
// member functions
//

// ------------ method called to produce the data  ------------
// template <typename T>
template <typename jetType>
void HLTJetCollectionsForBoostedLeptonPlusJets<jetType>::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  using namespace std;
  //(3)
  using namespace reco;
  //(3)
  typedef vector<RefVector<vector<jetType>, jetType, refhelper::FindUsingAdvance<vector<jetType>, jetType>>>
      JetCollectionVector;
  typedef vector<jetType> JetCollection;
  typedef edm::Ref<JetCollection> JetRef;
  typedef edm::RefVector<JetCollection> JetRefVector;

  Handle<trigger::TriggerFilterObjectWithRefs> PrevFilterOutput;
  iEvent.getByToken(m_theLeptonToken, PrevFilterOutput);

  //its easier on the if statement flow if I try everything at once, shouldnt add to timing
  // Electrons can be stored as objects of types TriggerCluster, TriggerElectron, or TriggerPhoton
  vector<reco::RecoChargedCandidateRef> muonCands;
  PrevFilterOutput->getObjects(trigger::TriggerMuon, muonCands);

  vector<Ref<reco::ElectronCollection>> eleCands;
  PrevFilterOutput->getObjects(trigger::TriggerElectron, eleCands);

  trigger::VRphoton photonCands;
  PrevFilterOutput->getObjects(trigger::TriggerPhoton, photonCands);

  vector<Ref<reco::RecoEcalCandidateCollection>> clusCands;
  PrevFilterOutput->getObjects(trigger::TriggerCluster, clusCands);

  Handle<JetCollection> theJetCollectionHandle;
  iEvent.getByToken(m_theJetToken, theJetCollectionHandle);

  typename JetCollection::const_iterator jet;

  unique_ptr<JetCollection> allSelections(new JetCollection);
  unique_ptr<JetCollectionVector> product(new JetCollectionVector);

  std::vector<size_t> usedCands;

  if (!muonCands.empty()) {  // muons
    for (jet = theJetCollectionHandle->begin(); jet != theJetCollectionHandle->end(); jet++) {
      //const jetType* referenceJet = &*jet;
      jetType cleanedJet = *jet;  //copy original jet
      for (size_t candNr = 0; candNr < muonCands.size(); candNr++) {
        if (std::find(usedCands.begin(), usedCands.end(), candNr) != usedCands.end())
          continue;
        if (deltaR((*muonCands[candNr]), cleanedJet) <= minDeltaR_) {
          std::vector<edm::Ptr<reco::PFCandidate>> pfConstituents = cleanedJet.getPFConstituents();
          for (std::vector<edm::Ptr<reco::PFCandidate>>::const_iterator i_candidate = pfConstituents.begin();
               i_candidate != pfConstituents.end();
               ++i_candidate) {
            if (deltaR((*muonCands[candNr]), (**i_candidate)) < 0.001) {
              cleanedJet.setP4(cleanedJet.p4() - muonCands[candNr]->p4());
              usedCands.push_back(candNr);
              break;
            }  //if constituent matched
          }    //for constituents
        }      //if dR<min
      }        //for cands
      allSelections->push_back(cleanedJet);
    }  //for jets
  }    //if cands

  if (!eleCands.empty()) {  // electrons
    for (jet = theJetCollectionHandle->begin(); jet != theJetCollectionHandle->end(); jet++) {
      //const jetType* referenceJet = &*jet;
      jetType cleanedJet = *jet;  //copy original jet
      for (size_t candNr = 0; candNr < eleCands.size(); candNr++) {
        if (std::find(usedCands.begin(), usedCands.end(), candNr) != usedCands.end())
          continue;
        if (deltaR((*eleCands[candNr]), cleanedJet) <= minDeltaR_) {
          std::vector<edm::Ptr<reco::PFCandidate>> pfConstituents = cleanedJet.getPFConstituents();
          for (std::vector<edm::Ptr<reco::PFCandidate>>::const_iterator i_candidate = pfConstituents.begin();
               i_candidate != pfConstituents.end();
               ++i_candidate) {
            if (deltaR((*eleCands[candNr]), (**i_candidate)) < 0.001) {
              cleanedJet.setP4(cleanedJet.p4() - eleCands[candNr]->p4());
              usedCands.push_back(candNr);
              break;
            }  //if constituent matched
          }    //for constituents
        }      //if dR<min
      }        //for cands
      allSelections->push_back(cleanedJet);
    }  //for jets
  }    //if cands

  if (!photonCands.empty()) {  // photons
    for (jet = theJetCollectionHandle->begin(); jet != theJetCollectionHandle->end(); jet++) {
      //const jetType* referenceJet = &*jet;
      jetType cleanedJet = *jet;  //copy original jet
      for (size_t candNr = 0; candNr < photonCands.size(); candNr++) {
        if (std::find(usedCands.begin(), usedCands.end(), candNr) != usedCands.end())
          continue;
        if (deltaR((*photonCands[candNr]), cleanedJet) <= minDeltaR_) {
          std::vector<edm::Ptr<reco::PFCandidate>> pfConstituents = cleanedJet.getPFConstituents();
          for (std::vector<edm::Ptr<reco::PFCandidate>>::const_iterator i_candidate = pfConstituents.begin();
               i_candidate != pfConstituents.end();
               ++i_candidate) {
            if (deltaR((*photonCands[candNr]), (**i_candidate)) < 0.001) {
              cleanedJet.setP4(cleanedJet.p4() - photonCands[candNr]->p4());
              usedCands.push_back(candNr);
              break;
            }  //if constituent matched
          }    //for constituents
        }      //if dR<min
      }        //for cands
      allSelections->push_back(cleanedJet);
    }  //for jets
  }    //if cands

  if (!clusCands.empty()) {  // trigger clusters
    for (jet = theJetCollectionHandle->begin(); jet != theJetCollectionHandle->end(); jet++) {
      //const jetType* referenceJet = &*jet;
      jetType cleanedJet = *jet;  //copy original jet
      for (size_t candNr = 0; candNr < clusCands.size(); candNr++) {
        if (std::find(usedCands.begin(), usedCands.end(), candNr) != usedCands.end())
          continue;
        if (deltaR((*clusCands[candNr]), cleanedJet) <= minDeltaR_) {
          std::vector<edm::Ptr<reco::PFCandidate>> pfConstituents = cleanedJet.getPFConstituents();
          for (std::vector<edm::Ptr<reco::PFCandidate>>::const_iterator i_candidate = pfConstituents.begin();
               i_candidate != pfConstituents.end();
               ++i_candidate) {
            if (deltaR((*clusCands[candNr]), (**i_candidate)) < 0.001) {
              cleanedJet.setP4(cleanedJet.p4() - clusCands[candNr]->p4());
              usedCands.push_back(candNr);
              break;
            }  //if constituent matched
          }    //for constituents
        }      //if dR<min
      }        //for cands
      allSelections->push_back(cleanedJet);
    }  //for jets
  }    //if cands

  NumericSafeGreaterByPt<jetType> compJets;
  // reorder cleaned jets
  std::sort(allSelections->begin(), allSelections->end(), compJets);
  edm::OrphanHandle<JetCollection> cleanedJetHandle = iEvent.put(std::move(allSelections));

  JetCollection const& jets = *cleanedJetHandle;

  JetRefVector cleanedJetRefs;

  for (unsigned iJet = 0; iJet < jets.size(); ++iJet) {
    cleanedJetRefs.push_back(JetRef(cleanedJetHandle, iJet));
  }

  product->emplace_back(cleanedJetRefs);
  iEvent.put(std::move(product));

  return;
}