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