开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
spatial.h
资源名称：mysql-4.1.16-win-src.zip　[点击查看]
上传用户：romrleung
上传日期：2022-05-23
资源大小：18897k
文件大小：14k
源码类别：

MySQL数据库

开发平台：

Visual C++

spatial.h：源码内容
  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  17. #ifndef _spatial_h
  18. #define _spatial_h
  20. #ifdef HAVE_SPATIAL
  22. const uint SRID_SIZE= 4;
  23. const uint SIZEOF_STORED_DOUBLE= 8;
  24. const uint POINT_DATA_SIZE= SIZEOF_STORED_DOUBLE*2; 
  25. const uint WKB_HEADER_SIZE= 1+4;
  26. const uint32 GET_SIZE_ERROR= ((uint32) -1);
  28. struct st_point_2d
  29. {
  30.   double x;
  31.   double y;
  32. };
  34. struct st_linear_ring
  35. {
  36.   uint32 n_points;
  37.   st_point_2d points;
  38. };
  40. /***************************** MBR *******************************/
  43. /*
  44.   It's ok that a lot of the functions are inline as these are only used once
  45.   in MySQL
  46. */
  48. struct MBR
  49. {
  50.   double xmin, ymin, xmax, ymax;
  52.   MBR()
  53.   {
  54.     xmin= ymin= DBL_MAX;
  55.     xmax= ymax= -DBL_MAX;
  56.   }
  58.   MBR(const double xmin_arg, const double ymin_arg,
  59.       const double xmax_arg, const double ymax_arg)
  60.     :xmin(xmin_arg), ymin(ymin_arg), xmax(xmax_arg), ymax(ymax_arg)
  61.   {}
  63.   MBR(const st_point_2d &min, const st_point_2d &max)
  64.     :xmin(min.x), ymin(min.y), xmax(max.x), ymax(max.y)
  65.   {}
  66.  
  67.   inline void add_xy(double x, double y)
  68.   {
  69.     /* Not using "else" for proper one point MBR calculation */
  70.     if (x < xmin)
  71.       xmin= x;
  72.     if (x > xmax)
  73.       xmax= x;
  74.     if (y < ymin)
  75.       ymin= y;
  76.     if (y > ymax)
  77.       ymax= y;
  78.   }
  79.   void add_xy(const char *px, const char *py)
  80.   {
  81.     double x, y;
  82.     float8get(x, px);
  83.     float8get(y, py);
  84.     add_xy(x,y);
  85.   }
  86.   void add_mbr(const MBR *mbr)
  87.   {
  88.     if (mbr->xmin < xmin)
  89.       xmin= mbr->xmin;
  90.     if (mbr->xmax > xmax)
  91.       xmax= mbr->xmax;
  92.     if (mbr->ymin < ymin)
  93.       ymin= mbr->ymin;
  94.     if (mbr->ymax > ymax)
  95.       ymax= mbr->ymax;
  96.   }
  98.   int equals(const MBR *mbr)
  99.   {
  100.     /* The following should be safe, even if we compare doubles */
  101.     return ((mbr->xmin == xmin) && (mbr->ymin == ymin) &&
  102.     (mbr->xmax == xmax) && (mbr->ymax == ymax));
  103.   }
  105.   int disjoint(const MBR *mbr)
  106.   {
  107.     /* The following should be safe, even if we compare doubles */
  108.     return ((mbr->xmin > xmax) || (mbr->ymin > ymax) ||
  109.     (mbr->xmax < xmin) || (mbr->ymax < ymin));
  110.   }
  112.   int intersects(const MBR *mbr)
  113.   {
  114.     return !disjoint(mbr);
  115.   }
  117.   int touches(const MBR *mbr)
  118.   {
  119.     /* The following should be safe, even if we compare doubles */
  120.     return ((((mbr->xmin == xmax) || (mbr->xmax == xmin)) && 
  121.      ((mbr->ymin >= ymin) && (mbr->ymin <= ymax) || 
  122.       (mbr->ymax >= ymin) && (mbr->ymax <= ymax))) ||
  123.     (((mbr->ymin == ymax) || (mbr->ymax == ymin)) &&
  124.      ((mbr->xmin >= xmin) && (mbr->xmin <= xmax) ||
  125.       (mbr->xmax >= xmin) && (mbr->xmax <= xmax))));
  126.   }
  128.   int within(const MBR *mbr)
  129.   {
  130.     /* The following should be safe, even if we compare doubles */
  131.     return ((mbr->xmin <= xmin) && (mbr->ymin <= ymin) &&
  132.     (mbr->xmax >= xmax) && (mbr->ymax >= ymax));
  133.   }
  135.   int contains(const MBR *mbr)
  136.   {
  137.     /* The following should be safe, even if we compare doubles */
  138.     return ((mbr->xmin >= xmin) && (mbr->ymin >= ymin) &&
  139.     (mbr->xmax <= xmax) && (mbr->ymax <= ymax));
  140.   }
  142.   bool inner_point(double x, double y) const
  143.   {
  144.     /* The following should be safe, even if we compare doubles */
  145.     return (xmin<x) && (xmax>x) && (ymin<y) && (ymax>x);
  146.   }
  148.   int overlaps(const MBR *mbr)
  149.   {
  150.     int lb= mbr->inner_point(xmin, ymin);
  151.     int rb= mbr->inner_point(xmax, ymin);
  152.     int rt= mbr->inner_point(xmax, ymax);
  153.     int lt= mbr->inner_point(xmin, ymax);
  155.     int a = lb+rb+rt+lt;
  156.     return (a>0) && (a<4) && (!within(mbr));
  157.   }
  158. };
  161. /***************************** Geometry *******************************/
  163. struct Geometry_buffer;
  165. class Geometry
  166. {
  167. public:
  168.   static void *operator new(size_t size, void *buffer)
  169.   {
  170.     return buffer;
  171.   }
  173.   static void operator delete(void *ptr, void *buffer)
  174.   {}
  176.   enum wkbType
  177.   {
  178.     wkb_point= 1,
  179.     wkb_linestring= 2,
  180.     wkb_polygon= 3,
  181.     wkb_multipoint= 4,
  182.     wkb_multilinestring= 5,
  183.     wkb_multipolygon= 6,
  184.     wkb_geometrycollection= 7,
  185.     wkb_end=7
  186.   };
  187.   enum wkbByteOrder
  188.   {
  189.     wkb_xdr= 0,    /* Big Endian */
  190.     wkb_ndr= 1     /* Little Endian */
  191.   };                                    
  193.   class Class_info
  194.   {
  195.   public:
  196.     LEX_STRING_WITH_INIT m_name;
  197.     int m_type_id;
  198.     void (*m_create_func)(void *);
  199.     Class_info(const char *name, int type_id, void(*create_func)(void *));
  200.   };
  202.   virtual const Class_info *get_class_info() const=0;
  203.   virtual uint32 get_data_size() const=0;
  204.   virtual bool init_from_wkt(Gis_read_stream *trs, String *wkb)=0;
  206.   /* returns the length of the wkb that was read */
  207.   virtual uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo,
  208.                              String *res)=0;
  209.   virtual bool get_data_as_wkt(String *txt, const char **end) const=0;
  210.   virtual bool get_mbr(MBR *mbr, const char **end) const=0;
  211.   virtual bool dimension(uint32 *dim, const char **end) const=0;
  212.   virtual int get_x(double *x) const { return -1; }
  213.   virtual int get_y(double *y) const { return -1; }
  214.   virtual int length(double *len) const  { return -1; }
  215.   virtual int area(double *ar, const char **end) const { return -1;}
  216.   virtual int is_closed(int *closed) const { return -1; }
  217.   virtual int num_interior_ring(uint32 *n_int_rings) const { return -1; }
  218.   virtual int num_points(uint32 *n_points) const { return -1; }
  219.   virtual int num_geometries(uint32 *num) const { return -1; }
  220.   virtual int start_point(String *point) const { return -1; }
  221.   virtual int end_point(String *point) const { return -1; }
  222.   virtual int exterior_ring(String *ring) const { return -1; }
  223.   virtual int centroid(String *point) const { return -1; }
  224.   virtual int point_n(uint32 num, String *result) const { return -1; }
  225.   virtual int interior_ring_n(uint32 num, String *result) const { return -1; }
  226.   virtual int geometry_n(uint32 num, String *result) const { return -1; }
  228. public:
  229.   static Geometry *create_by_typeid(Geometry_buffer *buffer, int type_id)
  230.   {
  231.     Class_info *ci;
  232.     if (!(ci= find_class((int) type_id)))
  233.       return NULL;
  234.     (*ci->m_create_func)((void *)buffer);
  235.     return my_reinterpret_cast(Geometry *)(buffer);
  236.   }
  238.   static Geometry *construct(Geometry_buffer *buffer,
  239.                              const char *data, uint32 data_len);
  240.   static Geometry *create_from_wkt(Geometry_buffer *buffer,
  241.    Gis_read_stream *trs, String *wkt,
  242.    bool init_stream=1);
  243.   static int create_from_wkb(Geometry_buffer *buffer,
  244.                              const char *wkb, uint32 len, String *res);
  245.   int as_wkt(String *wkt, const char **end)
  246.   {
  247.     uint32 len= get_class_info()->m_name.length;
  248.     if (wkt->reserve(len + 2, 512))
  249.       return 1;
  250.     wkt->qs_append(get_class_info()->m_name.str, len);
  251.     wkt->qs_append('(');
  252.     if (get_data_as_wkt(wkt, end))
  253.       return 1;
  254.     wkt->qs_append(')');
  255.     return 0;
  256.   }
  258.   inline void set_data_ptr(const char *data, uint32 data_len)
  259.   {
  260.     m_data= data;
  261.     m_data_end= data + data_len;
  262.   }
  264.   inline void shift_wkb_header()
  265.   {
  266.     m_data+= WKB_HEADER_SIZE;
  267.   }
  269.   bool envelope(String *result) const;
  270.   static Class_info *ci_collection[wkb_end+1];
  272. protected:
  273.   static Class_info *find_class(int type_id)
  274.   {
  275.     return ((type_id < wkb_point) || (type_id > wkb_end)) ?
  276.       NULL : ci_collection[type_id];
  277.   }  
  278.   static Class_info *find_class(const char *name, uint32 len);
  279.   const char *append_points(String *txt, uint32 n_points,
  280.     const char *data, uint32 offset) const;
  281.   bool create_point(String *result, const char *data) const;
  282.   bool create_point(String *result, double x, double y) const;
  283.   const char *get_mbr_for_points(MBR *mbr, const char *data, uint offset)
  284.     const;
  286.   inline bool no_data(const char *cur_data, uint32 data_amount) const
  287.   {
  288.     return (cur_data + data_amount > m_data_end);
  289.   }
  290.   const char *m_data;
  291.   const char *m_data_end;
  292. };
  295. /***************************** Point *******************************/
  296.  
  297. class Gis_point: public Geometry
  298. {
  299. public:
  300.   uint32 get_data_size() const;
  301.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  302.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  303.   bool get_data_as_wkt(String *txt, const char **end) const;
  304.   bool get_mbr(MBR *mbr, const char **end) const;
  305.   
  306.   int get_xy(double *x, double *y) const
  307.   {
  308.     const char *data= m_data;
  309.     if (no_data(data, SIZEOF_STORED_DOUBLE * 2))
  310.       return 1;
  311.     float8get(*x, data);
  312.     float8get(*y, data + SIZEOF_STORED_DOUBLE);
  313.     return 0;
  314.   }
  316.   int get_x(double *x) const
  317.   {
  318.     if (no_data(m_data, SIZEOF_STORED_DOUBLE))
  319.       return 1;
  320.     float8get(*x, m_data);
  321.     return 0;
  322.   }
  324.   int get_y(double *y) const
  325.   {
  326.     const char *data= m_data;
  327.     if (no_data(data, SIZEOF_STORED_DOUBLE * 2)) return 1;
  328.     float8get(*y, data + SIZEOF_STORED_DOUBLE);
  329.     return 0;
  330.   }
  332.   bool dimension(uint32 *dim, const char **end) const
  333.   {
  334.     *dim= 0;
  335.     *end= 0; /* No default end */
  336.     return 0;
  337.   }
  338.   const Class_info *get_class_info() const;
  339. };
  342. /***************************** LineString *******************************/
  344. class Gis_line_string: public Geometry
  345. {
  346. public:
  347.   uint32 get_data_size() const;
  348.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  349.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  350.   bool get_data_as_wkt(String *txt, const char **end) const;
  351.   bool get_mbr(MBR *mbr, const char **end) const;
  352.   int length(double *len) const;
  353.   int is_closed(int *closed) const;
  354.   int num_points(uint32 *n_points) const;
  355.   int start_point(String *point) const;
  356.   int end_point(String *point) const;
  357.   int point_n(uint32 n, String *result) const;
  358.   bool dimension(uint32 *dim, const char **end) const
  359.   {
  360.     *dim= 1;
  361.     *end= 0; /* No default end */
  362.     return 0;
  363.   }
  364.   const Class_info *get_class_info() const;
  365. };
  368. /***************************** Polygon *******************************/
  370. class Gis_polygon: public Geometry
  371. {
  372. public:
  373.   uint32 get_data_size() const;
  374.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  375.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  376.   bool get_data_as_wkt(String *txt, const char **end) const;
  377.   bool get_mbr(MBR *mbr, const char **end) const;
  378.   int area(double *ar, const char **end) const;
  379.   int exterior_ring(String *result) const;
  380.   int num_interior_ring(uint32 *n_int_rings) const;
  381.   int interior_ring_n(uint32 num, String *result) const;
  382.   int centroid_xy(double *x, double *y) const;
  383.   int centroid(String *result) const;
  384.   bool dimension(uint32 *dim, const char **end) const
  385.   {
  386.     *dim= 2;
  387.     *end= 0; /* No default end */
  388.     return 0;
  389.   }
  390.   const Class_info *get_class_info() const;
  391. };
  394. /***************************** MultiPoint *******************************/
  396. class Gis_multi_point: public Geometry
  397. {
  398. public:
  399.   uint32 get_data_size() const;
  400.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  401.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  402.   bool get_data_as_wkt(String *txt, const char **end) const;
  403.   bool get_mbr(MBR *mbr, const char **end) const;
  404.   int num_geometries(uint32 *num) const;
  405.   int geometry_n(uint32 num, String *result) const;
  406.   bool dimension(uint32 *dim, const char **end) const
  407.   {
  408.     *dim= 0;
  409.     *end= 0; /* No default end */
  410.     return 0;
  411.   }
  412.   const Class_info *get_class_info() const;
  413. };
  416. /***************************** MultiLineString *******************************/
  418. class Gis_multi_line_string: public Geometry
  419. {
  420. public:
  421.   uint32 get_data_size() const;
  422.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  423.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  424.   bool get_data_as_wkt(String *txt, const char **end) const;
  425.   bool get_mbr(MBR *mbr, const char **end) const;
  426.   int num_geometries(uint32 *num) const;
  427.   int geometry_n(uint32 num, String *result) const;
  428.   int length(double *len) const;
  429.   int is_closed(int *closed) const;
  430.   bool dimension(uint32 *dim, const char **end) const
  431.   {
  432.     *dim= 1;
  433.     *end= 0; /* No default end */
  434.     return 0;
  435.   }
  436.   const Class_info *get_class_info() const;
  437. };
  440. /***************************** MultiPolygon *******************************/
  442. class Gis_multi_polygon: public Geometry
  443. {
  444. public:
  445.   uint32 get_data_size() const;
  446.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  447.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  448.   bool get_data_as_wkt(String *txt, const char **end) const;
  449.   bool get_mbr(MBR *mbr, const char **end) const;
  450.   int num_geometries(uint32 *num) const;
  451.   int geometry_n(uint32 num, String *result) const;
  452.   int area(double *ar, const char **end) const;
  453.   int centroid(String *result) const;
  454.   bool dimension(uint32 *dim, const char **end) const
  455.   {
  456.     *dim= 2;
  457.     *end= 0; /* No default end */
  458.     return 0;
  459.   }
  460.   const Class_info *get_class_info() const;
  461. };
  464. /*********************** GeometryCollection *******************************/
  466. class Gis_geometry_collection: public Geometry
  467. {
  468. public:
  469.   uint32 get_data_size() const;
  470.   bool init_from_wkt(Gis_read_stream *trs, String *wkb);
  471.   uint init_from_wkb(const char *wkb, uint len, wkbByteOrder bo, String *res);
  472.   bool get_data_as_wkt(String *txt, const char **end) const;
  473.   bool get_mbr(MBR *mbr, const char **end) const;
  474.   int num_geometries(uint32 *num) const;
  475.   int geometry_n(uint32 num, String *result) const;
  476.   bool dimension(uint32 *dim, const char **end) const;
  477.   const Class_info *get_class_info() const;
  478. };
  480. const int geometry_buffer_size= sizeof(Gis_point);
  481. struct Geometry_buffer
  482. {
  483.   void *arr[(geometry_buffer_size - 1)/sizeof(void *) + 1];
  484. };
  486. #endif /*HAVE_SPATAIAL*/
  487. #endif