vtkPlusVirtualTextRecognizer.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 __vtkPlusVirtualTextRecognizer_h
#define __vtkPlusVirtualTextRecognizer_h

#include "PlusConfigure.h"
#include "vtkPlusDataCollectionExport.h"
#include "vtkPlusChannel.h"
#include "vtkPlusDevice.h"

namespace tesseract
{
  class TessBaseAPI;
}

typedef struct Pix PIX;

class vtkPlusDataCollectionExport vtkPlusVirtualTextRecognizer : public vtkPlusDevice
{
#ifdef PLUS_TEST_TextRecognizer
public:
#endif
  class TextFieldParameter
  {
  public:
    TextFieldParameter()
    {
      this->Origin[0] = 0;
      this->Origin[1] = 0;
      this->Origin[2] = 0;
      this->Size[0] = 0;
      this->Size[1] = 0;
      this->Size[2] = 1;
    }

  public:
    std::string LatestParameterValue;
    PIX* ReceivedFrame;
    vtkSmartPointer<vtkImageData> ScreenRegion;
    vtkPlusChannel* SourceChannel;
    std::string ParameterName;
    std::array<int, 3> Origin;
    std::array<int, 3> Size;
  };

public:
  typedef std::vector<TextFieldParameter*> FieldList;
  typedef FieldList::iterator FieldListIterator;
  typedef std::map<vtkPlusChannel*, FieldList > ChannelFieldListMap;
  typedef ChannelFieldListMap::iterator ChannelFieldListMapIterator;

public:
  static vtkPlusVirtualTextRecognizer* New();
  vtkTypeMacro(vtkPlusVirtualTextRecognizer, vtkPlusDevice);
  virtual void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  virtual PlusStatus InternalConnect();

  virtual PlusStatus InternalDisconnect();

  virtual PlusStatus ReadConfiguration(vtkXMLDataElement*);

  virtual PlusStatus WriteConfiguration(vtkXMLDataElement*);

  virtual PlusStatus NotifyConfigured();

  virtual bool IsTracker() const {return false;}
  virtual bool IsVirtual() const {return true;}

  vtkSetStdStringMacro(Language);
  vtkGetStdStringMacro(Language);

  vtkSetObjectMacro(OutputChannel, vtkPlusChannel);
  vtkGetObjectMacro(OutputChannel, vtkPlusChannel);

  vtkSetStdStringMacro(TessdataDirectory);
  vtkGetStdStringMacro(TessdataDirectory);

#ifdef PLUS_TEST_TextRecognizer
  ChannelFieldListMap& GetRecognitionFields();
#endif

protected:
  virtual PlusStatus InternalUpdate();

  void ClearConfiguration();

  void vtkImageDataToPix(igsioTrackedFrame& frame, TextFieldParameter* parameter);

  PlusStatus FindOrQueryFrame(igsioTrackedFrame& frame, std::map<double, int>& queriedFramesIndexes, TextFieldParameter* parameter,
                              std::vector<igsioTrackedFrame*>& queriedFrames);

  std::string                 Language;

  std::string                 TessdataDirectory;

  tesseract::TessBaseAPI*     TesseractAPI;

  vtkIGSIOTrackedFrameList*    TrackedFrames;

  ChannelFieldListMap         RecognitionFields;

  vtkPlusChannel*             OutputChannel;

protected:
  vtkPlusVirtualTextRecognizer();
  virtual ~vtkPlusVirtualTextRecognizer();

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

#endif //__vtkPlusVirtualTextRecognizer_h