def setup_D4_grid():
"""
Test functionality of routing when D4 is specified.
The elevation field in this test looks like:
1 2 3 4 5 6 7
1 2 3 0 5 0 7
1 2 3 4 0 0 7
1 2 3 0 5 6 7
1 2 0 0 0 6 7
1 2 3 0 5 6 7
1 2 3 4 5 6 7
"""
global frD8, frD4, lfD8, lfD4, mg1, mg2
global z, lake_nodes
mg1 = RasterModelGrid(7, 7, 1.)
mg2 = RasterModelGrid(7, 7, 1.)
z = mg1.node_x.copy() + 1.
lake_nodes = np.array([10, 16, 17, 18, 24, 32, 33, 38, 40])
z[lake_nodes] = 0.
mg1.add_field('node', 'topographic__elevation', z, units='-')
mg2.add_field('node', 'topographic__elevation', z, units='-')
frD8 = FlowRouter(mg1, method='D8')
frD4 = FlowRouter(mg2, method='D4')
lfD8 = DepressionFinderAndRouter(mg1, routing='D8')
lfD4 = DepressionFinderAndRouter(mg2, routing='D4')
def test_degenerate_drainage():
"""
This "hourglass" configuration should be one of the hardest to correctly
re-route.
"""
mg = RasterModelGrid(9, 5)
z_init = mg.node_x.copy()*0.0001 + 1.
lake_pits = np.array([7, 11, 12, 13, 17, 27, 31, 32, 33, 37])
z_init[lake_pits] = -1.
z_init[22] = 0. # the common spill pt for both lakes
z_init[21] = 0.1 # an adverse bump in the spillway
z_init[20] = -0.2 # the spillway
z = mg.add_field('node', 'topographic__elevation', z_init)
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
fr.route_flow()
lf.map_depressions()
correct_A = np.array([ 0., 0., 0., 0., 0.,
0., 1., 3., 1., 0.,
0., 5., 1., 2., 0.,
0., 1., 10., 1., 0.,
21., 21., 1., 1., 0.,
0., 1., 9., 1., 0.,
0., 3., 1., 2., 0.,
0., 1., 1., 1., 0.,
0., 0., 0., 0., 0.])
thelake = np.concatenate((lake_pits, [22])).sort()
assert_array_almost_equal(mg.at_node['drainage_area'], correct_A)
def setup_dans_grid():
"""
Create a 7x7 test grid with a well defined hole in it.
"""
from landlab import RasterModelGrid
from landlab.components.flow_routing import (FlowRouter,
DepressionFinderAndRouter)
global fr, lf, mg
global z, r_new, r_old, A_new, A_old, s_new, depr_outlet_target
mg = RasterModelGrid(7, 7, 1.)
z = np.array([
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 2.0, 0.0,
0.0, 2.0, 1.6, 1.5, 1.6, 2.0, 0.0, 0.0, 2.0, 1.7, 1.6, 1.7, 2.0, 0.0,
0.0, 2.0, 1.8, 2.0, 2.0, 2.0, 0.0, 0.0, 1.0, 0.6, 1.0, 1.0, 1.0, 0.0,
0.0, 0.0, -0.5, 0.0, 0.0, 0.0, 0.0
]).flatten()
r_old = np.array([
0, 1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 13, 14, 14, 17, 17, 17, 20, 20,
21, 21, 17, 17, 17, 27, 27, 28, 28, 37, 38, 39, 34, 34, 35, 44, 44, 44,
46, 41, 41, 42, 43, 44, 45, 46, 47, 48
]).flatten()
r_new = np.array([
0, 1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 13, 14, 14, 23, 23, 24, 20, 20,
21, 21, 30, 30, 24, 27, 27, 28, 28, 37, 38, 39, 34, 34, 35, 44, 44, 44,
46, 41, 41, 42, 43, 44, 45, 46, 47, 48
]).flatten()
A_old = np.array([[
0., 1., 1., 1., 1., 1., 0., 0., 1., 1., 1., 1., 1., 0., 1., 1., 1., 6.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 0.,
1., 2., 2., 2., 1., 1., 0., 0., 5., 0., 2., 0., 0.
]]).flatten()
A_new = np.array([[
0., 1., 1., 1., 1., 1., 0., 0., 1., 1., 1., 1., 1., 0., 1., 1., 1., 1.,
1., 1., 1., 1., 1., 3., 3., 1., 1., 1., 1., 1., 7., 1., 1., 1., 1., 0.,
1., 8., 2., 2., 1., 1., 0., 0., 11., 0., 2., 0., 0.
]]).flatten()
s_new = np.array([
0, 1, 8, 2, 9, 3, 10, 4, 11, 5, 12, 6, 7, 13, 14, 15, 20, 19, 21, 22,
27, 26, 28, 29, 34, 33, 35, 41, 40, 42, 43, 44, 36, 37, 30, 23, 16, 17,
24, 18, 25, 38, 31, 45, 46, 39, 32, 47, 48
]).flatten()
depr_outlet_target = np.array([
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, 30, 30,
30, XX, XX, XX, XX, 30, 30, 30, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX
]).flatten()
mg.add_field('node', 'topographic__elevation', z, units='-')
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
def d4_grid():
"""Test functionality of routing when D4 is specified.
The elevation field in this test looks like::
1 2 3 4 5 6 7
1 2 3 0 5 0 7
1 2 3 4 0 0 7
1 2 3 0 5 6 7
1 2 0 0 0 6 7
1 2 3 0 5 6 7
1 2 3 4 5 6 7
"""
mg1 = RasterModelGrid(7, 7, 1.)
mg2 = RasterModelGrid(7, 7, 1.)
z = mg1.node_x.copy() + 1.
lake_nodes = np.array([10, 16, 17, 18, 24, 32, 33, 38, 40])
z[lake_nodes] = 0.
mg1.add_field("node", "topographic__elevation", z, units="-")
mg2.add_field("node", "topographic__elevation", z, units="-")
frD8 = FlowRouter(mg1, method="D8")
frD4 = FlowRouter(mg2, method="D4")
lfD8 = DepressionFinderAndRouter(mg1, routing="D8")
lfD4 = DepressionFinderAndRouter(mg2, routing="D4")
class DansGrid(object):
pass
d4_grid = DansGrid()
d4_grid.mg1 = mg1
d4_grid.mg2 = mg2
d4_grid.z = z
d4_grid.lake_nodes = lake_nodes
d4_grid.frD8 = frD8
d4_grid.frD4 = frD4
d4_grid.lfD8 = lfD8
d4_grid.lfD4 = lfD4
return d4_grid
def test_three_pits():
"""
A test to ensure the component correctly handles cases where there are
multiple pits.
"""
mg = RasterModelGrid(10,10,1.)
z = mg.add_field('node', 'topographic__elevation', mg.node_x.copy())
# a sloping plane
#np.random.seed(seed=0)
#z += np.random.rand(100)/10000.
# punch some holes
z[33] = 1.
z[43] = 1.
z[37] = 4.
z[74:76] = 1.
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
fr.route_flow()
lf.map_depressions()
flow_sinks_target = np.zeros(100, dtype=bool)
flow_sinks_target[mg.boundary_nodes] = True
# no internal sinks now:
assert_array_equal(mg.at_node['flow__sink_flag'], flow_sinks_target)
# test conservation of mass:
assert_almost_equal(mg.at_node['drainage_area'
].reshape((10,10))[1:-1,1].sum(), 8.**2)
# ^all the core nodes
# test the actual flow field:
nA = np.array([ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
8., 8., 7., 6., 5., 4., 3., 2., 1., 0.,
2., 2., 1., 1., 2., 1., 1., 1., 1., 0.,
26., 26., 25., 15., 11., 10., 9., 8., 1., 0.,
2., 2., 1., 9., 2., 1., 1., 1., 1., 0.,
2., 2., 1., 1., 5., 4., 3., 2., 1., 0.,
2., 2., 1., 1., 1., 1., 3., 2., 1., 0.,
20., 20., 19., 18., 17., 12., 3., 2., 1., 0.,
2., 2., 1., 1., 1., 1., 3., 2., 1., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
assert_array_equal(mg.at_node['drainage_area'], nA)
#test a couple more properties:
lc = np.empty(100, dtype=int)
lc.fill(XX)
lc[33] = 33
lc[43] = 33
lc[37] = 37
lc[74:76] = 74
assert_array_equal(lf.lake_map, lc)
assert_array_equal(lf.lake_codes, [33, 37, 74])
assert_equal(lf.number_of_lakes, 3)
assert_array_almost_equal(lf.lake_areas, [2., 1., 2.])
assert_array_almost_equal(lf.lake_volumes, [2., 2., 4.])
def initialize(self, input_stream=None):
"""
The BMI-style initialize method takes an optional input_stream
parameter, which may be either a ModelParameterDictionary object or
an input stream from which a ModelParameterDictionary can read values.
"""
# Create a ModelParameterDictionary for the inputs
if input_stream is None:
inputs = None
elif type(input_stream) == ModelParameterDictionary:
inputs = input_stream
else:
inputs = ModelParameterDictionary(input_stream)
# Make sure the grid includes elevation data. This means either:
# 1. The grid has a node field called 'topographic__elevation', or
# 2. The input file has an item called 'ELEVATION_FIELD_NAME' *and*
# a field by this name exists in the grid.
try:
self._elev = self._grid.at_node['topographic__elevation']
except FieldError:
try:
self.topo_field_name = inputs.read_string('ELEVATION_' +
'FIELD_NAME')
except AttributeError:
print('Error: Because your grid does not have a node field')
print('called "topographic__elevation", you need to pass the')
print('name of a text input file or ModelParameterDictionary,')
print('and this file or dictionary needs to include the name')
print('of another field in your grid that contains your')
print('elevation data.')
raise AttributeError
except MissingKeyError:
print('Error: Because your grid does not have a node field')
print('called "topographic__elevation", your input file (or')
print('ModelParameterDictionary) must include an entry with')
print('the key "ELEVATION_FIELD_NAME", which gives the name')
print('of a field in your grid that contains your elevation')
print('data.')
raise MissingKeyError('ELEVATION_FIELD_NAME')
try:
self._elev = self._grid.at_node[self.topo_field_name]
except AttributeError:
print('Your grid does not seem to have a node field called',
self.topo_field_name)
else:
self.topo_field_name = 'topographic__elevation'
# create the only new output field:
self.sed_fill_depth = self._grid.add_zeros('node',
'sediment_fill__depth',
noclobber=False)
self._lf = DepressionFinderAndRouter(self._grid, routing=self._routing)
self._fr = FlowRouter(self._grid, method=self._routing)
def setup_dans_grid2():
"""
Create a 7x7 test grid with a well defined hole in it, AT THE EDGE.
"""
from landlab import RasterModelGrid
from landlab.components.flow_routing import (FlowRouter,
DepressionFinderAndRouter)
global fr, lf, mg
global z, r_new, r_old, A_new, A_old, s_new, depr_outlet_target
mg = RasterModelGrid((7, 7), (1., 1.))
z = mg.node_x.copy()
guard_sides = np.concatenate((np.arange(7, 14), np.arange(35, 42)))
edges = np.concatenate((np.arange(7), np.arange(42, 49)))
hole_here = np.array(([15, 16, 22, 23, 29, 30]))
z[guard_sides] = z[13]
z[edges] = -2. # force flow outwards from the tops of the guards
z[hole_here] = -1.
#print(z)
# A_new = np.array([[[0., 1., 1., 1., 1., 1., 0.,
# 0., 1., 1., 1., 1., 1., 0.,
# 15., 9., 4., 3., 2., 1., 0.,
# 0., 6., 4., 3., 2., 1., 0.,
# 0., 1., 4., 3., 2., 1., 0.,
# 0., 1., 1., 1., 1., 1., 0.,
# 0., 1., 1., 1., 1., 1., 0.]]]).flatten()
A_new = np.array([[[
0., 1., 1., 1., 1., 1., 0., 0., 1., 1., 1., 1., 1., 0., 15., 5., 4.,
3., 2., 1., 0., 0., 10., 4., 3., 2., 1., 0., 0., 1., 4., 3., 2., 1.,
0., 0., 1., 1., 1., 1., 1., 0., 0., 1., 1., 1., 1., 1., 0.
]]]).flatten()
depr_outlet_target = np.array([
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, 14, 14, XX,
XX, XX, XX, XX, 14, 14, XX, XX, XX, XX, XX, 14, 14, XX, XX, XX, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX, XX
]).flatten()
mg.add_field('node', 'topographic__elevation', z, units='-')
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
def test_lake_mapper():
"""
Create a test grid and run a series of tests.
"""
# Make a test grid
rmg = create_test_grid()
# Instantiate a lake mapper
# (Note that we don't need to send it an input file name, because our grid
# already has a topographic__elevation field)
lm = DepressionFinderAndRouter(rmg)
# Run it on our test grid
lm.map_depressions()
# Run tests
check_fields1(rmg)
check_array_values1(rmg, lm)
check_fields2(rmg)
check_array_values2(rmg, lm)
def test_composite_pits():
"""
A test to ensure the component correctly handles cases where there are
multiple pits, inset into each other.
"""
mg = RasterModelGrid(10, 10, 1.)
z = mg.add_field('node', 'topographic__elevation', mg.node_x.copy())
# a sloping plane
#np.random.seed(seed=0)
#z += np.random.rand(100)/10000.
# punch one big hole
z.reshape((10,10))[3:8,3:8] = 0.
# dig a couple of inset holes
z[57] = -1.
z[44] = -2.
z[54] = -10.
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
fr.route_flow()
lf.map_depressions()
flow_sinks_target = np.zeros(100, dtype=bool)
flow_sinks_target[mg.boundary_nodes] = True
# no internal sinks now:
assert_array_equal(mg.at_node['flow__sink_flag'], flow_sinks_target)
# test conservation of mass:
assert_almost_equal(mg.at_node['drainage_area'
].reshape((10,10))[1:-1,1].sum(), 8.**2)
# ^all the core nodes
# test the actual flow field:
nA = np.array([ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
8., 8., 7., 6., 5., 4., 3., 2., 1., 0.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 0.,
1., 1., 1., 4., 2., 2., 8., 4., 1., 0.,
1., 1., 1., 8., 3., 15., 3., 2., 1., 0.,
1., 1., 1., 13., 25., 6., 3., 2., 1., 0.,
1., 1., 1., 45., 3., 3., 5., 2., 1., 0.,
50., 50., 49., 3., 2., 2., 2., 4., 1., 0.,
1., 1., 1., 1., 1., 1., 1., 1., 1., 0.,
0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
assert_array_equal(mg.at_node['drainage_area'], nA)
# the lake code map:
lc = np.array([XX, XX, XX, XX, XX, XX, XX, XX, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX,
XX, XX, XX, 57, 57, 57, 57, 57, XX, XX,
XX, XX, XX, 57, 57, 57, 57, 57, XX, XX,
XX, XX, XX, 57, 57, 57, 57, 57, XX, XX,
XX, XX, XX, 57, 57, 57, 57, 57, XX, XX,
XX, XX, XX, 57, 57, 57, 57, 57, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX,
XX, XX, XX, XX, XX, XX, XX, XX, XX, XX])
#test the remaining properties:
assert_equal(lf.lake_outlets.size, 1)
assert_equal(lf.lake_outlets[0], 72)
outlets_in_map = np.unique(lf.depression_outlet_map)
assert_equal(outlets_in_map.size, 2)
assert_equal(outlets_in_map[1], 72)
assert_equal(lf.number_of_lakes, 1)
assert_equal(lf.lake_codes[0], 57)
assert_array_equal(lf.lake_map, lc)
assert_almost_equal(lf.lake_areas[0], 25.)
assert_almost_equal(lf.lake_volumes[0], 63.)
# create the fields in the grid
mg.add_zeros("topographic__elevation", at="node")
z = mg.zeros(at="node") + init_elev
# z += mg.node_x*0.001
mg["node"]["topographic__elevation"] = z + numpy.random.rand(len(z)) / 1000.
# make some K values in a field to test
mg.at_node["K_values"] = 0.1 + numpy.random.rand(nrows * ncols) / 10.
print("Running ...")
time_on = time.time()
# instantiate the components:
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(mg, "./drive_sp_params.txt")
lf = DepressionFinderAndRouter(mg)
# load the Fastscape module too, to allow direct comparison
fsp = FastscapeEroder(mg, "./drive_sp_params.txt")
# perform the loop:
elapsed_time = 0. # total time in simulation
while elapsed_time < time_to_run:
# for i in range(10):
print(elapsed_time)
if elapsed_time + dt > time_to_run:
print("Short step!")
dt = time_to_run - elapsed_time
mg = fr.route_flow(method="D8")
lf.map_depressions()
# print 'Area: ', numpy.max(mg.at_node['drainage_area'])
# mg = fsp.erode(mg)
def test_edge_draining():
"""
This tests when the lake attempts to drain from an edge, where an issue
is suspected.
"""
# Create a 7x7 test grid with a well defined hole in it, AT THE EDGE.
mg = RasterModelGrid((7, 7), (1., 1.))
z = mg.node_x.copy()
guard_sides = np.concatenate((np.arange(7, 14), np.arange(35, 42)))
edges = np.concatenate((np.arange(7), np.arange(42, 49)))
hole_here = np.array(([15, 16, 22, 23, 29, 30]))
z[guard_sides] = z[13]
z[edges] = -2. # force flow outwards from the tops of the guards
z[hole_here] = -1.
A_new = np.array([[[
0.,
1.,
1.,
1.,
1.,
1.,
0.,
0.,
1.,
1.,
1.,
1.,
1.,
0.,
15.,
5.,
4.,
3.,
2.,
1.,
0.,
0.,
10.,
4.,
3.,
2.,
1.,
0.,
0.,
1.,
4.,
3.,
2.,
1.,
0.,
0.,
1.,
1.,
1.,
1.,
1.,
0.,
0.,
1.,
1.,
1.,
1.,
1.,
0.,
]]]).flatten()
depr_outlet_target = np.array([
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
14,
14,
XX,
XX,
XX,
XX,
XX,
14,
14,
XX,
XX,
XX,
XX,
XX,
14,
14,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
XX,
]).flatten()
mg.add_field("node", "topographic__elevation", z, units="-")
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
fr.route_flow()
lf.map_depressions()
assert mg.at_node["drainage_area"] == approx(A_new)
assert lf.depression_outlet_map == approx(depr_outlet_target)
def dans_grid3():
"""
Create a 7x7 test grid with a well defined hole in it.
"""
mg = RasterModelGrid(7, 7, 1.)
z = np.array([
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
[0.0, 2.0, 2.0, 2.0, 2.0, 2.0, 0.0],
[0.0, 2.0, 1.6, 1.5, 1.6, 2.0, 0.0],
[0.0, 2.0, 1.7, 1.6, 1.7, 2.0, 0.0],
[0.0, 2.0, 1.8, 2.0, 2.0, 2.0, 0.0],
[0.0, 1.0, 0.6, 1.0, 1.0, 1.0, 0.0],
[0.0, 0.0, -0.5, 0.0, 0.0, 0.0, 0.0],
]).flatten()
r_old = np.array([
[0, 1, 2, 3, 4, 5, 6],
[7, 1, 2, 3, 4, 5, 13],
[14, 14, 17, 17, 17, 20, 20],
[21, 21, 17, 17, 17, 27, 27],
[28, 28, 37, 38, 39, 34, 34],
[35, 44, 44, 44, 46, 41, 41],
[42, 43, 44, 45, 46, 47, 48],
]).flatten()
r_new = np.array([
[0, 1, 2, 3, 4, 5, 6],
[7, 1, 2, 3, 4, 5, 13],
[14, 14, 23, 24, 24, 20, 20],
[21, 21, 30, 30, 24, 27, 27],
[28, 28, 37, 38, 39, 34, 34],
[35, 44, 44, 44, 46, 41, 41],
[42, 43, 44, 45, 46, 47, 48],
]).flatten()
A_old = np.array([
[0., 1., 1., 1., 1., 1., 0.],
[0., 1., 1., 1., 1., 1., 0.],
[1., 1., 1., 6., 1., 1., 1.],
[1., 1., 1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1., 1., 1.],
[0., 1., 2., 2., 2., 1., 1.],
[0., 0., 5., 0., 2., 0., 0.],
]).flatten()
A_new = np.array([
[0., 1., 1., 1., 1., 1., 0.],
[0., 1., 1., 1., 1., 1., 0.],
[1., 1., 1., 1., 1., 1., 1.],
[1., 1., 2., 4., 1., 1., 1.],
[1., 1., 7., 1., 1., 1., 1.],
[0., 1., 8., 2., 2., 1., 1.],
[0., 0., 11., 0., 2., 0., 0.],
]).flatten()
s_new = np.array([
[0, 1, 8, 2, 9, 3, 10],
[4, 11, 5, 12, 6, 7, 13],
[14, 15, 20, 19, 21, 22, 27],
[26, 28, 29, 34, 33, 35, 41],
[40, 42, 43, 44, 36, 37, 30],
[23, 16, 24, 17, 18, 25, 38],
[31, 45, 46, 39, 32, 47, 48],
]).flatten()
links_old = np.array([
[-1, -1, -1, -1, -1, -1, -1],
[-1, 7, 8, 9, 10, 11, -1],
[-1, 26, 28, -1, 29, 31, -1],
[-1, 39, 113, 35, 114, 44, -1],
[-1, 52, 60, 61, 62, 57, -1],
[-1, 146, 73, 149, 75, 70, -1],
[-1, -1, -1, -1, -1, -1, -1],
]).flatten()
links_new = np.array([
[-1, -1, -1, -1, -1, -1, -1],
[-1, 7, 8, 9, 10, 11, -1],
[-1, 26, 34, 35, 115, 31, -1],
[-1, 39, 47, 125, 42, 44, -1],
[-1, 52, 60, 61, 62, 57, -1],
[-1, 146, 73, 149, 75, 70, -1],
[-1, -1, -1, -1, -1, -1, -1],
]).flatten()
depr_outlet_target = np.array([
[XX, XX, XX, XX, XX, XX, XX],
[XX, XX, XX, XX, XX, XX, XX],
[XX, XX, 30, 30, 30, XX, XX],
[XX, XX, 30, 30, 30, XX, XX],
[XX, XX, XX, XX, XX, XX, XX],
[XX, XX, XX, XX, XX, XX, XX],
[XX, XX, XX, XX, XX, XX, XX],
]).flatten()
mg.add_field("node", "topographic__elevation", z, units="-")
fr = FlowRouter(mg)
lf = DepressionFinderAndRouter(mg)
class DansGrid(object):
pass
dans_grid = DansGrid()
dans_grid.mg = mg
dans_grid.fr = fr
dans_grid.lf = lf
dans_grid.z = z
dans_grid.r_new = r_new
dans_grid.r_old = r_old
dans_grid.A_new = A_new
dans_grid.A_old = A_old
dans_grid.s_new = s_new
dans_grid.depr_outlet_target = depr_outlet_target
dans_grid.links_old = links_old
dans_grid.links_new = links_new
return dans_grid
