WebSVN – seema-scanner – Diff – /src/SMVideoWidget.cpp


#include "SMVideoWidget.h"
 
SMVideoWidget::SMVideoWidget(QWidget *parent) : QLabel(parent){
 
}
 
static QImage cvMat2qImage(cv::Mat mat){
 
    // 8-bits unsigned, raw bayer image
    if(mat.type()==CV_8UC1) {
        cv::Mat rgb;
        cv::cvtColor(mat, rgb, CV_BayerBG2RGB);
        QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB888);
        img = img.copy();
        return img;
    // 8-bit unsigned rgb image
    } else if(mat.type()==CV_8UC3) {
        // Copy input Mat
        QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
        return img;
    // 16-bit unsigned gray-scale image
    } else if(mat.type()==CV_16UC1) {
        mat.convertTo(mat, CV_8UC1, 1.0/256.0);
        return cvMat2qImage(mat);
    // 16-bit unsigned rgb image
    } else if(mat.type()==CV_16UC3) {
        mat.convertTo(mat, CV_8UC3, 1.0/256.0);
        return cvMat2qImage(mat);
    // 32bit floating point gray-scale image
    } else if(mat.type()==CV_32FC1) {
        cv::Mat rgb(mat.size(), CV_32FC3);
 
        cv::cvtColor(mat, rgb, cv::COLOR_GRAY2RGB);
 
        QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB32);
        img = img.copy();
        return img;
    } else {
        std::cerr << "SMVideoWidget: cv::Mat could not be converted to QImage!";
        return QImage();
    }
 
    // 8-bit unsigned gray-scale image
//    } else if(mat.type()==CV_8UC1) {
//        // Set the color table (used to tranMVate colour indexes to qRgb values)
//        QVector<QRgb> colorTable;
//        for (int i=0; i<256; i++)
//            colorTable.push_back(qRgb(i,i,i));
//        // Copy input Mat
//        QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_Indexed8);
//        img.setColorTable(colorTable);
//        return img;
}
 
 
void SMVideoWidget::showImageCV(cv::Mat image){
 
    QImage qimage = cvMat2qImage(image);
 
    // correct size only if label has no borders/frame!
    int w = this->width();
    int h = this->height();
 
    pixmap = QPixmap::fromImage(qimage);
    this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));
 
}
 
void SMVideoWidget::resizeEvent(QResizeEvent *event){
 
    if(!pixmap.isNull()){
        // correct size only if label has no borders/frame!
        int w = event->size().width();
        int h = event->size().height();
        this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));
    }
}
 

Subversion Repositories seema-scanner

(root)/src/SMVideoWidget.cpp – Rev 113 → 114

