TTClusterBuilder.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_CLUSTER_BUILDER_H
#define L1_TRACK_TRIGGER_CLUSTER_BUILDER_H

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/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/TTClusterAlgorithm.h"
#include "L1Trigger/TrackTrigger/interface/TTClusterAlgorithmRecord.h"
#include "Geometry/CommonTopologies/interface/Topology.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"

#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"

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

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

  private:
    edm::ESHandle< TTClusterAlgorithm< T > > theClusterFindingAlgoHandle;
    std::vector< edm::EDGetTokenT< edm::DetSetVector< Phase2TrackerDigi > > > rawHitTokens;
    unsigned int                             ADCThreshold;  
    bool                                     storeLocalCoord;

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

    void RetrieveRawHits( std::map< DetId, std::vector< T > > &mRawHits,
                          const edm::Event& iEvent );

}; 

template< typename T >
TTClusterBuilder< T >::TTClusterBuilder( const edm::ParameterSet& iConfig )
{
  ADCThreshold     = iConfig.getParameter< unsigned int >("ADCThreshold");
  storeLocalCoord  = iConfig.getParameter< bool >("storeLocalCoord");

  std::vector< edm::InputTag > rawHitInputTags  = iConfig.getParameter< std::vector< edm::InputTag > >("rawHits");
  for ( auto it = rawHitInputTags.begin(); it != rawHitInputTags.end(); ++it ) {
    rawHitTokens.push_back(consumes< edm::DetSetVector< Phase2TrackerDigi > >(*it));
  }
  
  produces< edmNew::DetSetVector< TTCluster< T > > >( "ClusterInclusive" );
}

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

template< typename T >
void TTClusterBuilder< T >::beginRun( const edm::Run& run, const edm::EventSetup& iSetup )
{
  iSetup.get< TTClusterAlgorithmRecord >().get( theClusterFindingAlgoHandle );

}

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

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

template< >
void TTClusterBuilder< Ref_Phase2TrackerDigi_ >::RetrieveRawHits( std::map< DetId, std::vector< Ref_Phase2TrackerDigi_ > > &mRawHits,
                                                          const edm::Event& iEvent );

#endif