Initial commit

Waldbraende/sentinel_helpers.py

@@ -0,0 +1,36 @@
+import zipfile
+
+
+def band_paths(p, bands, resolution=None):
+
+    '''
+
+    Given a zip file 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 and can
+
+    thus be directly opened by rasterio.
+
+    '''
+
+    with zipfile.ZipFile(p) as f:
+
+        files = f.namelist()
+
+        rasters = [f for f in files if f.endswith('.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)