TTStubAssociator.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_STUB_ASSOCIATOR_H
#define L1_TRACK_TRIGGER_STUB_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/TTStub.h"

#include "L1Trigger/TrackTrigger/interface/classNameFinder.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTClusterAssociationMap.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTStubAssociationMap.h"
#include "SimTracker/TrackTriggerAssociation/plugins/TTClusterAssociator.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"

#include "Geometry/CommonTopologies/interface/Topology.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"

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

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

    ~TTStubAssociator() override;

  private:
    std::vector< edm::InputTag > TTStubsInputTags;
    std::vector< edm::InputTag > TTClusterTruthInputTags;

    std::vector<edm::EDGetTokenT< edmNew::DetSetVector< TTStub< T > > > > TTStubsTokens;
    std::vector<edm::EDGetTokenT< TTClusterAssociationMap< T > > > TTClusterTruthTokens;

    edm::ESHandle<TrackerGeometry> theTrackerGeometry;
    edm::ESHandle<TrackerTopology> theTrackerTopology;

    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 >
TTStubAssociator< T >::TTStubAssociator( const edm::ParameterSet& iConfig )
{
  TTStubsInputTags = iConfig.getParameter< std::vector< edm::InputTag > >( "TTStubs" );
  TTClusterTruthInputTags = iConfig.getParameter< std::vector< edm::InputTag > >( "TTClusterTruth" );


  for ( auto iTag =  TTClusterTruthInputTags.begin(); iTag!=  TTClusterTruthInputTags.end(); iTag++ )
  {
    TTClusterTruthTokens.push_back(consumes< TTClusterAssociationMap< T > >(*iTag));
  }

  for ( auto iTag =  TTStubsInputTags.begin(); iTag!=  TTStubsInputTags.end(); iTag++ )
  {
    TTStubsTokens.push_back(consumes< edmNew::DetSetVector< TTStub< T > > >(*iTag));

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

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

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

  iSetup.get<TrackerTopologyRcd>().get(theTrackerTopology);
  iSetup.get<TrackerDigiGeometryRecord>().get(theTrackerGeometry);
}

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

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

#endif