Конвертировать Yuv420 в rgb и показать на растровом изображении qt
Я хочу преобразовать изображение yuv в RGB для отображения на растровом изображении QT. Сейчас у меня есть то, что я вижу пустое растровое изображение.
int convertYUVtoRGB(int y, int u, int v) {
int r,g,b;
r = y + (int)(1.402f*v);
g = y - (int)(0.344f*u +0.714f*v);
b = y + (int)(1.772f*u);
r = r>255? 255 : r<0 ? 0 : r;
g = g>255? 255 : g<0 ? 0 : g;
b = b>255? 255 : b<0 ? 0 : b;
return 0xff000000 | (b<<16) | (g<<8) | r;
}unsigned char * convertYUV420_NV21toRGB8888(unsigned char data[78080], int width, int height) {
int size = width*height;
int offset = size;
unsigned char * pixels = new unsigned char[size];
int u, v, y1, y2, y3, y4;
// i percorre os Y and the final pixels
// k percorre os pixles U e V
for(int i=0, k=0; i < size; i+=2, k+=2) {
y1 = data[i ]&0xff;
y2 = data[i+1]&0xff;
y3 = data[width+i ]&0xff;
y4 = data[width+i+1]&0xff;
u = data[offset+k ]&0xff;
v = data[offset+k+1]&0xff;
u = u-128;
v = v-128;
pixels[i ] = convertYUVtoRGB(y1, u, v);
pixels[i+1] = convertYUVtoRGB(y2, u, v);
pixels[width+i ] = convertYUVtoRGB(y3, u, v);
pixels[width+i+1] = convertYUVtoRGB(y4, u, v);
if (i!=0 && (i+2)%width==0)
i+=width;
}
return pixels;
}
int main(int argc, char *argv[])
{
QApplication a(argc, argv);
unsigned char * buffer;
unsigned char * image = NULL;
QPixmap pixmap;
QImage img(320, 122, QImage::Format_ARGB32_Premultiplied);
img.fill(QColor(Qt::white).rgb());
ifstream is;
is.open ("bunny.yuv", ios::binary );
// is.seekg (0, ios::end);
// length = is.tellg();
// is.seekg (0, ios::beg);
buffer = new unsigned char[78081];
is.read((char * )buffer,78080);
is.close();
/*
for (int x = 0; x < 10; ++x) {
for (int y = 0; y < 10; ++y) {
img.setPixel(x, y, qRgb(0, 0, 0));
}
}
*/for(int i = 0; i < 78080; i++)
qDebug() << buffer[i]<< endl;
image = convertYUV420_NV21toRGB8888(buffer,320,122);
QByteArray byteImage((const char *)image);
pixmap.loadFromData(byteImage);QLabel myLabel;
myLabel.setPixmap(pixmap);
myLabel.setGeometry(20,100,320,122);
myLabel.show();
return a.exec();
}
РЕДАКТИРОВАТЬ: я добавил правильный формат и скобки. Все та же проблема.
С уважением,
Решение
И когда вы сделали расчет в посте Джейсона — лучший формат для растрового изображения — QImage :: Format_ARGB32_Premultiplied — формат Blue byte Green byte Red byte 0xff
Другие решения
Это не будет делать то, что вы думаете:
r = y + (int)1.402f*v;
g = y - (int)(0.344f*u +0.714f*v);
b = y + (int)1.772f*u;
Поскольку вы пропустили круглые скобки и таким образом приводите свои константы с плавающей точкой к целым числам до вычисления.
А затем, преобразовав ваши данные YUV в RGB, упакованные в int, вы сохраняете этот int в символ …
Так:
unsigned char * pixels = new unsigned char[size];
Вероятно, должно быть:
unsigned int* pixels = new unsigned int[size];
И очевидно, что это также влияет на тип возвращаемого значения функции. Вы также можете изменить тип вашей функции YUV -> RGB на unsigned, просто для согласованности.
В convertYUV420_NV21toRGB8888, pixels массив без знака, но вы пытаетесь назначить int к этому. Все, что вы получите, это младший байт, и большая часть байтов в вашем выводе останется неустановленной. Вам нужно полностью переосмыслить способ написания вывода.