HLTL1TMuonSelector.cc

Go to the documentation of this file.

//-------------------------------------------------
//
//
//--------------------------------------------------

// Class Header
#include "HLTL1TMuonSelector.h"

// Framework
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"

using namespace std;
using namespace edm;
using namespace l1t;

// constructors
HLTL1TMuonSelector::HLTL1TMuonSelector(const edm::ParameterSet& iConfig)
    : theSource(iConfig.getParameter<InputTag>("InputObjects")),
      theL1MinPt(iConfig.getParameter<double>("L1MinPt")),
      theL1MaxEta(iConfig.getParameter<double>("L1MaxEta")),
      theL1MinQuality(iConfig.getParameter<unsigned int>("L1MinQuality")),
      centralBxOnly_(iConfig.getParameter<bool>("CentralBxOnly")) {
  muCollToken_ = consumes<MuonBxCollection>(theSource);

  produces<MuonBxCollection>();
}

// destructor
HLTL1TMuonSelector::~HLTL1TMuonSelector() = default;

void HLTL1TMuonSelector::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("InputObjects", edm::InputTag("hltGmtStage2Digis"));
  desc.add<double>("L1MinPt", -1.);
  desc.add<double>("L1MaxEta", 5.0);
  desc.add<unsigned int>("L1MinQuality", 0);
  desc.add<bool>("CentralBxOnly", true);
  descriptions.add("hltL1TMuonSelector", desc);
}

void HLTL1TMuonSelector::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  const std::string metname = "Muon|RecoMuon|HLTL1TMuonSelector";

  unique_ptr<MuonBxCollection> output(new MuonBxCollection());

  // Muon particles
  edm::Handle<MuonBxCollection> muColl;
  iEvent.getByToken(muCollToken_, muColl);
  LogTrace(metname) << "Number of muons " << muColl->size() << endl;

  for (int ibx = muColl->getFirstBX(); ibx <= muColl->getLastBX(); ++ibx) {
    if (centralBxOnly_ && (ibx != 0))
      continue;
    for (auto it = muColl->begin(ibx); it != muColl->end(ibx); it++) {
      unsigned int quality = it->hwQual();
      int valid_charge = it->hwChargeValid();

      float pt = it->pt();
      float eta = it->eta();
      float theta = 2 * atan(exp(-eta));
      float phi = it->phi();
      int charge = it->charge();
      // Set charge=0 for the time being if the valid charge bit is zero
      if (!valid_charge)
        charge = 0;

      if (pt < theL1MinPt || fabs(eta) > theL1MaxEta)
        continue;

      LogTrace(metname) << "L1 Muon Found";
      LogTrace(metname) << "Pt = " << pt << " GeV/c";
      LogTrace(metname) << "eta = " << eta;
      LogTrace(metname) << "theta = " << theta << " rad";
      LogTrace(metname) << "phi = " << phi << " rad";
      LogTrace(metname) << "charge = " << charge;

      if (quality <= theL1MinQuality)
        continue;
      LogTrace(metname) << "quality = " << quality;

      output->push_back(ibx, *it);
    }
  }

  iEvent.put(std::move(output));
}

// declare this class as a framework plugin
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(HLTL1TMuonSelector);