DiagramPrefsDialog.cpp

Go to the documentation of this file.

 
#include "DiagramPrefsDialog.h"
 
#include <QCheckBox>
#include <QFileDialog>
#include <QMessageBox>
 
#include "Base/OGSError.h"
#include "DetailWindow.h"
#include "DiagramList.h"
#include "GeoLib/Station.h"
#include "GetDateTime.h"
 
DiagramPrefsDialog::DiagramPrefsDialog(const GeoLib::Station* stn,
                                       const QString& listName,
                                       // DatabaseConnection* db,
                                       QDialog* parent)
    : QDialog(parent), _window(nullptr)
{
    setAttribute(Qt::WA_DeleteOnClose);
 
    setupUi(this);
    stationNameLabel->setText(QString::fromStdString(stn->getName()));
    stationTypeLabel->setText(listName);
}
 
DiagramPrefsDialog::DiagramPrefsDialog(GeoLib::Station* stn, QDialog* parent)
    : QDialog(parent), _window(nullptr)
{
    setupUi(this);
    stationNameLabel->setText(QString::fromStdString(stn->getName()));
    stationTypeLabel->setText("");
    DiagramList::readList(stn->getSensorData(), _list);
 
    fromDateLine->setText(
        QString::number(stn->getSensorData()->getStartTime()));
    toDateLine->setText(QString::number(stn->getSensorData()->getEndTime()));
    this->createVisibilityCheckboxes();
}
 
DiagramPrefsDialog::DiagramPrefsDialog(const QString& filename,
                                       DetailWindow* window,
                                       QDialog* parent)
    : QDialog(parent), _window(window)
{
    QFileInfo fi(filename);
    setupUi(this);
    stationNameLabel->setText(fi.baseName());
    stationTypeLabel->setText("");
    this->loadFile(filename);
}
 
DiagramPrefsDialog::~DiagramPrefsDialog()
{
    this->destroy();
}
 
void DiagramPrefsDialog::accept()
{
    QDateTime start_date(getDateTime(fromDateLine->text()));
    QDateTime end_date(getDateTime(toDateLine->text()));
 
    if (start_date == QDateTime() || end_date == QDateTime() ||
        start_date > end_date || _list.empty())
    {
        OGSError::box("No data found...");
        return;
    }
 
    if (_list[0]->size() == 0)
    {
        OGSError::box("Invalid station data.");
        this->done(QDialog::Rejected);
    }
 
    // Data has been loaded.
    // If loading lists beyond the first one fails at least nothing terrible
    // will happen.
    bool window_is_empty(false);
    if (_window == nullptr)
    {
        _window = new DetailWindow();
        _window->setAttribute(Qt::WA_DeleteOnClose);
        window_is_empty = true;
    }
 
    for (std::size_t i = 0; i < _list.size(); i++)
    {
        if (_visability[i]->isChecked())
        {
            _list[i]->truncateToRange(start_date, end_date);
            _window->addList(_list[i]);
            window_is_empty = false;
        }
    }
 
    if (!window_is_empty)
    {
        _window->show();
        this->done(QDialog::Accepted);
    }
    else
    {
        delete _window;
        _window = nullptr;
        OGSError::box("No dataset selected.");
    }
}
 
void DiagramPrefsDialog::reject()
{
    this->done(QDialog::Rejected);
}
 
void DiagramPrefsDialog::on_loadFileButton_clicked()
{
    QString fileName =
        QFileDialog::getOpenFileName(this,
                                     "Select time series file to open",
                                     "",
                                     "Time series files (*.stn *.txt)");
    if (!fileName.isEmpty())
    {
        loadFile(fileName);
    }
}
 
int DiagramPrefsDialog::loadFile(const QString& filename)
{
    if (DiagramList::readList(filename, _list))
    {
        for (auto& item : _list)
        {
            // item->setName(stationTypeLabel->text() + ": " +
            // stationNameLabel->text());
            item->setXLabel("Time");
            // item->setYLabel("Water Level");
            item->setXUnit("day");
            // item->setYUnit("metres");
            item->setColor(QColor(Qt::red));
        }
        fromDateLine->setText(_list[0]->getStartDate().toString("dd.MM.yyyy"));
        QDateTime endDate = _list[0]->getStartDate().addSecs(
            static_cast<int>(_list[0]->maxXValue()));
        toDateLine->setText(endDate.toString("dd.MM.yyyy"));
        this->createVisibilityCheckboxes();
        return 1;
    }
 
    OGSError::box("Error reading file.");
    return 0;
}
 
int DiagramPrefsDialog::loadList(
    const std::vector<std::pair<QDateTime, float>>& coords)
{
    if (!coords.empty())
    {
        auto* l = new DiagramList;
        l->setName(stationTypeLabel->text() + ": " + stationNameLabel->text());
        l->setXLabel("Time");
        // l->setYLabel("Water Level");
        l->setXUnit("day");
        // l->setYUnit("metres");
        l->setColor(QColor(Qt::red));
        l->setList(coords);
        _list.push_back(l);
        return 1;
    }
    return 0;
}
 
void DiagramPrefsDialog::createVisibilityCheckboxes()
{
    for (auto& item : _list)
    {
        QCheckBox* box = new QCheckBox(item->getName());
        box->setChecked(true);
        this->CheckBoxLayout->addWidget(box);
        _visability.push_back(box);
    }
}