rio-mucho 0.0.1

Install

From pypi:

pip install rio-mucho --pre

From github (usually for a branch / dev):

pip install pip install git+ssh://git@github.com/mapbox/rio-mucho.git@<branch>

Development:

git clone git@github.com:mapbox/rio-mucho.git
cd rio-mucho
pip install -e .

Usage

with riomucho.RioMucho([{inputs}], {output}, {run function},
    windows={windows},
    global_args={global arguments},
    kwargs={kwargs to write}) as rios:

    rios.run({processes})

def basic_run({data}, {window}, {ij}, {global args}):
    ## do something
    return {out}

global_args = {
    'divide_value': 2
}

Example

import riomucho, rasterio, numpy

def basic_run(data, window, ij, g_args):
    ## do something
    out = np.array(
        [d[0] /= global_args['divide'] for d in data]
        )
    return out

# get windows from an input
with rasterio.open('/tmp/test_1.tif') as src:
    ## grabbing the windows as an example. Default behavior is identical.
    windows = [[window, ij] for ij, window in src.block_windows()]
    kwargs = src.meta
    # since we are only writing to 2 bands
    kwargs.update(count=2)

global_args = {
    'divide': 2
}

processes = 4

# run it
with riomucho.RioMucho(['input1.tif','input2.tif'], 'output.tif', basic_run,
    windows=windows,
    global_args=global_args,
    kwargs=kwargs) as rm:

    rm.run(processes)

Utility functions

`riomucho.utils.array_stack([array, array, array,…])

Given a list of ({depth}, {rows}, {cols}) numpy arrays, stack into a single (l{list length * each image depth}, {rows}, {cols}) array. This is useful for handling variation between rgb inputs of a single file, or separate files for each.

files = ['rgb.tif']
open_files = [rasterio.open(f) for f in files]
rgb = `riomucho.utils.array_stack([src.read() for src in open_files])

files = ['r.tif', 'g.tif', 'b.tif']
open_files = [rasterio.open(f) for f in files]
rgb = `riomucho.utils.array_stack([src.read() for src in open_files])