TriggerSummaryProducerAOD.h

Go to the documentation of this file.

#ifndef HLTcore_TriggerSummaryProducerAOD_h
#define HLTcore_TriggerSummaryProducerAOD_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/GetterOfProducts.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Provenance/interface/ProductID.h"

#include "DataFormats/HLTReco/interface/TriggerTypeDefs.h"
#include "DataFormats/HLTReco/interface/TriggerObject.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"

#include "DataFormats/L1Trigger/interface/L1HFRingsFwd.h"
#include "DataFormats/L1Trigger/interface/L1EtMissParticleFwd.h"
#include "DataFormats/METReco/interface/METFwd.h"
#include "DataFormats/METReco/interface/CaloMETFwd.h"
#include "DataFormats/METReco/interface/PFMETFwd.h"

#include "DataFormats/RecoCandidate/interface/RecoEcalCandidateFwd.h"
#include "DataFormats/EgammaCandidates/interface/ElectronFwd.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/Candidate/interface/CompositeCandidateFwd.h"
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/CaloMETCollection.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidateFwd.h"
#include "DataFormats/L1Trigger/interface/L1EmParticleFwd.h"
#include "DataFormats/L1Trigger/interface/L1MuonParticleFwd.h"
#include "DataFormats/L1Trigger/interface/L1JetParticleFwd.h"
#include "DataFormats/L1Trigger/interface/L1EtMissParticleFwd.h"
#include "DataFormats/L1Trigger/interface/L1HFRingsFwd.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/TauReco/interface/PFTauFwd.h"

#include <map>
#include <set>
#include <string>
#include <vector>

#include <functional>
#include "tbb/concurrent_unordered_set.h"
#include <regex>

namespace edm {
  class EventSetup;
}

namespace edm {
  class ConfigurationDescriptions;
}

//
// class declaration
//

struct InputTagHash {
  std::size_t operator()(const edm::InputTag& inputTag) const {
    std::hash<std::string> Hash;
    // bit-wise xor
    return Hash(inputTag.label()) ^ Hash(inputTag.instance()) ^ Hash(inputTag.process());
  }
};
struct GlobalInputTags {
  GlobalInputTags() : filterTagsGlobal_(), collectionTagsGlobal_() {}
  mutable tbb::concurrent_unordered_set<edm::InputTag, InputTagHash> filterTagsGlobal_;
  mutable tbb::concurrent_unordered_set<edm::InputTag, InputTagHash> collectionTagsGlobal_;
};

class TriggerSummaryProducerAOD : public edm::stream::EDProducer<edm::GlobalCache<GlobalInputTags>> {
public:
  explicit TriggerSummaryProducerAOD(const edm::ParameterSet&, const GlobalInputTags*);
  ~TriggerSummaryProducerAOD() override;
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endStream() override;
  static void globalEndJob(const GlobalInputTags*);

  // additional
  static std::unique_ptr<GlobalInputTags> initializeGlobalCache(edm::ParameterSet const&) {
    return std::unique_ptr<GlobalInputTags>(new GlobalInputTags());
  };

  template <typename C>
  void fillTriggerObjectCollections(const edm::Event&, edm::GetterOfProducts<C>&);

  template <typename T>
  void fillTriggerObject(const T&);
  void fillTriggerObject(const l1extra::L1HFRings&);
  void fillTriggerObject(const l1extra::L1EtMissParticle&);
  void fillTriggerObject(const reco::PFMET&);
  void fillTriggerObject(const reco::CaloMET&);
  void fillTriggerObject(const reco::MET&);

  template <typename C>
  void fillFilterObjectMembers(const edm::Event&,
                               const edm::InputTag& tag,
                               const trigger::Vids&,
                               const std::vector<edm::Ref<C>>&);

  template <typename C>
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<C>&);
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<l1extra::L1HFRingsCollection>&);
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<l1extra::L1EtMissParticleCollection>&);
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<reco::PFMETCollection>&);
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<reco::CaloMETCollection>&);
  void fillFilterObjectMember(const int&, const int&, const edm::Ref<reco::METCollection>&);

private:
  bool throw_;
  std::string pn_;
  std::vector<std::regex> moduleLabelPatternsToMatch_;
  std::vector<std::regex> moduleLabelPatternsToSkip_;

  struct OrderInputTag {
    bool ignoreProcess_;
    OrderInputTag(bool ignoreProcess) : ignoreProcess_(ignoreProcess){};
    inline bool operator()(const edm::InputTag& l, const edm::InputTag& r) const {
      int c = l.label().compare(r.label());
      if (0 == c) {
        if (ignoreProcess_) {
          return l.instance() < r.instance();
        }
        c = l.instance().compare(r.instance());
        if (0 == c) {
          return l.process() < r.process();
        }
      }
      return c < 0;
    };
  };
  typedef std::set<edm::InputTag, OrderInputTag> InputTagSet;

  InputTagSet filterTagsEvent_;
  InputTagSet filterTagsStream_;

  InputTagSet collectionTagsEvent_;
  InputTagSet collectionTagsStream_;

  trigger::TriggerObjectCollection toc_;
  std::vector<std::string> tags_;
  std::map<edm::ProductID, unsigned int> offset_;

  trigger::Keys keys_;
  trigger::Vids ids_;

  std::vector<bool> maskFilters_;

  edm::GetterOfProducts<trigger::TriggerFilterObjectWithRefs> getTriggerFilterObjectWithRefs_;
  edm::GetterOfProducts<reco::RecoEcalCandidateCollection> getRecoEcalCandidateCollection_;
  edm::GetterOfProducts<reco::ElectronCollection> getElectronCollection_;
  edm::GetterOfProducts<reco::RecoChargedCandidateCollection> getRecoChargedCandidateCollection_;
  edm::GetterOfProducts<reco::CaloJetCollection> getCaloJetCollection_;
  edm::GetterOfProducts<reco::CompositeCandidateCollection> getCompositeCandidateCollection_;
  edm::GetterOfProducts<reco::METCollection> getMETCollection_;
  edm::GetterOfProducts<reco::CaloMETCollection> getCaloMETCollection_;
  edm::GetterOfProducts<reco::PFMETCollection> getPFMETCollection_;
  edm::GetterOfProducts<reco::IsolatedPixelTrackCandidateCollection> getIsolatedPixelTrackCandidateCollection_;
  edm::GetterOfProducts<l1extra::L1EmParticleCollection> getL1EmParticleCollection_;
  edm::GetterOfProducts<l1extra::L1MuonParticleCollection> getL1MuonParticleCollection_;
  edm::GetterOfProducts<l1extra::L1JetParticleCollection> getL1JetParticleCollection_;
  edm::GetterOfProducts<l1extra::L1EtMissParticleCollection> getL1EtMissParticleCollection_;
  edm::GetterOfProducts<l1extra::L1HFRingsCollection> getL1HFRingsCollection_;
  edm::GetterOfProducts<reco::PFJetCollection> getPFJetCollection_;
  edm::GetterOfProducts<reco::PFTauCollection> getPFTauCollection_;
  edm::GetterOfProducts<l1t::MuonBxCollection> getL1TMuonParticleCollection_;
  edm::GetterOfProducts<l1t::EGammaBxCollection> getL1TEGammaParticleCollection_;
  edm::GetterOfProducts<l1t::JetBxCollection> getL1TJetParticleCollection_;
  edm::GetterOfProducts<l1t::TauBxCollection> getL1TTauParticleCollection_;
  edm::GetterOfProducts<l1t::EtSumBxCollection> getL1TEtSumParticleCollection_;
};
#endif