Blame view

FFNvDecoder/cuda_kernels.h 579 Bytes
a359bb4f   Hu Chunming   调通可正常运行 -- 初版
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
  #pragma once
  #include "cuda_runtime.h"
  #include "device_launch_parameters.h"
  
  #include <stdio.h>
  #include <stdlib.h>
  
  #include <string.h>
  #include <math.h>
  
  #include <cuda.h>
  
  typedef enum
  {
  	ITU601 = 1,
  	ITU709 = 2
  } e_ColorSpace;
  
  namespace cuda_common
  {
  	cudaError_t setColorSpace(e_ColorSpace CSC, float hue);
  
  	cudaError_t NV12ToRGBnot(CUdeviceptr d_srcNV12, size_t nSourcePitch, unsigned char* d_dstRGB, int width, int height);
  	cudaError_t CUDAToBGR(CUdeviceptr dataY, CUdeviceptr dataUV, size_t pitchY, size_t pitchUV, unsigned char* d_dstRGB, int width, int height);
  }