meshattributeassigndialog.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_DIALOGS_MESHATTRIBUTEASSIGNDIALOG_H
#define OPENSWMMVIS_UI_DIALOGS_MESHATTRIBUTEASSIGNDIALOG_H
 
#include "mesh/meshinfil.h"
#include "mesh/naturalnbinterpolator.h"
 
#include <QByteArray>
#include <QDialog>
#include <QFutureWatcher>
#include <QPointF>
#include <QPointer>
#include <QSet>
#include <QString>
#include <QStringList>
#include <QVector>
 
class MapCanvas;
class QComboBox;
class QDoubleSpinBox;
class QGroupBox;
class QLabel;
class QProgressBar;
class QPushButton;
class QRadioButton;
class QSpinBox;
class QCheckBox;
class QTableWidget;
class SelectionManager;
class SWMM2DMeshLayer;
 
namespace openswmmvis::ui {
 
class MeshAttributeAssignDialog : public QDialog
{
    Q_OBJECT
 
public:
    enum class Source { Raster, Vector };
 
    enum class Mode {
        SingleNumeric,     
        MultiNumeric,      
        ClassifiedInfil    
    };
 
    enum class Sampling {
        Centroid,          
        OverlayAuto,       
        AreaWeightedMean,  
        Majority,          
        NaturalNeighbour   
    };
 
    enum class WriteTarget {
        CellOverrides,     
        RegionDefaults     
    };
 
    // -----------------------------------------------------------------------
    // Data exchanged with the sampling worker.
    //
    // Both structs are plain data: no QObject, no layer pointer, no GDAL
    // handle. collectJob() fills the Job on the GUI thread; the worker
    // re-opens every source by path on its own thread.
    // -----------------------------------------------------------------------
 
    struct Job
    {
        Mode     mode     = Mode::SingleNumeric;
        Source   source   = Source::Raster;
        Sampling sampling = Sampling::Centroid;
 
        QVector<int>     triangles;
        QVector<QPointF> centroids;
        QVector<QPointF> triVerts;
 
        QString meshCrsWkt;          
 
        // ---- Raster source ----
        QString      rasterPath;
        QVector<int> bands;          
        int          keyBand1 = 1;   
        int          keyBand2 = 0;   
        double       scale    = 1.0;
        double       offset   = 0.0;
 
        // ---- Vector source ----
        QString         vectorPath;
        QString         vectorLayerName;
        QString         vectorFilterExpr;
        QStringList     fields;      
        QString         keyField1;   
        QString         keyField2;   
        bool            filterBySelection = false;
        QSet<long long> selectedIds;
 
        // ---- Numeric targets ----
        QVector<QByteArray> targetKeys;
        QVector<double>     targetMin;   
        QVector<double>     targetMax;   
 
        // ---- Classified infiltration ----
        mesh::InfilLookupTable table;
 
        // ---- Natural neighbour ----
        mesh::NaturalNeighbourInterpolator::Variant nnVariant =
            mesh::NaturalNeighbourInterpolator::Variant::Sibson;
    };
 
    struct SampleResult
    {
        QVector<int> triangles;             
        QVector<QVector<double>> values;
        QVector<mesh::InfilRow> rows;
        QStringList keys;
 
        int  skippedNoData     = 0;  
        int  skippedNonNumeric = 0;  
        int  skippedRange      = 0;  
        int  unmatchedKeys     = 0;  
        int  scanned           = 0;  
        bool cancelled         = false;
        QString resolvedSampling;
        QString error;               
    };
 
    MeshAttributeAssignDialog(SWMM2DMeshLayer  *meshLayer,
                              MapCanvas        *canvas,
                              SelectionManager *selection,
                              Source            initialSource,
                              const QString    &depthUnitLabel,
                              QWidget          *parent = nullptr);
 
public slots:
    void reject() override;
 
private slots:
    void onSourceChanged();
    void onTargetChanged();
    void onModeChanged();
    void onSamplingChanged();
    void onPreview();
    void onApply();
    void onBrowseRaster();
    void onCancelOrClose();
    void onSampleFinished();
 
    // Multi-target table
    void onAddTargetRow();
    void onRemoveTargetRow();
 
    // Classified lookup table
    void onAddLookupRow();
    void onRemoveLookupRow();
    void onLoadLookupCsv();
    void onSaveLookupCsv();
    void onTwoKeyToggled();
 
private:
    void buildUi(Source initialSource, const QString &depthUnitLabel);
    void buildTargetGroup(const QString &depthUnitLabel);
    void buildLookupGroup();
    void populateLayerCombos();
    void refreshVectorFields();
    void updateButtons();
    void setRunning(bool running);
 
