]> git.openstreetmap.org Git - nominatim.git/blob - sql/partition-functions.src.sql
enable 3 spacing rules again, no PHP file needed changes
[nominatim.git] / sql / partition-functions.src.sql
1 create or replace function getNearFeatures(in_partition INTEGER, feature 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(feature, centroid)) as distance, isguess, postcode, centroid FROM (
10         SELECT * FROM location_area_large_-partition- WHERE ST_Intersects(geometry, feature) and rank_search < maxrank
11         UNION ALL
12         SELECT * FROM location_area_country WHERE ST_Intersects(geometry, feature) and rank_search < maxrank
13       ) as location_area
14       GROUP BY place_id, keywords, rank_address, rank_search, isguess, postcode, centroid
15       ORDER BY rank_address, isin_tokens && keywords desc, isguess asc,
16         ST_Distance(feature, 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, postcode TEXT,
60   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
61 DECLARE
62 BEGIN
63   IF in_rank_address = 0 THEN
64     RETURN TRUE;
65   END IF;
66
67   IF in_rank_search <= 4 THEN
68     INSERT INTO location_area_country (partition, place_id, country_code, keywords, rank_search, rank_address, isguess, centroid, geometry)
69       values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry);
70     RETURN TRUE;
71   END IF;
72
73 -- start
74   IF in_partition = -partition- THEN
75     INSERT INTO location_area_large_-partition- (partition, place_id, country_code, keywords, rank_search, rank_address, isguess, postcode, centroid, geometry)
76       values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, postcode, in_centroid, in_geometry);
77     RETURN TRUE;
78   END IF;
79 -- end
80
81   RAISE EXCEPTION 'Unknown partition %', in_partition;
82   RETURN FALSE;
83 END
84 $$
85 LANGUAGE plpgsql;
86
87 create or replace function getNearestNamedFeature(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_token INTEGER) RETURNS setof nearfeature AS $$
88 DECLARE
89   r nearfeature%rowtype;
90 BEGIN
91
92 -- start
93   IF in_partition = -partition- THEN
94     FOR r IN 
95       SELECT place_id, name_vector, address_rank, search_rank,
96           ST_Distance(centroid, point) as distance, null as isguess
97           FROM search_name_-partition-
98           WHERE name_vector @> ARRAY[isin_token]
99           AND search_rank < maxrank
100       UNION ALL
101       SELECT place_id, name_vector, address_rank, search_rank,
102           ST_Distance(centroid, point) as distance, null as isguess
103           FROM search_name_country
104           WHERE name_vector @> ARRAY[isin_token]
105           AND search_rank < maxrank
106       ORDER BY distance ASC limit 1
107     LOOP
108       RETURN NEXT r;
109     END LOOP;
110     RETURN;
111   END IF;
112 -- end
113
114   RAISE EXCEPTION 'Unknown partition %', in_partition;
115 END
116 $$
117 LANGUAGE plpgsql;
118
119 create or replace function getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[]) 
120   RETURNS setof nearfeature AS $$
121 DECLARE
122   r nearfeature%rowtype;
123 BEGIN
124
125 -- start
126   IF in_partition = -partition- THEN
127     FOR r IN 
128       SELECT place_id, name_vector, address_rank, search_rank,
129           ST_Distance(centroid, point) as distance, null as isguess
130           FROM search_name_-partition-
131           WHERE name_vector @> isin_token
132           AND ST_DWithin(centroid, point, 0.01) 
133           AND search_rank between 26 and 27
134       ORDER BY distance ASC limit 1
135     LOOP
136       RETURN NEXT r;
137     END LOOP;
138     RETURN;
139   END IF;
140 -- end
141
142   RAISE EXCEPTION 'Unknown partition %', in_partition;
143 END
144 $$
145 LANGUAGE plpgsql;
146
147 create or replace function getNearestNamedPlaceFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[]) 
148   RETURNS setof nearfeature AS $$
149 DECLARE
150   r nearfeature%rowtype;
151 BEGIN
152
153 -- start
154   IF in_partition = -partition- THEN
155     FOR r IN 
156       SELECT place_id, name_vector, address_rank, search_rank,
157           ST_Distance(centroid, point) as distance, null as isguess
158           FROM search_name_-partition-
159           WHERE name_vector @> isin_token
160           AND ST_DWithin(centroid, point, 0.03) 
161           AND search_rank between 16 and 22
162       ORDER BY distance ASC limit 1
163     LOOP
164       RETURN NEXT r;
165     END LOOP;
166     RETURN;
167   END IF;
168 -- end
169
170   RAISE EXCEPTION 'Unknown partition %', in_partition;
171 END
172 $$
173 LANGUAGE plpgsql;
174
175
176 create or replace function insertSearchName(
177   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), 
178   in_name_vector INTEGER[], in_nameaddress_vector INTEGER[],
179   in_rank_search INTEGER, in_rank_address INTEGER, in_importance FLOAT,
180   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
181 DECLARE
182 BEGIN
183
184   DELETE FROM search_name WHERE place_id = in_place_id;
185   INSERT INTO search_name (place_id, search_rank, address_rank, importance, country_code, name_vector, nameaddress_vector, centroid)
186     values (in_place_id, in_rank_search, in_rank_address, in_importance, in_country_code, in_name_vector, in_nameaddress_vector, in_centroid);
187
188   IF in_rank_search <= 4 THEN
189     DELETE FROM search_name_country WHERE place_id = in_place_id;
190     IF in_rank_address > 0 THEN
191       INSERT INTO search_name_country (place_id, search_rank, address_rank, name_vector, centroid)
192         values (in_place_id, in_rank_search, in_rank_address, in_name_vector, in_geometry);
193     END IF;
194     RETURN TRUE;
195   END IF;
196
197 -- start
198   IF in_partition = -partition- THEN
199     DELETE FROM search_name_-partition- values WHERE place_id = in_place_id;
200     IF in_rank_address > 0 THEN
201       INSERT INTO search_name_-partition- (place_id, search_rank, address_rank, name_vector, centroid)
202         values (in_place_id, in_rank_search, in_rank_address, in_name_vector, in_geometry);
203     END IF;
204     RETURN TRUE;
205   END IF;
206 -- end
207
208   RAISE EXCEPTION 'Unknown partition %', in_partition;
209   RETURN FALSE;
210 END
211 $$
212 LANGUAGE plpgsql;
213
214 create or replace function deleteSearchName(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
215 DECLARE
216 BEGIN
217
218   DELETE from search_name WHERE place_id = in_place_id;
219   DELETE from search_name_country WHERE place_id = in_place_id;
220
221 -- start
222   IF in_partition = -partition- THEN
223     DELETE from search_name_-partition- WHERE place_id = in_place_id;
224     RETURN TRUE;
225   END IF;
226 -- end
227
228   RAISE EXCEPTION 'Unknown partition %', in_partition;
229
230   RETURN FALSE;
231 END
232 $$
233 LANGUAGE plpgsql;
234
235 create or replace function insertLocationRoad(
236   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
237 DECLARE
238 BEGIN
239
240 -- start
241   IF in_partition = -partition- THEN
242     DELETE FROM location_road_-partition- where place_id = in_place_id;
243     INSERT INTO location_road_-partition- (partition, place_id, country_code, geometry)
244       values (in_partition, in_place_id, in_country_code, in_geometry);
245     RETURN TRUE;
246   END IF;
247 -- end
248
249   RAISE EXCEPTION 'Unknown partition %', in_partition;
250   RETURN FALSE;
251 END
252 $$
253 LANGUAGE plpgsql;
254
255 create or replace function deleteRoad(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
256 DECLARE
257 BEGIN
258
259 -- start
260   IF in_partition = -partition- THEN
261     DELETE FROM location_road_-partition- where place_id = in_place_id;
262     RETURN TRUE;
263   END IF;
264 -- end
265
266   RAISE EXCEPTION 'Unknown partition %', in_partition;
267
268   RETURN FALSE;
269 END
270 $$
271 LANGUAGE plpgsql;
272
273 create or replace function getNearestRoadFeature(in_partition INTEGER, point GEOMETRY) RETURNS setof nearfeature AS $$
274 DECLARE
275   r nearfeature%rowtype;
276   search_diameter FLOAT;  
277 BEGIN
278
279 -- start
280   IF in_partition = -partition- THEN
281     search_diameter := 0.00005;
282     WHILE search_diameter < 0.1 LOOP
283       FOR r IN 
284         SELECT place_id, null, null, null,
285             ST_Distance(geometry, point) as distance, null as isguess
286             FROM location_road_-partition-
287             WHERE ST_DWithin(geometry, point, search_diameter) 
288         ORDER BY distance ASC limit 1
289       LOOP
290         RETURN NEXT r;
291         RETURN;
292       END LOOP;
293       search_diameter := search_diameter * 2;
294     END LOOP;
295     RETURN;
296   END IF;
297 -- end
298
299   RAISE EXCEPTION 'Unknown partition %', in_partition;
300 END
301 $$
302 LANGUAGE plpgsql;
303
304 create or replace function getNearestParellelRoadFeature(in_partition INTEGER, line GEOMETRY) RETURNS setof nearfeature AS $$
305 DECLARE
306   r nearfeature%rowtype;
307   search_diameter FLOAT;  
308   p1 GEOMETRY;
309   p2 GEOMETRY;
310   p3 GEOMETRY;
311 BEGIN
312
313   IF st_geometrytype(line) not in ('ST_LineString') THEN
314     RETURN;
315   END IF;
316
317   p1 := ST_LineInterpolatePoint(line,0);
318   p2 := ST_LineInterpolatePoint(line,0.5);
319   p3 := ST_LineInterpolatePoint(line,1);
320
321 -- start
322   IF in_partition = -partition- THEN
323     search_diameter := 0.0005;
324     WHILE search_diameter < 0.01 LOOP
325       FOR r IN 
326         SELECT place_id, null, null, null,
327             ST_Distance(geometry, line) as distance, null as isguess
328             FROM location_road_-partition-
329             WHERE ST_DWithin(line, geometry, search_diameter)
330             ORDER BY (ST_distance(geometry, p1)+
331                       ST_distance(geometry, p2)+
332                       ST_distance(geometry, p3)) ASC limit 1
333       LOOP
334         RETURN NEXT r;
335         RETURN;
336       END LOOP;
337       search_diameter := search_diameter * 2;
338     END LOOP;
339     RETURN;
340   END IF;
341 -- end
342
343   RAISE EXCEPTION 'Unknown partition %', in_partition;
344 END
345 $$
346 LANGUAGE plpgsql;