HLTScoutingPFProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    HLTrigger/JetMET
// Class:      HLTScoutingPFProducer
// 
//
// Original Author:  Dustin James Anderson
//         Created:  Fri, 12 Jun 2015 15:49:20 GMT
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BTauReco/interface/JetTag.h"

#include "DataFormats/Scouting/interface/ScoutingPFJet.h"
#include "DataFormats/Scouting/interface/ScoutingParticle.h"
#include "DataFormats/Scouting/interface/ScoutingVertex.h"

#include "DataFormats/Math/interface/deltaR.h"

class HLTScoutingPFProducer : public edm::global::EDProducer<> {
    public:
        explicit HLTScoutingPFProducer(const edm::ParameterSet&);
        ~HLTScoutingPFProducer() override;

        static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

    private:
        void produce(edm::StreamID sid, edm::Event & iEvent, edm::EventSetup const & setup) const final;

        const edm::EDGetTokenT<reco::PFJetCollection> pfJetCollection_;
        const edm::EDGetTokenT<reco::JetTagCollection> pfJetTagCollection_;
        const edm::EDGetTokenT<reco::PFCandidateCollection> pfCandidateCollection_;
        const edm::EDGetTokenT<reco::VertexCollection> vertexCollection_;
        const edm::EDGetTokenT<reco::PFMETCollection> metCollection_;
        const edm::EDGetTokenT<double> rho_;

        const double pfJetPtCut;
        const double pfJetEtaCut;
        const double pfCandidatePtCut;

        const bool doJetTags;
        const bool doCandidates;
        const bool doMet;
};

//
// constructors and destructor
//
HLTScoutingPFProducer::HLTScoutingPFProducer(const edm::ParameterSet& iConfig):
    pfJetCollection_(consumes<reco::PFJetCollection>(iConfig.getParameter<edm::InputTag>("pfJetCollection"))),
    pfJetTagCollection_(consumes<reco::JetTagCollection>(iConfig.getParameter<edm::InputTag>("pfJetTagCollection"))),
    pfCandidateCollection_(consumes<reco::PFCandidateCollection>(iConfig.getParameter<edm::InputTag>("pfCandidateCollection"))),
    vertexCollection_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexCollection"))),
    metCollection_(consumes<reco::PFMETCollection>(iConfig.getParameter<edm::InputTag>("metCollection"))),
    rho_(consumes<double>(iConfig.getParameter<edm::InputTag>("rho"))),
    pfJetPtCut(iConfig.getParameter<double>("pfJetPtCut")),
    pfJetEtaCut(iConfig.getParameter<double>("pfJetEtaCut")),
    pfCandidatePtCut(iConfig.getParameter<double>("pfCandidatePtCut")),
    doJetTags(iConfig.getParameter<bool>("doJetTags")),
    doCandidates(iConfig.getParameter<bool>("doCandidates")),
    doMet(iConfig.getParameter<bool>("doMet"))
{
    //register products
    produces<ScoutingPFJetCollection>();
    produces<ScoutingParticleCollection>();
    produces<double>("rho");
    produces<double>("pfMetPt");
    produces<double>("pfMetPhi");
}

HLTScoutingPFProducer::~HLTScoutingPFProducer() = default;

