QImage frameToImage(const QVideoFrame& frame)
{
QVideoFrame cloneFrame(frame);
cloneFrame.map(QAbstractVideoBuffer::ReadOnly);
// 获取帧的大小和像素格式
QSize frameSize = frame.size();
QVideoFrame::PixelFormat frameFormat = cloneFrame.pixelFormat();
// 根据像素格式选择转换方法
if (frameFormat == QVideoFrame::Format_YUV420P)
{
// 将YUV420P格式的数据转换为RGB格式
int width = frameSize.width();
int height = frameSize.height();
int uIndex = width * height;
int vIndex = uIndex + ((width * height) >> 2);
const uchar* yData = cloneFrame.bits();
QImage image(frameSize, QImage::Format_RGB888);
for (int y = 0; y < height; y++) {
for (int x = 0; x < width; x++) {
// Y分量
double Y = (double)yData[y * width + x];
// U分量
double U = (double)yData[uIndex + (y / 2) * (width / 2) + (x / 2)] - 128;
// V分量
double V = (double)yData[vIndex + (y / 2) * (width / 2) + (x / 2)] - 128;
// 转换公式
int R = (int)(Y + 1.13983 * V);
int G = (int)(Y - 0.39466 * U - 0.58060 * V);
int B = (int)(Y + 2.03211 * U);
R = qBound(0, R, 255);
G = qBound(0, G, 255);
B = qBound(0, B, 255);
QRgb rgbValue = qRgb(R, G, B);
image.setPixel(x, y, rgbValue);
}
}
return image;
}
// 如果帧格式不是YUV420P,则返回空图像
return QImage();
}
c语言实现QT YUV420P格式的QVideoFrame流转化为QImage
最新推荐文章于 2024-06-18 11:32:51 发布