mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QFileDialog>
#include <QStandardItemModel>
#include <QMessageBox>
#include <QProcess>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    QStringList rulelist;
    rulelist << "point" << "pointOnSurface" << "centroid" << "mean" << "stdev" << "median" << "proportion" << "min" << "max" << "mode" << "sum";
    ui->rule->addItems(rulelist);
    QStringList formatlist;
    formatlist << "SQLite" << "ESRI Shapefile";
    ui->f->addItems(formatlist);

    setDefaults();
}

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

void MainWindow::setDefaults()
{
    //tab input/output
    ui->input->clear();
    ui->sample->clear();
    ui->bndnodata->clear();
    ui->srcnodata->clear();
    ui->polygon->setChecked(false);
    ui->f->setCurrentIndex(0);
    ui->output->clear();
    //tab extract
    ui->bname->setText("b");
    ui->cname->setText("label");
    ui->rule->setCurrentIndex(0);
    ui->nclass->clear();
    QStringList labels;
    setClassTable(labels);
    ui->threshold->clear();
}

void MainWindow::on_actionInput_triggered()
{
    QString qsinput = QFileDialog::getOpenFileName(this, "Input");
    ui->input->setText(qsinput);
}

void MainWindow::on_actionSample_triggered()
{
    QString qssample = QFileDialog::getOpenFileName(this, "Sample");
    ui->sample->setText(qssample);
}

void MainWindow::on_actionOutput_triggered()
{
    QString qsoutput = QFileDialog::getSaveFileName(this,"Output image","","*.*");
    ui->output->setText(qsoutput);
}

void MainWindow::on_toolButton_input_clicked()
{
    on_actionInput_triggered();
}

void MainWindow::on_toolButton_sample_clicked()
{
    on_actionSample_triggered();
}

void MainWindow::on_toolButton_output_clicked()
{
    on_actionOutput_triggered();
}

void MainWindow::setClassTable(const QStringList &labels)
{
    QStandardItemModel *model = new QStandardItemModel(labels.size(),2,this); //2 Rows and 3 Columns
    model->setHorizontalHeaderItem(0, new QStandardItem(QString("label name")));
    model->setHorizontalHeaderItem(1, new QStandardItem(QString("select(%)")));
    for(int ilabel=0;ilabel<labels.size();++ilabel){
        QStandardItem *firstCol = new QStandardItem(QString(labels[ilabel]));
        model->setItem(ilabel,0,firstCol);
        QStandardItem *secondCol = new QStandardItem(QString::number(100));
        model->setItem(ilabel,1,secondCol);
    }
    ui->tableView_labels->setModel(model);
}

void MainWindow::on_pushButton_run_clicked()
{
    try{
        ui->commandLineEdit->clear();
        ui->consoleEdit->clear();

        QString program = "pkextract";
        if(ui->sample->text().isEmpty())
            MainWindow::on_actionSample_triggered();
        if(ui->sample->text().isEmpty()){
            QString qsError="No sample image file selected";
            throw(qsError);
        }

        if(ui->input->text().isEmpty())
            MainWindow::on_actionInput_triggered();
        if(ui->input->text().isEmpty()){
            QString qsError="No input image file selected";
            throw(qsError);
        }

        if(ui->output->text().isEmpty())
            MainWindow::on_actionOutput_triggered();
        if(ui->output->text().isEmpty()){
            QString qsError="No output image file selected";
            throw(qsError);
        }

        program+=" --f "+ui->f->currentText();

        if(ui->polygon->isChecked())
            program+=" --polygon";

        program+=" --rule "+ui->rule->currentText();
//        QList<QComboBox*> qcomboBoxList = this->findChildren<QComboBox *>();

//        for(QList<QComboBox*>::ConstIterator qcbit=qcomboBoxList.begin();qcbit!=qcomboBoxList.end();++qcbit){
//            QString qsOption;
//            qsOption+=" --";
//            qsOption+=(*qcbit)->objectName();
//            program+=qsOption;
//            program+=" ";
//            program+=QString::number((*qcbit)->currentIndex());
//        }

        QList<QLineEdit*> qlineEditList = this->findChildren<QLineEdit *>();

        for(QList<QLineEdit*>::ConstIterator qlbit=qlineEditList.begin();qlbit!=qlineEditList.end();++qlbit){
            if(!((*qlbit)->text().isEmpty())){
                QString qsOption;
                qsOption+=" --";
                qsOption+=(*qlbit)->objectName();
                qsOption+=" ";
                qsOption+=(*qlbit)->text();
                program+=qsOption;
            }
        }

        //class table
        for(int irow=0;irow<ui->tableView_labels->model()->rowCount();++irow){
            QString qsOption;
            qsOption+=" --class ";
            qsOption+=ui->tableView_labels->model()->data(ui->tableView_labels->model()->index(irow,0)).toString();
            qsOption+=" --threshold ";
            qsOption+=ui->tableView_labels->model()->data(ui->tableView_labels->model()->index(irow,1)).toString();
            program+=qsOption;
        }

        ui->commandLineEdit->insert(program);

//        QProcess *myProcess = new QProcess(parent);
        QProcess *myProcess = new QProcess(this);
        myProcess->start(program);
        myProcess->setProcessChannelMode(QProcess::MergedChannels);
        this->setCursor(Qt::WaitCursor);
        myProcess->waitForFinished(-1);
        this->setCursor(Qt::ArrowCursor);
//        QString p_stderr = myProcess->readyReadStandardError();
//        if(!p_stderr.isEmpty()){
//            QMessageBox msgBox;
//            msgBox.setText(p_stderr);
//            msgBox.exec();
//        }
        QString p_stdout = myProcess->readAll();
        ui->consoleEdit->insertPlainText(p_stdout);
        delete myProcess;
    }
    catch(QString qsError){
        QMessageBox msgBox;
        msgBox.setText(qsError);
        msgBox.exec();
    }
}

void MainWindow::on_toolButton_createTable_clicked()
{
    int nclass=ui->nclass->text().toInt();
    QStringList labels;
    for(int iclass=1;iclass<=nclass;++iclass){
        QString lstring="label";
        lstring+=QString::number(iclass);
        labels << lstring;
    }
    setClassTable(labels);
}

void MainWindow::on_pushButton_restore_clicked()
{
    setDefaults();
}