TTCluster.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_CLUSTER_FORMAT_H
#define L1_TRACK_TRIGGER_CLUSTER_FORMAT_H

#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/Common/interface/DetSet.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/MeasurementPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h" 

template< typename T >
class TTCluster
{
  public:
    TTCluster();
    TTCluster( std::vector< T > aHits,
               DetId aDetId,
               unsigned int aStackMember,
               bool storeLocal );

    ~TTCluster();


    std::vector< T > getHits() const                   { return theHits; }
    void             setHits( std::vector< T > aHits ) { theHits = aHits; }

    DetId        getDetId() const         { return theDetId; }
    void         setDetId( DetId aDetId ) { theDetId = aDetId; }
    unsigned int getStackMember() const                      { return theStackMember; }
    void         setStackMember( unsigned int aStackMember ) { theStackMember = aStackMember; }

    std::vector< int > findRows() const;
    std::vector< int > findCols() const;
    void setCoordinates( std::vector< int > a, std::vector< int > b ) { theRows = a; theCols = b; }
    std::vector< int > getRows() const { return theRows; }
    std::vector< int > getCols() const { return theCols; }

    unsigned int findWidth() const;

    MeasurementPoint findHitLocalCoordinates( unsigned int hitIdx ) const;
    MeasurementPoint findAverageLocalCoordinates() const;
    MeasurementPoint findAverageLocalCoordinatesCentered() const;

    std::string print( unsigned int i = 0 ) const;

  private:
    std::vector< T >  theHits;
    DetId             theDetId;
    unsigned int      theStackMember;

    std::vector< int > theRows;
    std::vector< int > theCols;

}; 

template< typename T >
TTCluster< T >::TTCluster()
{
  theHits.clear();
  theDetId = 0;
  theStackMember = 0;

  theRows.clear();
  theCols.clear();
}

template< typename T >
TTCluster< T >::TTCluster( std::vector< T > aHits,
                           DetId aDetId,
                           unsigned int aStackMember,
                           bool storeLocal )
{
  this->setHits( aHits );
  this->setDetId( aDetId );
  this->setStackMember( aStackMember );    

  theRows.clear();
  theCols.clear();
  if ( storeLocal )
  {
    this->setCoordinates( this->findRows(), this->findCols() );
  }
}

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

template< >
unsigned int TTCluster< edm::Ref< edm::DetSetVector< Phase2TrackerDigi >, Phase2TrackerDigi > >::findWidth() const;

template< >
MeasurementPoint TTCluster< edm::Ref< edm::DetSetVector< Phase2TrackerDigi >, Phase2TrackerDigi > >::findHitLocalCoordinates( unsigned int hitIdx ) const;

template< >
MeasurementPoint TTCluster< edm::Ref< edm::DetSetVector< Phase2TrackerDigi >, Phase2TrackerDigi > >::findAverageLocalCoordinates() const;

template< typename T > 
std::vector< int > TTCluster< T >::findRows() const
{
  std::vector< int > temp;
  return temp;
}

template< typename T > 
std::vector< int > TTCluster< T >::findCols() const
{
  std::vector< int > temp;
  return temp;
}

template< >
std::vector< int > TTCluster< edm::Ref< edm::DetSetVector< Phase2TrackerDigi >, Phase2TrackerDigi > >::findRows() const;

template< >
std::vector< int > TTCluster< edm::Ref< edm::DetSetVector< Phase2TrackerDigi >, Phase2TrackerDigi > >::findCols() const;

template< typename T >
std::string TTCluster< T >::print( unsigned int i ) const
{
  std::string padding("");
  for ( unsigned int j = 0; j != i; ++j )
  {
    padding+="\t";
  }

  std::stringstream output;
  output<<padding<<"TTCluster:\n";
  padding+='\t';
  output << padding << "DetId: " << theDetId.rawId() << '\n';
  output << padding << "member: " << theStackMember << ", cluster size: " << theHits.size() << '\n';
  return output.str();
}

template< typename T >
std::ostream& operator << ( std::ostream& os, const TTCluster< T >& aTTCluster )
{
  return ( os << aTTCluster.print() );
}

#endif