    void fillParamCombo(QComboBox *combo, const QString &depthUnitLabel) const;
 
    [[nodiscard]] Mode     currentMode() const;
    [[nodiscard]] Sampling currentSampling() const;
 
    [[nodiscard]] QVector<int> scopeTriangles() const;
 
    [[nodiscard]] QVector<QPointF> centroidsFor(const QVector<int> &tris) const;
 
    [[nodiscard]] bool collectJob(Job *job, QString *err) const;
 
    void startSampling(bool apply);
 
    [[nodiscard]] QVector<QByteArray> collectTargetKeys() const;
 
    void setTargetRowSourceWidget(int row);
 
    [[nodiscard]] mesh::InfilLookupTable collectLookupTable() const;
    void                                 applyLookupTable(const mesh::InfilLookupTable &t);
    void                                 rebuildLookupColumns();
    void maskLookupRow(int row);
 
    [[nodiscard]] QString summarise(const SampleResult &r) const;
 
    void applyResult(const SampleResult &r);
    void applyNumericResult(const SampleResult &r);
    void applyInfilResult(const SampleResult &r);
    void applyInfilDefaultsResult(const SampleResult &r);
 
    QPointer<SWMM2DMeshLayer>  m_mesh;
    QPointer<MapCanvas>        m_canvas;
    QPointer<SelectionManager> m_selection;
 
    QComboBox      *m_modeCombo     = nullptr;
    QComboBox      *m_targetCombo   = nullptr;   // SingleNumeric
    QGroupBox      *m_singleGroup   = nullptr;
    QGroupBox      *m_multiGroup    = nullptr;
    QTableWidget   *m_targetTable   = nullptr;   // MultiNumeric
    QPushButton    *m_addTargetBtn  = nullptr;
    QPushButton    *m_delTargetBtn  = nullptr;
    QGroupBox      *m_lookupGroup   = nullptr;   // ClassifiedInfil
    QTableWidget   *m_lookupTable   = nullptr;
    QComboBox      *m_key1Combo     = nullptr;
    QComboBox      *m_key2Combo     = nullptr;
    QSpinBox       *m_key1BandSpin  = nullptr;
    QSpinBox       *m_key2BandSpin  = nullptr;
    QCheckBox      *m_twoKeyCheck   = nullptr;
    QLabel         *m_key1Label     = nullptr;
    QLabel         *m_key2Label     = nullptr;
 
    QRadioButton   *m_srcRaster     = nullptr;
    QRadioButton   *m_srcVector     = nullptr;
    QComboBox      *m_rasterCombo   = nullptr;
    QPushButton    *m_browseBtn     = nullptr;
    QSpinBox       *m_bandSpin      = nullptr;
    QDoubleSpinBox *m_scaleSpin     = nullptr;
    QDoubleSpinBox *m_offsetSpin    = nullptr;
    QComboBox      *m_vectorCombo   = nullptr;
    QComboBox      *m_fieldCombo    = nullptr;
    QCheckBox      *m_selectedOnly  = nullptr;
 
    QComboBox      *m_samplingCombo = nullptr;
    QComboBox      *m_nnVariantCombo = nullptr;
    QLabel         *m_nnVariantLbl  = nullptr;
 
    QRadioButton   *m_writeOverrides = nullptr;
    QRadioButton   *m_writeDefaults  = nullptr;
    QCheckBox      *m_keepInherited  = nullptr;
    QGroupBox      *m_writeGroup     = nullptr;
 
    QRadioButton   *m_scopeAll      = nullptr;
    QRadioButton   *m_scopeSelected = nullptr;
    QLabel         *m_statusLbl     = nullptr;
    QProgressBar   *m_progress      = nullptr;
    QPushButton    *m_previewBtn    = nullptr;
    QPushButton    *m_applyBtn      = nullptr;
    QPushButton    *m_closeBtn      = nullptr;
 
    QString m_browsedRasterPath;
    QString m_depthUnitLabel;
 
    QFutureWatcher<SampleResult> *m_watcher = nullptr;
    Job m_runJob;
    bool m_applyPending = false;
};
 
} // namespace openswmmvis::ui
 
#endif // OPENSWMMVIS_UI_DIALOGS_MESHATTRIBUTEASSIGNDIALOG_H