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();

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

   private:
      virtual void beginStream(edm::StreamID) override;
      virtual void produce(edm::Event&, const edm::EventSetup&) override;
      virtual 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;

   auto_ptr<vector<int>> jetMom ( new vector<int> );
   auto_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(jetMom,"jetPt");
   
   if (doRefs_) iEvent.put(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);
}

//define this as a plug-in
DEFINE_FWK_MODULE(TestBXVectorRefProducer);