Blame view

src/cuda_kernels.h 579 Bytes
e41a52bb   Hu Chunming   1.优化数据读取线程;2. 添加A...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
  #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
  {
3d2ab595   Hu Chunming   支持gb28181
21
  	cudaError_t setColorSpace(e_ColorSpace CSC, float hue);
e41a52bb   Hu Chunming   1.优化数据读取线程;2. 添加A...
22
23
24
25
  
  	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);
  }