NV12转化为RGB

最新推荐文章于 2024-08-08 15:01:59 发布

tkp2014

最新推荐文章于 2024-08-08 15:01:59 发布

阅读量8k

点赞数 1

分类专栏： JPEG 文章标签： NV12 RGB JPEG

本文链接：https://blog.csdn.net/tkp2014/article/details/42837635

版权

这个示例展示了如何使用CUDA将NV12（YUV 4:2:0 planar）源转换为ARGB格式。通过定义CUDA内核函数，包括YUV到RGB的转换和颜色空间调整，将NV12格式的数据转换为ARGB格式的像素。

摘要由CSDN通过智能技术生成

/*
NV12ToARGB color space conversion CUDA kernel

This sample uses CUDA to perform a simple NV12 (YUV 4:2:0 planar)
source and converts to output in ARGB format
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "cudaProcessFrame.h"

__constant__ uint32 constAlpha;

#define MUL(x,y) (x*y)
__constant__ float constHueColorSpaceMat[9];

__device__ void YUV2RGB(uint32 *yuvi, float *red, float *green, float *blue)
{
float luma, chromaCb, chromaCr;

// Prepare for hue adjustment
luma = (float)yuvi[0];
chromaCb = (float)((int32)yuvi[1] - 512.0f);
chromaCr = (float)((int32)yuvi[2] - 512.0f);

// Convert YUV To RGB with hue adjustment
*red = MUL(luma, constHueColorSpaceMat[0]) +
MUL(chromaCb, constHueColorSpaceMat[1]) +
MUL(chromaCr, constHueColorSpaceMat[2]);
*green= MUL(luma, constHueColorSpaceMat[3]) +
MUL(chromaCb, constHueColorSpaceMat[4]) +
MUL(chromaCr, constHueColorSpaceMat[5]);
*blue = MUL(luma, constHueColorSpaceMat[6]) +
MUL(chromaCb, constHueColorSpaceMat[7]) +
MUL(chromaCr, constHueColorSpaceMat[8]);
}

__device__ uint32 RGBAPACK_8bit(float red, float green, float blue, uint32 alpha)
{
uint32 ARGBpixel = 0;

// Clamp final 10 bit results
red = min(max(red, 0.0f), 255.0f);
green = min(max(green, 0.0f), 255.0f);
blue = min(max(blue, 0.0f), 255.0f);