Source repositorynvenc_d3d11_on_cuda.h

Go to the documentation of this file.

#pragma once
#ifdef _WIN32
  // local includes
  #include "nvenc_d3d11.h"
  #include "nvenc_sdk.h"
 
namespace NVENC_NAMESPACE {

  class nvenc_d3d11_on_cuda final: public nvenc_d3d11 {
  public:
    explicit nvenc_d3d11_on_cuda(ID3D11Device *d3d_device, ::nvenc::shared_dll dll);
    ~nvenc_d3d11_on_cuda() override;
 
    ID3D11Texture2D *get_input_texture() override;
 
  private:
    bool init_library() override;
 
    bool create_and_register_input_buffer() override;
 
    bool synchronize_input_buffer() override;
 
    bool cuda_succeeded(CUresult result);
 
    bool cuda_failed(CUresult result);
 
    struct autopop_context {
      autopop_context(nvenc_d3d11_on_cuda &parent, CUcontext pushed_context):
          parent(parent),
          pushed_context(pushed_context) {
      }
 
      ~autopop_context();
 
      explicit operator bool() const {
        return pushed_context != nullptr;
      }
 
      nvenc_d3d11_on_cuda &parent;
      CUcontext pushed_context = nullptr;
    };
 
    autopop_context push_context();
 
    const ID3D11DevicePtr d3d_device;
    ID3D11Texture2DPtr d3d_input_texture;
 
    struct {
      tcuInit *cuInit;
      tcuD3D11GetDevice *cuD3D11GetDevice;
      tcuCtxCreate_v2 *cuCtxCreate;
      tcuCtxDestroy_v2 *cuCtxDestroy;
      tcuCtxPushCurrent_v2 *cuCtxPushCurrent;
      tcuCtxPopCurrent_v2 *cuCtxPopCurrent;
      tcuMemAllocPitch_v2 *cuMemAllocPitch;
      tcuMemFree_v2 *cuMemFree;
      tcuGraphicsD3D11RegisterResource *cuGraphicsD3D11RegisterResource;
      tcuGraphicsUnregisterResource *cuGraphicsUnregisterResource;
      tcuGraphicsMapResources *cuGraphicsMapResources;
      tcuGraphicsUnmapResources *cuGraphicsUnmapResources;
      tcuGraphicsSubResourceGetMappedArray *cuGraphicsSubResourceGetMappedArray;
      tcuMemcpy2D_v2 *cuMemcpy2D;
      HMODULE dll;
    } cuda_functions = {};
 
    CUresult last_cuda_error = CUDA_SUCCESS;
    CUcontext cuda_context = nullptr;
    CUgraphicsResource cuda_d3d_input_texture = nullptr;
    CUdeviceptr cuda_surface = 0;
    size_t cuda_surface_pitch = 0;
  };
 
}  // namespace NVENC_NAMESPACE
#endif