Types.h

Go to the documentation of this file.

/*
 * Copyright 2022 HEAVY.AI, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <memory>
#include <stdexcept>
#include <string>
#include <vector>

#include "Shared/sqltypes.h"

class OGRGeometry;
class OGRCoordinateTransformation;

namespace Geospatial {

struct WkbView {
  uint8_t const* ptr_;
  size_t size_;
};

class GeoTypesError : public std::runtime_error {
 public:
  explicit GeoTypesError(const std::string& type, const int ogr_err)
      : std::runtime_error("Geo" + type +
                           " Error: " + GeoTypesError::OGRErrorToStr(ogr_err)) {}
  explicit GeoTypesError(const std::string& type, const std::string& err)
      : std::runtime_error("Geo" + type + " Error: " + err) {}

 private:
  static std::string OGRErrorToStr(const int ogr_err);
};

class GeoTypesFactory;

class GeoBase {
 public:
  GeoBase() : owns_geom_obj_(true) {}
  virtual ~GeoBase();

  std::string getWktString() const;
  bool getWkb(std::vector<uint8_t>&) const;
  enum class GeoType {
    kPOINT,
    kLINESTRING,
    kPOLYGON,
    kMULTIPOLYGON,
    kGEOMETRY,
    kGEOMETRYCOLLECTION,
    kMULTILINESTRING,
    kMULTIPOINT
  };
  enum class GeoOp {
    kPROJECTION = 0,
    kINTERSECTION = 1,
    kDIFFERENCE = 2,
    kUNION = 3,
    kBUFFER = 4,
    kISVALID = 5,
    kISEMPTY = 6,
    kEQUALS = 7,
    kCONCAVEHULL = 8,
    kCONVEXHULL = 9
  };
  virtual GeoType getType() const = 0;
  const OGRGeometry* getOGRGeometry() const { return geom_; }

  int32_t getBestPlanarSRID() const;
  bool transform(int32_t srid0, int32_t srid1);
  bool transform(SQLTypeInfo& ti);
  static std::shared_ptr<OGRCoordinateTransformation> getTransformation(int32_t srid0,
                                                                        int32_t srid1);

  virtual bool operator==(const GeoBase& other) const;

  bool isEmpty() const;

  std::unique_ptr<GeoBase> run(GeoOp op, const GeoBase& other) const;
  std::unique_ptr<GeoBase> optimized_run(GeoOp op, const GeoBase& other) const;
  std::unique_ptr<GeoBase> run(GeoOp op, double param) const;
  bool run(GeoOp op) const;

  virtual std::unique_ptr<GeoBase> clone() const = 0;

 protected:
  GeoBase(OGRGeometry* geom, const bool owns_geom_obj)
      : geom_(geom), owns_geom_obj_(owns_geom_obj) {}
  OGRGeometry* geom_ = nullptr;
  bool owns_geom_obj_;

  static int createFromWktString(const std::string& wkt, OGRGeometry** geom);
  static int createFromWkbView(OGRGeometry** geom, WkbView const);

  friend class GeoTypesFactory;
};

class GeoPoint : public GeoBase {
 public:
  GeoPoint(const std::vector<double>& coords);
  GeoPoint(const std::string& wkt);

  void getColumns(std::vector<double>& coords) const;
  GeoType getType() const final { return GeoType::kPOINT; }

  std::unique_ptr<GeoBase> clone() const override;

 protected:
  GeoPoint(OGRGeometry* geom, const bool owns_geom_obj) : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

class GeoMultiPoint : public GeoBase {
 public:
  GeoMultiPoint(const std::vector<double>& coords);
  GeoMultiPoint(const std::string& wkt);

  void getColumns(std::vector<double>& coords, std::vector<double>& bounds) const;
  GeoType getType() const final { return GeoType::kMULTIPOINT; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoMultiPoint(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

class GeoLineString : public GeoBase {
 public:
  GeoLineString(const std::vector<double>& coords);
  GeoLineString(const std::string& wkt);

  void getColumns(std::vector<double>& coords, std::vector<double>& bounds) const;
  GeoType getType() const final { return GeoType::kLINESTRING; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoLineString(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

class GeoMultiLineString : public GeoBase {
 public:
  GeoMultiLineString(const std::vector<double>& coords,
                     const std::vector<int32_t>& linestring_sizes);
  GeoMultiLineString(const std::string& wkt);

  void getColumns(std::vector<double>& coords,
                  std::vector<int32_t>& linestring_sizes,
                  std::vector<double>& bounds) const;
  GeoType getType() const final { return GeoType::kMULTILINESTRING; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoMultiLineString(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

class GeoPolygon : public GeoBase {
 public:
  GeoPolygon(const std::vector<double>& coords, const std::vector<int32_t>& ring_sizes);
  GeoPolygon(const std::string& wkt);

  void getColumns(std::vector<double>& coords,
                  std::vector<int32_t>& ring_sizes,
                  std::vector<double>& bounds) const;
  GeoType getType() const final { return GeoType::kPOLYGON; }

  int32_t getNumInteriorRings() const;

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoPolygon(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

class GeoMultiPolygon : public GeoBase {
 public:
  GeoMultiPolygon(const std::vector<double>& coords,
                  const std::vector<int32_t>& ring_sizes,
                  const std::vector<int32_t>& poly_rings);
  GeoMultiPolygon(const std::string& wkt);

  void getColumns(std::vector<double>& coords,
                  std::vector<int32_t>& ring_sizes,
                  std::vector<int32_t>& poly_rings,
                  std::vector<double>& bounds) const;
  GeoType getType() const final { return GeoType::kMULTIPOLYGON; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoMultiPolygon(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {}

  friend class GeoTypesFactory;
};

// TODO: with addition of MULTILINESTING and generic GEOMETRY, GEOMETRYCOLLECTION types
// need to rename poly-specific ring_sizes and poly_rings arrays and columns to meta
// names. First meta layer above the coords: meta1 array will contain component sizes
//   - for MULTILINESTRING it will hold linestring sizes
//   - for POLYGON and MULTIPOLYGON it will hold ring sizes
// Second meta layer above the coords: meta2 array will contain larger component sizes
//   - for MULTIPOLYGON it will hold poly sizes in terms of rings (ring counts)
// Thrid meta layer above the coords: meta3 array will contain geometry kinds
//   - for GEOMETRY it will hold the single geometry kind
//   - for GEOMETRYCOLLECTION it will hold geometry kinds included in the collection

class GeoGeometry : public GeoBase {
 public:
  GeoGeometry(const std::vector<double>& coords,
              const std::vector<int32_t>& ring_sizes,
              const std::vector<int32_t>& poly_rings,
              const std::vector<int32_t>& geo_kinds){};
  GeoGeometry(const std::string& wkt){};

  void getColumns(std::vector<double>& coords,
                  std::vector<int32_t>& ring_sizes,
                  std::vector<int32_t>& poly_rings,
                  std::vector<int32_t>& geo_kinds,
                  std::vector<double>& bounds) const {};
  GeoType getType() const final { return GeoType::kGEOMETRY; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoGeometry(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {
    if (!isEmpty()) {
      throw GeoTypesError("GeoTypesFactory", "Non-empty GEOMETRY");
    }
  }

  friend class GeoTypesFactory;
};

class GeoGeometryCollection : public GeoBase {
 public:
  GeoGeometryCollection(const std::vector<double>& coords,
                        const std::vector<int32_t>& ring_sizes,
                        const std::vector<int32_t>& poly_rings,
                        const std::vector<int32_t>& geo_kinds){};
  GeoGeometryCollection(const std::string& wkt);

  void getColumns(std::vector<double>& coords,
                  std::vector<int32_t>& ring_sizes,
                  std::vector<int32_t>& poly_rings,
                  std::vector<int32_t>& geo_kinds,
                  std::vector<double>& bounds) const {};
  GeoType getType() const final { return GeoType::kGEOMETRYCOLLECTION; }

  std::unique_ptr<GeoBase> clone() const final;

 protected:
  GeoGeometryCollection(OGRGeometry* geom, const bool owns_geom_obj)
      : GeoBase(geom, owns_geom_obj) {
    if (!isEmpty()) {
      throw GeoTypesError("GeoTypesFactory", "Non-empty GEOMETRYCOLLECTION");
    }
  }

  friend class GeoTypesFactory;
};

class GeoTypesFactory {
 public:
  static OGRGeometry* createOGRGeometry(const std::string& wkt_or_wkb_hex);

  static std::unique_ptr<GeoBase> createGeoType(const std::string& wkt_or_wkb_hex);
  static std::unique_ptr<GeoBase> createGeoType(const WkbView);
  static std::unique_ptr<GeoBase> createGeoType(OGRGeometry* geom);

  static bool getGeoColumns(const std::string& wkt_or_wkb_hex,
                            SQLTypeInfo& ti,
                            std::vector<double>& coords,
                            std::vector<double>& bounds,
                            std::vector<int>& ring_sizes,
                            std::vector<int>& poly_rings,
                            const bool promote_poly_to_mpoly = false);

  static bool getGeoColumns(const std::vector<uint8_t>& wkb,
                            SQLTypeInfo& ti,
                            std::vector<double>& coords,
                            std::vector<double>& bounds,
                            std::vector<int>& ring_sizes,
                            std::vector<int>& poly_rings,
                            const bool promote_poly_to_mpoly = false);

  static bool getGeoColumns(OGRGeometry* geom,
                            SQLTypeInfo& ti,
                            std::vector<double>& coords,
                            std::vector<double>& bounds,
                            std::vector<int>& ring_sizes,
                            std::vector<int>& poly_rings,
                            const bool promote_poly_to_mpoly = false);

  static bool getGeoColumns(const std::vector<std::string>* wkt_or_wkb_hex_column,
                            SQLTypeInfo& ti,
                            std::vector<std::vector<double>>& coords_column,
                            std::vector<std::vector<double>>& bounds_column,
                            std::vector<std::vector<int>>& ring_sizes_column,
                            std::vector<std::vector<int>>& poly_rings_column,
                            const bool promote_poly_to_mpoly = false);

  static void getNullGeoColumns(SQLTypeInfo& ti,
                                std::vector<double>& coords,
                                std::vector<double>& bounds,
                                std::vector<int>& ring_sizes,
                                std::vector<int>& poly_rings,
                                const bool promote_poly_to_mpoly = false);

 private:
  static std::unique_ptr<Geospatial::GeoBase> createGeoTypeImpl(
      OGRGeometry* geom,
      const bool owns_geom_obj = true);
  static void getGeoColumnsImpl(const std::unique_ptr<GeoBase>& geospatial_base,
                                SQLTypeInfo& ti,
                                std::vector<double>& coords,
                                std::vector<double>& bounds,
                                std::vector<int>& ring_sizes,
                                std::vector<int>& poly_rings,
                                const bool promote_poly_to_mpoly = false);
};

}  // namespace Geospatial