plotvariablepickerdialog.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_DIALOGS_PLOTVARIABLEPICKERDIALOG_H
#define OPENSWMMVIS_UI_DIALOGS_PLOTVARIABLEPICKERDIALOG_H
 
#include "plot/irunlayer.h"
#include "plot/plotattribute.h"
 
#include <QDialog>
#include <QVector>
 
class QDialogButtonBox;
class QLineEdit;
class QTreeWidget;
class QTreeWidgetItem;
 
namespace openswmmvis::ui {
 
class PlotVariablePickerDialog : public QDialog
{
    Q_OBJECT
 
public:
    struct Entry {
        openswmmvis::plot::ObjectRef     ref;
        openswmmvis::plot::PlotAttribute attribute =
            openswmmvis::plot::PlotAttribute::Unknown;
        QString                          species;
 
        openswmmvis::plot::ResultDescriptor descriptor() const
        {
            return species.isEmpty()
                ? openswmmvis::plot::ResultDescriptor::forAttribute(attribute)
                : openswmmvis::plot::ResultDescriptor::forSpecies(species);
        }
    };
 
    explicit PlotVariablePickerDialog(
        const QVector<openswmmvis::plot::ObjectRef> &features,
        const openswmmvis::plot::IRunLayer *availability,
        openswmmvis::plot::UnitSystem units,
        QWidget *parent = nullptr);
 
    [[nodiscard]] QVector<Entry> checkedEntries() const;
 
private:
    void buildTree(const QVector<openswmmvis::plot::ObjectRef> &features,
                   const openswmmvis::plot::IRunLayer *availability,
                   openswmmvis::plot::UnitSystem units);
    void addAttributeLeaf(QTreeWidgetItem *group,
                          const openswmmvis::plot::ObjectRef &ref,
                          openswmmvis::plot::PlotAttribute attr,
                          const openswmmvis::plot::IRunLayer *availability,
                          openswmmvis::plot::UnitSystem units);
    void addDescriptorLeaf(QTreeWidgetItem *group,
                           const openswmmvis::plot::ObjectRef &ref,
                           const openswmmvis::plot::ResultDescriptor &d,
                           const openswmmvis::plot::IRunLayer *availability,
                           openswmmvis::plot::UnitSystem units);
    void setAllChecked(bool checked);
    void invertChecked();
    void applyFilter(const QString &text);
    void updateOkEnabled();
 
    QTreeWidget      *m_tree    = nullptr;
    QLineEdit        *m_filter  = nullptr;
    QDialogButtonBox *m_buttons = nullptr;
};
 
} // namespace openswmmvis::ui
 
#endif // OPENSWMMVIS_UI_DIALOGS_PLOTVARIABLEPICKERDIALOG_H