geodata.h

#ifndef __GEODATA_H__
#define __GEODATA_H__
 
#include "RasterTileExtractor.h"
#include "VectorExtractor.h"
#include "defines.h"
#include "geofeatures.h"
#include "geoimage.h"
#include "godot_cpp/variant/dictionary.hpp"
#include "godot_cpp/variant/variant.hpp"
 
namespace godot {
 
// Forward decaration
class EXPORT GeoDataset;
 
class EXPORT GeoFeatureLayer : public RefCounted {
    GDCLASS(GeoFeatureLayer, RefCounted)
 
  protected:
    static void _bind_methods();
 
  public:
    GeoFeatureLayer() = default;
    ~GeoFeatureLayer() = default; // No need to delete anything here - OGRLayers are part of
                                  // the dataset and deleted with it.
 
    bool is_valid();
 
    Dictionary get_file_info();
 
    Ref<GeoDataset> get_dataset();
 
    int get_epsg_code();
 
    Vector3 get_center();
 
    Ref<GeoFeature> get_feature_by_id(int id);
 
    Array get_all_features();
 
    Ref<GeoFeature> create_feature();
 
    void remove_feature(Ref<GeoFeature> feature);
 
    void clear_cache();
 
    void save_override();
 
    void save_new(String file_path);
 
    Array get_features_near_position(double pos_x, double pos_y, double radius, int max_features);
 
    Array get_features_in_square(double top_left_x, double top_left_y, double size_meters,
                               int max_features);
 
    Array get_features_by_attribute_filter(String filter);
 
    Ref<GeoFeature> get_specialized_feature(std::shared_ptr<Feature> raw_feature);
 
    void set_native_layer(std::shared_ptr<NativeLayer> new_layer);
 
    void set_origin_dataset(Ref<GeoDataset> dataset);
 
    String name;
 
  private:
    std::shared_ptr<NativeLayer> layer;
    std::map<std::shared_ptr<Feature>, Ref<GeoFeature>> feature_cache;
    Ref<GeoDataset> origin_dataset;
    ExtentData extent_data;
};
 
class EXPORT GeoRasterLayer : public RefCounted {
    GDCLASS(GeoRasterLayer, RefCounted)
 
  protected:
    static void _bind_methods();
 
  public:
    GeoRasterLayer() : origin_dataset(nullptr) {}
    ~GeoRasterLayer() = default;
 
    bool is_valid();
 
    bool has_write_access();
 
    Dictionary get_file_info();
 
    int get_epsg_code();
 
    Image::Format get_format();
 
    int get_band_count();
 
    Array get_band_descriptions();
 
    Ref<GeoDataset> get_dataset();
 
    Ref<GeoRasterLayer> clone();
 
    Ref<GeoImage> get_image(double top_left_x, double top_left_y, double size_meters, int img_size,
                            GeoImage::INTERPOLATION interpolation_type);
 
    Ref<GeoImage> get_band_image(double top_left_x, double top_left_y, double size_meters, int img_size,
                            GeoImage::INTERPOLATION interpolation_type, int band_index);
 
    float get_value_at_position(double pos_x, double pos_y);
 
    float get_value_at_position_with_resolution(double pos_x, double pos_y,
                                                double pixel_size_meters);
 
    void set_value_at_position(double pos_x, double pos_y, Variant value);
 
    void smooth_add_value_at_position(double pos_x, double pos_y, double summand, double radius);
 
    void overlay_image_at_position(double pos_x, double pos_y, Ref<Image> image, double scale);
 
    Rect2 get_extent();
 
    Vector3 get_center();
 
    float get_min();
 
    float get_max();
 
    float get_pixel_size();
 
    void load_from_file(String file_path, bool write_access);
 
    void set_native_dataset(std::shared_ptr<NativeDataset> new_dataset);
 
    void set_origin_dataset(Ref<GeoDataset> dataset);
 
    bool write_access;
 
    String name;
 
  private:
    Ref<GeoDataset> origin_dataset;
    std::shared_ptr<NativeDataset> dataset;
    ExtentData extent_data;
};
 
class EXPORT GeoDataset : public RefCounted {
    GDCLASS(GeoDataset, RefCounted)
 
  protected:
    static void _bind_methods();
 
  public:
    GeoDataset() = default;
    ~GeoDataset();
 
    bool is_valid();
 
    Dictionary get_file_info();
 
    bool has_write_access();
 
    Array get_raster_layers();
 
    Array get_feature_layers();
 
    Ref<GeoRasterLayer> get_raster_layer(String name);
 
    Ref<GeoFeatureLayer> get_feature_layer(String name);
 
    Ref<GeoFeatureLayer> get_sql_feature_layer(String query);
 
    void load_from_file(String file_path, bool write_access);
 
    void set_native_dataset(std::shared_ptr<NativeDataset> new_dataset);
 
    bool write_access;
 
    std::shared_ptr<NativeDataset> dataset;
 
    String name;
};
 
} // namespace godot
 
#endif // __GEODATA_H__