stud/IV/aggregate-rivers.sql
Motiejus Jakštys 395c52d4f4 more debugging
2021-04-17 13:49:05 +03:00

41 lines
1.6 KiB
PL/PgSQL

/* Aggregates rivers by name and proximity. */
drop function if exists aggregate_rivers;
create function aggregate_rivers() returns table(osm_id bigint, name text, way geometry) as $$
declare
c record;
cc record;
changed boolean;
begin
while (select count(1) from aggregate_rivers_tmp) > 0 loop
select * from aggregate_rivers_tmp limit 1 into c;
delete from aggregate_rivers_tmp a where a.osm_id = c.osm_id;
changed = true;
while changed loop
changed = false;
for cc in (select * from aggregate_rivers_tmp a where a.name = c.name and st_dwithin(a.way, c.way, 500)) loop
c.way = st_linemerge(st_union(c.way, cc.way));
delete from aggregate_rivers_tmp a where a.osm_id = cc.osm_id;
changed = true;
end loop;
end loop; -- while changed
return query select c.osm_id, c.name, c.way;
end loop; -- count(1) from aggregate_rivers_tmp > 0
return;
end
$$ language plpgsql;
drop index if exists aggregate_rivers_tmp_id;
drop index if exists aggregate_rivers_tmp_gix;
drop table if exists aggregate_rivers_tmp;
create temporary table aggregate_rivers_tmp (osm_id bigint, name text, way geometry);
create index aggregate_rivers_tmp_id on aggregate_rivers_tmp(osm_id);
create index aggregate_rivers_tmp_gix on aggregate_rivers_tmp using gist(way) include(name);
insert into aggregate_rivers_tmp
select p.osm_id, p.name, p.way from planet_osm_line p
where waterway in ('river', 'stream', 'canal') and :where;
drop table if exists agg_rivers;
create table agg_rivers as (select * from aggregate_rivers() where st_length(way) >= 50000);
drop table aggregate_rivers_tmp;