[Expand rasterio windows] Get rasterio windows that expand the shape of raster blocks #rasterio

get_ravioli_windows.py

import sys
import click
import rasterio


def expand(window, size=1):
    r, c = window
    return ((r[0] - size, r[1] + size), (c[0] - size, c[1] + size))


def check_window(window, w, h):
    r, c = window
    return ((max(0, r[0]), min(h, r[1])), (max(0, c[0]), min(w, c[1])))


@click.command()
@click.argument(
    'raster-fn',
    type=click.Path(exists=True),
    required=True
)
@click.option(
    '--expand-size',
    type=click.INT,
    default=1,
    help='Number of pixels to expand window'
)

def main(raster_fn, expand_size):
    src = rasterio.open(raster_fn)
    for _, w in src.block_windows():
        new_w = check_window(expand(w, expand_size), src.width, src.height)
        print 'Original: ', w
        print 'Expanded: ', new_w
        print


if __name__ == '__main__':
    sys.exit(main())