TTStubBuilder.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_STUB_BUILDER_H
#define L1_TRACK_TRIGGER_STUB_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 "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"

#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithm.h"
#include "L1Trigger/TrackTrigger/interface/TTStubAlgorithmRecord.h"

#include "DataFormats/Common/interface/DetSetVectorNew.h"

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

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

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

    ~TTStubBuilder() override;

  private:
    edm::ESHandle< TTStubAlgorithm< T > > theStubFindingAlgoHandle;
    edm::EDGetTokenT< edmNew::DetSetVector< TTCluster< T > > > clustersToken;
    bool ForbidMultipleStubs;

    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;

    static bool SortStubBendPairs( const std::pair< unsigned int, double >& left, const std::pair< unsigned int, double >& right );
    static bool SortStubsBend( const TTStub< T >& left, const TTStub< T >& right );
}; 

template< typename T >
TTStubBuilder< T >::TTStubBuilder( const edm::ParameterSet& iConfig )
{
  clustersToken = consumes< edmNew::DetSetVector< TTCluster< T > > >(iConfig.getParameter< edm::InputTag >( "TTClusters" ));
  ForbidMultipleStubs = iConfig.getParameter< bool >( "OnlyOnePerInputCluster" );
  produces< edmNew::DetSetVector< TTCluster< T > > >( "ClusterAccepted" );
  produces< edmNew::DetSetVector< TTStub< T > > >( "StubAccepted" );
  produces< edmNew::DetSetVector< TTStub< T > > >( "StubRejected" );
}

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

template< typename T >
void TTStubBuilder< T >::beginRun( const edm::Run& run, const edm::EventSetup& iSetup )
{
  iSetup.get< TTStubAlgorithmRecord >().get( theStubFindingAlgoHandle );
}

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

template< typename T >
bool TTStubBuilder< T >::SortStubBendPairs( const std::pair< unsigned int, double >& left, const std::pair< unsigned int, double >& right )
{
  return fabs(left.second) < fabs(right.second);
}

template< typename T >
bool TTStubBuilder< T >::SortStubsBend( const TTStub< T >& left, const TTStub< T >& right )
{
  return fabs(left.getTriggerBend()) < fabs(right.getTriggerBend());
}

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

#endif