featurestore.h

Go to the documentation of this file.

 
#ifndef FEATURESTORE_H
#define FEATURESTORE_H
 
#include "feature/featuregeometry.h"
#include "feature/featuretypes.h"
 
#include <QString>
#include <QStringList>
#include <QVector>
 
class GDALDataset;
class OGRLayer;
class OGRFeature;
 
namespace openswmmvis::feature {
 
class FeatureStore
{
public:
    FeatureStore();
    ~FeatureStore();
 
    FeatureStore(const FeatureStore &) = delete;
    FeatureStore &operator=(const FeatureStore &) = delete;
 
    // ----- Container-level operations (static; open nothing on success) ----
 
    [[nodiscard]] static bool ensureGeoPackage(const QString &gpkgPath,
                                               QString *error = nullptr);
 
    [[nodiscard]] static QStringList tables(const QString &gpkgPath);
 
    [[nodiscard]] static bool tableExists(const QString &gpkgPath,
                                          const QString &table);
 
    [[nodiscard]] static QString uniqueTableName(const QString &gpkgPath,
                                                 const QString &desired);
 
    [[nodiscard]] static bool createTable(const QString &gpkgPath,
                                          const QString &table,
                                          GeometryType   type,
                                          bool           hasZ,
                                          const Schema  &schema,
                                          const QString &srsWkt,
                                          QString       *error = nullptr);
 
    [[nodiscard]] static bool dropTable(const QString &gpkgPath,
                                        const QString &table,
                                        QString       *error = nullptr);
 
    // ----- Session --------------------------------------------------------
 
    [[nodiscard]] bool open(const QString &gpkgPath, const QString &table,
                            QString *error = nullptr);
 
    void attach(GDALDataset *ds, OGRLayer *layer,
                const QString &gpkgPath, const QString &table);
 
    void close();
    [[nodiscard]] bool isOpen() const { return m_layer != nullptr; }
 
    [[nodiscard]] QString gpkgPath()  const { return m_gpkgPath; }
    [[nodiscard]] QString tableName() const { return m_table; }
 
    // ----- Introspection --------------------------------------------------
 
    [[nodiscard]] Schema schema() const;
    [[nodiscard]] GeometryType geometryType() const;
    [[nodiscard]] bool hasZ() const;
    [[nodiscard]] QString srsWkt() const;
    [[nodiscard]] int count() const;
 
    // ----- Reads ----------------------------------------------------------
 
    [[nodiscard]] QVector<FeatureId> featureIds() const;
    [[nodiscard]] bool feature(FeatureId id, Feature &out) const;
    [[nodiscard]] QVector<Feature> allFeatures() const;
 
    // ----- Writes ---------------------------------------------------------
 
    [[nodiscard]] FeatureId addFeature(const Feature &f, QString *error = nullptr);
 
    [[nodiscard]] bool setGeometry(FeatureId id, const FeatureGeometry &g,
                                   QString *error = nullptr);
    [[nodiscard]] bool setAttributes(FeatureId id, const QVariantMap &attrs,
                                     QString *error = nullptr);
    [[nodiscard]] bool setFeature(FeatureId id, const Feature &f,
                                  QString *error = nullptr);
    [[nodiscard]] bool removeFeature(FeatureId id, QString *error = nullptr);
 
    // ----- Schema evolution ----------------------------------------------
 
    [[nodiscard]] bool addField(const FieldDef &f, QString *error = nullptr);
    [[nodiscard]] bool removeField(const QString &name, QString *error = nullptr);
 
    bool flush();
 
private:
    [[nodiscard]] OGRFeature *fetch(FeatureId id) const;
    void applyAttributes(OGRFeature *feat, const QVariantMap &attrs) const;
    [[nodiscard]] QVariantMap readAttributes(const OGRFeature *feat) const;
 
    GDALDataset *m_dataset = nullptr;   
    OGRLayer    *m_layer   = nullptr;   
    bool         m_ownsDataset = true;
    QString      m_gpkgPath;
    QString      m_table;
};
 
}   // namespace openswmmvis::feature
 
#endif // FEATURESTORE_H