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


#include "SMVideoWidget.h"
#include "cvtools.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, raw bayer image
    } else if(mat.type()==CV_16UC1) {
        cv::Mat mat8bit = mat.clone();
        cvtools::rshift(mat8bit, 8);
        mat8bit.convertTo(mat8bit, CV_8UC1);
        cv::Mat rgb;
        cv::cvtColor(mat8bit, rgb, CV_BayerBG2RGB);
        QImage img((const uchar*)rgb.data, rgb.cols, rgb.rows, rgb.step, QImage::Format_RGB888);
        img = img.copy();
        return img;
    // 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){
    if(image.rows){
        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));
    }
}
 

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

Subversion Repositories seema-scanner

(root)/src/SMVideoWidget.cpp – Rev 120 → 207