mapdialog.cpp

Go to the documentation of this file.

/*******************************************************************************
    Copyright (C) 2014-2016 Wright State University - All Rights Reserved
    Daniel P. Foose - Maintainer/Lead Developer

    This file is part of Vespucci.

    Vespucci is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Vespucci is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Vespucci.  If not, see <http://www.gnu.org/licenses/>.
*******************************************************************************/
#include "mapdialog.h"
#include "ui_mapdialog.h"

MapDialog::MapDialog(MainWindow *parent, QStringList data_keys, QSharedPointer<VespucciWorkspace> ws) :
    QDialog(parent),
    ui(new Ui::MapDialog)
{
    ui->setupUi(this);
    workspace_ = ws;
    data_keys_ = data_keys;
    main_window_ = parent;

    if (data_keys_.size() > 3 || data_keys_.size() < 2 || !workspace_->Mappable(data_keys)){
        QMessageBox::warning(this, "Cannot map this object", "Object is not mappable");
        hide();
        close();
        return;
    }

    uword col_count;
    try{
        col_count = workspace_->GetMatrix(data_keys_).n_cols;
    }catch(exception e){
        main_window_->DisplayExceptionWarning("MapDialog constructor, VespucciWorkspace::GetMatrix", e);
        close();
    }

    try{
        dataset_ = workspace_->data_model()->GetDataset(data_keys_[0]);
    }catch(exception e){
        main_window_->DisplayExceptionWarning("MapDialog constructor, DataModel::GetDataset", e);
        close();
    }

    ui->columnSpinBox->setRange(1, col_count);
    ui->nameLineEdit->setText(data_keys_.last());
    uword unique_ct = vec(unique(workspace_->GetMatrix(data_keys_).col(0))).n_elem;
    QStringList color_list = workspace_->GradientNames(unique_ct < 10);
    ui->gradientComboBox->addItems(color_list);
}

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

void MapDialog::on_buttonBox_accepted()
{
    uword unique_ct = vec(unique(workspace_->GetMatrix(data_keys_).col(0))).n_elem;
    uint column_index = ui->columnSpinBox->value() - 1;
    QString map_name = ui->nameLineEdit->text();
    vec data;
    bool ok = true;
    try{
        data = workspace_->data_model()->GetMatrix(data_keys_).col(column_index);
    }catch(exception e){
        main_window_->DisplayExceptionWarning("DataModel::GetMatrix or mat::col", e);
        ok = false;
    }

    if (ok){
        QCPColorGradient gradient = workspace_->GetGradient(ui->gradientComboBox->currentText(), unique_ct);

        try{
            if (data_keys_.size() == 2)
                dataset_->CreateMap(map_name,
                                    data_keys_[1],
                                    column_index,
                                    gradient);
            else
                dataset_->CreateMap(map_name,
                                    data_keys_[1],
                                    data_keys_[2],
                                    column_index,
                                    gradient);
        }catch(exception e){
            main_window_->DisplayExceptionWarning("VespucciDataset::CreateMap()", e);
        }
    }
    close();
}

void MapDialog::on_columnSpinBox_valueChanged(int arg1)
{
    mat data;
    bool ok = true;
    try{
        data = workspace_->data_model()->GetMatrix(data_keys_).col(arg1-1);
    }catch(exception e){
        main_window_->DisplayExceptionWarning("DataModel::GetMatrix or mat::col", e);
        ok = false;
        close();
    }
    if (ok){
        vec unique_values = unique(data);
        QStringList color_list = workspace_->GradientNames(unique_values.n_rows < 10);
        ui->gradientComboBox->clear();
        ui->gradientComboBox->addItems(color_list);
    }
}