Merge branch 'master' into openstreetbugs
[rails.git] / db / functions / maptile.c
1 #ifndef USE_MYSQL
2 #ifndef USE_PGSQL
3 #error One of USE_MYSQL or USE_PGSQL must be defined
4 #endif
5 #endif
6
7 #include <math.h>
8
9 /* The real maptile-for-point functionality is here */
10
11 static long long internal_maptile_for_point(double lat, double lon, long long zoom)
12 {
13    double       scale = pow(2, zoom);
14    double       r_per_d = M_PI / 180;
15    unsigned int x;
16    unsigned int y;
17
18    x = floor((lon + 180.0) * scale / 360.0);
19    y = floor((1 - log(tan(lat * r_per_d) + 1.0 / cos(lat * r_per_d)) / M_PI) * scale / 2.0);
20
21    return (x << zoom) | y;
22 }
23
24 #ifdef USE_MYSQL
25 #ifdef USE_PGSQL
26 #error ONLY one of USE_MYSQL and USE_PGSQL should be defined
27 #endif
28
29 #include <my_global.h>
30 #include <my_sys.h>
31 #include <m_string.h>
32 #include <mysql.h>
33
34 my_bool maptile_for_point_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
35 {
36    if ( args->arg_count != 3 ||
37         args->arg_type[0] != INT_RESULT ||
38         args->arg_type[1] != INT_RESULT ||
39         args->arg_type[2] != INT_RESULT )
40    {
41       strcpy( message, "Your maptile_for_point arguments are bogus!" );
42       return 1;
43    }
44
45    return 0;
46 }
47
48 void maptile_for_point_deinit(UDF_INIT *initid)
49 {
50    return;
51 }
52
53 long long maptile_for_point(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error)
54 {
55    double       lat = *(long long *)args->args[0] / 10000000.0;
56    double       lon = *(long long *)args->args[1] / 10000000.0;
57    long long    zoom = *(long long *)args->args[2];
58
59    return internal_maptile_for_point(lat, lon, zoom);
60 }
61 #endif
62
63 #ifdef USE_PGSQL
64 #ifdef USE_MYSQL
65 #error ONLY one of USE_MYSQL and USE_PGSQL should be defined
66 #endif
67
68 #include <postgres.h>
69 #include <fmgr.h>
70
71 Datum
72 maptile_for_point(PG_FUNCTION_ARGS)
73 {
74   double lat = PG_GETARG_INT64(0) / 10000000.0;
75   double lon = PG_GETARG_INT64(1) / 10000000.0;
76   int zoom = PG_GETARG_INT32(2);
77
78   PG_RETURN_INT32(internal_maptile_for_point(lat, lon, zoom));
79 }
80
81 PG_FUNCTION_INFO_V1(maptile_for_point);
82
83 /*
84  * To bind this into PGSQL, try something like:
85  *
86  * CREATE FUNCTION maptile_for_point(int8, int8, int4) RETURNS int4
87  *  AS '/path/to/rails-port/db/functions/libpgosm', 'maptile_for_point'
88  *  LANGUAGE C STRICT;
89  *
90  * (without all the *s)
91  */
92
93 #ifdef PG_MODULE_MAGIC
94 PG_MODULE_MAGIC;
95 #endif
96
97 #endif