PndHypCluster.h

Go to the documentation of this file.

#ifndef PNDHYPCLUSTER_H
#define PNDHYPCLUSTER_H

#include "TObject.h"
#include <vector>
#include <iostream>

#include "PndHypHit.h" // contains SensorSide enumeration

class PndHypCluster : public TObject {
 public:
  PndHypCluster() : fClusterList(), fSide(SensorSide::kTOP){};
  ~PndHypCluster(){};
  PndHypCluster(std::vector<Int_t> list);

  void SetClusterList(std::vector<Int_t> list) { fClusterList = list; }
  std::vector<Int_t> GetClusterList() const { return fClusterList; };
  Int_t GetClusterSize() const { return fClusterList.size(); };
  Int_t GetDigiIndex(Int_t i) const { return fClusterList[i]; }

  SensorSide GetSensorSide() const { return fSide; };
  void SetSensorSide(SensorSide s) { fSide = s; }
  bool DigiBelongsToCluster(Int_t digiIndex);
  void Print();

  friend std::ostream &operator<<(std::ostream &out, PndHypCluster &cl)
  {
    std::vector<Int_t> list = cl.GetClusterList();
    out << "Hits in Cluster:" << std::endl;
    for (unsigned int i = 0; i < list.size(); i++)
      out << list[i] << std::endl;
    out << std::endl;

    return out;
  }

 private:
  std::vector<Int_t> fClusterList;
  SensorSide fSide;

  ClassDef(PndHypCluster, 1);
};

#endif