Source code for rafem.subside

#! /usr/local/bin/python

import numpy as np


def linear_subsidence(n, riv_i, riv_j, ch_depth, sub_rate, sub_start, SL):
    """ Subside cells in rows beyond start location by linear rate.
    Updated version could be to subside at an increasing rate towards ocean. """

    subside_cells = np.zeros_like(n)
    subside_cells[sub_start:, :] = 1

    subaerial_elev = np.copy(n)
    subaerial_elev[riv_i, riv_j] += ch_depth

    subside_cells[subaerial_elev <= SL] = 0

    n[subside_cells == 1] -= sub_rate