mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "m_dialog.h"
#include <QPainter>
#include <QDebug>
#include <QFile>
#include <QDockWidget>
#include <QTextEdit>
#include <QSplitter>
#include <QFileDialog>
#include <QMessageBox>
#include <QtCharts>

#include "imagepro.h"

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    setGeometry(300,300,800,600);
    createDockWidget();

//    QFile file("F:\\Code\\BiliCG\\BiliCG\\bmp256.bmp");
//    if (!file.open(QIODevice::ReadOnly))
//    {
//        qDebug() << "Open Error";
//    }
//    QDataStream in(&file);
//    in.setByteOrder(QDataStream::LittleEndian);
//    uchar header;
//    in >> header;
//    qDebug() << QString(header);
//    in >> header;
//    qDebug() << QString(header);
//    uint32_t size = 0;
//    in >> size;
//    qDebug() << size;
//    file.close();
    connect(ImagePro::Instance(), &ImagePro::showSrc, srcWidget, &SrcImageWidget::showPic);
    connect(ImagePro::Instance(), &ImagePro::showDst, dstWidget, &DstImageWidget::showPic);
    connect(ImagePro::Instance(), &ImagePro::disableUpdate, dstWidget, &DstImageWidget::disableUpdate);
}

MainWindow::~MainWindow()
{
    delete ui;
    qDebug() << "~MainWindow";
}

void MainWindow::createDockWidget()
{
    srcWidget = new SrcImageWidget(this);
    dstWidget = new DstImageWidget(this);
    QSplitter *qs = new QSplitter(this);
    qs->setOrientation(Qt::Horizontal);
    qs->addWidget(srcWidget);
    qs->addWidget(dstWidget);
    setCentralWidget(qs);
}


void MainWindow::on_actionOpen_Image_triggered()
{
    qDebug() << "Open file";
    QString fileName = QFileDialog::getOpenFileName(this,
        tr("Open Image"), QString(), tr("Image Files (*.png *.jpg *.jpeg *.bmp *.gif)"));
    qDebug() << fileName;
    if(fileName.isEmpty())
    {
        return;
    }
    if(!ImagePro::Instance()->setPic(fileName))
    {
        QMessageBox::warning(this, tr("Error"), tr("Failed to open image file!\nPlease check if the file exists and is a valid image."));
        return;
    }
}

void MainWindow::createBarView(double *serial, int size)
{
    if(serial == nullptr || size <= 0)
        return;
    
    QBarSet *set0 = new QBarSet(tr("Histogram"));
    set0->setColor(Qt::black);
    QStringList categories;
    
    // 准备数据，只显示前256个值（灰度值范围）
    int displaySize = qMin(size, 256);
    for(int i = 0; i < displaySize; ++i)
    {
        int value = static_cast<int>(serial[i] * 1000);
        *set0 << value;
        categories << QString::number(i);
    }

    QBarSeries *series = new QBarSeries();
    series->append(set0);

    QChart *chart = new QChart();
    chart->addSeries(series);
    chart->setTitle(tr("Histogram Chart"));
    chart->setAnimationOptions(QChart::SeriesAnimations);

    QBarCategoryAxis *axisX = new QBarCategoryAxis();
    axisX->append(categories);
    chart->addAxis(axisX, Qt::AlignBottom);
    series->attachAxis(axisX);

    QValueAxis *axisY = new QValueAxis();
    axisY->setRange(0, 1000);
    axisY->setTitleText(tr("Frequency"));
    chart->addAxis(axisY, Qt::AlignLeft);
    series->attachAxis(axisY);

    chart->legend()->setVisible(true);
    chart->legend()->setAlignment(Qt::AlignBottom);

    QChartView *chartView = new QChartView(chart);
    chartView->setRenderHint(QPainter::Antialiasing);
    chartView->setWindowTitle(tr("Histogram"));
    chartView->resize(800, 400);
    chartView->show();
}

void MainWindow::on_actionGray_triggered()
{
    ImagePro::Instance()->doProcess(GRAY);
}

void MainWindow::on_actionBinary_triggered()
{
     M_Dialog md(this,true,true);
     if(md.exec() == QDialog::Accepted)
     {
         ImagePro::Instance()->doProcess(BINARY,false,md.getValue());
     }
}

void MainWindow::on_actionHis_triggered()
{
    createBarView(ImagePro::Instance()->calHistogram(),256);
}

void MainWindow::on_actionHistogram_Equalization_triggered()
{
    createBarView(ImagePro::Instance()->his_equal(),256);
}

void MainWindow::on_actionGua_triggered()
{
    ImagePro::Instance()->doProcess(GAUSSIAN_BLUR);
}

void MainWindow::on_actionsobel_triggered()
{
    M_Dialog md(this, true, false);
    bool enhanced = (md.exec() == QDialog::Accepted);
    ImagePro::Instance()->doProcess(SOBEL, enhanced);
}

void MainWindow::on_horizontal_triggered()
{
    ImagePro::Instance()->doProcess(H_SHARPEN);
}

void MainWindow::on_vertical_triggered()
{
    ImagePro::Instance()->doProcess(V_SHARPEN);
}

void MainWindow::on_actiondual_triggered()
{
    ImagePro::Instance()->doProcess(DUAL_SHARPEN);
}

void MainWindow::on_actionmid_triggered()
{
     ImagePro::Instance()->doProcess(MED_FILTER);
}


void MainWindow::on_actionsquare_mask_triggered()
{
    ImagePro::Instance()->doProcess(SQUARE_MASK);
}


void MainWindow::on_actionFourier_triggered()
{
    ImagePro::Instance()->doProcess(FOURIER);
}

void MainWindow::on_actionInv_Fourier_triggered()
{
    ImagePro::Instance()->doProcess(INV_FOURIER);
}

void MainWindow::on_actionswitch_triggered()
{
    ImagePro::Instance()->swap();
}

void MainWindow::on_actionfushi_triggered()
{
    ImagePro::Instance()->doProcess(EROSION);
}

void MainWindow::on_actionpengzhang_triggered()
{
    ImagePro::Instance()->doProcess(DILATION);
}