vtkPlusSavedDataSource.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 __vtkPlusSavedDataSource_h
#define __vtkPlusSavedDataSource_h

#include "vtkPlusDataCollectionExport.h"

#include "vtkPlusDevice.h"

class vtkPlusBuffer;

class vtkPlusDataCollectionExport vtkPlusSavedDataSource;

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

  virtual PlusStatus ReadConfiguration( vtkXMLDataElement* config );
  virtual PlusStatus WriteConfiguration( vtkXMLDataElement* config );

  vtkSetStringMacro( SequenceFile );
  vtkGetStringMacro( SequenceFile );

  void SetLoopTimeRange( double loopStartTime, double loopStopTime );

  void GetLoopTimeRange( double& loopStartTime, double& loopStopTime );

  vtkGetMacro( RepeatEnabled, bool );
  vtkSetMacro( RepeatEnabled, bool );
  vtkBooleanMacro( RepeatEnabled, bool );

  vtkGetMacro( UseAllFrameFields, bool );
  vtkSetMacro( UseAllFrameFields, bool );
  vtkBooleanMacro( UseAllFrameFields, bool );

  vtkGetMacro( UseOriginalTimestamps, bool );
  vtkSetMacro( UseOriginalTimestamps, bool );
  vtkBooleanMacro( UseOriginalTimestamps, bool );

  vtkGetObjectMacro( LocalVideoBuffer, vtkPlusBuffer );

  virtual bool IsTracker() const;

  virtual PlusStatus NotifyConfigured();

  PlusStatus Probe();

protected:
  vtkPlusSavedDataSource();
  virtual ~vtkPlusSavedDataSource();

  virtual PlusStatus InternalConnect();

  virtual PlusStatus InternalConnectVideo( vtkIGSIOTrackedFrameList* savedDataBuffer );

  virtual PlusStatus InternalConnectTracker( vtkIGSIOTrackedFrameList* savedDataBuffer );

  virtual PlusStatus InternalDisconnect();

  PlusStatus InternalUpdate();

  PlusStatus InternalUpdateCurrentTimestamp( BufferItemUidType frameToBeAddedUid, int frameToBeAddedLoopIndex );

  PlusStatus InternalUpdateOriginalTimestamp( BufferItemUidType frameToBeAddedUid, int frameToBeAddedLoopIndex );

  BufferItemUidType GetClosestFrameUidWithinTimeRange( double time_Local, double startTime_Local, double stopTime_Local );

  vtkPlusBuffer* GetLocalTrackerBuffer();

  vtkPlusDataSource* GetOutputDataSource();

  vtkPlusBuffer* GetLocalBuffer();

  void DeleteLocalBuffers();

protected:
  int FrameBufferRowAlignment;

  char* SequenceFile;

  bool RepeatEnabled;

  double LoopStartTime_Local;

  double LoopStopTime_Local;

  vtkPlusBuffer* LocalVideoBuffer;

  std::map<std::string, vtkPlusBuffer*> LocalTrackerBuffers;

  bool UseAllFrameFields;

  bool UseOriginalTimestamps;

  BufferItemUidType LastAddedFrameUid;

  int LastAddedLoopIndex;

  BufferItemUidType LoopFirstFrameUid;

  BufferItemUidType LoopLastFrameUid;

  enum SimulatedStreamType
  {
    TRACKER_STREAM, /*< The device provides a tracker stream */
    VIDEO_STREAM   
  };

  SimulatedStreamType SimulatedStream;

private:
  static vtkPlusSavedDataSource* Instance;
  vtkPlusSavedDataSource( const vtkPlusSavedDataSource& ); // Not implemented.
  void operator=( const vtkPlusSavedDataSource& ); // Not implemented.
};

#endif