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 d60189245bd3f2ef5c09fe61a91bc87beab01868..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 <quad_tile.h>
 
 my_bool tile_for_point_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
 {
@@ -29,3 +42,35 @@ long long tile_for_point(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *
 
    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