TTClusterAlgorithm_neighbor.h

Go to the documentation of this file.

#ifndef CLUSTERING_ALGORITHM_NEIGHBOR_H
#define CLUSTERING_ALGORITHM_NEIGHBOR_H

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"

#include "L1Trigger/TrackTrigger/interface/TTClusterAlgorithm.h"
#include "L1Trigger/TrackTrigger/interface/TTClusterAlgorithmRecord.h"

#include <string>
#include <cstdlib>
#include <map>

template< typename T >
class TTClusterAlgorithm_neighbor : public TTClusterAlgorithm< T >
{
  private:
    
  public:
    TTClusterAlgorithm_neighbor( )
      : TTClusterAlgorithm< T >( __func__ ){}

    ~TTClusterAlgorithm_neighbor() override{}

    void Cluster( std::vector< std::vector< T > > &output,
                  const std::vector< T > &input) const override;

    bool isANeighbor( const T& center, const T& mayNeigh) const;
    void addNeighbors( std::vector< T >& cluster, const std::vector< T >& input, unsigned int start, std::vector<bool> &masked ) const;

}; 

template< >
void TTClusterAlgorithm_neighbor< Ref_Phase2TrackerDigi_ >::Cluster( std::vector<std::vector< Ref_Phase2TrackerDigi_ > > &output,
                                                             const std::vector< Ref_Phase2TrackerDigi_ > &input ) const;

template< >
bool TTClusterAlgorithm_neighbor< Ref_Phase2TrackerDigi_ >::isANeighbor( const Ref_Phase2TrackerDigi_& center,
                                                                 const Ref_Phase2TrackerDigi_& mayNeigh ) const;

template< >
void TTClusterAlgorithm_neighbor< Ref_Phase2TrackerDigi_ >::addNeighbors( std::vector< Ref_Phase2TrackerDigi_ >& cluster,
                                                                  const std::vector< Ref_Phase2TrackerDigi_ >& input,
                                                  unsigned int startVal,
                                                                  std::vector< bool >& used) const;






template< typename T >
class ES_TTClusterAlgorithm_neighbor : public edm::ESProducer
{
  private:

  public:
    ES_TTClusterAlgorithm_neighbor( const edm::ParameterSet & p )
    {
      setWhatProduced( this );
    }

    ~ES_TTClusterAlgorithm_neighbor() override{}

    std::unique_ptr< TTClusterAlgorithm< T > > produce( const TTClusterAlgorithmRecord & record )
    { 
      TTClusterAlgorithm< T >* TTClusterAlgo =
        new TTClusterAlgorithm_neighbor< T >( );

      return std::unique_ptr< TTClusterAlgorithm< T > >( TTClusterAlgo );
    }

}; 

#endif