DatabaseStatements.h

Go to the documentation of this file.

#ifndef DATABASESTATEMENTS_H
#define DATABASESTATEMENTS_H

#include <memory>
#include <vector>
#include <string>
#include <unordered_map>
#include <regex>

#include <sqlite3.h>

#include "SegmentationParameters.h"
#include "DatabaseTable.h"
#include "SpatialiteBlob.h"

namespace agroeye {
namespace handlers {
    
using agroeye::operations::StatisticsTypes;
using agroeye::general::geometry::SpatialiteBlobWkb;
    
class GeneralStatement {
public:    
    void setId(const uint32_t& id);    
    void process(); 
    void flush();   
protected:
    GeneralStatement() {};
    GeneralStatement(std::shared_ptr<sqlite3> aPtrSql3) : ptrSql3(aPtrSql3) {};
    GeneralStatement(GeneralStatement&&) = default;
    GeneralStatement& operator=(GeneralStatement&&) = default;
    
    void prepare(); 
    std::shared_ptr<sqlite3> ptrSql3 {nullptr}; 
    std::unique_ptr<sqlite3_stmt, int(*)(sqlite3_stmt*)> ptrStmt {nullptr, [](sqlite3_stmt* pStmt) {return sqlite3_finalize(pStmt);}};  
    std::string buffer;     
    size_t id_position {0}; 
};
    
class GeometryStatement : public GeneralStatement {
public:
    GeometryStatement() {};
    GeometryStatement(std::shared_ptr<sqlite3> ptrSql3, const DbTable& table); 
    GeometryStatement(GeometryStatement&&) = default;
    GeometryStatement& operator=(GeometryStatement&&) = default;
    void setGeometry(SpatialiteBlobWkb&& blob);
private:    
    size_t the_geom_position {0};
};


class StatisticStatement : public GeneralStatement {
public:
    StatisticStatement() {};
    StatisticStatement(std::shared_ptr<sqlite3> aPtrSql3, const DbTable& table);
    StatisticStatement(StatisticStatement&&) = default;
    StatisticStatement& operator=(StatisticStatement&&) = default;
    
    void setBandValue(const int& band, const double& value);

private:    
    using BandPosition = std::pair<int, int>;
    int extractBandValue(const std::string& bandName) const;    
    std::vector<BandPosition> bandPositionList;  
};

class PreparedStatement {
public:
    PreparedStatement() {}; 
    PreparedStatement(GeometryStatement&& aGeomStmt) : geomStmt(std::move(aGeomStmt)) {};
    
    void addStatStmt(const StatisticsTypes& type, StatisticStatement&& statStmt);
    
    void setId(uint32_t id);
    
    void setGeometry(SpatialiteBlobWkb&& blob);
    
    void setStatBandValue(const StatisticsTypes& type, const int& band, const double& value);
    void process(); 
    void flush();   
protected:
    GeometryStatement geomStmt;
    std::vector<std::pair<StatisticsTypes, StatisticStatement>> statStmts;
};

} // namespace handlers2
} // namespace agroeye

#endif