terrainsizefield.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_MESH_TERRAINSIZEFIELD_H
#define OPENSWMMVIS_MESH_TERRAINSIZEFIELD_H
 
#include <QPointF>
#include <QString>
#include <QVector>
 
#include <functional>
 
class GDALDataset;
 
namespace mesh
{
 
struct TerrainSizeOptions {
    double tolerance = 0.0; 
    int maxLevel = 10;      
    int outLevel = 2;       
    qint64 maxBandBytes = 256ll * 1024 * 1024; 
 
    std::function<void(const float *, int, int, int)> rowSink;
    bool rowsOnly = false; 
    int decimation = 1;
};
 
class TerrainSizeField
{
public:
    bool build(GDALDataset *ds, int band, int col0, int row0, int cols, int rows,
               const TerrainSizeOptions &opt, const std::function<bool(double)> &progress = {});
 
    bool buildFromFile(const QString &path, int band, double gx0, double gy0, double gx1,
                       double gy1, const TerrainSizeOptions &opt,
                       const std::function<bool(double)> &progress = {});
 
    [[nodiscard]] double sizeAtGeo(double x, double y) const;
    [[nodiscard]] double pixelSize() const { return m_pixelSize; }
    [[nodiscard]] QPointF windowPixelToGeo(double px, double py) const;
 
    bool buildFromGrid(const float *z, int cols, int rows, const TerrainSizeOptions &opt);
 
    [[nodiscard]] double sizePixelsAt(double px, double py) const;
 
    [[nodiscard]] bool isValid() const { return m_outCols > 0 && m_outRows > 0; }
    [[nodiscard]] int outCols() const { return m_outCols; }
    [[nodiscard]] int outRows() const { return m_outRows; }
    [[nodiscard]] int outLevel() const { return m_outLevel; }
    [[nodiscard]] int levelsUsed() const { return m_maxLevel; }
    [[nodiscard]] const QVector<float> &cells() const { return m_h; }
    [[nodiscard]] QString errorMsg() const { return m_errorMsg; }
 
private:
    void processBand(const float *z, int cols, int bandRows, int bandRow0, double tol);
    void dilateMinimum();
 
    QVector<float> m_h; 
    int m_outCols = 0, m_outRows = 0;
    int m_outLevel = 2, m_maxLevel = 10;
    int m_col0 = 0, m_row0 = 0, m_cols = 0, m_rows = 0;
    int m_decimation = 1;                 
    double m_geo[6] = {0, 1, 0, 0, 0, 1}; 
    double m_invGeo[6] = {0, 1, 0, 0, 0, 1};
    double m_pixelSize = 0.0;
    QString m_errorMsg;
};
 
} // namespace mesh
 
#endif // OPENSWMMVIS_MESH_TERRAINSIZEFIELD_H