#ifndef MAINWINDOW_H
#define MAINWINDOW_H
#include <QMainWindow>
#include <QMessageBox>
#include <QLabel>
#include <QImage>
#include <QFileDialog>
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <QDebug>
#include <iostream>
using namespace cv;
namespace Ui {
class MainWindow;
}
class MainWindow : public QMainWindow
{
    Q_OBJECT
public:
    explicit MainWindow(QWidget *parent = 0);
    ~MainWindow();
    QString openFileName;//保存图片路径
    QImage img;          //QImage存储图片
    QImage scaleImg;     //为了显示在label中压缩后的图片
private:
    Ui::MainWindow *ui;
private slots:
    void openFileSlot();//qt打开图片
    void cannySlot();//执行canny算子并显示
    Mat QImage2cvMat(QImage image);//QImage转Mat
    QImage cvMat2QImage(const Mat& mat);//Mat转QImage
};
#endif // MAINWINDOW_H

#include "mainwindow.h"
#include "ui_mainwindow.h"
MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    QObject::connect(ui->openButton,SIGNAL(clicked()),this,SLOT(openFileSlot()));
    QObject::connect(ui->cannyButton,SIGNAL(clicked()),this,SLOT(cannySlot()));
}
MainWindow::~MainWindow()
{
    delete ui;
}
//打开图片按钮对应功能
void MainWindow::openFileSlot()
{
    //得到文件名
    openFileName=QFileDialog::getOpenFileName(this,"打开文件",QDir::currentPath());
    if(! ( img.load(openFileName) ) ) //加载图像
    {
        QMessageBox::information(this,
                                 "打开图像失败",
                                 "打开图像失败!");
        return;
    }
    //使图片合适地放到label中
    if((img.width()/img.height())>(ui->srcLabel->width()/ui->srcLabel->height()))
        scaleImg = img.scaledToWidth(ui->srcLabel->width(),Qt::FastTransformation);
    else
        scaleImg = img.scaledToHeight(ui->srcLabel->height(),Qt::FastTransformation);
    ui->srcLabel->setPixmap(QPixmap::fromImage(scaleImg));
}
//canny算子按钮对应功能
void MainWindow::cannySlot()
{
    if(openFileName.isEmpty())//检测是否选中了文件
    {
        QMessageBox::information(this,"警告！","没图片算个卵！");
        return;
    }
    Mat matImg = QImage2cvMat(img);
    Mat matImgGray,edge;
    cvtColor(matImg,matImgGray,COLOR_RGB2GRAY);
    //3*3降噪
    blur(matImgGray,edge,Size(3,3));
//    GaussianBlur(matImgGray,edge,Size(3,3),0,0);//高斯降噪
	//边缘处理20,60可变
    Canny(edge,edge,20,60,3);
    //使图片合适地放到label中
    if((img.width()/img.height())>(ui->dstLabel->width()/ui->dstLabel->height()))
        scaleImg = img.scaledToWidth(ui->dstLabel->width(),Qt::FastTransformation);
    else
        scaleImg = img.scaledToHeight(ui->dstLabel->height(),Qt::FastTransformation);
    ui->dstLabel->setPixmap(QPixmap::fromImage(scaleImg));
}
//QImage转Mat
Mat MainWindow::QImage2cvMat(QImage image)
{
    cv::Mat mat;
    qDebug() << image.format();
    switch(image.format())
    {
    case QImage::Format_ARGB32:
    case QImage::Format_RGB32:
    case QImage::Format_ARGB32_Premultiplied:
        mat = cv::Mat(image.height(), image.width(), CV_8UC4, (void*)image.constBits(), image.bytesPerLine());
        break;
    case QImage::Format_RGB888:
        mat = cv::Mat(image.height(), image.width(), CV_8UC3, (void*)image.constBits(), image.bytesPerLine());
        cv::cvtColor(mat, mat, CV_BGR2RGB);
        break;
    case QImage::Format_Indexed8:
        mat = cv::Mat(image.height(), image.width(), CV_8UC1, (void*)image.constBits(), image.bytesPerLine());
        break;
    }
    return mat;
}
//Mat转QImage
QImage MainWindow::cvMat2QImage(const cv::Mat& mat)
{
    // 8-bits unsigned, NO. OF CHANNELS = 1
    if(mat.type() == CV_8UC1)
    {
        QImage image(mat.cols, mat.rows, QImage::Format_Indexed8);
        // Set the color table (used to translate colour indexes to qRgb values)
        image.setColorCount(256);
        for(int i = 0; i < 256; i++)
        {
            image.setColor(i, qRgb(i, i, i));
        }
        // Copy input Mat
        uchar *pSrc = mat.data;
        for(int row = 0; row < mat.rows; row ++)
        {
            uchar *pDest = image.scanLine(row);
            memcpy(pDest, pSrc, mat.cols);
            pSrc += mat.step;
        }
        return image;
    }
    // 8-bits unsigned, NO. OF CHANNELS = 3
    else if(mat.type() == CV_8UC3)
    {
        // Copy input Mat
        const uchar *pSrc = (const uchar*)mat.data;
        // Create QImage with same dimensions as input Mat
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
        return image.rgbSwapped();
    }
    else if(mat.type() == CV_8UC4)
    {
        qDebug() << "CV_8UC4";
        // Copy input Mat
        const uchar *pSrc = (const uchar*)mat.data;
        // Create QImage with same dimensions as input Mat
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);
        return image.copy();
    }
    else
    {
        qDebug() << "ERROR: Mat could not be converted to QImage.";
        return QImage();
    }
}