GeoRaster.h

#ifndef RASTERTILEEXTRACTOR_GEORASTER_H
#define RASTERTILEEXTRACTOR_GEORASTER_H
 
#include "defines.h"
#include <cstdint>
#include <mutex>
 
struct RasterIOHelper {
    int clamped_pixel_offset_x;
    int clamped_pixel_offset_y;
    int min_raster_size; // Currently unused
    int remainder_x_left;
    int remainder_y_top;
    int target_height;
    int target_width;
    int usable_height;
    int usable_width;
 
    static std::mutex rasterio_mutex;
 
    RasterIOHelper();
 
    ~RasterIOHelper();
};
 
// Forward declaration of GDALDataset from <gdal/gdal_priv.h>
class GDALDataset;
 
class GeoRaster {
  public:
    enum FORMAT { // size (bits)    | data type         |   Number of chanels
        RGB,      //    8           | int               |   3
        RGBA,     //    8           | int               |   4
        RF,       //    32<=X<=64   | float             |   X >= 1
        BYTE,     //    8           | int               |   X >= 1
        MIXED,    //    8<=X<=64    | int and/or float  |   X >= 2
        UNKNOWN   //    unknown     | unknown           |   X >= 1
    };
 
    GeoRaster(GDALDataset *data, int interpolation_type);
 
    GeoRaster(GDALDataset *data, int pixel_offset_x, int pixel_offset_y,
              int source_window_size_pixels, int destination_window_size_pixels,
              int interpolation_type);
 
    ~GeoRaster() = default;
 
    static FORMAT get_format_for_dataset(GDALDataset *data);
    
    void *get_as_array();
 
    void *get_band_as_array(int band_index);
 
    int get_size_in_bytes();
 
    FORMAT get_format();
 
    FORMAT get_band_format(int band_index);
 
    int get_pixel_size_x();
 
    int get_pixel_size_y();
 
    uint64_t *get_histogram();
 
    int *get_most_common(int number_of_elements);
 
  private:
    GDALDataset *data;
 
    FORMAT format;
 
    int pixel_offset_x;
 
    int pixel_offset_y;
 
    int source_window_size_pixels;
 
    int destination_window_size_pixels;
 
    int interpolation_type;
 
    RasterIOHelper get_raster_io_helper();
};
 
#endif // RASTERTILEEXTRACTOR_GEORASTER_H