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 "common/RefFrame.h"
#include "algorithms/tracking/HoughFinder/HFCluster.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);

  float Theta() const { return _theta; };
  float R() const { return _r; };

  unsigned int cl1ID() { return _cl1ID; };
  unsigned int cl2ID() { return _cl2ID; };

  bool ContainsCluster(const Cluster &cluster) const {
    return (cluster.Hits()[cluster.SeedHitIndex()].VolumeID() == _cl1ID ||
            cluster.Hits()[cluster.SeedHitIndex()].VolumeID() == _cl2ID);
  };

private:
  unsigned int _cl1ID;
  unsigned int _cl2ID;

  float _theta;
  float _r;
};

inline bool operator==(const HoughTrackStub &lhs, const HoughTrackStub &rhs) {
  return ((lhs._cl1ID == rhs._cl1ID && lhs._cl2ID == rhs._cl2ID) ||
          (lhs._cl1ID == rhs._cl2ID && lhs._cl1ID == rhs._cl2ID));
}

} // namespace Herd

#endif