HLTCaloTowerHtMhtProducer.cc

Go to the documentation of this file.

#include "HLTrigger/JetMET/interface/HLTCaloTowerHtMhtProducer.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "DataFormats/Common/interface/Handle.h"


// Constructor
HLTCaloTowerHtMhtProducer::HLTCaloTowerHtMhtProducer(const edm::ParameterSet & iConfig) :
  usePt_                  ( iConfig.getParameter<bool>("usePt") ),
  minPtTowerHt_             ( iConfig.getParameter<double>("minPtTowerHt") ),
  minPtTowerMht_            ( iConfig.getParameter<double>("minPtTowerMht") ),
  maxEtaTowerHt_            ( iConfig.getParameter<double>("maxEtaTowerHt") ),
  maxEtaTowerMht_           ( iConfig.getParameter<double>("maxEtaTowerMht") ),
  towersLabel_              ( iConfig.getParameter<edm::InputTag>("towersLabel") ) {
    m_theTowersToken = consumes<CaloTowerCollection>(towersLabel_);
    
    // Register the products
    produces<reco::METCollection>();
}

// Destructor
HLTCaloTowerHtMhtProducer::~HLTCaloTowerHtMhtProducer() = default;

// Fill descriptions
void HLTCaloTowerHtMhtProducer::fillDescriptions(edm::ConfigurationDescriptions & descriptions) {
    // Current default is for hltHtMht
    edm::ParameterSetDescription desc;
    desc.add<bool>("usePt", false);
    desc.add<double>("minPtTowerHt", 1.);
    desc.add<double>("minPtTowerMht", 1.);
    desc.add<double>("maxEtaTowerHt", 5.);
    desc.add<double>("maxEtaTowerMht", 5.);
    desc.add<edm::InputTag>("towersLabel", edm::InputTag("hltTowerMakerForAll"));
    descriptions.add("hltCaloTowerHtMhtProducer", desc);
}

// Produce the products
void HLTCaloTowerHtMhtProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {

    // Create a pointer to the products
    std::unique_ptr<reco::METCollection> result(new reco::METCollection());

    edm::Handle<CaloTowerCollection> towers;
    iEvent.getByToken(m_theTowersToken, towers);

    double ht = 0., mhx = 0., mhy = 0.;

    if (towers->size() > 0) {
        for(auto const & j : *towers) {
            double pt = usePt_ ? j.pt() : j.et();
            double eta = j.eta();
            double phi = j.phi();
            double px = usePt_ ? j.px() : j.et() * cos(phi);
            double py = usePt_ ? j.py() : j.et() * sin(phi);

            if (pt > minPtTowerHt_ && std::abs(eta) < maxEtaTowerHt_) {
                ht += pt;
            }

            if (pt > minPtTowerMht_ && std::abs(eta) < maxEtaTowerMht_) {
                mhx -= px;
                mhy -= py;
            }
        }
    }

    reco::MET::LorentzVector p4(mhx, mhy, 0, sqrt(mhx*mhx + mhy*mhy));
    reco::MET::Point vtx(0, 0, 0);
    reco::MET htmht(ht, p4, vtx);
    result->push_back(htmht);
    
    // Put the products into the Event
    iEvent.put(std::move(result));
}