Add postgres implementation of tile_for_point function.

[rails.git] / db / functions / quadtile.c

diff --git a/db/functions/quadtile.c b/db/functions/quadtile.c
index c0eb1ec3a38fdc57e00b74714bd398a505fb8a76..29758edc6cf7b53c556acfc03aab9e98b56ef6bf 100644 (file)
--- a/db/functions/quadtile.c
+++ b/db/functions/quadtile.c
@@ -1,8 +1,21 @@
+#ifndef USE_MYSQL
+#ifndef USE_PGSQL
+#error One of USE_MYSQL or USE_PGSQL must be defined
+#endif
+#endif
+
+#include <math.h>
+#include <quad_tile.h>
+
+#ifdef USE_MYSQL
+#ifdef USE_PGSQL
+#error ONLY one of USE_MYSQL and USE_PGSQL should be defined
+#endif
+

 #include <my_global.h>
 #include <my_sys.h>
 #include <m_string.h>
 #include <mysql.h>
 #include <my_global.h>
 #include <my_sys.h>
 #include <m_string.h>
 #include <mysql.h>

-#include <quad_tile.h>

 
 my_bool tile_for_point_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
 {
 
 my_bool tile_for_point_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
 {


@@ -27,5 +40,37 @@ long long tile_for_point(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *
    long long lat = *(long long *)args->args[0];
    long long lon = *(long long *)args->args[1];
 
    long long lat = *(long long *)args->args[0];
    long long lon = *(long long *)args->args[1];
 

-   return xy2tile(lon2x(lon / 1000000.0), lat2y(lat / 1000000.0));
+   return xy2tile(lon2x(lon / 10000000.0), lat2y(lat / 10000000.0));
+}
+#endif
+
+#ifdef USE_PGSQL
+#ifdef USE_MYSQL
+#error ONLY one of USE_MYSQL and USE_PGSQL should be defined
+#endif
+
+#include <postgres.h>
+#include <fmgr.h>
+
+Datum
+tile_for_point(PG_FUNCTION_ARGS)
+{
+  double lat = PG_GETARG_INT32(0) / 10000000.0;
+  double lon = PG_GETARG_INT32(1) / 10000000.0;
+
+  PG_RETURN_INT64(xy2tile(lon2x(lon), lat2y(lat)));

 }
 }

+
+PG_FUNCTION_INFO_V1(tile_for_point);
+
+/*
+ * To bind this into PGSQL, try something like:
+ *
+ * CREATE FUNCTION tile_for_point(int4, int4) RETURNS int8
+ *  AS '/path/to/rails-port/db/functions/libpgosm', 'tile_for_point'
+ *  LANGUAGE C STRICT;
+ *
+ * (without all the *s)
+ */
+
+#endif