webrtc-sys 0.3.34

Unsafe bindings to libwebrtc
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#ifndef WEBRTC_NVIDIA_I420_BUFFER_CUDA_H_
#define WEBRTC_NVIDIA_I420_BUFFER_CUDA_H_

#include <cuda.h>

#include <cstdint>

#include "NvEncoder/NvEncoderCuda.h"
#include "api/video/i420_buffer.h"

namespace webrtc {

inline void CopyHostPlaneToDevice(CUdeviceptr dst,
                                  uint32_t dst_pitch,
                                  const uint8_t* src,
                                  uint32_t src_pitch,
                                  uint32_t width_bytes,
                                  uint32_t height) {
  CUDA_MEMCPY2D copy = {0};
  copy.srcMemoryType = CU_MEMORYTYPE_HOST;
  copy.srcHost = src;
  copy.srcPitch = src_pitch;
  copy.dstMemoryType = CU_MEMORYTYPE_DEVICE;
  copy.dstDevice = dst;
  copy.dstPitch = dst_pitch;
  copy.WidthInBytes = width_bytes;
  copy.Height = height;
  CUDA_DRVAPI_CALL(cuMemcpy2D(&copy));
}

inline void CopyI420BufferToDeviceFrame(CUcontext context,
                                        const I420BufferInterface& buffer,
                                        const NvEncInputFrame& input_frame) {
  if (input_frame.bufferFormat != NV_ENC_BUFFER_FORMAT_IYUV ||
      input_frame.numChromaPlanes != 2) {
    NVENC_THROW_ERROR("NVENC I420 upload requires an IYUV input frame",
                      NV_ENC_ERR_INVALID_PARAM);
  }

  CUDA_DRVAPI_CALL(cuCtxPushCurrent(context));

  const CUdeviceptr dst_y = reinterpret_cast<CUdeviceptr>(input_frame.inputPtr);
  const CUdeviceptr dst_u = dst_y + input_frame.chromaOffsets[0];
  const CUdeviceptr dst_v = dst_y + input_frame.chromaOffsets[1];

  CopyHostPlaneToDevice(dst_y, input_frame.pitch, buffer.DataY(),
                        buffer.StrideY(), buffer.width(), buffer.height());
  CopyHostPlaneToDevice(dst_u, input_frame.chromaPitch, buffer.DataU(),
                        buffer.StrideU(), buffer.ChromaWidth(),
                        buffer.ChromaHeight());
  CopyHostPlaneToDevice(dst_v, input_frame.chromaPitch, buffer.DataV(),
                        buffer.StrideV(), buffer.ChromaWidth(),
                        buffer.ChromaHeight());

  CUDA_DRVAPI_CALL(cuCtxPopCurrent(NULL));
}

}  // namespace webrtc

#endif  // WEBRTC_NVIDIA_I420_BUFFER_CUDA_H_