HoughTrackStub.h

Go to the documentation of this file.

/*
 * HoughTrackStub.h
 *
 *  Created on: 30 Dec 2018
 *      Author: Valerio Formato
 */
 
#ifndef HOUGHTRACKSTUB_H_
#define HOUGHTRACKSTUB_H_
 
#include "herd/analysis/algorithms/tracking/HoughFinderTools/HFCluster.h"
#include "herd/analysis/common/RefFrame.h"
 
namespace Herd {
 
class HoughTrackStub {
public:
  HoughTrackStub(const HFCluster &cl1, const HFCluster &cl2, RefFrame::Axis dir1, RefFrame::Axis dir2);
 
  friend bool operator==(const HoughTrackStub &lhs, const HoughTrackStub &rhs);
 
  [[nodiscard]] float Theta() const { return m_theta; };
  [[nodiscard]] float R() const { return m_r; };
 
  unsigned int cl1ID() { return m_cl1ID; };
  unsigned int cl2ID() { return m_cl2ID; };
 
  [[nodiscard]] bool ContainsCluster(const Cluster &cluster) const {
    return (cluster.Hits()[cluster.SeedHitIndex()].VolumeID() == m_cl1ID ||
            cluster.Hits()[cluster.SeedHitIndex()].VolumeID() == m_cl2ID);
  };
 
private:
  unsigned int m_cl1ID;
  unsigned int m_cl2ID;
 
  float m_theta;
  float m_r;
};
 
inline bool operator==(const HoughTrackStub &lhs, const HoughTrackStub &rhs) {
  return ((lhs.m_cl1ID == rhs.m_cl1ID && lhs.m_cl2ID == rhs.m_cl2ID) ||
          (lhs.m_cl1ID == rhs.m_cl2ID && lhs.m_cl2ID == rhs.m_cl1ID));
}
 
} // namespace Herd
 
#endif