vtkPlusVirtualVolumeReconstructor.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 __vtkPlusVirtualVolumeReconstructor_h
#define __vtkPlusVirtualVolumeReconstructor_h

#include "vtkPlusDataCollectionExport.h"

#include "vtkPlusDevice.h"
#include <string>

class vtkPlusVolumeReconstructor;

class vtkPlusDataCollectionExport vtkPlusVirtualVolumeReconstructor : public vtkPlusDevice
{
public:
  static vtkPlusVirtualVolumeReconstructor* New();
  vtkTypeMacro(vtkPlusVirtualVolumeReconstructor, vtkPlusDevice);
  void PrintSelf(ostream& os, vtkIndent indent);

  virtual PlusStatus GetReconstructedVolumeFromFile(const std::string& inputSeqFilename, vtkImageData* reconstructedVolume, std::string& errorMessage);

  PlusStatus GetReconstructedVolume(vtkImageData* reconstructedVolume, std::string& outErrorMessage, bool applyHoleFilling = true);

  PlusStatus UpdateTransformRepository(vtkIGSIOTransformRepository* sharedTransformRepository);

  vtkGetMacro(EnableReconstruction, bool);
  void SetEnableReconstruction(bool aValue);

  virtual PlusStatus Reset();

  vtkSetStdStringMacro(OutputVolFilename);
  vtkGetStdStringMacro(OutputVolFilename);

  vtkSetStdStringMacro(OutputVolDeviceName);
  vtkGetStdStringMacro(OutputVolDeviceName);

  void SetOutputOrigin(double* origin);

  void SetOutputSpacing(double* spacing);

  void SetOutputExtent(int* extent);

  vtkGetMacro(TotalFramesRecorded, long int);

protected:

  virtual PlusStatus ReadConfiguration(vtkXMLDataElement*);

  virtual PlusStatus WriteConfiguration(vtkXMLDataElement*);

  virtual PlusStatus InternalUpdate();

  virtual PlusStatus NotifyConfigured();

  virtual double GetAcquisitionRate() const;

  virtual int OutputChannelCount() const;

  virtual void InternalWriteOutputChannels(vtkXMLDataElement* rootXMLElement);

  void SetRequestedFrameRate(double aValue);
  vtkGetMacro(RequestedFrameRate, double);

  virtual vtkPlusDataCollector* GetDataCollector() { return this->DataCollector; }

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

  virtual PlusStatus InternalConnect();
  virtual PlusStatus InternalDisconnect();

  PlusStatus AddFrames(vtkIGSIOTrackedFrameList* trackedFrameList);

  double GetSamplingPeriodSec();

  vtkPlusVirtualVolumeReconstructor();
  virtual ~vtkPlusVirtualVolumeReconstructor();

protected:
  double m_LastAlreadyRecordedFrameTimestamp;

  double m_NextFrameToBeRecordedTimestamp;

  const int m_SamplingFrameRate;

  double RequestedFrameRate;

  /* Time waited in update */
  double m_TimeWaited;
  double m_LastUpdateTime;

  double LastRecordedFrameTimestamp;

  long int TotalFramesRecorded;  // hard drive will probably fill up before a regular int is hit, but still...

  vtkSmartPointer<vtkPlusVolumeReconstructor> VolumeReconstructor;
  vtkSmartPointer<vtkIGSIOTransformRepository> TransformRepository;

  bool EnableReconstruction;

  std::string OutputVolFilename;
  std::string OutputVolDeviceName;

  vtkSmartPointer<vtkIGSIORecursiveCriticalSection> VolumeReconstructorAccessMutex;

private:
  vtkPlusVirtualVolumeReconstructor(const vtkPlusVirtualVolumeReconstructor&);   // Not implemented.
  void operator=(const vtkPlusVirtualVolumeReconstructor&);   // Not implemented.
};

#endif