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

Subversion Repositories seema-scanner

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