Blame view
Convert.cpp
2.25 KB
3f2992b2c
V1.0
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
#include <QAbstractVideoBuffer>
#include <QDebug>
#include <opencv2/imgproc/imgproc.hpp>
#include "Convert.h"
using namespace cv;
QImage Convert::fromYuvToRgb(QVideoFrame frame)
{
int y, u, v;
int c, d, e;
int r, g, b;
int height = frame.height();
int width = frame.width();
int size = height * width;
QImage image(width, height, QImage::Format_RGB888);
frame.map(QAbstractVideoBuffer::ReadOnly);
uchar *ptr = frame.bits();
for (int i(0); i < height; i++) {
for (int j(0); j < width; j++) {
// retrieve Y, U, V components
y = *(ptr + i * width + j);
u = *(ptr + size + i / 2 * width / 2 + j / 2);
v = *(ptr + size + size / 4 + i / 2 * width / 2 + j / 2);
// compute R, G, B components
c = y - 16;
d = u - 128;
e = v - 128;
r = (298*c + 409*e + 128) >> 8;
g = (298*c - 100*d - 208*e + 128) >> 8;
b = (298*c + 516*d + 128) >> 8;
// clamp R, G, B components
r = (r > 255) ? 255 : r;
g = (g > 255) ? 255 : g;
b = (b > 255) ? 255 : b;
r = (r < 0) ? 0 : r;
g = (g < 0) ? 0 : g;
b = (b < 0) ? 0 : b;
image.setPixel(j, i, qRgb(r, g, b));
}
}
frame.unmap();
return image;
}
Mat Convert::fromYuvToBgrMat(QVideoFrame frame)
{
QImage qimage = Convert::fromYuvToRgb(frame).rgbSwapped();
Mat image(frame.height(), frame.width(), CV_8UC3, qimage.bits());
return image.clone();
}
QImage Convert::fromYuvToGray(QVideoFrame frame)
{
int y;
int height = frame.height();
int width = frame.width();
int size = height * width;
QImage image(width, height, QImage::Format_Indexed8);
frame.map(QAbstractVideoBuffer::ReadOnly);
uchar *fPtr = frame.bits(); // pointer to video frame pixels
uchar *iPtr = image.bits(); // pointer to image pixels
for (int i(0); i < size; i++) {
// retrieve Y component
y = *fPtr;
// set image pixel
*iPtr = y;
// increment frame and image pointers
fPtr++;
iPtr++;
}
frame.unmap();
return image;
}
QImage Convert::fromBGRMatToQImage(const Mat &frame)
{
Mat tmp;
cvtColor(frame, tmp, CV_BGR2RGB);
QImage dst1((uchar *) tmp.data, tmp.cols, tmp.rows, tmp.step, QImage::Format_RGB888);
QImage dst2(dst1);
dst2.detach();
return dst2;
}
|