TTTrackAssociator.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_TRACK_ASSOCIATOR_H
#define L1_TRACK_TRIGGER_TRACK_ASSOCIATOR_H

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/L1TrackTrigger/interface/TTTrack.h"

#include "L1Trigger/TrackTrigger/interface/classNameFinder.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTStubAssociationMap.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTTrackAssociationMap.h"
#include "SimTracker/TrackTriggerAssociation/plugins/TTStubAssociator.h"
#include "SimTracker/TrackTriggerAssociation/plugins/TTClusterAssociator.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"

#include <memory>
#include <map>
#include <vector>

template <typename T>
class TTTrackAssociator : public edm::stream::EDProducer<> {
public:
  explicit TTTrackAssociator(const edm::ParameterSet& iConfig);

  ~TTTrackAssociator() override;

private:
  std::vector<edm::InputTag> TTTracksInputTags;

  std::vector<edm::EDGetTokenT<std::vector<TTTrack<T> > > > TTTracksTokens;

  edm::EDGetTokenT<TTStubAssociationMap<T> > TTStubTruthToken;
  edm::EDGetTokenT<TTClusterAssociationMap<T> > TTClusterTruthToken;

  bool TTTrackAllowOneFalse2SStub;

  void beginRun(const edm::Run& run, const edm::EventSetup& iSetup) override;
  void endRun(const edm::Run& run, const edm::EventSetup& iSetup) override;
  void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;

};  

template <typename T>
TTTrackAssociator<T>::TTTrackAssociator(const edm::ParameterSet& iConfig) {
  TTTracksInputTags = iConfig.getParameter<std::vector<edm::InputTag> >("TTTracks");
  TTClusterTruthToken = consumes<TTClusterAssociationMap<T> >(iConfig.getParameter<edm::InputTag>("TTClusterTruth"));
  TTStubTruthToken = consumes<TTStubAssociationMap<T> >(iConfig.getParameter<edm::InputTag>("TTStubTruth"));
  TTTrackAllowOneFalse2SStub = iConfig.getParameter<bool>("TTTrackAllowOneFalse2SStub");
  if (TTTrackAllowOneFalse2SStub) {
    edm::LogInfo("TTTrackAssociator< ") << "Allow track if no more than one 2S stub doesn't match truth.";
  } else {
    edm::LogInfo("TTTrackAssociator< ") << "All 2S stubs must match truth.";
  }

  for (auto iTag = TTTracksInputTags.begin(); iTag != TTTracksInputTags.end(); iTag++) {
    TTTracksTokens.push_back(consumes<std::vector<TTTrack<T> > >(*iTag));

    produces<TTTrackAssociationMap<T> >((*iTag).instance());
  }
}

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

template <typename T>
void TTTrackAssociator<T>::beginRun(const edm::Run& run, const edm::EventSetup& iSetup) {
  edm::LogInfo("TTStubAssociator< ") << "TTTrackAssociator< " << templateNameFinder<T>() << " > loaded.";
}

template <typename T>
void TTTrackAssociator<T>::endRun(const edm::Run& run, const edm::EventSetup& iSetup) {}

template <>
void TTTrackAssociator<Ref_Phase2TrackerDigi_>::produce(edm::Event& iEvent, const edm::EventSetup& iSetup);

#endif