PlusFidLineFinder.h

Go to the documentation of this file.

/*=Plus=header=begin======================================================
Program: Plus
Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
See License.txt for details.
=========================================================Plus=header=end*/

#ifndef _FIDUCIAL_LINE_FINDER_H
#define _FIDUCIAL_LINE_FINDER_H

#include "PlusFidPatternRecognitionCommon.h"
#include "PlusConfigure.h"

class vtkXMLDataElement;

class vtkPlusCalibrationExport PlusFidLineFinder
{
public:
  PlusFidLineFinder();
  virtual ~PlusFidLineFinder();

  void SetFrameSize(const FrameSizeType& frameSize);

  void SetCandidateFidValues(const std::vector<PlusFidDot>& value) { m_CandidateFidValues = value; };

  void SetDotsVector(const std::vector<PlusFidDot>& value) { m_DotsVector = value; };

  void SetPatterns(const std::vector<PlusFidPattern*>& value) { m_Patterns = value; };

  void SetApproximateSpacingMmPerPixel(double value) { m_ApproximateSpacingMmPerPixel = value; };

  void SetMinThetaDegrees(double angleDeg);

  void SetMaxThetaDegrees(double angleDeg);

  void SetCollinearPointsMaxDistanceFromLineMm(double value) { m_CollinearPointsMaxDistanceFromLineMm = value; };

  PlusStatus ReadConfiguration(vtkXMLDataElement* rootConfigElement);

  void FindLines();

  std::vector<std::vector<PlusFidLine>>& GetLinesVector();

  double GetMaxThetaRad() { return m_MaxThetaRad; };

  double GetMinThetaRad() { return m_MinThetaRad; };

  std::vector<PlusFidPattern*>& GetPatterns() { return m_Patterns; };

  std::vector<PlusNWire> GetNWires();

  void SetImageToPhantomTransform(double* matrixElements);

  void Clear();

protected:
  void ComputeParameters();

  void FindLinesNPoints();

  void FindLines2Points();

  static double SegmentLength(const PlusFidDot& dot1, const PlusFidDot& dot2);

  double ComputeDistancePointLine(const PlusFidDot& dot, const PlusFidLine& line);

  void ComputeLine(PlusFidLine& line);

  static double ComputeAngleRad(const PlusFidDot& dot1, const PlusFidDot& dot2);

  bool AcceptLine(PlusFidLine& line);

  bool AcceptAngleRad(double angleRad);

  //Accessors and mutators

  double* GetImageNormalVectorInPhantomFrameMaximumRotationAngleDeg() { return m_ImageNormalVectorInPhantomFrameMaximumRotationAngleDeg; };
  void SetImageNormalVectorInPhantomFrameMaximumRotationAngleDeg(double* anglesDeg);

  double* GetImageToPhantomTransform() { return m_ImageToPhantomTransform; };

protected:
  FrameSizeType m_FrameSize;
  double m_ApproximateSpacingMmPerPixel;
  double m_ImageNormalVectorInPhantomFrameMaximumRotationAngleDeg[6];
  double m_ImageToPhantomTransform[16];

  // line length and line pair distance errors in percent - read from phantom definition
  double m_MaxLinePairDistanceErrorPercent;
  double m_CollinearPointsMaxDistanceFromLineMm;
  double m_MinThetaRad;
  double m_MaxThetaRad;

  std::vector<PlusFidDot> m_CandidateFidValues; // pointer to the fiducial candidates coordinates

  std::vector<PlusFidDot> m_DotsVector;
  std::vector< std::vector<PlusFidLine> > m_LinesVector;

  std::vector<PlusFidPattern*> m_Patterns;
};

#endif // _FIDUCIAL_LINE_FINDER_H