]> 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, 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 getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[]) 
88   RETURNS setof nearfeature AS $$
89 DECLARE
90   r nearfeature%rowtype;
91 BEGIN
92
93 -- start
94   IF in_partition = -partition- THEN
95     FOR r IN 
96       SELECT place_id, name_vector, address_rank, search_rank,
97           ST_Distance(centroid, point) as distance, null as isguess
98           FROM search_name_-partition-
99           WHERE name_vector && isin_token
100           AND ST_DWithin(centroid, point, 0.015)
101           AND search_rank between 26 and 27
102       ORDER BY distance ASC limit 1
103     LOOP
104       RETURN NEXT r;
105     END LOOP;
106     RETURN;
107   END IF;
108 -- end
109
110   RAISE EXCEPTION 'Unknown partition %', in_partition;
111 END
112 $$
113 LANGUAGE plpgsql;
114
115 create or replace function getNearestNamedPlaceFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[]) 
116   RETURNS setof nearfeature AS $$
117 DECLARE
118   r nearfeature%rowtype;
119 BEGIN
120
121 -- start
122   IF in_partition = -partition- THEN
123     FOR r IN 
124       SELECT place_id, name_vector, address_rank, search_rank,
125           ST_Distance(centroid, point) as distance, null as isguess
126           FROM search_name_-partition-
127           WHERE name_vector && isin_token
128           AND ST_DWithin(centroid, point, 0.04)
129           AND search_rank between 16 and 22
130       ORDER BY distance ASC limit 1
131     LOOP
132       RETURN NEXT r;
133     END LOOP;
134     RETURN;
135   END IF;
136 -- end
137
138   RAISE EXCEPTION 'Unknown partition %', in_partition;
139 END
140 $$
141 LANGUAGE plpgsql;
142
143
144 create or replace function insertSearchName(
145   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), 
146   in_name_vector INTEGER[], in_nameaddress_vector INTEGER[],
147   in_rank_search INTEGER, in_rank_address INTEGER, in_importance FLOAT,
148   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
149 DECLARE
150 BEGIN
151
152   DELETE FROM search_name WHERE place_id = in_place_id;
153   INSERT INTO search_name (place_id, search_rank, address_rank, importance, country_code, name_vector, nameaddress_vector, centroid)
154     values (in_place_id, in_rank_search, in_rank_address, in_importance, in_country_code, in_name_vector, in_nameaddress_vector, in_centroid);
155
156 -- start
157   IF in_partition = -partition- THEN
158     DELETE FROM search_name_-partition- values WHERE place_id = in_place_id;
159     IF in_rank_address > 0 THEN
160       INSERT INTO search_name_-partition- (place_id, search_rank, address_rank, name_vector, centroid)
161         values (in_place_id, in_rank_search, in_rank_address, in_name_vector, in_geometry);
162     END IF;
163     RETURN TRUE;
164   END IF;
165 -- end
166
167   RAISE EXCEPTION 'Unknown partition %', in_partition;
168   RETURN FALSE;
169 END
170 $$
171 LANGUAGE plpgsql;
172
173 create or replace function deleteSearchName(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
174 DECLARE
175 BEGIN
176
177   DELETE from search_name WHERE place_id = in_place_id;
178
179 -- start
180   IF in_partition = -partition- THEN
181     DELETE from search_name_-partition- WHERE place_id = in_place_id;
182     RETURN TRUE;
183   END IF;
184 -- end
185
186   RAISE EXCEPTION 'Unknown partition %', in_partition;
187
188   RETURN FALSE;
189 END
190 $$
191 LANGUAGE plpgsql;
192
193 create or replace function insertLocationRoad(
194   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
195 DECLARE
196 BEGIN
197
198 -- start
199   IF in_partition = -partition- THEN
200     DELETE FROM location_road_-partition- where place_id = in_place_id;
201     INSERT INTO location_road_-partition- (partition, place_id, country_code, geometry)
202       values (in_partition, in_place_id, in_country_code, in_geometry);
203     RETURN TRUE;
204   END IF;
205 -- end
206
207   RAISE EXCEPTION 'Unknown partition %', in_partition;
208   RETURN FALSE;
209 END
210 $$
211 LANGUAGE plpgsql;
212
213 create or replace function deleteRoad(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
214 DECLARE
215 BEGIN
216
217 -- start
218   IF in_partition = -partition- THEN
219     DELETE FROM location_road_-partition- where place_id = in_place_id;
220     RETURN TRUE;
221   END IF;
222 -- end
223
224   RAISE EXCEPTION 'Unknown partition %', in_partition;
225
226   RETURN FALSE;
227 END
228 $$
229 LANGUAGE plpgsql;
230
231 create or replace function getNearestRoadFeature(in_partition INTEGER, point GEOMETRY) RETURNS setof nearfeature AS $$
232 DECLARE
233   r nearfeature%rowtype;
234   search_diameter FLOAT;  
235 BEGIN
236
237 -- start
238   IF in_partition = -partition- THEN
239     search_diameter := 0.00005;
240     WHILE search_diameter < 0.1 LOOP
241       FOR r IN 
242         SELECT place_id, null, null, null,
243             ST_Distance(geometry, point) as distance, null as isguess
244             FROM location_road_-partition-
245             WHERE ST_DWithin(geometry, point, search_diameter) 
246         ORDER BY distance ASC limit 1
247       LOOP
248         RETURN NEXT r;
249         RETURN;
250       END LOOP;
251       search_diameter := search_diameter * 2;
252     END LOOP;
253     RETURN;
254   END IF;
255 -- end
256
257   RAISE EXCEPTION 'Unknown partition %', in_partition;
258 END
259 $$
260 LANGUAGE plpgsql;
261
262 create or replace function getNearestParellelRoadFeature(in_partition INTEGER, line GEOMETRY) RETURNS setof nearfeature AS $$
263 DECLARE
264   r nearfeature%rowtype;
265   search_diameter FLOAT;  
266   p1 GEOMETRY;
267   p2 GEOMETRY;
268   p3 GEOMETRY;
269 BEGIN
270
271   IF st_geometrytype(line) not in ('ST_LineString') THEN
272     RETURN;
273   END IF;
274
275   p1 := ST_LineInterpolatePoint(line,0);
276   p2 := ST_LineInterpolatePoint(line,0.5);
277   p3 := ST_LineInterpolatePoint(line,1);
278
279 -- start
280   IF in_partition = -partition- THEN
281     search_diameter := 0.0005;
282     WHILE search_diameter < 0.01 LOOP
283       FOR r IN 
284         SELECT place_id, null, null, null,
285             ST_Distance(geometry, line) as distance, null as isguess
286             FROM location_road_-partition-
287             WHERE ST_DWithin(line, geometry, search_diameter)
288             ORDER BY (ST_distance(geometry, p1)+
289                       ST_distance(geometry, p2)+
290                       ST_distance(geometry, p3)) ASC limit 1
291       LOOP
292         RETURN NEXT r;
293         RETURN;
294       END LOOP;
295       search_diameter := search_diameter * 2;
296     END LOOP;
297     RETURN;
298   END IF;
299 -- end
300
301   RAISE EXCEPTION 'Unknown partition %', in_partition;
302 END
303 $$
304 LANGUAGE plpgsql;