vtkPlusLineSegmentationAlgo.h

Go to the documentation of this file.

#ifndef __vtkPlusLineSegmentationAlgo_h
#define __vtkPlusLineSegmentationAlgo_h

#include "itkImage.h"
#include "vtkPlusCalibrationExport.h"
#include "vtkObject.h"
#include <deque>

//class igsioTrackedFrame; 
//class vtkIGSIOTrackedFrameList;

class vtkPlusCalibrationExport vtkPlusLineSegmentationAlgo : public vtkObject
{
public:
  struct LineParameters 
  {
    bool lineDetected;
    double lineOriginPoint_Image[2];
    double lineDirectionVector_Image[2];

    double Slope() const { return lineDirectionVector_Image[1] / lineDirectionVector_Image[0]; }
  };

  typedef unsigned char CharPixelType;
  typedef itk::Image<CharPixelType, 2> CharImageType;

  static vtkPlusLineSegmentationAlgo* New();
  vtkTypeMacro(vtkPlusLineSegmentationAlgo, vtkObject);
  virtual void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  PlusStatus ReadConfiguration(vtkXMLDataElement* aConfig);

  void SetTrackedFrameList(vtkIGSIOTrackedFrameList& aTrackedFrameList);
  void SetTrackedFrame(igsioTrackedFrame& aTrackedFrame);

  void SetSignalTimeRange(double rangeMin, double rangeMax);

  void SetClipRectangle(int clipRectangleOriginPix[2], int clipRectangleSizePix[2]);

  PlusStatus Update();

  void GetDetectedTimestamps(std::deque<double>& timestamps);

  void GetDetectedPositions(std::deque<double>& positions);

  void GetDetectedLineParameters(std::vector<LineParameters>& parameters);

  void SetSaveIntermediateImages(bool saveIntermediateImages);

  void SetIntermediateFilesOutputDirectory(const std::string& outputDirectory);

  PlusStatus Reset();

  vtkGetMacro(PlotIntensityProfile, bool);
  vtkSetMacro(PlotIntensityProfile, bool);

protected:
  vtkPlusLineSegmentationAlgo();
  virtual ~vtkPlusLineSegmentationAlgo();

  PlusStatus VerifyVideoInput();

  PlusStatus ComputeVideoPositionMetric();

  PlusStatus FindPeakStart(std::deque<int>& intensityProfile, int maxFromLargestArea, int startOfMaxArea, double& startOfPeak);

  PlusStatus FindLargestPeak(std::deque<int>& intensityProfile, int& maxFromLargestArea, int& maxFromLargestAreaIndex, int& startOfMaxArea);

  PlusStatus ComputeCenterOfGravity(std::deque<int>& intensityProfile, int startOfMaxArea, double& centerOfGravity);

  void ComputeLineParameters(std::vector<itk::Point<double, 2> >& data, LineParameters& outputParameters);

  void PlotIntArray(const std::deque<int>& intensityValues);

  void PlotDoubleArray(const std::deque<double>& intensityValues);

  void SaveIntermediateImage(int frameNumber, CharImageType::Pointer scanlineImage, double x_0, double y_0, double r_x, double r_y, int numOfValidScanlines, const std::vector<itk::Point<double, 2> >& intensityPeakPositions);

  void LimitToClipRegion(CharImageType::RegionType& region);

protected:
  vtkSmartPointer<vtkIGSIOTrackedFrameList> m_TrackedFrameList;

  std::deque<double> m_SignalValues;
  std::deque<double> m_SignalTimestamps;
  std::vector<LineParameters> m_LineParameters;

  bool m_SaveIntermediateImages;

  std::string IntermediateFilesOutputDirectory;

  bool PlotIntensityProfile;

  double m_SignalTimeRangeMin;
  double m_SignalTimeRangeMax;

  CharImageType::IndexValueType m_ClipRectangleOrigin[2];

  CharImageType::SizeValueType m_ClipRectangleSize[2];

private:
  vtkPlusLineSegmentationAlgo(const vtkPlusLineSegmentationAlgo&);
  void operator=(const vtkPlusLineSegmentationAlgo&);
};

#endif //  __vtkPlusLineSegmentationAlgo_h