TTClusterAssociator.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_CLUSTER_ASSOCIATOR_H
#define L1_TRACK_TRIGGER_CLUSTER_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 "L1Trigger/TrackTrigger/interface/classNameFinder.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTClusterAssociationMap.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"


#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithm.h"
#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithmRecord.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 TTClusterAssociator : public edm::stream::EDProducer<>
{
  public:
    explicit TTClusterAssociator( const edm::ParameterSet& iConfig );

    ~TTClusterAssociator() override;

  private:
    edm::Handle< edm::DetSetVector< PixelDigiSimLink > >   thePixelDigiSimLinkHandle;
    edm::Handle< edm::SimTrackContainer >                  theSimTrackHandle;
    edm::Handle< std::vector< TrackingParticle > >         TrackingParticleHandle;

    std::vector< edm::InputTag > TTClustersInputTags;

    edm::EDGetTokenT< edm::DetSetVector< PixelDigiSimLink > >              digisimLinkToken;
    edm::EDGetTokenT< edm::SimTrackContainer >                             simTrackToken;
    edm::EDGetTokenT< std::vector< TrackingParticle > >                    tpToken;
    //std::vector< edm::EDGetTokenT< edm::DetSetVector< TTCluster< T > > > > TTClustersTokens;
    std::vector<edm::EDGetTokenT< edmNew::DetSetVector< TTCluster< T > > > > TTClustersTokens;

    //    const StackedTrackerGeometry                           *theStackedTrackers;
    //unsigned int                                           ADCThreshold;


    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 >
TTClusterAssociator< T >::TTClusterAssociator( const edm::ParameterSet& iConfig )
{
  simTrackToken = consumes< edm::SimTrackContainer >(iConfig.getParameter< edm::InputTag >( "simTrackHits" ));
  digisimLinkToken = consumes< edm::DetSetVector< PixelDigiSimLink > >(iConfig.getParameter< edm::InputTag >( "digiSimLinks" ));
  tpToken = consumes< std::vector< TrackingParticle >  >(iConfig.getParameter< edm::InputTag >( "trackingParts" ));


  TTClustersInputTags = iConfig.getParameter< std::vector< edm::InputTag > >( "TTClusters" );

  for ( auto iTag =  TTClustersInputTags.begin(); iTag!=  TTClustersInputTags.end(); iTag++ )
  {
    TTClustersTokens.push_back(consumes< edmNew::DetSetVector< TTCluster< T > > >(*iTag));

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

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

template< typename T >
void TTClusterAssociator< T >::beginRun( const edm::Run& run, const edm::EventSetup& iSetup )
{
  iSetup.get<TrackerDigiGeometryRecord>().get(theTrackerGeometry);
  iSetup.get<TrackerTopologyRcd>().get(theTrackerTopology);

  edm::LogInfo("TTClusterAssociator< ") << templateNameFinder< T >() << " > loaded.";
}

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

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

#endif