mirror of
https://github.com/heyarne/earth-observation-for-journalism.git
synced 2026-05-06 19:13:40 +02:00
44 lines
No EOL
1.5 KiB
Python
44 lines
No EOL
1.5 KiB
Python
import glob
|
|
import urllib.parse
|
|
from pathlib import Path
|
|
import zipfile
|
|
|
|
import geopandas as gp
|
|
|
|
|
|
def band_paths(p, bands, resolution=None):
|
|
'''
|
|
|
|
Given a zip file or folder at `p`, returns the paths inside p to the raster files containing
|
|
information for the given bands. Because some bands are available in more than one
|
|
resolution, this can be filtered by prodiding a third parameter (e.g. resolution='10m').
|
|
|
|
The returned paths are formatted in the zip scheme as per Apache Commons VFS if necessary
|
|
and can be directly opened by rasterio.
|
|
'''
|
|
if p.endswith('.zip'):
|
|
with zipfile.ZipFile(p) as f:
|
|
files = f.namelist()
|
|
rasters = [f for f in files if f.endswith('.jp2')]
|
|
else:
|
|
rasters = glob.glob(Path(p) / '**/*.jp2')
|
|
rasters = [r for r in rasters for b in bands if b in r]
|
|
if resolution:
|
|
rasters = [r for r in rasters if resolution in r]
|
|
|
|
rasters = ['zip+file://' + p + '!/' + r for r in rasters]
|
|
return rasters
|
|
|
|
|
|
def rgb_paths(zip_file, resolution='10m'):
|
|
return band_paths(zip_file, ['B02', 'B03', 'B04'], resolution)
|
|
|
|
|
|
def search_osm(place):
|
|
'''
|
|
Returns a GeoDataFrame with results from OpenStreetMap Nominatim for the given search string.
|
|
This allows us to fetch detailed geometries for virtually any place on earth.
|
|
'''
|
|
urlescaped_place = urllib.parse.quote(place)
|
|
search_url = 'https://nominatim.openstreetmap.org/search/?q={}&format=geojson&polygon_geojson=1'.format(urlescaped_place)
|
|
return gp.read_file(search_url) |