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;

    //    edm::InputTag TTClusterTruthInputTag;
    //    edm::InputTag TTStubTruthInputTag;

    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" ));

  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