vtkPlusNVidiaDVPVideoSourceWin32.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 __vtkPlusNvidiaDVPVideoSource_h
#define __vtkPlusNvidiaDVPVideoSource_h

#include "vtkPlusDataCollectionExport.h"

#include "vtkPlusDevice.h"
#include "nvCommon.h"
#include "nvSDIin.h"
#include "nvConfigure.h"

class CNvGpu;

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

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

  virtual bool IsTracker() const;

  virtual PlusStatus NotifyConfigured();

  vtkGetMacro(EnableGPUCPUCopy, bool);

protected:
  vtkSetMacro(EnableGPUCPUCopy, bool);

  vtkPlusNvidiaDVPVideoSource();
  virtual ~vtkPlusNvidiaDVPVideoSource();

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

  HRESULT SetupSDIDevices();
  GLboolean SetupGL();
  GLboolean CleanupGL();
  HRESULT CleanupSDIDevices();
  HRESULT StartSDIPipeline();
  HRESULT StopSDIPipeline();
  GLenum CaptureVideo();
  HRESULT CopyGPUToCPU();
  void Shutdown();

protected:
  HRESULT SetupSDIinGL();
  HRESULT SetupSDIinDevices();
  HRESULT CleanupSDIinGL();

protected:
  CNvGpu* NvGPU;

  HDC HandleDC;

  HGLRC HandleGLRC;

  nvOptions NvOptions;

  CNvSDIin NvSDIin;

  FrameSizeType VideoSize;

  unsigned char* CPUFrame;

  GLuint VideoBufferObject[MAX_VIDEO_STREAMS];
  GLuint VideoBufferFormat;
  int NumFrames;

  uint64_t FrameNumber;

  bool EnableGPUCPUCopy;

  vtkPlusDataSource* OutputDataSource;

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

#endif