X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/6ad397d4a971820e06ea4117e8fe8372e8bba17f..72be143cef05bf77655cebbfb0f82b8d3426872e:/lib-sql/functions/utils.sql diff --git a/lib-sql/functions/utils.sql b/lib-sql/functions/utils.sql index ff2f037d..50116566 100644 --- a/lib-sql/functions/utils.sql +++ b/lib-sql/functions/utils.sql @@ -73,6 +73,26 @@ END; $$ LANGUAGE plpgsql IMMUTABLE; + +CREATE OR REPLACE FUNCTION get_rel_node_members(members JSONB, memberLabels TEXT[]) + RETURNS SETOF BIGINT + AS $$ +DECLARE + member JSONB; +BEGIN + FOR member IN SELECT * FROM jsonb_array_elements(members) + LOOP + IF member->>'type' = 'N' and member->>'role' = ANY(memberLabels) THEN + RETURN NEXT (member->>'ref')::bigint; + END IF; + END LOOP; + + RETURN; +END; +$$ +LANGUAGE plpgsql IMMUTABLE; + + -- Copy 'name' to or from the default language. -- -- \param country_code Country code of the object being named. @@ -144,18 +164,44 @@ CREATE OR REPLACE FUNCTION get_country_code(place geometry) DECLARE place_centre GEOMETRY; nearcountry RECORD; + countries TEXT[]; BEGIN place_centre := ST_PointOnSurface(place); -- RAISE WARNING 'get_country_code, start: %', ST_AsText(place_centre); -- Try for a OSM polygon - FOR nearcountry IN - SELECT country_code from location_area_country - WHERE country_code is not null and st_covers(geometry, place_centre) limit 1 - LOOP - RETURN nearcountry.country_code; - END LOOP; + SELECT array_agg(country_code) FROM location_area_country + WHERE country_code is not null and st_covers(geometry, place_centre) + INTO countries; + + IF array_length(countries, 1) = 1 THEN + RETURN countries[1]; + END IF; + + IF array_length(countries, 1) > 1 THEN + -- more than one country found, confirm against the fallback data what to choose + FOR nearcountry IN + SELECT country_code FROM country_osm_grid + WHERE ST_Covers(geometry, place_centre) AND country_code = ANY(countries) + ORDER BY area ASC + LOOP + RETURN nearcountry.country_code; + END LOOP; + -- Still nothing? Choose the country code with the smallest partition number. + -- And failing that, just go by the alphabet. + FOR nearcountry IN + SELECT cc, + (SELECT partition FROM country_name WHERE country_code = cc) as partition + FROM unnest(countries) cc + ORDER BY partition, cc + LOOP + RETURN nearcountry.cc; + END LOOP; + + -- Should never be reached. + RETURN countries[1]; + END IF; -- RAISE WARNING 'osm fallback: %', ST_AsText(place_centre); @@ -416,6 +462,20 @@ END; $$ LANGUAGE plpgsql IMMUTABLE; +CREATE OR REPLACE FUNCTION simplify_large_polygons(geometry GEOMETRY) + RETURNS GEOMETRY + AS $$ +BEGIN + IF ST_GeometryType(geometry) in ('ST_Polygon','ST_MultiPolygon') + and ST_MemSize(geometry) > 3000000 + THEN + geometry := ST_SimplifyPreserveTopology(geometry, 0.0001); + END IF; + RETURN geometry; +END; +$$ +LANGUAGE plpgsql IMMUTABLE; + CREATE OR REPLACE FUNCTION place_force_delete(placeid BIGINT) RETURNS BOOLEAN