// ------------ method called to produce the data  ------------
void HLTScoutingPFProducer::produce(edm::StreamID sid, edm::Event & iEvent, edm::EventSetup const & setup) const
{
    using namespace edm;

    //get vertices
     Handle<reco::VertexCollection> vertexCollection;
     std::unique_ptr<ScoutingVertexCollection> outVertices(new ScoutingVertexCollection());
     if(iEvent.getByToken(vertexCollection_, vertexCollection)){
        for(auto &vtx : *vertexCollection){
            outVertices->emplace_back(
                          vtx.x(), vtx.y(), vtx.z(), vtx.zError(), vtx.xError(), vtx.yError(), vtx.tracksSize(), vtx.chi2(), vtx.ndof(), vtx.isValid()
                    );
        }
      }

    //get rho
    Handle<double> rho;
    std::unique_ptr<double> outRho(new double(-999));
    if(iEvent.getByToken(rho_, rho)){
        outRho.reset(new double(*rho));
    }

    //get MET
    Handle<reco::PFMETCollection> metCollection;
    std::unique_ptr<double> outMetPt(new double(-999));
    std::unique_ptr<double> outMetPhi(new double(-999));
    if(doMet && iEvent.getByToken(metCollection_, metCollection)){
        outMetPt.reset(new double(metCollection->front().pt()));
        outMetPhi.reset(new double(metCollection->front().phi()));
    }

    //get PF candidates
    Handle<reco::PFCandidateCollection> pfCandidateCollection;
    std::unique_ptr<ScoutingParticleCollection> outPFCandidates(new ScoutingParticleCollection());
    if(doCandidates && iEvent.getByToken(pfCandidateCollection_, pfCandidateCollection)){
        for(auto &cand : *pfCandidateCollection){
            if(cand.pt() > pfCandidatePtCut){
                int vertex_index = -1;
                int index_counter = 0;
                double dr2 = 0.0001;
                for (auto &vtx: *outVertices) {
                    double tmp_dr2 = pow(vtx.x() - cand.vx(), 2) + pow(vtx.y() - cand.vy(), 2)
                        + pow(vtx.z() - cand.vz(), 2);
                    if (tmp_dr2 < dr2) {
                        dr2 = tmp_dr2;
                        vertex_index = index_counter;
                    }
                    if (dr2 == 0.0)
                        break;
                    ++index_counter;
                }
                outPFCandidates->emplace_back(
                        cand.pt(), cand.eta(), cand.phi(), cand.mass(), cand.pdgId(), vertex_index
                        );
            }
        }
    }

    //get PF jets
    Handle<reco::PFJetCollection> pfJetCollection;
    std::unique_ptr<ScoutingPFJetCollection> outPFJets(new ScoutingPFJetCollection());
    if(iEvent.getByToken(pfJetCollection_, pfJetCollection)){
        //get PF jet tags
        Handle<reco::JetTagCollection> pfJetTagCollection;
        bool haveJetTags = false;
        if(doJetTags && iEvent.getByToken(pfJetTagCollection_, pfJetTagCollection)){
            haveJetTags = true;
        }

        for(auto &jet : *pfJetCollection){
            if(jet.pt() < pfJetPtCut || fabs(jet.eta()) > pfJetEtaCut) continue;
            //find the jet tag corresponding to the jet
            float tagValue = -20;
            float minDR2 = 0.01;
            if(haveJetTags){
                for(auto &tag : *pfJetTagCollection){
                    float dR2 = reco::deltaR2(jet, *(tag.first));
                    if(dR2 < minDR2){
                        minDR2 = dR2;
                        tagValue = tag.second;
                    }
                }
            }
            //get the PF constituents of the jet
            std::vector<int> candIndices;
            if(doCandidates){
                for(auto &cand : jet.getPFConstituents()){
                    if(cand->pt() > pfCandidatePtCut){
                        //search for the candidate in the collection
                        float minDR2 = 0.0001;
                        int matchIndex = -1;
                        int outIndex = 0;
                        for(auto &outCand : *outPFCandidates){
                            float dR2 = pow(cand->eta() - outCand.eta(), 2) + pow(cand->phi() - outCand.phi(), 2);
                            if(dR2 < minDR2){
                                minDR2 = dR2;
                                matchIndex = outIndex;
                            }
                            if(minDR2 == 0){
                                break;
                            }
                            outIndex++;
                        }
                        candIndices.push_back(matchIndex);
                    }
                }
            }
            outPFJets->emplace_back(
                    jet.pt(), jet.eta(), jet.phi(), jet.mass(), jet.jetArea(),
                    jet.chargedHadronEnergy(), jet.neutralHadronEnergy(), jet.photonEnergy(),
                    jet.electronEnergy(), jet.muonEnergy(), jet.HFHadronEnergy(), jet.HFEMEnergy(),
                    jet.chargedHadronMultiplicity(), jet.neutralHadronMultiplicity(), jet.photonMultiplicity(),
                    jet.electronMultiplicity(), jet.muonMultiplicity(), 
                    jet.HFHadronMultiplicity(), jet.HFEMMultiplicity(), 
                    jet.hoEnergy(), tagValue, 0.0, candIndices
                    );
        }
    }

    //put output
    iEvent.put(std::move(outPFCandidates));
    iEvent.put(std::move(outPFJets));
    iEvent.put(std::move(outRho), "rho");
    iEvent.put(std::move(outMetPt), "pfMetPt");
    iEvent.put(std::move(outMetPhi), "pfMetPhi");
}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void HLTScoutingPFProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.add<edm::InputTag>("pfJetCollection",edm::InputTag("hltAK4PFJets"));
    desc.add<edm::InputTag>("pfJetTagCollection",edm::InputTag("hltCombinedSecondaryVertexBJetTagsPF"));
    desc.add<edm::InputTag>("pfCandidateCollection", edm::InputTag("hltParticleFlow"));
    desc.add<edm::InputTag>("vertexCollection", edm::InputTag("hltPixelVertices"));
    desc.add<edm::InputTag>("metCollection", edm::InputTag("hltPFMETProducer"));
    desc.add<edm::InputTag>("rho", edm::InputTag("hltFixedGridRhoFastjetAll"));
    desc.add<double>("pfJetPtCut", 20.0);
    desc.add<double>("pfJetEtaCut", 3.0);
    desc.add<double>("pfCandidatePtCut", 0.6);
    desc.add<bool>("doJetTags", true);
    desc.add<bool>("doCandidates", true);
    desc.add<bool>("doMet", true);
    descriptions.add("hltScoutingPFProducer", desc);
}

// declare this class as a framework plugin
DEFINE_FWK_MODULE(HLTScoutingPFProducer);