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&);
  ~HLTCollectionProducer() override;
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
  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