Rev 113		Rev 114
1	`#include "SMVideoWidget.h"`	1	`#include "SMVideoWidget.h"`
2		2
3	`SMVideoWidget::SMVideoWidget(QWidget *parent) : QLabel(parent){`	3	`SMVideoWidget::SMVideoWidget(QWidget *parent) : QLabel(parent){`
4		4
5	`}`	5	`}`
6		6
7	`static QImage cvMat2qImage(cv::Mat mat){`	7	`static QImage cvMat2qImage(cv::Mat mat){`
8		8
9	`// 8-bits unsigned, raw bayer image`	9	`// 8-bits unsigned, raw bayer image`
10	`if(mat.type()==CV_8UC1) {`	10	`if(mat.type()==CV_8UC1) {`
11	`cv::Mat rgb;`	11	`cv::Mat rgb;`
12	`cv::cvtColor(mat, rgb, CV_BayerBG2RGB);`	12	`cv::cvtColor(mat, rgb, CV_BayerBG2RGB);`
13	`QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB888);`	13	`QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB888);`
-		14	`img = img.copy();`
14	`return img;`	15	`return img;`
15	`// 8-bit unsigned rgb image`	16	`// 8-bit unsigned rgb image`
16	`} else if(mat.type()==CV_8UC3) {`	17	`} else if(mat.type()==CV_8UC3) {`
17	`// Copy input Mat`	18	`// Copy input Mat`
18	`QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);`	19	`QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);`
19	`return img;`	20	`return img;`
20	`// 16-bit unsigned gray-scale image`	21	`// 16-bit unsigned gray-scale image`
21	`} else if(mat.type()==CV_16UC1) {`	22	`} else if(mat.type()==CV_16UC1) {`
22	`mat.convertTo(mat, CV_8UC1, 1.0/256.0);`	23	`mat.convertTo(mat, CV_8UC1, 1.0/256.0);`
23	`return cvMat2qImage(mat);`	24	`return cvMat2qImage(mat);`
24	`// 16-bit unsigned rgb image`	25	`// 16-bit unsigned rgb image`
25	`} else if(mat.type()==CV_16UC3) {`	26	`} else if(mat.type()==CV_16UC3) {`
26	`mat.convertTo(mat, CV_8UC3, 1.0/256.0);`	27	`mat.convertTo(mat, CV_8UC3, 1.0/256.0);`
27	`return cvMat2qImage(mat);`	28	`return cvMat2qImage(mat);`
28	`// 32bit floating point gray-scale image`	29	`// 32bit floating point gray-scale image`
29	`} else if(mat.type()==CV_32FC1) {`	30	`} else if(mat.type()==CV_32FC1) {`
30	`cv::Mat rgb(mat.size(), CV_32FC3);`	31	`cv::Mat rgb(mat.size(), CV_32FC3);`
31	`rgb.addref();`	-
32	`cv::cvtColor(mat, rgb, cv::COLOR_GRAY2RGB);`	32	`cv::cvtColor(mat, rgb, cv::COLOR_GRAY2RGB);`
33	`// Copy input Mat`	-
34	`QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB32);`	33	`QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB32);`
35	`rgb.release();`	34	`img = img.copy();`
36	`return img;`	35	`return img;`
37	`} else {`	36	`} else {`
38	`std::cerr << "SMVideoWidget: cv::Mat could not be converted to QImage!";`	37	`std::cerr << "SMVideoWidget: cv::Mat could not be converted to QImage!";`
39	`return QImage();`	38	`return QImage();`
40	`}`	39	`}`
41		40
42	`// 8-bit unsigned gray-scale image`	41	`// 8-bit unsigned gray-scale image`
43	`// } else if(mat.type()==CV_8UC1) {`	42	`// } else if(mat.type()==CV_8UC1) {`
44	`// // Set the color table (used to tranMVate colour indexes to qRgb values)`	43	`// // Set the color table (used to tranMVate colour indexes to qRgb values)`
45	`// QVector<QRgb> colorTable;`	44	`// QVector<QRgb> colorTable;`
46	`// for (int i=0; i<256; i++)`	45	`// for (int i=0; i<256; i++)`
47	`// colorTable.push_back(qRgb(i,i,i));`	46	`// colorTable.push_back(qRgb(i,i,i));`
48	`// // Copy input Mat`	47	`// // Copy input Mat`
49	`// QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_Indexed8);`	48	`// QImage img((const uchar*)mat.data, mat.cols, mat.rows, mat.step, QImage::Format_Indexed8);`
50	`// img.setColorTable(colorTable);`	49	`// img.setColorTable(colorTable);`
51	`// return img;`	50	`// return img;`
52	`}`	51	`}`
53		52
54		53
55	`void SMVideoWidget::showImageCV(cv::Mat image){`	54	`void SMVideoWidget::showImageCV(cv::Mat image){`
56		55
57	`QImage qimage = cvMat2qImage(image);`	56	`QImage qimage = cvMat2qImage(image);`
58		57
59	`// correct size only if label has no borders/frame!`	58	`// correct size only if label has no borders/frame!`
60	`int w = this->width();`	59	`int w = this->width();`
61	`int h = this->height();`	60	`int h = this->height();`
62		61
63	`pixmap = QPixmap::fromImage(qimage);`	62	`pixmap = QPixmap::fromImage(qimage);`
64	`this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));`	63	`this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));`
65		64
66	`}`	65	`}`
67		66
68	`void SMVideoWidget::resizeEvent(QResizeEvent *event){`	67	`void SMVideoWidget::resizeEvent(QResizeEvent *event){`
69		68
70	`if(!pixmap.isNull()){`	69	`if(!pixmap.isNull()){`
71	`// correct size only if label has no borders/frame!`	70	`// correct size only if label has no borders/frame!`
72	`int w = event->size().width();`	71	`int w = event->size().width();`
73	`int h = event->size().height();`	72	`int h = event->size().height();`
74	`this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));`	73	`this->setPixmap(pixmap.scaled(w,h,Qt::KeepAspectRatio));`
75	`}`	74	`}`
76	`}`	75	`}`
77		76