metadataset.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 "Data/Dataset/metadataset.h"


MetaDataset::MetaDataset(QString name,
                         MainWindow *main_window,
                         QString *directory,
                         QString method_description,
                         MetaMethod::Method method,
                         QList<QSharedPointer<VespucciDataset> > parent_datasets)
    : VespucciDataset(name, main_window, directory)
{
    parent_datasets_ = parent_datasets;

    if(!ParentsValid()){
        throw std::runtime_error("Improper input to MetaDataset constructor");
        cerr << "Improper input to MetaDataset constructor\n";

    }
    method_ = method;
    method_description_ = method_description;
    mat spectra;
    vec wavelength = parent_datasets_[0]->wavelength();
    vec x;
    vec y;
    switch(method_) {
    case MetaMethod::AverageSpectra :
        try{
            spectra = ProcessAverage(x, y);
        }
        catch(std::exception e){
            throw std::runtime_error("MetaDataset::ProcessAverage");
        }

        break;
    case MetaMethod::ConcatenateDatasets :
        try{
            spectra = Concatenate(x, y);
        }
        catch(std::exception e){
            throw std::runtime_error("MetaDataset::Concatenate");
        }
        break;
    default:
        throw std::runtime_error("Improper input to MetaDataset");
    }

    try{
        SetData(spectra, wavelength, x, y);
        SetParentDatasetIndices(parent_coordinates_);
    }
    catch(std::exception e){
        throw std::runtime_error("Failure to set data in MetaDataset constructor");
    }
    vec indices_temp(spectra.n_cols);
    for (uword i = 0; i < indices_temp.n_elem; ++i){
        indices_temp(i) = i;
    }
    SetIndices(indices_temp);
}

mat MetaDataset::ProcessAverage(vec &x, vec &y)
{
    mat spectra;
    vec parent(1);
    mat indices_buf(1, 3);
    for (int i = 0; i < parent_datasets_.size(); ++i){
        spectra.insert_cols(spectra.n_cols, parent_datasets_[i]->AverageSpectrum(false));
        parent(0) = i + 1;
        indices_buf(0, 0) = i + 1;
        indices_buf(0, 1) = 0;
        indices_buf(0, 2) = 0;
        x.insert_rows(x.n_rows, parent);
        y.insert_rows(y.n_rows, parent);
        parent_coordinates_.insert_rows(parent_coordinates_.n_rows, indices_buf);
    }
    return spectra;
}

mat MetaDataset::Concatenate(vec &x, vec &y)
{
    mat spectra;
    mat indices_buf;
    indices_buf.set_size(1, 3);
    cout << "Concatenate()\n";

    for (int i = 0; i < parent_datasets_.size(); ++i){
        spectra.insert_cols(spectra.n_cols, parent_datasets_[i]->spectra());
        indices_buf.set_size(parent_datasets_[i]->x().n_rows, 3);
        indices_buf.col(0) = (i+1)*ones(indices_buf.n_rows);
        indices_buf.col(1) = parent_datasets_[i]->x();
        indices_buf.col(2) = parent_datasets_[i]->y();
        parent_coordinates_.insert_rows(parent_coordinates_.n_rows, indices_buf);
        x.insert_rows(x.n_rows, indices_buf.col(1));
        y.insert_rows(y.n_rows, indices_buf.col(2));
    }
    return spectra;
}

bool MetaDataset::ParentsValid()
{
    uword size = parent_datasets_[0]->wavelength_ptr()->n_elem;
    for (int i = 0; i<parent_datasets_.size(); ++i)
        if (parent_datasets_[i]->wavelength_ptr()->n_elem != size)
            return false;

    return true;
}

vec *MetaDataset::parents()
{
    return &parents_;
}