TestBXVectorRefProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    HLTrigger/TestBXVectorRefProducer
// Class:      TestBXVectorRefProducer
//
//
// Original Author:  Vladimir Rekovic
//         Created:  Fri, 12 Feb 2016 09:56:04 GMT
//
//
 
// system include files
#include <memory>
 
// user include files
#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/L1Trigger/interface/BXVector.h"
#include "DataFormats/L1Trigger/interface/Jet.h"
 
//
// class declaration
//
 
class TestBXVectorRefProducer : public edm::stream::EDProducer<> {
public:
  explicit TestBXVectorRefProducer(const edm::ParameterSet&);
  ~TestBXVectorRefProducer() override;
 
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
private:
  void beginStream(edm::StreamID) override;
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endStream() override;
 
  //virtual void beginRun(edm::Run const&, edm::EventSetup const&) override;
  //virtual void endRun(edm::Run const&, edm::EventSetup const&) override;
  //virtual void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;
  //virtual void endLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;
 
  // ----------member data ---------------------------
  bool doRefs_;
  edm::InputTag src_;
  edm::EDGetTokenT<l1t::JetBxCollection> token_;
};
 
//
// constants, enums and typedefs
//
 
//
// static data member definitions
//
 
//
// constructors and destructor
//
TestBXVectorRefProducer::TestBXVectorRefProducer(const edm::ParameterSet& iConfig) {
  //now do what ever other initialization is needed
  src_ = iConfig.getParameter<edm::InputTag>("src");
  doRefs_ = iConfig.getParameter<bool>("doRefs");
  token_ = consumes<l1t::JetBxCollection>(src_);
 
  //register your products
  produces<vector<int>>("jetPt").setBranchAlias("jetPt");
 
  if (doRefs_) {
    produces<l1t::JetRefVector>("l1tJetRef").setBranchAlias("l1tJetRef");
  }
}
 
TestBXVectorRefProducer::~TestBXVectorRefProducer() {
  // do anything here that needs to be done at destruction time
  // (e.g. close files, deallocate resources etc.)
}
 
//
// member functions
//
 
// ------------ method called to produce the data  ------------
void TestBXVectorRefProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
 
  unique_ptr<vector<int>> jetMom(new vector<int>);
  unique_ptr<l1t::JetRefVector> jetRef(new l1t::JetRefVector);
 
  // retrieve the tracks
  Handle<l1t::JetBxCollection> jets;
  iEvent.getByToken(token_, jets);
  if (!jets.isValid())
    return;
 
  const int size = jets->size();
  jetMom->reserve(size);
 
  l1t::JetBxCollection::const_iterator iter;
 
  for (iter = jets->begin(0); iter != jets->end(0); ++iter) {
    jetMom->push_back(iter->pt());
 
    l1t::JetRef myref(jets, jets->key(iter));
    jetRef->push_back(myref);
 
  }  // end for
 
  iEvent.put(std::move(jetMom), "jetPt");
 
  if (doRefs_)
    iEvent.put(std::move(jetRef), "l1tJetRef");
 
  return;
}
 
// ------------ method called once each stream before processing any runs, lumis or events  ------------
void TestBXVectorRefProducer::beginStream(edm::StreamID) {}
 
// ------------ method called once each stream after processing all runs, lumis and events  ------------
void TestBXVectorRefProducer::endStream() {}
 
// ------------ method called when starting to processes a run  ------------
/*
void
TestBXVectorRefProducer::beginRun(edm::Run const&, edm::EventSetup const&)
{
}
*/
 
// ------------ method called when ending the processing of a run  ------------
/*
void
TestBXVectorRefProducer::endRun(edm::Run const&, edm::EventSetup const&)
{
}
*/
 
// ------------ method called when starting to processes a luminosity block  ------------
/*
void
TestBXVectorRefProducer::beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&)
{
}
*/
 
// ------------ method called when ending the processing of a luminosity block  ------------
/*
void
TestBXVectorRefProducer::endLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&)
{
}
*/
 
// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void TestBXVectorRefProducer::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);
}
 
// declare this class as a framework plugin
DEFINE_FWK_MODULE(TestBXVectorRefProducer);