L1TMuonBarrelKalmanStubProducer.cc

Go to the documentation of this file.

#include <memory>
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/StreamID.h"
 
#include "DataFormats/L1DTTrackFinder/interface/L1MuDTChambPhDigi.h"
#include "DataFormats/L1DTTrackFinder/interface/L1MuDTChambPhContainer.h"
#include "DataFormats/L1DTTrackFinder/interface/L1MuDTChambThDigi.h"
#include "DataFormats/L1DTTrackFinder/interface/L1MuDTChambThContainer.h"
 
#include "L1Trigger/L1TMuonBarrel/interface/L1TMuonBarrelKalmanStubProcessor.h"
 
//For masks
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "CondFormats/L1TObjects/interface/L1TMuonBarrelParams.h"
#include "CondFormats/DataRecord/interface/L1TMuonBarrelParamsRcd.h"
#include "CondFormats/L1TObjects/interface/L1MuDTTFMasks.h"
#include "CondFormats/DataRecord/interface/L1MuDTTFMasksRcd.h"
 
//
// class declaration
//
 
class L1TMuonBarrelKalmanStubProducer : public edm::stream::EDProducer<> {
public:
  explicit L1TMuonBarrelKalmanStubProducer(const edm::ParameterSet&);
  ~L1TMuonBarrelKalmanStubProducer() override;
 
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
private:
  void beginStream(edm::StreamID) override;
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endStream() override;
  edm::EDGetTokenT<L1MuDTChambPhContainer> srcPhi_;
  edm::EDGetTokenT<L1MuDTChambThContainer> srcTheta_;
  L1TMuonBarrelKalmanStubProcessor* proc_;
  int verbose_;
  edm::ESHandle<L1TMuonBarrelParams> bmtfParamsHandle_;
};
 
//
// constants, enums and typedefs
//
 
//
// static data member definitions
//
 
//
// constructors and destructor
//
L1TMuonBarrelKalmanStubProducer::L1TMuonBarrelKalmanStubProducer(const edm::ParameterSet& iConfig)
    : srcPhi_(consumes<L1MuDTChambPhContainer>(iConfig.getParameter<edm::InputTag>("srcPhi"))),
      srcTheta_(consumes<L1MuDTChambThContainer>(iConfig.getParameter<edm::InputTag>("srcTheta"))),
      proc_(new L1TMuonBarrelKalmanStubProcessor(iConfig)),
      verbose_(iConfig.getParameter<int>("verbose")) {
  produces<L1MuKBMTCombinedStubCollection>();
}
 
L1TMuonBarrelKalmanStubProducer::~L1TMuonBarrelKalmanStubProducer() {
  // do anything here that needs to be done at destruction time
  // (e.g. close files, deallocate resources etc.)
  if (proc_ != nullptr)
    delete proc_;
}
 
//
// member functions
//
 
// ------------ method called to produce the data  ------------
void L1TMuonBarrelKalmanStubProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  Handle<L1MuDTChambPhContainer> phiIn;
  iEvent.getByToken(srcPhi_, phiIn);
 
  Handle<L1MuDTChambThContainer> thetaIn;
  iEvent.getByToken(srcTheta_, thetaIn);
 
  //Get parameters
 
  const L1TMuonBarrelParamsRcd& bmtfParamsRcd = iSetup.get<L1TMuonBarrelParamsRcd>();
  bmtfParamsRcd.get(bmtfParamsHandle_);
  const L1TMuonBarrelParams& bmtfParams = *bmtfParamsHandle_.product();
 
  L1MuKBMTCombinedStubCollection stubs = proc_->makeStubs(phiIn.product(), thetaIn.product(), bmtfParams);
  if (verbose_ == 1)
    for (const auto& stub : stubs) {
      printf("Stub: wheel=%d sector=%d station =%d tag=%d eta1=%d qeta1=%d eta2=%d qeta2=%d\n",
             stub.whNum(),
             stub.scNum(),
             stub.stNum(),
             stub.tag(),
             stub.eta1(),
             stub.qeta1(),
             stub.eta2(),
             stub.qeta2());
    }
 
  if (verbose_ == 2) {
    std::cout << "NEW" << std::endl;
    for (uint sector = 0; sector < 12; ++sector)
      proc_->makeInputPattern(phiIn.product(), thetaIn.product(), sector);
  }
 
  iEvent.put(std::make_unique<L1MuKBMTCombinedStubCollection>(stubs));
}
 
// ------------ method called once each stream before processing any runs, lumis or events  ------------
void L1TMuonBarrelKalmanStubProducer::beginStream(edm::StreamID) {}
 
// ------------ method called once each stream after processing all runs, lumis and events  ------------
void L1TMuonBarrelKalmanStubProducer::endStream() {}
 
void L1TMuonBarrelKalmanStubProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}
 
//define this as a plug-in
DEFINE_FWK_MODULE(L1TMuonBarrelKalmanStubProducer);