File indexing completed on 2024-06-02 03:49:30

 /*
     File                 : BatchEditValueLabelsDialog.cpp
     Project              : LabPlot
     Description          : Dialog to modify multiply value labels in a batch mode
     --------------------------------------------------------------------
     SPDX-FileCopyrightText: 2021 Alexander Semke <alexander.semke@web.de>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "BatchEditValueLabelsDialog.h"
 #include "backend/core/Settings.h"
 #include "backend/core/column/Column.h"
 #include "backend/core/datatypes/DateTime2StringFilter.h"
 #include "backend/lib/macros.h"
 
 #include <QDialogButtonBox>
 #include <QLabel>
 #include <QPushButton>
 #include <QTextEdit>
 #include <QVBoxLayout>
 #include <QWindow>
 
 #include <KLocalizedString>
 
 #include <KWindowConfig>
 
 /*!
     \class BatchEditValueLabelsDialog
     \brief Dialog to add a new the value label
 
     \ingroup kdefrontend
  */
 BatchEditValueLabelsDialog::BatchEditValueLabelsDialog(QWidget* parent)
     : QDialog(parent)
     , teValueLabels(new QTextEdit()) {
     setWindowTitle(i18nc("@title:window", "Edit Value Labels"));
 
     auto* layout = new QVBoxLayout(this);
 
     auto* label = new QLabel(i18n("Value Labels:"));
     layout->addWidget(label);
 
     layout->addWidget(teValueLabels);
 
     auto* btnBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
     connect(btnBox, &QDialogButtonBox::accepted, this, &BatchEditValueLabelsDialog::save);
     connect(btnBox, &QDialogButtonBox::accepted, this, &BatchEditValueLabelsDialog::accept);
     connect(btnBox, &QDialogButtonBox::rejected, this, &BatchEditValueLabelsDialog::reject);
     layout->addWidget(btnBox);
 
     // restore saved settings if available
     KConfigGroup conf = Settings::group(QLatin1String("BatchEditValueLabelsDialog"));
 
     create(); // ensure there's a window created
     if (conf.exists()) {
         KWindowConfig::restoreWindowSize(windowHandle(), conf);
         resize(windowHandle()->size()); // workaround for QTBUG-40584
     } else
         resize(QSize(200, 0).expandedTo(minimumSize()));
 }
 
 BatchEditValueLabelsDialog::~BatchEditValueLabelsDialog() {
     // save the current settings
     KConfigGroup conf = Settings::group(QLatin1String("BatchEditValueLabelsDialog"));
     KWindowConfig::saveWindowSize(windowHandle(), conf);
 }
 
 void BatchEditValueLabelsDialog::setColumns(const QList<Column*>& columns) {
     m_columns = columns;
 
     if (m_columns.isEmpty())
         return;
 
     m_column = m_columns.first();
 
     // show the available value labels for the first columm
     if (m_column->valueLabelsInitialized()) {
         QString text;
 
         switch (m_column->labelsMode()) {
         case AbstractColumn::ColumnMode::Double: {
             const auto* labels = m_column->valueLabels();
             if (!labels)
                 return;
             auto it = labels->constBegin();
             while (it != labels->constEnd()) {
                 if (!text.isEmpty())
                     text += QLatin1Char('\n');
                 text += QString::number(it->value) + QLatin1String(" = ") + it->label;
                 ++it;
             }
             break;
         }
         case AbstractColumn::ColumnMode::Integer: {
             const auto* labels = m_column->intValueLabels();
             if (!labels)
                 return;
             auto it = labels->constBegin();
             while (it != labels->constEnd()) {
                 if (!text.isEmpty())
                     text += QLatin1Char('\n');
                 text += QString::number(it->value) + QLatin1String(" = ") + it->label;
                 ++it;
             }
             break;
         }
         case AbstractColumn::ColumnMode::BigInt: {
             const auto* labels = m_column->bigIntValueLabels();
             if (!labels)
                 return;
             auto it = labels->constBegin();
             while (it != labels->constEnd()) {
                 if (!text.isEmpty())
                     text += QLatin1Char('\n');
                 text += QString::number(it->value) + QLatin1String(" = ") + it->label;
                 ++it;
             }
             break;
         }
         case AbstractColumn::ColumnMode::Text: {
             const auto* labels = m_column->textValueLabels();
             if (!labels)
                 return;
             auto it = labels->constBegin();
             while (it != labels->constEnd()) {
                 if (!text.isEmpty())
                     text += QLatin1Char('\n');
                 text += it->value + QLatin1String(" = ") + it->label;
                 ++it;
             }
             break;
         }
         case AbstractColumn::ColumnMode::Month:
         case AbstractColumn::ColumnMode::Day:
         case AbstractColumn::ColumnMode::DateTime: {
             const auto* labels = m_column->dateTimeValueLabels();
             if (!labels)
                 return;
             const auto* filter = static_cast<DateTime2StringFilter*>(m_column->outputFilter());
             const auto& dateTimeFormat = filter->format();
             auto it = labels->constBegin();
             while (it != labels->constEnd()) {
                 if (!text.isEmpty())
                     text += QLatin1Char('\n');
                 text += it->value.toString(dateTimeFormat) + QLatin1String(" = ") + it->label;
                 ++it;
             }
             break;
         }
         }
 
         teValueLabels->setText(text);
     } else {
         QString text = i18n("Provide the list of value-label pairs like in the following example:");
         text += QLatin1String("\n1 = true\n0 = false");
         teValueLabels->setPlaceholderText(text);
     }
 }
 
 void BatchEditValueLabelsDialog::save() const {
     // remove all already available labels first
     for (auto* column : m_columns)
         column->valueLabelsRemoveAll();
 
     // add new labels
     const auto numberLocale = QLocale();
     QString label;
     QString valueStr;
     bool ok;
     const auto mode = m_column->columnMode();
     const QString& text = teValueLabels->toPlainText();
     const auto& lines = text.split(QLatin1Char('\n'));
 
     for (const auto& line : lines) {
         auto pair = line.split(QLatin1Char('='));
         if (pair.size() != 2)
             continue;
 
         valueStr = pair.at(0).simplified();
         label = pair.at(1).simplified();
         if (valueStr.isEmpty() || label.isEmpty())
             continue;
 
         switch (mode) {
         case AbstractColumn::ColumnMode::Double: {
             double value = numberLocale.toDouble(valueStr, &ok);
             if (!ok)
                 continue;
             for (auto* col : m_columns)
                 col->addValueLabel(value, label);
             break;
         }
         case AbstractColumn::ColumnMode::Integer: {
             int value = numberLocale.toInt(valueStr, &ok);
             if (!ok)
                 continue;
             for (auto* col : m_columns)
                 col->addValueLabel(value, label);
             break;
         }
         case AbstractColumn::ColumnMode::BigInt: {
             qint64 value = numberLocale.toLongLong(valueStr, &ok);
             if (!ok)
                 continue;
             for (auto* col : m_columns)
                 col->addValueLabel(value, label);
             break;
         }
         case AbstractColumn::ColumnMode::Text: {
             for (auto* col : m_columns)
                 col->addValueLabel(valueStr, label);
             break;
         }
         case AbstractColumn::ColumnMode::Month:
         case AbstractColumn::ColumnMode::Day:
         case AbstractColumn::ColumnMode::DateTime: {
             const auto* filter = static_cast<DateTime2StringFilter*>(m_column->outputFilter());
             const QDateTime& value = QDateTime::fromString(valueStr, filter->format());
             if (!value.isValid())
                 continue;
             for (auto* col : m_columns)
                 col->addValueLabel(value, label);
             break;
         }
         }
     }
 }