mirror of
https://github.com/MariaDB/server.git
synced 2025-08-01 03:47:19 +03:00
Added GIS extension
This commit is contained in:
bar@gw.udmsearch.izhnet.ru
476
sql/spatial.h
Normal file
476
sql/spatial.h
Normal file
@ -0,0 +1,476 @@
|
||||
#ifndef _spatial_h
|
||||
#define _spatial_h
|
||||
|
||||
#include "gstream.h"
|
||||
|
||||
const int POINT_DATA_SIZE = 8+8;
|
||||
const int WKB_HEADER_SIZE = 1+4;
|
||||
|
||||
struct stPoint2D
|
||||
{
|
||||
double x;
|
||||
double y;
|
||||
};
|
||||
|
||||
struct stLinearRing
|
||||
{
|
||||
size_t n_points;
|
||||
stPoint2D points;
|
||||
};
|
||||
|
||||
/***************************** MBR *******************************/
|
||||
|
||||
struct MBR
|
||||
{
|
||||
MBR()
|
||||
{
|
||||
xmin=DBL_MAX;
|
||||
ymin=DBL_MAX;
|
||||
xmax=-DBL_MAX;
|
||||
ymax=-DBL_MAX;
|
||||
}
|
||||
|
||||
MBR(const double &_xmin, const double &_ymin, const double &_xmax, const double &_ymax)
|
||||
{
|
||||
xmin=_xmin;
|
||||
ymin=_ymin;
|
||||
xmax=_xmax;
|
||||
ymax=_ymax;
|
||||
}
|
||||
|
||||
MBR(const stPoint2D &min, const stPoint2D &max)
|
||||
{
|
||||
xmin=min.x;
|
||||
ymin=min.y;
|
||||
xmax=max.x;
|
||||
ymax=max.y;
|
||||
}
|
||||
|
||||
double xmin;
|
||||
double ymin;
|
||||
double xmax;
|
||||
double ymax;
|
||||
|
||||
void add_xy(double x, double y)
|
||||
{ /* Not using "else" for proper one point MBR calculation */
|
||||
if (x<xmin)
|
||||
{
|
||||
xmin=x;
|
||||
}
|
||||
if (x>xmax)
|
||||
{
|
||||
xmax=x;
|
||||
}
|
||||
if (y<ymin)
|
||||
{
|
||||
ymin=y;
|
||||
}
|
||||
if (y>ymax)
|
||||
{
|
||||
ymax=y;
|
||||
}
|
||||
}
|
||||
|
||||
void add_xy(double *px, double *py)
|
||||
{ /* Not using "else" for proper one point MBR calculation */
|
||||
double x, y;
|
||||
float8get(x, px);
|
||||
float8get(y, py);
|
||||
if (x<xmin)
|
||||
{
|
||||
xmin=x;
|
||||
}
|
||||
if (x>xmax)
|
||||
{
|
||||
xmax=x;
|
||||
}
|
||||
if (y<ymin)
|
||||
{
|
||||
ymin=y;
|
||||
}
|
||||
if (y>ymax)
|
||||
{
|
||||
ymax=y;
|
||||
}
|
||||
}
|
||||
|
||||
void add_mbr(const MBR *mbr)
|
||||
{
|
||||
if (mbr->xmin<xmin)
|
||||
{
|
||||
xmin=mbr->xmin;
|
||||
}
|
||||
if (mbr->xmax>xmax)
|
||||
{
|
||||
xmax=mbr->xmax;
|
||||
}
|
||||
if (mbr->ymin<ymin)
|
||||
{
|
||||
ymin=mbr->ymin;
|
||||
}
|
||||
if (mbr->ymax>ymax)
|
||||
{
|
||||
ymax=mbr->ymax;
|
||||
}
|
||||
}
|
||||
|
||||
int equals(const MBR *mbr)
|
||||
{
|
||||
return (mbr->xmin==xmin)&&(mbr->ymin==ymin)&&(mbr->xmax==xmax)&&(mbr->ymax==ymax);
|
||||
}
|
||||
|
||||
int disjoint(const MBR *mbr)
|
||||
{
|
||||
return (mbr->xmin>xmax)||(mbr->ymin>ymax)||(mbr->xmax<xmin)||(mbr->ymax<ymin);
|
||||
}
|
||||
|
||||
int intersects(const MBR *mbr)
|
||||
{
|
||||
return !disjoint(mbr);
|
||||
}
|
||||
|
||||
int touches(const MBR *mbr)
|
||||
{
|
||||
return (((mbr->xmin==xmax) || (mbr->xmax==xmin)) &&
|
||||
((mbr->ymin>=ymin) && (mbr->ymin<=ymax) ||
|
||||
(mbr->ymax>=ymin) && (mbr->ymax<=ymax))) ||
|
||||
(((mbr->ymin==ymax) || (mbr->ymax==ymin)) &&
|
||||
((mbr->xmin>=xmin) && (mbr->xmin<=xmax) ||
|
||||
(mbr->xmax>=xmin)&&(mbr->xmax<=xmax)));
|
||||
}
|
||||
|
||||
int within(const MBR *mbr)
|
||||
{
|
||||
return (mbr->xmin<=xmin) && (mbr->ymin<=ymin) &&
|
||||
(mbr->xmax>=xmax) && (mbr->ymax>=ymax);
|
||||
}
|
||||
|
||||
int contains(const MBR *mbr)
|
||||
{
|
||||
return (mbr->xmin>=xmin) && (mbr->ymin>=ymin) &&
|
||||
(mbr->xmax<=xmax) && (mbr->ymax<=ymax);
|
||||
}
|
||||
|
||||
bool inner_point(double x, double y) const
|
||||
{
|
||||
return (xmin<x) && (xmax>x) && (ymin<y) && (ymax>x);
|
||||
}
|
||||
|
||||
int overlaps(const MBR *mbr)
|
||||
{
|
||||
int lb = mbr->inner_point(xmin, ymin);
|
||||
int rb = mbr->inner_point(xmax, ymin);
|
||||
int rt = mbr->inner_point(xmax, ymax);
|
||||
int lt = mbr->inner_point(xmin, ymax);
|
||||
|
||||
int a = lb+rb+rt+lt;
|
||||
return (a>0) && (a<4) && (!within(mbr));
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
/***************************** Geometry *******************************/
|
||||
|
||||
class Geometry;
|
||||
|
||||
typedef int (Geometry::*GF_InitFromText)(GTextReadStream *, String *);
|
||||
typedef int (Geometry::*GF_GetDataAsText)(String *) const;
|
||||
typedef size_t (Geometry::*GF_GetDataSize)() const;
|
||||
typedef int (Geometry::*GF_GetMBR)(MBR *) const;
|
||||
|
||||
typedef int (Geometry::*GF_GetD)(double *) const;
|
||||
typedef int (Geometry::*GF_GetI)(int *) const;
|
||||
typedef int (Geometry::*GF_GetUI)(uint32 *) const;
|
||||
typedef int (Geometry::*GF_GetWS)(String *) const;
|
||||
typedef int (Geometry::*GF_GetUIWS)(uint32, String *) const;
|
||||
|
||||
#define GEOM_METHOD_PRESENT(geom_obj, method)\
|
||||
(geom_obj.m_vmt->method != &Geometry::method)
|
||||
|
||||
class Geometry
|
||||
{
|
||||
public:
|
||||
enum wkbType
|
||||
{
|
||||
wkbPoint = 1,
|
||||
wkbLineString = 2,
|
||||
wkbPolygon = 3,
|
||||
wkbMultiPoint = 4,
|
||||
wkbMultiLineString = 5,
|
||||
wkbMultiPolygon = 6,
|
||||
wkbGeometryCollection = 7
|
||||
};
|
||||
enum wkbByteOrder
|
||||
{
|
||||
wkbXDR = 0, /* Big Endian */
|
||||
wkbNDR = 1 /* Little Endian */
|
||||
};
|
||||
|
||||
|
||||
class GClassInfo
|
||||
{
|
||||
public:
|
||||
GF_InitFromText init_from_text;
|
||||
GF_GetDataAsText get_data_as_text;
|
||||
GF_GetDataSize get_data_size;
|
||||
GF_GetMBR get_mbr;
|
||||
GF_GetD get_x;
|
||||
GF_GetD get_y;
|
||||
GF_GetD length;
|
||||
GF_GetD area;
|
||||
|
||||
GF_GetI is_closed;
|
||||
|
||||
GF_GetUI num_interior_ring;
|
||||
GF_GetUI num_points;
|
||||
GF_GetUI num_geometries;
|
||||
GF_GetUI dimension;
|
||||
|
||||
GF_GetWS start_point;
|
||||
GF_GetWS end_point;
|
||||
GF_GetWS exterior_ring;
|
||||
GF_GetWS centroid;
|
||||
|
||||
GF_GetUIWS point_n;
|
||||
GF_GetUIWS interior_ring_n;
|
||||
GF_GetUIWS geometry_n;
|
||||
|
||||
int m_type_id;
|
||||
const char *m_name;
|
||||
GClassInfo *m_next_rt;
|
||||
};
|
||||
GClassInfo *m_vmt;
|
||||
|
||||
const GClassInfo *get_class_info() const { return m_vmt; }
|
||||
size_t get_data_size() const { return (this->*m_vmt->get_data_size)(); }
|
||||
|
||||
int init_from_text(GTextReadStream *trs, String *wkb)
|
||||
{ return (this->*m_vmt->init_from_text)(trs, wkb); }
|
||||
|
||||
int get_data_as_text(String *txt) const
|
||||
{ return (this->*m_vmt->get_data_as_text)(txt); }
|
||||
|
||||
int get_mbr(MBR *mbr) const { return (this->*m_vmt->get_mbr)(mbr); }
|
||||
int dimension(uint32 *dim) const
|
||||
{ return (this->*m_vmt->dimension)(dim); }
|
||||
|
||||
int get_x(double *x) const { return (this->*m_vmt->get_x)(x); }
|
||||
int get_y(double *y) const { return (this->*m_vmt->get_y)(y); }
|
||||
int length(double *len) const { return (this->*m_vmt->length)(len); }
|
||||
int area(double *ar) const { return (this->*m_vmt->area)(ar); }
|
||||
|
||||
int is_closed(int *closed) const { return (this->*m_vmt->is_closed)(closed); }
|
||||
|
||||
int num_interior_ring(uint32 *n_int_rings) const
|
||||
{ return (this->*m_vmt->num_interior_ring)(n_int_rings); }
|
||||
int num_points(uint32 *n_points) const
|
||||
{ return (this->*m_vmt->num_points)(n_points); }
|
||||
|
||||
int num_geometries(uint32 *num) const
|
||||
{ return (this->*m_vmt->num_geometries)(num); }
|
||||
|
||||
int start_point(String *point) const
|
||||
{ return (this->*m_vmt->start_point)(point); }
|
||||
int end_point(String *point) const
|
||||
{ return (this->*m_vmt->end_point)(point); }
|
||||
int exterior_ring(String *ring) const
|
||||
{ return (this->*m_vmt->exterior_ring)(ring); }
|
||||
int centroid(String *point) const
|
||||
{ return (this->*m_vmt->centroid)(point); }
|
||||
|
||||
int point_n(uint32 num, String *result) const
|
||||
{ return (this->*m_vmt->point_n)(num, result); }
|
||||
int interior_ring_n(uint32 num, String *result) const
|
||||
{ return (this->*m_vmt->interior_ring_n)(num, result); }
|
||||
int geometry_n(uint32 num, String *result) const
|
||||
{ return (this->*m_vmt->geometry_n)(num, result); }
|
||||
|
||||
public:
|
||||
int create_from_wkb(const char *data, uint32 data_len);
|
||||
int create_from_wkt(GTextReadStream *trs, String *wkt, int init_stream=1);
|
||||
int init(int type_id)
|
||||
{
|
||||
m_vmt = find_class(type_id);
|
||||
return !m_vmt;
|
||||
}
|
||||
int new_geometry(const char *name, size_t len)
|
||||
{
|
||||
m_vmt = find_class(name, len);
|
||||
return !m_vmt;
|
||||
}
|
||||
|
||||
int as_wkt(String *wkt) const
|
||||
{
|
||||
if(wkt->reserve(strlen(get_class_info()->m_name) + 2, 512))
|
||||
return 1;
|
||||
wkt->qs_append(get_class_info()->m_name);
|
||||
wkt->qs_append('(');
|
||||
if(get_data_as_text(wkt))
|
||||
return 1;
|
||||
wkt->qs_append(')');
|
||||
return 0;
|
||||
}
|
||||
|
||||
void init_from_wkb(const char *data, uint32 data_len)
|
||||
{
|
||||
m_data = data;
|
||||
m_data_end = data + data_len;
|
||||
}
|
||||
|
||||
void shift_wkb_header()
|
||||
{
|
||||
m_data += WKB_HEADER_SIZE;
|
||||
}
|
||||
|
||||
int envelope(String *result) const;
|
||||
|
||||
protected:
|
||||
static GClassInfo *find_class(int type_id);
|
||||
static GClassInfo *find_class(const char *name, size_t len);
|
||||
|
||||
bool no_data(const char *cur_data, uint32 data_amount) const
|
||||
{
|
||||
return cur_data + data_amount > m_data_end;
|
||||
}
|
||||
|
||||
const char *m_data;
|
||||
const char *m_data_end;
|
||||
};
|
||||
|
||||
/***************************** Point *******************************/
|
||||
|
||||
class GPoint: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int get_xy(double *x, double *y) const
|
||||
{
|
||||
const char *data = m_data;
|
||||
if(no_data(data, sizeof(double)) * 2) return 1;
|
||||
float8get(*x, data);
|
||||
float8get(*y, data + sizeof(double));
|
||||
return 0;
|
||||
}
|
||||
|
||||
int get_x(double *x) const
|
||||
{
|
||||
if(no_data(m_data, sizeof(double))) return 1;
|
||||
float8get(*x, m_data);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int get_y(double *y) const
|
||||
{
|
||||
const char *data = m_data;
|
||||
if(no_data(data, sizeof(double)) * 2) return 1;
|
||||
float8get(*y, data + sizeof(double));
|
||||
return 0;
|
||||
}
|
||||
|
||||
int dimension(uint32 *dim) const { *dim = 0; return 0; }
|
||||
};
|
||||
|
||||
/***************************** LineString *******************************/
|
||||
|
||||
class GLineString: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int length(double *len) const;
|
||||
int is_closed(int *closed) const;
|
||||
int num_points(uint32 *n_points) const;
|
||||
int start_point(String *point) const;
|
||||
int end_point(String *point) const;
|
||||
int point_n(uint32 n, String *result) const;
|
||||
int dimension(uint32 *dim) const { *dim = 1; return 0; }
|
||||
// IsRing
|
||||
};
|
||||
|
||||
/***************************** Polygon *******************************/
|
||||
|
||||
class GPolygon: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int area(double *ar) const;
|
||||
int exterior_ring(String *result) const;
|
||||
int num_interior_ring(uint32 *n_int_rings) const;
|
||||
int interior_ring_n(uint32 num, String *result) const;
|
||||
int centroid_xy(double *x, double *y) const;
|
||||
int centroid(String *result) const;
|
||||
int dimension(uint32 *dim) const { *dim = 2; return 0; }
|
||||
// PointOnSurface
|
||||
};
|
||||
|
||||
/***************************** MultiPoint *******************************/
|
||||
|
||||
class GMultiPoint: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
int dimension(uint32 *dim) const { *dim = 0; return 0; }
|
||||
};
|
||||
|
||||
/***************************** MultiLineString *******************************/
|
||||
|
||||
class GMultiLineString: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int length(double *len) const;
|
||||
int is_closed(int *closed) const;
|
||||
int dimension(uint32 *dim) const { *dim = 1; return 0; }
|
||||
};
|
||||
|
||||
/***************************** MultiPolygon *******************************/
|
||||
|
||||
class GMultiPolygon: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int area(double *ar) const;
|
||||
int centroid(String *result) const;
|
||||
int dimension(uint32 *dim) const { *dim = 2; return 0; }
|
||||
// PointOnSurface
|
||||
};
|
||||
|
||||
/***************************** GeometryCollection *******************************/
|
||||
|
||||
class GGeometryCollection: public Geometry
|
||||
{
|
||||
public:
|
||||
size_t get_data_size() const;
|
||||
int init_from_text(GTextReadStream *trs, String *wkb);
|
||||
int get_data_as_text(String *txt) const;
|
||||
int get_mbr(MBR *mbr) const;
|
||||
|
||||
int num_geometries(uint32 *num) const;
|
||||
int geometry_n(uint32 num, String *result) const;
|
||||
|
||||
int dimension(uint32 *dim) const;
|
||||
};
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user