DtEncoderIO.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2018 VT-MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/

#pragma once
#include <vrvCore/DtDe.h>
#include <vrvVideoStream/vrvVideoStream.h>
#include <vrvVideoStream/DtNvQueue.h>
#include <vrvVideoStream/DtReadFramebufferProcessData.h>
#include <osg/Image>
// CUDA includes
#include "cuda_runtime.h"
#include <cuda_gl_interop.h>

// CUDA utilities and system includes
#include <vrvVideoStream/DtHelperCuda.h>
#include <cuda.h>
#include <cuda_runtime_api.h>

#include <NvCodec/NvEncoder/nvEncodeAPI.h>

namespace makVrv
{
#define MAX_ENCODE_QUEUE 32

   typedef struct DT_DLL_VRVVIDEOSTREAM _EncodeInputBuffer
   {
      _EncodeInputBuffer()
         : dwWidth(0)
         , dwHeight(0)
         , pNV12devPtr(0)
         , uNV12Stride(0)
         , nvRegisteredResource(0)
         , hInputSurface(0)
         , bufferFmt(NV_ENC_BUFFER_FORMAT_ARGB)
      {

      }
      unsigned int      dwWidth;
      unsigned int      dwHeight;
      CUdeviceptr       pNV12devPtr;
      uint32_t          uNV12Stride;
      void*             nvRegisteredResource;
      NV_ENC_INPUT_PTR  hInputSurface;
      NV_ENC_BUFFER_FORMAT bufferFmt;
   } EncodeInputBuffer;

   typedef struct DT_DLL_VRVVIDEOSTREAM _EncodeOutputBuffer
   {
      _EncodeOutputBuffer()
         : hBitstreamBuffer(0)
      {

      }
      NV_ENC_OUTPUT_PTR     hBitstreamBuffer;
   } EncodeOutputBuffer;

   typedef struct DT_DLL_VRVVIDEOSTREAM _EncodeBuffer
   {
      EncodeOutputBuffer      stOutputBfr;
      EncodeInputBuffer       stInputBfr;
   } EncodeBuffer;

   class DT_DLL_VRVVIDEOSTREAM DtEncoderIO
   {
   public:
      DtEncoderIO();

      virtual ~DtEncoderIO();

      void deinitialize();

      bool initialize(DtDe* theDe, const osg::ref_ptr<osg::Image>& image, unsigned int frameRate);

      void flipVertically(CUdeviceptr dev, uint32_t stride, int height);

      void submitPBO(unsigned int pboId, osg::ref_ptr<osg::Image>& image, DtReadFramebufferProcessData& fbData, unsigned int frameRate);

   protected:

      void deinitCUDA();

      bool initCUDA(const osg::ref_ptr<osg::Image>& image, unsigned int frameRate);

      void deinitNVidiaEncoder();

      bool initNVidiaEncoder(const osg::ref_ptr<osg::Image>& image, unsigned int frameRate);

      int myCudaDeviceId;

      int myHcuDevice;

      CUcontext myHcuContext;

      EncodeBuffer myEncodeBuffer[MAX_ENCODE_QUEUE];

      DtNvQueue<EncodeBuffer> myEncodeBufferQueue;

      void * myNVEncoder;

      uint64_t myEncodeIdx;

      CUdeviceptr myCUDATempDevicePtr;

      uint32_t    myTempStride;

      unsigned int myFrameRate;

      DtDe* myDe;
   };
}