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