// mainwindow.cpp
#include "mainwindow.h"
#include "ui_mainwindow.h"

MainWindow::MainWindow(QWidget *parent)
   : QMainWindow(parent)
   , ui(new Ui::MainWindow)
{
   ui->setupUi(this);

   // 设定信号与槽 连接
   connect(ui->btn_test1,&QPushButton::clicked,this,&MainWindow::testFunc1);
   connect(ui->btn_test2,&QPushButton::clicked,this,&MainWindow::testFunc2);

   // 初始化 ui
   ui->pt_log->clear();  // 清除框内输出
}

MainWindow::~MainWindow()
{
   delete ui;
}

void MainWindow::testFunc1(void)
{
   ui->pt_log->appendPlainText("你点击了 测试按钮 1 ");
}

void MainWindow::testFunc2(void)
{
   ui->pt_log->appendPlainText("你点击了 测试按钮 2");
} 


// mainwindow.h
#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>

QT_BEGIN_NAMESPACE
namespace Ui { class MainWindow; }
QT_END_NAMESPACE

class MainWindow : public QMainWindow
{
   Q_OBJECT

public slots:
   void testFunc1(void);
   void testFunc2(void);

public:
   MainWindow(QWidget *parent = nullptr);
   ~MainWindow();

private:
   Ui::MainWindow *ui;
};
#endif // MAINWINDOW_H

// main.cpp
#include "mainwindow.h"
#include <QApplication>

// 运行主窗口 用于显示界面 ui
int main(int argc, char *argv[])
{
   QApplication a(argc, argv);
   MainWindow w;
   w.show();
   return a.exec();
}

// 图片路径
QString lena_img = "../testimages/lena.png";
void MainWindow::testFunc1(void)
{
    QPixmap pixmap;
    pixmap.load(lena_img);

    ui->lb_src->setPixmap(pixmap);
    ui->pt_log->appendPlainText("左侧使用 QPixmap load 图像数据1 ");
}

void MainWindow::testFunc2(void)
{
    QImage image(lena_img);
    ui->lb_dst->setPixmap(QPixmap::fromImage(image));
    ui->pt_log->appendPlainText("右侧使用 QImage 转换成 QPixmap 进行显示2 ");
}

/**
 * @fn  QImage CvMat2QImage(const cv::Mat & mat)
 *
 * @brief   将opencv mat 转换成 QT image
 *
 * @author  IRIS_Chen
 * @date    2019/12/19
 *
 * @param   mat The matrix
 *
 * @return  A QImage
 */
QImage CvMat2QImage(const cv::Mat &mat)
{
    // 图像的通道
    int channel = mat.channels();

    // 设立一个表 直接查询 其中 0 2 是无效值 1 3 4 对应的转换值
    const std::map<int, QImage::Format> img_cvt_map {
        { 1, QImage::Format_Grayscale8 },
        { 3, QImage::Format_RGB888 },
        { 4, QImage::Format_ARGB32 }
    };

    QImage image(mat.data, mat.cols, mat.rows,
                 static_cast<int>(mat.step),
                 img_cvt_map.at(channel));

    // 三通道图像 值做 通道转换
    return channel == 3 ? image.rgbSwapped() : image;
}

/**
* @fn  static cv::Mat QImage2CvMat(const QImage &image);
*
* @brief   QT Image 转换成 cv Mat 结构
*
* @author  IRIS_Chen
* @date    2019/12/19
*
* @param   image   The image
*
* @return  A cv::Mat
*/
cv::Mat QImage2CvMat(const QImage &image)
{
    cv::Mat mat;
    const std::map<QImage::Format, int> img_cvt_map{
        { QImage::Format_Grayscale8, 1 },
        { QImage::Format_RGB888, 3 },
        { QImage::Format_ARGB32, 4}
    };

    return cv::Mat(image.height(), image.width(),img_cvt_map.at(image.format()));
}

// 图片路径
QString lena_img = "../testimages/lena.png";
void MainWindow::testFunc1(void)
{
    QPixmap pixmap;
    pixmap.load(lena_img);

    // 在图上绘制文字
    QPainter painter(&pixmap);
    painter.setPen(QColor(Qt::yellow));
    painter.drawText(100,100,"QT QPixmap");


    ui->lb_src->setPixmap(pixmap);
    ui->pt_log->appendPlainText("左侧使用 QPixmap load 图像数据1 ");
}

void MainWindow::testFunc2(void)
{
    cv::Mat mat = cv::imread("../testimages/lena.png");
    // 在图上显示文字
    cv::putText(mat,"OpenCV Mat",cv::Point(100,100),cv::FONT_HERSHEY_COMPLEX,1.0, cv::Scalar(0, 255, 255));

    QImage image = CvMat2QImage(mat);

    ui->lb_dst->setPixmap(QPixmap::fromImage(image));
    ui->pt_log->appendPlainText("右侧使用 Mat --> QImage --> QPixmap 进行显示2 ");
}