图像转换:RGB24转换为YUV420P

代码片段:利用FFmpeg中libswsscale库 完成RGB24格式图像到YUV420P格式的转换

//************************************
// Method:    rgb24_to_yuv420p
// FullName:  rgb24_to_yuv420p
// Access:    public 
// Returns:   int			返回值为0表示成功,否则失败
// Qualifier:				将RGB24格式图像转化为YUV420P格式,并进行相应的缩放
// Parameter: void * src	输入缓冲
// Parameter: int srcW		源宽
// Parameter: int srcH		源高
// Parameter: int srcSz		输入RGB图像数据大小=srcW*srcH*3
// Parameter: void * dst	输出缓冲
// Parameter: int dstW		目标宽
// Parameter: int dstH		目标高
// Parameter: int dstSz		输出的YUV图像数据大小=dstW*dstH*1.5
//************************************
int rgb24_to_yuv420p(void* src, int srcW, int srcH, int srcSz, void* dst, int dstW, int dstH, int dstSz)
{
	if (!src || !dst || srcSz <= 0 || srcW <= 0 || srcH <= 0 || dstSz <= 0||dstW<=0||dstH<=0)
	{
		return -1;
	}
	if (srcSz != srcW*srcH * 3 || dstSz != dstW*dstH*1.5)
	{
		return -1;
	}
	auto* ctxt = sws_getContext(srcW, srcH, AV_PIX_FMT_RGB24, dstW, dstH, AV_PIX_FMT_YUV420P, 0, 0, 0, 0);
	const int srcStride[1] = { 3 * srcW };
	const int dstStride[4] = { dstW, dstW / 2, dstW / 2, 0 };
	uint8_t * dstEx[4] = { 0 };
	dstEx[0] = (uint8_t *)dst;
	dstEx[1] = dstEx[0] + dstW*dstH;
	dstEx[2] = dstEx[1] + dstW*dstH / 4;
	sws_scale(ctxt, (const uint8_t * const *)&src, srcStride, 0, srcH,dstEx, dstStride);
	sws_freeContext(ctxt);
	return 0;
}


你可能感兴趣的:(FFmpeg)