nvenc_dynamic_factory.h

Go to the documentation of this file.

 
#pragma once
#ifdef _WIN32
 
// standard includes
  #include <functional>
  #include <memory>
 
  // local includes
  #include "nvenc_d3d11_interface.h"
  #include "nvenc_shared_dll.h"
  #include "nvenc_version.h"
 
namespace nvenc {
 
  struct nvenc_runtime_api {
    using load_driver_fn = std::function<shared_dll()>;  
 
    using get_symbol_fn = std::function<FARPROC(HMODULE, const char *)>;
 
    load_driver_fn load_driver;  
    get_symbol_fn get_symbol;  
  };
 
  class nvenc_dynamic_factory {
  public:
    using create_encoder_fn =
      std::function<std::unique_ptr<nvenc_d3d11_interface>(ID3D11Device *, shared_dll)>;
 
    nvenc_dynamic_factory(
      shared_dll dll,
      nvenc_sdk_version sdk_version,
      create_encoder_fn create_native,
      create_encoder_fn create_on_cuda
    );
 
    static std::shared_ptr<nvenc_dynamic_factory> get();
 
    static std::shared_ptr<nvenc_dynamic_factory> get(const nvenc_runtime_api &runtime_api);
 
    std::unique_ptr<nvenc_d3d11_interface> create_nvenc_d3d11_native(ID3D11Device *d3d_device) const;
 
    std::unique_ptr<nvenc_d3d11_interface> create_nvenc_d3d11_on_cuda(ID3D11Device *d3d_device) const;
 
    nvenc_sdk_version sdk_version() const;
 
  private:
    shared_dll dll;
    nvenc_sdk_version selected_sdk_version;
    create_encoder_fn create_native;
    create_encoder_fn create_on_cuda;
  };
 
}  // namespace nvenc
#endif