vespucciworkspace.h

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/>.
*******************************************************************************/

#ifndef VESPUCCIWORKSPACE_H

#define VESPUCCIWORKSPACE_H

#include "Data/Dataset/vespuccidataset.h"

#include "Math/VespucciMath.h"
#include "Global/vespucci.h"
#include <qcustomplot.h>
#include "GUI/mainwindow.h"
#include "GUI/QAbstractItemModel/datasettreemodel.h"
#include "Global/enums.h"
#include "Global/datamodel.h"
#include "Global/global.h"
#include "Data/Analysis/analysisresults.h"
class VespucciDataset;
class MainWindow;
class DatasetTreeModel;
class AnalysisResults;

class VespucciWorkspace
{
public:
    VespucciWorkspace(QString settings_file);
    ~VespucciWorkspace();
    QStringList dataset_names() const;
    //QStringList map_names();

    //adds or removes dataset or map to relevant lists
    void AddDataset(QSharedPointer<VespucciDataset> dataset);
    void RemoveDataset(QString name);
    void UpdateModel();

    void RemoveDatasetAt(const QModelIndex &parent);
    //void RemoveMapAt(int i);
    QSharedPointer<VespucciDataset> DatasetAt(const QModelIndex &parent);

    int dataset_loading_count() const;
    //int map_loading_count();

    void set_directory(QString directory);
    QString directory() const;
    QString *directory_ptr();

    void SetPointers(MainWindow *main_window, DatasetTreeModel *tree_model);

    MainWindow* main_window();
    double GetWavelengthMin(const QString &key) const;
    double GetWavelengthMax(const QString &key) const;

    QSharedPointer<VespucciDataset> GetDataset(const QString &key) const;
    QSharedPointer<AnalysisResults> GetAnalysisResults(const QString &dataset_key, const QString &results_key) const;
    QSharedPointer<MapData> GetMap(const QString &dataset_key, const QString &map_key) const;
    QSharedPointer<MapData> GetMap(const QStringList &map_keys);
    const mat & GetAuxiliaryMatrix(const QString &dataset_key, const QString &matrix_key) const;
    const mat & GetResultsMatrix(const QString &dataset_key, const QString &results_key, const QString &matrix_key) const;
    const mat &GetCoreMatrix(const QString &dataset_key, const QString &matrix_key) const;



    QCPRange *global_data_range();
    QCPColorGradient *global_gradient();

    void AddGlobalGradient(QString name, QString gradient_key, double lower, double upper);
    void RecalculateGlobalGradient(QString name);
    void RemoveColorRange(QString name);
    QStringList GlobalGradientKeys();
    QMap<QString, Vespucci::GlobalGradient> global_gradients();
    QCPColorGradient GetGradient(QString key, int count = 0);
    Vespucci::GlobalGradient GetGlobalGradient(QString key);
    QStringList GradientNames(bool include_cluster = false);


    void ClearDatasets();

    QList<QSharedPointer<VespucciDataset> > *datasets();
    unsigned int UpdateCount();
    DatasetTreeModel *dataset_tree_model() const;

    void ResetSettings();
    void CheckSettings();
    QSettings *settings();


    DataModel *data_model();

    const mat & GetMatrix(const QStringList &keys) const;
    bool HasMatrix(const QStringList &keys) const;

    bool Mappable(const QStringList &keys) const;
    bool Plottable(const QStringList &keys) const;

private:

    void UpdateTreeModel();

    //pointers to main window and necessary widgets

    MainWindow *main_window_;

    DatasetTreeModel *dataset_tree_model_;

    QString directory_; //= QDir::homePath();


    unsigned int dataset_loading_count_;

    QCPRange global_data_range_;

    QCPColorGradient global_color_gradient_;

    QSettings settings_;

    DataModel *data_model_;

    QMap<QString, QCPColorGradient> gradients_;
    QMap<QString, Vespucci::GlobalGradient> global_gradients_;
};

#endif // VESPUCCIWORKSPACE_H