]> git.openstreetmap.org Git - nominatim.git/blob - sql/partition-functions.src.sql
Merge remote-tracking branch 'upstream/master'
[nominatim.git] / sql / partition-functions.src.sql
1 create or replace function getNearFeatures(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_tokens INT[]) RETURNS setof nearfeaturecentr AS $$
2 DECLARE
3   r nearfeaturecentr%rowtype;
4 BEGIN
5
6 -- start
7   IF in_partition = -partition- THEN
8     FOR r IN 
9       SELECT place_id, keywords, rank_address, rank_search, min(ST_Distance(point, centroid)) as distance, isguess, centroid FROM (
10         SELECT * FROM location_area_large_-partition- WHERE ST_Contains(geometry, point) and rank_search < maxrank
11         UNION ALL
12         SELECT * FROM location_area_country WHERE ST_Contains(geometry, point) and rank_search < maxrank
13       ) as location_area
14       GROUP BY place_id, keywords, rank_address, rank_search, isguess, centroid
15       ORDER BY rank_address, isin_tokens && keywords desc, isguess asc,
16         ST_Distance(point, centroid) * 
17           CASE 
18                WHEN rank_address = 16 AND rank_search = 15 THEN 0.2 -- capital city
19                WHEN rank_address = 16 AND rank_search = 16 THEN 0.25 -- city
20                WHEN rank_address = 16 AND rank_search = 17 THEN 0.5 -- town
21                ELSE 1 END ASC -- everything else
22     LOOP
23       RETURN NEXT r;
24     END LOOP;
25     RETURN;
26   END IF;
27 -- end
28
29   RAISE EXCEPTION 'Unknown partition %', in_partition;
30 END
31 $$
32 LANGUAGE plpgsql;
33
34 create or replace function deleteLocationArea(in_partition INTEGER, in_place_id BIGINT, in_rank_search INTEGER) RETURNS BOOLEAN AS $$
35 DECLARE
36 BEGIN
37
38   IF in_rank_search <= 4 THEN
39     DELETE from location_area_country WHERE place_id = in_place_id;
40     RETURN TRUE;
41   END IF;
42
43 -- start
44   IF in_partition = -partition- THEN
45     DELETE from location_area_large_-partition- WHERE place_id = in_place_id;
46     RETURN TRUE;
47   END IF;
48 -- end
49
50   RAISE EXCEPTION 'Unknown partition %', in_partition;
51
52   RETURN FALSE;
53 END
54 $$
55 LANGUAGE plpgsql;
56
57 create or replace function insertLocationAreaLarge(
58   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_keywords INTEGER[], 
59   in_rank_search INTEGER, in_rank_address INTEGER, in_estimate BOOLEAN, 
60   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
61 DECLARE
62 BEGIN
63
64   IF in_rank_search <= 4 THEN
65     INSERT INTO location_area_country values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry);
66     RETURN TRUE;
67   END IF;
68
69 -- start
70   IF in_partition = -partition- THEN
71     INSERT INTO location_area_large_-partition- values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry);
72     RETURN TRUE;
73   END IF;
74 -- end
75
76   RAISE EXCEPTION 'Unknown partition %', in_partition;
77   RETURN FALSE;
78 END
79 $$
80 LANGUAGE plpgsql;
81
82 create or replace function getNearestNamedFeature(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_token INTEGER) RETURNS setof nearfeature AS $$
83 DECLARE
84   r nearfeature%rowtype;
85 BEGIN
86
87 -- start
88   IF in_partition = -partition- THEN
89     FOR r IN 
90       SELECT place_id, name_vector, address_rank, search_rank,
91           ST_Distance(centroid, point) as distance, null as isguess
92           FROM search_name_-partition-
93           WHERE name_vector @> ARRAY[isin_token]
94           AND search_rank < maxrank
95       UNION ALL
96       SELECT place_id, name_vector, address_rank, search_rank,
97           ST_Distance(centroid, point) as distance, null as isguess
98           FROM search_name_country
99           WHERE name_vector @> ARRAY[isin_token]
100           AND search_rank < maxrank
101       ORDER BY distance ASC limit 1
102     LOOP
103       RETURN NEXT r;
104     END LOOP;
105     RETURN;
106   END IF;
107 -- end
108
109   RAISE EXCEPTION 'Unknown partition %', in_partition;
110 END
111 $$
112 LANGUAGE plpgsql;
113
114 create or replace function getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER) 
115   RETURNS setof nearfeature AS $$
116 DECLARE
117   r nearfeature%rowtype;
118 BEGIN
119
120 -- start
121   IF in_partition = -partition- THEN
122     FOR r IN 
123       SELECT place_id, name_vector, address_rank, search_rank,
124           ST_Distance(centroid, point) as distance, null as isguess
125           FROM search_name_-partition-
126           WHERE name_vector @> ARRAY[isin_token]
127           AND ST_DWithin(centroid, point, 0.01) 
128           AND search_rank between 22 and 27
129       ORDER BY distance ASC limit 1
130     LOOP
131       RETURN NEXT r;
132     END LOOP;
133     RETURN;
134   END IF;
135 -- end
136
137   RAISE EXCEPTION 'Unknown partition %', in_partition;
138 END
139 $$
140 LANGUAGE plpgsql;
141
142 create or replace function getNearestPostcode(in_partition INTEGER, point GEOMETRY) 
143   RETURNS TEXT AS $$
144 DECLARE
145   out_postcode TEXT;
146 BEGIN
147
148 -- start
149   IF in_partition = -partition- THEN
150     SELECT postcode
151         FROM location_area_large_-partition- join placex using (place_id)
152         WHERE st_contains(location_area_large_-partition-.geometry, point)
153         AND class = 'place' and type = 'postcode' 
154       ORDER BY st_distance(location_area_large_-partition-.centroid, point) ASC limit 1
155       INTO out_postcode;
156     RETURN out_postcode;
157   END IF;
158 -- end
159
160   RAISE EXCEPTION 'Unknown partition %', in_partition;
161 END
162 $$
163 LANGUAGE plpgsql;
164
165 create or replace function insertSearchName(
166   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), 
167   in_name_vector INTEGER[], in_nameaddress_vector INTEGER[],
168   in_rank_search INTEGER, in_rank_address INTEGER, in_importance FLOAT,
169   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
170 DECLARE
171 BEGIN
172
173   DELETE FROM search_name WHERE place_id = in_place_id;
174   INSERT INTO search_name values (in_place_id, in_rank_search, in_rank_address, in_importance, in_country_code, 
175     in_name_vector, in_nameaddress_vector, in_centroid);
176
177   IF in_rank_search <= 4 THEN
178     DELETE FROM search_name_country WHERE place_id = in_place_id;
179     INSERT INTO search_name_country values (in_place_id, in_rank_search, in_rank_address, 
180       in_name_vector, in_geometry);
181     RETURN TRUE;
182   END IF;
183
184 -- start
185   IF in_partition = -partition- THEN
186     DELETE FROM search_name_-partition- values WHERE place_id = in_place_id;
187     INSERT INTO search_name_-partition- values (in_place_id, in_rank_search, in_rank_address, 
188       in_name_vector, in_geometry);
189     RETURN TRUE;
190   END IF;
191 -- end
192
193   RAISE EXCEPTION 'Unknown partition %', in_partition;
194   RETURN FALSE;
195 END
196 $$
197 LANGUAGE plpgsql;
198
199 create or replace function deleteSearchName(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
200 DECLARE
201 BEGIN
202
203   DELETE from search_name WHERE place_id = in_place_id;
204   DELETE from search_name_country WHERE place_id = in_place_id;
205
206 -- start
207   IF in_partition = -partition- THEN
208     DELETE from search_name_-partition- WHERE place_id = in_place_id;
209     RETURN TRUE;
210   END IF;
211 -- end
212
213   RAISE EXCEPTION 'Unknown partition %', in_partition;
214
215   RETURN FALSE;
216 END
217 $$
218 LANGUAGE plpgsql;
219
220 create or replace function insertLocationRoad(
221   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
222 DECLARE
223 BEGIN
224
225 -- start
226   IF in_partition = -partition- THEN
227     DELETE FROM location_road_-partition- where place_id = in_place_id;
228     INSERT INTO location_road_-partition- values (in_partition, in_place_id, in_country_code, in_geometry);
229     RETURN TRUE;
230   END IF;
231 -- end
232
233   RAISE EXCEPTION 'Unknown partition %', in_partition;
234   RETURN FALSE;
235 END
236 $$
237 LANGUAGE plpgsql;
238
239 create or replace function deleteRoad(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
240 DECLARE
241 BEGIN
242
243 -- start
244   IF in_partition = -partition- THEN
245     DELETE FROM location_road_-partition- where place_id = in_place_id;
246     RETURN TRUE;
247   END IF;
248 -- end
249
250   RAISE EXCEPTION 'Unknown partition %', in_partition;
251
252   RETURN FALSE;
253 END
254 $$
255 LANGUAGE plpgsql;
256
257 create or replace function getNearestRoadFeature(in_partition INTEGER, point GEOMETRY) RETURNS setof nearfeature AS $$
258 DECLARE
259   r nearfeature%rowtype;
260   search_diameter FLOAT;  
261 BEGIN
262
263 -- start
264   IF in_partition = -partition- THEN
265     search_diameter := 0.00005;
266     WHILE search_diameter < 0.1 LOOP
267       FOR r IN 
268         SELECT place_id, null, null, null,
269             ST_Distance(geometry, point) as distance, null as isguess
270             FROM location_road_-partition-
271             WHERE ST_DWithin(geometry, point, search_diameter) 
272         ORDER BY distance ASC limit 1
273       LOOP
274         RETURN NEXT r;
275         RETURN;
276       END LOOP;
277       search_diameter := search_diameter * 2;
278     END LOOP;
279     RETURN;
280   END IF;
281 -- end
282
283   RAISE EXCEPTION 'Unknown partition %', in_partition;
284 END
285 $$
286 LANGUAGE plpgsql;
287
288 create or replace function getNearestParellelRoadFeature(in_partition INTEGER, line GEOMETRY) RETURNS setof nearfeature AS $$
289 DECLARE
290   r nearfeature%rowtype;
291   search_diameter FLOAT;  
292   p1 GEOMETRY;
293   p2 GEOMETRY;
294   p3 GEOMETRY;
295 BEGIN
296
297   IF st_geometrytype(line) not in ('ST_LineString') THEN
298     RETURN;
299   END IF;
300
301   p1 := ST_Line_Interpolate_Point(line,0);
302   p2 := ST_Line_Interpolate_Point(line,0.5);
303   p3 := ST_Line_Interpolate_Point(line,1);
304
305 -- start
306   IF in_partition = -partition- THEN
307     search_diameter := 0.0005;
308     WHILE search_diameter < 0.01 LOOP
309       FOR r IN 
310         SELECT place_id, null, null, null,
311             ST_Distance(geometry, line) as distance, null as isguess
312             FROM location_road_-partition-
313             WHERE ST_DWithin(line, geometry, search_diameter)
314             ORDER BY (ST_distance(geometry, p1)+
315                       ST_distance(geometry, p2)+
316                       ST_distance(geometry, p3)) ASC limit 1
317       LOOP
318         RETURN NEXT r;
319         RETURN;
320       END LOOP;
321       search_diameter := search_diameter * 2;
322     END LOOP;
323     RETURN;
324   END IF;
325 -- end
326
327   RAISE EXCEPTION 'Unknown partition %', in_partition;
328 END
329 $$
330 LANGUAGE plpgsql;