HLTCollectionProducer.h

Go to the documentation of this file.

#ifndef HLTCollectionProducer_h
#define HLTCollectionProducer_h

//
// Package:    HLTCollectionProducer
// Class:      HLTCollectionProducer
// 
//
// Original Author:  Ian Tomalin
//

#include <string>
#include <vector>

// user include files
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/RefVector.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
#include "HLTrigger/HLTcore/interface/defaultModuleLabel.h"

//
// class declaration
//

template <typename T> 
class HLTCollectionProducer : public edm::global::EDProducer<> {

  public:
    explicit HLTCollectionProducer(const edm::ParameterSet&);
    virtual ~HLTCollectionProducer();
    static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);
    virtual void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
  
  private:
    const edm::InputTag                                          hltObjectTag_;
    const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> hltObjectToken_;
    const std::vector<int> triggerTypes_;
};


//
// constructors and destructor
//
template <typename T>
HLTCollectionProducer<T>::HLTCollectionProducer(const edm::ParameterSet& iConfig) :
  hltObjectTag_ ( iConfig.getParameter<edm::InputTag>("HLTObject") ),
  hltObjectToken_(consumes<trigger::TriggerFilterObjectWithRefs>(hltObjectTag_)),
  triggerTypes_ ( iConfig.getParameter<std::vector<int> >("TriggerTypes") )
{
  produces< std::vector<T> >();
}

template <typename T>
HLTCollectionProducer<T>::~HLTCollectionProducer() {}

template<typename T>
void 
HLTCollectionProducer<T>::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag> ("HLTObject", edm::InputTag("TriggerFilterObjectWithRefs"));
  std::vector<int> triggerTypes;
  desc.add<std::vector<int> > ("TriggerTypes",triggerTypes);
  descriptions.add(defaultModuleLabel<HLTCollectionProducer<T>>(), desc);
}

//
// member functions
//

// ------------ method called to produce the data  ------------
template <typename T>
void
HLTCollectionProducer<T>::produce(edm::StreamID iStreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const
{

  std::unique_ptr<std::vector<T> > collection ( new std::vector<T>() );

  // get hold of collection of TriggerFilterObjectsWithRefs
  edm::Handle<trigger::TriggerFilterObjectWithRefs> hltObject;
  iEvent.getByToken(hltObjectToken_, hltObject);
  std::vector<edm::Ref<std::vector<T> > > objectRefs;

  for (size_t t=0; t<triggerTypes_.size(); ++t) {
    objectRefs.clear();
    hltObject->getObjects( triggerTypes_[t], objectRefs );
    for (size_t i = 0; i < objectRefs.size(); ++i) {
      collection->push_back(*objectRefs[i]);
    }
  }
  
  iEvent.put(std::move(collection));

}

#endif // HLTCollectionProducer_h