bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

build_shapes: Re-include Corse and Malta

Browse Source
This commit is contained in:
Jonas Hörsch 2018-08-27 19:11:36 +02:00
parent cd44a92185
commit 21fd124d6e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/build_shapes.py
View File

@ -14,7 +14,7 @@ try:
except ImportError: except ImportError:
from countrycode import countrycode from countrycode import countrycode
def _simplify_polys(polys, minarea=0.1, tolerance=0.01): def _simplify_polys(polys, minarea=0.1, tolerance=0.01, filterremote=True):
if isinstance(polys, MultiPolygon): if isinstance(polys, MultiPolygon):
polys = sorted(polys, key=attrgetter('area'), reverse=True) polys = sorted(polys, key=attrgetter('area'), reverse=True)
mainpoly = polys[0] mainpoly = polys[0]
@ -22,7 +22,7 @@ def _simplify_polys(polys, minarea=0.1, tolerance=0.01):
if mainpoly.area > minarea: if mainpoly.area > minarea:
polys = MultiPolygon([p polys = MultiPolygon([p
for p in takewhile(lambda p: p.area > minarea, polys) for p in takewhile(lambda p: p.area > minarea, polys)
if mainpoly.distance(p) < mainlength]) if not filterremote or (mainpoly.distance(p) < mainlength)])
else: else:
polys = mainpoly polys = mainpoly
return polys.simplify(tolerance=tolerance) return polys.simplify(tolerance=tolerance)
@ -49,7 +49,7 @@ def eez(country_shapes):
df = gpd.read_file(snakemake.input.eez) df = gpd.read_file(snakemake.input.eez)
df = df.loc[df['ISO_3digit'].isin(cntries3)] df = df.loc[df['ISO_3digit'].isin(cntries3)]
df['name'] = countrycode(df['ISO_3digit'], origin='iso3c', target='iso2c') df['name'] = countrycode(df['ISO_3digit'], origin='iso3c', target='iso2c')
s = df.set_index('name').geometry.map(_simplify_polys) s = df.set_index('name').geometry.map(lambda s: _simplify_polys(s, filterremote=False))
return gpd.GeoSeries({k:v for k,v in s.iteritems() if v.distance(country_shapes[k]) < 1e-3}) return gpd.GeoSeries({k:v for k,v in s.iteritems() if v.distance(country_shapes[k]) < 1e-3})
def country_cover(country_shapes, eez_shapes=None): def country_cover(country_shapes, eez_shapes=None):