PlusFidPatternRecognition.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 _FID_PATTERN_RECOGNITION_H
#define _FID_PATTERN_RECOGNITION_H

#include "PlusFidPatternRecognitionCommon.h"
#include "PlusFidSegmentation.h"
#include "PlusFidLineFinder.h"
#include "PlusFidLabeling.h"

#include "vtkXMLDataElement.h"

//class igsioTrackedFrame; 
//class vtkIGSIOTrackedFrameList;

class vtkPlusCalibrationExport PlusFidPatternRecognition
{

public:
  enum PatternRecognitionError
  {
    PATTERN_RECOGNITION_ERROR_NO_ERROR,
    PATTERN_RECOGNITION_ERROR_UNKNOWN,
    PATTERN_RECOGNITION_ERROR_TOO_MANY_CANDIDATES
  };

  PlusFidPatternRecognition();
  virtual ~PlusFidPatternRecognition();

  PlusStatus ReadConfiguration(vtkXMLDataElement* rootConfigElement);

  PlusStatus RecognizePattern(vtkIGSIOTrackedFrameList* trackedFrameList, PatternRecognitionError& patternRecognitionError, int* numberOfSuccessfullySegmentedImages = NULL, std::vector<unsigned int>* segmentedFramesIndices = NULL);

  PlusStatus RecognizePattern(igsioTrackedFrame* trackedFrame, PatternRecognitionError& patternRecognitionError, unsigned int frameIndex);

  PlusStatus RecognizePattern(igsioTrackedFrame* trackedFrame, PlusPatternRecognitionResult& patternRecognitionResult, PatternRecognitionError& patternRecognitionError, unsigned int frameIndex);

  void DrawDots(PlusFidSegmentation::PixelType* image);

  void DrawResults(PlusFidSegmentation::PixelType* image);

  PlusFidSegmentation* GetFidSegmentation() { return  & m_FidSegmentation; };

  PlusFidLineFinder* GetFidLineFinder() { return & m_FidLineFinder; };

  PlusFidLabeling* GetFidLabeling() { return & m_FidLabeling; };

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

  void SetMaxLineLengthToleranceMm(double value);

  void SetNumberOfMaximumFiducialPointCandidates(int aMax);

  PlusStatus ReadPhantomDefinition(vtkXMLDataElement* rootConfigElement);

protected:

  PlusFidSegmentation           m_FidSegmentation;
  PlusFidLineFinder             m_FidLineFinder;
  PlusFidLabeling               m_FidLabeling;
  std::vector<PlusFidPattern*>  m_Patterns;

  double                        m_MaxLineLengthToleranceMm;
};

//-----------------------------------------------------------------------------

#endif //_FID_PATTERN_RECOGNITION_H