def instantiate_rain_generator(self):
"""Instantiate component used to generate storm sequence."""
# Handle option for duration.
if self.opt_stochastic_duration:
self.rain_generator = PrecipitationDistribution(
mean_storm_duration=self.mean_storm_duration,
mean_interstorm_duration=self.mean_interstorm_duration,
mean_storm_depth=self.mean_storm_depth,
total_t=self.clock.stop,
delta_t=self.clock.step,
random_seed=self.seed,
)
self.run_for = self.run_for_stochastic # override base method
else:
from scipy.special import gamma
self.rain_generator = PrecipitationDistribution(
mean_storm_duration=1.0,
mean_interstorm_duration=1.0,
mean_storm_depth=1.0,
random_seed=self.seed,
)
self.scale_factor = self.rainfall__mean_rate / gamma(
1.0 + (1.0 / self.shape_factor))
if (isinstance(self.number_of_sub_time_steps,
(int, np.integer)) is False):
raise ValueError(
("number_of_sub_time_steps must be of type integer."))
self.n_sub_steps = self.number_of_sub_time_steps
def instantiate_rain_generator(self):
"""Instantiate RainGenerator."""
# Handle option for duration.
self.opt_stochastic_duration = (self.params['opt_stochastic_duration'])
if self.opt_stochastic_duration:
self.rain_generator = \
PrecipitationDistribution(mean_storm_duration=self.params['mean_storm_duration'],
mean_interstorm_duration=self.params['mean_interstorm_duration'],
mean_storm_depth=self.params['mean_storm_depth'],
total_t=self.params['run_duration'],
delta_t=self.params['dt'],
random_seed=int(self.params['random_seed']))
self.run_for = self.run_for_stochastic # override base method
else:
from scipy.special import gamma
mean_storm__intensity = (self._length_factor) * self.params[
'mean_storm__intensity'] # has units length per time
intermittency_factor = self.params['intermittency_factor']
self.rain_generator = \
PrecipitationDistribution(mean_storm_duration=1.0,
mean_interstorm_duration=1.0,
mean_storm_depth=1.0,
random_seed=int(self.params['random_seed']))
self.intermittency_factor = intermittency_factor
self.mean_storm__intensity = mean_storm__intensity
self.shape_factor = self.params['precip_shape_factor']
self.scale_factor = (self.mean_storm__intensity /
gamma(1.0 + (1.0 / self.shape_factor)))
self.n_sub_steps = int(self.params['number_of_sub_time_steps'])
def __init__(self, input_file=None, params=None):
"""Initialize the StochasticDischargeHortonianModel."""
# Call ErosionModel's init
super(StochasticDischargeHortonianModel,
self).__init__(input_file=input_file, params=params)
# Instantiate components
self.flow_router = FlowRouter(self.grid, **self.params)
self.lake_filler = DepressionFinderAndRouter(self.grid, **self.params)
self.rain_generator = \
PrecipitationDistribution(delta_t=self.params['dt'],
total_time=self.params['run_duration'],
**self.params)
# Add a field for discharge
if 'surface_water__discharge' not in self.grid.at_node:
self.grid.add_zeros('node', 'surface_water__discharge')
self.discharge = self.grid.at_node['surface_water__discharge']
# Get the infiltration-capacity parameter
self.infilt = self.params['infiltration_capacity']
# Run flow routing and lake filler (only once, because we are not
# not changing topography)
self.flow_router.run_one_step()
self.lake_filler.map_depressions()
def initialize(data, grid, grid1):
"""Initialize random plant type field.
Plant types are defined as the following:
* GRASS = 0
* SHRUB = 1
* TREE = 2
* BARE = 3
* SHRUBSEEDLING = 4
* TREESEEDLING = 5
"""
grid1.at_cell['vegetation__plant_functional_type'] = compose_veg_grid(
grid1,
percent_bare=data['percent_bare_initial'],
percent_grass=data['percent_grass_initial'],
percent_shrub=data['percent_shrub_initial'],
percent_tree=data['percent_tree_initial'])
# Assign plant type for representative ecohydrologic simulations
grid.at_cell['vegetation__plant_functional_type'] = np.arange(6)
grid1.at_node['topographic__elevation'] = np.full(grid1.number_of_nodes,
1700.)
grid.at_node['topographic__elevation'] = np.full(grid.number_of_nodes,
1700.)
precip_dry = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_dry'],
mean_interstorm_duration=data['mean_interstorm_dry'],
mean_storm_depth=data['mean_storm_depth_dry'])
precip_wet = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_wet'],
mean_interstorm_duration=data['mean_interstorm_wet'],
mean_storm_depth=data['mean_storm_depth_wet'])
radiation = Radiation(grid)
pet_tree = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_tree'],
delta_d=data['DeltaD'])
pet_shrub = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_shrub'],
delta_d=data['DeltaD'])
pet_grass = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_grass'],
delta_d=data['DeltaD'])
soil_moisture = SoilMoisture(grid, **data) # Soil Moisture object
vegetation = Vegetation(grid, **data) # Vegetation object
vegca = VegCA(grid1, **data) # Cellular automaton object
# Initializing inputs for Soil Moisture object
grid.at_cell['vegetation__live_leaf_area_index'] = (
1.6 * np.ones(grid.number_of_cells))
grid.at_cell['soil_moisture__initial_saturation_fraction'] = (
0.59 * np.ones(grid.number_of_cells))
return (precip_dry, precip_wet, radiation, pet_tree, pet_shrub, pet_grass,
soil_moisture, vegetation, vegca)
def test_stoch_sp_raster_record_state():
"""
Initialize HydrologyEventStreamPower on a raster grid.
Use several storm-interstorm pairs and make sure state recorded
as expected.
"""
mg = RasterModelGrid((3, 3), xy_spacing=10.0)
mg.set_status_at_node_on_edges(
right=mg.BC_NODE_IS_CLOSED,
top=mg.BC_NODE_IS_CLOSED,
left=mg.BC_NODE_IS_CLOSED,
bottom=mg.BC_NODE_IS_FIXED_VALUE,
)
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
wt = mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
pd = PrecipitationDistribution(
mg,
mean_storm_duration=10,
mean_interstorm_duration=100,
mean_storm_depth=1e-3,
total_t=200,
)
pd.seed_generator(seedval=1)
hm = HydrologyEventStreamPower(mg,
precip_generator=pd,
groundwater_model=gdp)
wt0 = wt.copy()
hm.run_step_record_state()
times = np.array([
0.0,
hm.storm_dts[0],
hm.storm_dts[0] + hm.interstorm_dts[0],
hm.storm_dts[0] + hm.interstorm_dts[0] + hm.storm_dts[1],
hm.storm_dts[0] + hm.interstorm_dts[0] + hm.storm_dts[1] +
hm.interstorm_dts[1],
])
intensities = np.zeros(5)
intensities[0] = hm.intensities[0]
intensities[2] = hm.intensities[1]
assert_equal(hm.time, times)
assert_equal(hm.intensity, intensities)
assert_equal(hm.qs_all.shape, (5, 9))
assert_equal(hm.Q_all.shape, (5, 9))
assert_equal(hm.wt_all.shape, (5, 9))
assert_equal(hm.qs_all[0, :], np.zeros(9))
assert_equal(hm.Q_all[0, :], np.zeros(9))
assert_equal(hm.wt_all[0, :], wt0)
def Initialize_(data, grid, grid1):
# Plant types are defined as following:
# GRASS = 0; SHRUB = 1; TREE = 2; BARE = 3;
# SHRUBSEEDLING = 4; TREESEEDLING = 5
# Initialize random plant type field
grid1['cell']['vegetation__plant_functional_type'] = compose_veg_grid(
grid1,
percent_bare=data['percent_bare_initial'],
percent_grass=data['percent_grass_initial'],
percent_shrub=data['percent_shrub_initial'],
percent_tree=data['percent_tree_initial'])
# Assign plant type for representative ecohydrologic simulations
grid['cell']['vegetation__plant_functional_type'] = np.arange(0, 6)
grid1['node']['topographic__elevation'] = (1700. *
np.ones(grid1.number_of_nodes))
grid['node']['topographic__elevation'] = (1700. *
np.ones(grid.number_of_nodes))
PD_D = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_dry'],
mean_interstorm_duration=data['mean_interstorm_dry'],
mean_storm_depth=data['mean_storm_depth_dry'])
PD_W = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_wet'],
mean_interstorm_duration=data['mean_interstorm_wet'],
mean_storm_depth=data['mean_storm_depth_wet'])
Rad = Radiation(grid)
PET_Tree = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_tree'],
delta_d=data['DeltaD'])
PET_Shrub = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_shrub'],
delta_d=data['DeltaD'])
PET_Grass = PotentialEvapotranspiration(grid,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_grass'],
delta_d=data['DeltaD'])
SM = SoilMoisture(grid, **data) # Soil Moisture object
VEG = Vegetation(grid, **data) # Vegetation object
vegca = VegCA(grid1, **data) # Cellular automaton object
# # Initializing inputs for Soil Moisture object
grid['cell']['vegetation__live_leaf_area_index'] = (
1.6 * np.ones(grid.number_of_cells))
grid['cell']['soil_moisture__initial_saturation_fraction'] = (
0.59 * np.ones(grid.number_of_cells))
# Initializing Soil Moisture
return PD_D, PD_W, Rad, PET_Tree, PET_Shrub, PET_Grass, SM, \
VEG, vegca
def test_stoch_sp_threshold_above_threshold():
"""
Test the stochastic event model with stream power threshold in which
the one core node is set up to exceed erosion threshold for the value
of Q that it attains. This can be checked by comparing the accumulated q
to the threshold value needed for erosion Q0.
"""
mg = RasterModelGrid((3, 3), xy_spacing=10.0)
mg.set_status_at_node_on_edges(
right=mg.BC_NODE_IS_CLOSED,
top=mg.BC_NODE_IS_CLOSED,
left=mg.BC_NODE_IS_CLOSED,
bottom=mg.BC_NODE_IS_FIXED_VALUE,
)
elev = mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
wt = mg.add_ones("node", "water_table__elevation")
elev[4] += 0.01
wt[:] = elev
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
pd = PrecipitationDistribution(
mg,
mean_storm_duration=10,
mean_interstorm_duration=100,
mean_storm_depth=1e-3,
total_t=100,
)
pd.seed_generator(seedval=1)
hm = HydrologyEventThresholdStreamPower(
mg,
precip_generator=pd,
groundwater_model=gdp,
sp_coefficient=1e-5,
sp_threshold=1e-12,
)
hm.run_step()
storm_dt = 1.4429106411 # storm duration
storm_q = 0.0244046740 # accumulated q before threshold effect subtracted
interstorm_q = 0.0 # interstorm q is zero in this case
assert_almost_equal(
hm.q_eff[4],
0.5 * (max(interstorm_q - hm.Q0[4], 0) + max(storm_q - hm.Q0[4], 0)) *
storm_dt / hm.T_h,
)
def test_storms():
dt = 500.0
uplift = 0.0001
mean_duration = 100.0
mean_interstorm = 400.0
mean_depth = 5.0
storm_run_time = 3000000.0
delta_t = 500.0
mg = RasterModelGrid((10, 10), xy_spacing=1000.0)
mg.add_zeros("topographic__elevation", at="node")
z = mg.zeros(at="node")
mg["node"]["topographic__elevation"] = z + np.random.rand(len(z)) / 1000.0
mg.add_zeros("water__unit_flux_in", at="node")
precip = PrecipitationDistribution(
mean_storm_duration=mean_duration,
mean_interstorm_duration=mean_interstorm,
mean_storm_depth=mean_depth,
total_t=storm_run_time,
delta_t=delta_t,
)
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(
mg,
K_sp=0.0001,
m_sp=0.5,
n_sp=1.0,
threshold_sp=1.0,
discharge_field="surface_water__discharge",
)
for (
interval_duration,
rainfall_rate,
) in precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.0:
mg.at_node["water__unit_flux_in"].fill(rainfall_rate)
fr.run_one_step()
sp.run_one_step(dt)
mg.at_node["topographic__elevation"][mg.core_nodes] += (
uplift * interval_duration)
def test_storms():
input_file_string = os.path.join(_THIS_DIR, "drive_sp_params_storms.txt")
inputs = ModelParameterDictionary(input_file_string, auto_type=True)
nrows = inputs.read_int("nrows")
ncols = inputs.read_int("ncols")
dx = inputs.read_float("dx")
dt = inputs.read_float("dt")
uplift = inputs.read_float("uplift_rate")
mean_duration = inputs.read_float("mean_storm")
mean_interstorm = inputs.read_float("mean_interstorm")
mean_depth = inputs.read_float("mean_depth")
storm_run_time = inputs.read_float("storm_run_time")
delta_t = inputs.read_float("delta_t")
mg = RasterModelGrid(nrows, ncols, xy_spacing=dx)
mg.add_zeros("topographic__elevation", at="node")
z = mg.zeros(at="node")
mg["node"]["topographic__elevation"] = z + np.random.rand(len(z)) / 1000.
mg.add_zeros("water__unit_flux_in", at="node")
precip = PrecipitationDistribution(
mean_storm_duration=mean_duration,
mean_interstorm_duration=mean_interstorm,
mean_storm_depth=mean_depth,
total_t=storm_run_time,
delta_t=delta_t,
)
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(mg, **inputs)
for (
interval_duration,
rainfall_rate,
) in precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.:
mg.at_node["water__unit_flux_in"].fill(rainfall_rate)
fr.run_one_step()
sp.run_one_step(dt)
mg.at_node["topographic__elevation"][mg.core_nodes] += (
uplift * interval_duration
)
def test_storms():
input_file_string = os.path.join(_THIS_DIR, "drive_sp_params_storms.txt")
inputs = ModelParameterDictionary(input_file_string, auto_type=True)
nrows = inputs.read_int("nrows")
ncols = inputs.read_int("ncols")
dx = inputs.read_float("dx")
dt = inputs.read_float("dt")
uplift = inputs.read_float("uplift_rate")
mean_duration = inputs.read_float("mean_storm")
mean_interstorm = inputs.read_float("mean_interstorm")
mean_depth = inputs.read_float("mean_depth")
storm_run_time = inputs.read_float("storm_run_time")
delta_t = inputs.read_float("delta_t")
mg = RasterModelGrid((nrows, ncols), xy_spacing=dx)
mg.add_zeros("topographic__elevation", at="node")
z = mg.zeros(at="node")
mg["node"]["topographic__elevation"] = z + np.random.rand(len(z)) / 1000.0
mg.add_zeros("water__unit_flux_in", at="node")
precip = PrecipitationDistribution(
mean_storm_duration=mean_duration,
mean_interstorm_duration=mean_interstorm,
mean_storm_depth=mean_depth,
total_t=storm_run_time,
delta_t=delta_t,
)
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(mg, **inputs)
for (
interval_duration,
rainfall_rate,
) in precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.0:
mg.at_node["water__unit_flux_in"].fill(rainfall_rate)
fr.run_one_step()
sp.run_one_step(dt)
mg.at_node["topographic__elevation"][mg.core_nodes] += (
uplift * interval_duration)
def test_storms():
input_file_string = os.path.join(_THIS_DIR, 'drive_sp_params_storms.txt')
inputs = ModelParameterDictionary(input_file_string, auto_type=True)
nrows = inputs.read_int('nrows')
ncols = inputs.read_int('ncols')
dx = inputs.read_float('dx')
dt = inputs.read_float('dt')
time_to_run = inputs.read_float('run_time')
uplift = inputs.read_float('uplift_rate')
mean_duration = inputs.read_float('mean_storm')
mean_interstorm = inputs.read_float('mean_interstorm')
mean_depth = inputs.read_float('mean_depth')
storm_run_time = inputs.read_float('storm_run_time')
delta_t = inputs.read_float('delta_t')
mg = RasterModelGrid(nrows, ncols, dx)
mg.add_zeros('topographic__elevation', at='node')
z = mg.zeros(at='node')
mg['node']['topographic__elevation'] = z + np.random.rand(len(z)) / 1000.
mg.add_zeros('water__unit_flux_in', at='node')
precip = PrecipitationDistribution(
mean_storm_duration=mean_duration,
mean_interstorm_duration=mean_interstorm,
mean_storm_depth=mean_depth,
total_t=storm_run_time,
delta_t=delta_t)
fr = FlowAccumulator(mg, flow_director='D8')
sp = StreamPowerEroder(mg, **inputs)
for (interval_duration, rainfall_rate) in \
precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.:
mg.at_node['water__unit_flux_in'].fill(rainfall_rate)
fr.run_one_step()
sp.run_one_step(dt)
mg.at_node['topographic__elevation'][
mg.core_nodes] += uplift * interval_duration
def test_stoch_sp_threshold_hex():
"""
Initialize HydrologyEventStreamPower on a hex grid.
Use single storm-interstorm pair and make sure it returns the
quantity calculated. This is not an analytical solution, just
the value that is returned when using gdp and adaptive
timestep solver. Confirms that hex grid returns the same value
as raster grid, adjusted for cell area. Confirms that when streampower
threshold is zero (Default), returns the same values as
HydrologyEventStreamPower.
"""
mg = HexModelGrid((3, 3), node_layout="rect", spacing=10.0)
mg.status_at_node[mg.status_at_node == 1] = 4
mg.status_at_node[0] = 1
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
pd = PrecipitationDistribution(
mg,
mean_storm_duration=10,
mean_interstorm_duration=100,
mean_storm_depth=1e-3,
total_t=100,
)
pd.seed_generator(seedval=1)
hm = HydrologyEventThresholdStreamPower(mg,
precip_generator=pd,
groundwater_model=gdp,
routing_method="Steepest")
hm.run_step()
assert_almost_equal(hm.q_eff[4], 0.00017614 * np.sqrt(3) / 2)
assert_almost_equal(
hm.q_an[4],
0.00017614 * np.sqrt(3) / 2 / np.sqrt(np.sqrt(3) / 2 * 100))
def test_storms():
input_file_string = os.path.join(_THIS_DIR, 'drive_sp_params_storms.txt')
inputs = ModelParameterDictionary(input_file_string, auto_type=True)
nrows = inputs.read_int('nrows')
ncols = inputs.read_int('ncols')
dx = inputs.read_float('dx')
dt = inputs.read_float('dt')
time_to_run = inputs.read_float('run_time')
uplift = inputs.read_float('uplift_rate')
mean_duration = inputs.read_float('mean_storm')
mean_interstorm = inputs.read_float('mean_interstorm')
mean_depth = inputs.read_float('mean_depth')
storm_run_time = inputs.read_float('storm_run_time')
delta_t = inputs.read_float('delta_t')
mg = RasterModelGrid(nrows, ncols, dx)
mg.add_zeros('topographic__elevation', at='node')
z = mg.zeros(at='node')
mg['node']['topographic__elevation'] = z + np.random.rand(len(z)) / 1000.
mg.add_zeros('water__unit_flux_in', at='node')
precip = PrecipitationDistribution(mean_storm_duration = mean_duration,
mean_interstorm_duration = mean_interstorm,
mean_storm_depth = mean_depth,
total_t = storm_run_time, delta_t = delta_t)
fr = FlowAccumulator(mg, flow_director='D8')
sp = StreamPowerEroder(mg, **inputs)
for (interval_duration, rainfall_rate) in \
precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.:
mg.at_node['water__unit_flux_in'].fill(rainfall_rate)
fr.run_one_step()
sp.run_one_step(dt)
mg.at_node['topographic__elevation'][
mg.core_nodes] += uplift * interval_duration
def test_stoch_sp_threshold_raster_null():
"""
Initialize HydrologyEventThresholdStreamPower on a raster grid.
Use single storm-interstorm pair and make sure it returns the
quantity calculated. This is not an analytical solution, just
the value that is returned when using gdp and adaptive
timestep solver. Confirms that when streampower threshold is
zero (Default), returns the same values as HydrologyEventStreamPower.
"""
mg = RasterModelGrid((3, 3), xy_spacing=10.0)
mg.set_status_at_node_on_edges(
right=mg.BC_NODE_IS_CLOSED,
top=mg.BC_NODE_IS_CLOSED,
left=mg.BC_NODE_IS_CLOSED,
bottom=mg.BC_NODE_IS_FIXED_VALUE,
)
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
pd = PrecipitationDistribution(
mg,
mean_storm_duration=10,
mean_interstorm_duration=100,
mean_storm_depth=1e-3,
total_t=100,
)
pd.seed_generator(seedval=1)
hm = HydrologyEventThresholdStreamPower(mg,
precip_generator=pd,
groundwater_model=gdp)
hm.run_step()
assert_almost_equal(hm.q_eff[4], 0.00017614)
assert_almost_equal(hm.q_an[4], 0.00017614 / 10.0)
class StochasticErosionModel(ErosionModel):
"""Base class for stochastic-precipitation terrainbento models.
A **StochasticErosionModel** inherits from
:py:class:`ErosionModel` and provides functionality needed by all
stochastic-precipitation models.
This is a base class that handles processes related to the generation of
preciptiation events.
Two primary options are avaliable for the stochastic erosion models. When
``opt_stochastic_duration=True`` the model will use the
`PrecipitationDistribution <https://landlab.readthedocs.io/en/latest/landlab.components.uniform_precip.html>`_
Landlab component to generate a random storm duration, interstorm duration,
and precipitation intensity or storm depth from an exponential
distribution. When this option is selected, the following parameters are
used:
- mean_storm_duration
- mean_interstorm_duration
- mean_storm_depth
When ``opt_stochastic_duration==False`` the model will have uniform
timesteps but generate rainfall from a stretched exponential distribution.
The duration indicated by the parameter ``step`` will first be split into a
series of sub-timesteps based on the parameter
``number_of_sub_time_steps``, and then each of these sub-timesteps will
experience a duration of rain and no-rain based on the value of
``rainfall_intermittency_factor``. The duration of rain and no-rain will
not change, but the intensity of rain will vary based on a stretched
exponential distribution described by the shape factor
``rainfall__shape_factor`` and with a scale factor calculated so that the
mean of the distribution has the value given by ``rainfall__mean_rate``.
The following parameters are used:
- rainfall__shape_factor
- number_of_sub_time_steps
- rainfall_intermittency_factor
- rainfall__mean_rate
The hydrology uses calculation of drainage area using the user-specified
routing method. It then performs one of two options, depending on the
user's choice of ``opt_stochastic_duration`` (True or False).
If the user requests stochastic duration, the model iterates through a sequence
of storm and interstorm periods. Storm depth is drawn at random from a gamma
distribution, and storm duration from an exponential distribution; storm
intensity is then depth divided by duration. This sequencing is implemented by
overriding the run_for method.
If the user does not request stochastic duration (indicated by setting
``opt_stochastic_duration`` to ``False``), then the default
(**erosion_model** base class) **run_for** method is used. Whenever
**run_one_step** is called, storm intensity is generated at random from an
exponential distribution with mean given by the parameter
``rainfall__mean_rate``. The stream power component is run for only a
fraction of the time step duration step, as specified by the parameter
``rainfall_intermittency_factor``. For example, if ``step`` is 10 years and
the intermittency factor is 0.25, then the stream power component is run
for only 2.5 years.
In either case, given a storm precipitation intensity :math:`P`, the runoff
production rate :math:`R` [L/T] is calculated using:
.. math::
R = P - I (1 - \exp ( -P / I ))
where :math:`I` is the soil infiltration capacity. At the sub-grid scale, soil
infiltration capacity is assumed to have an exponential distribution of which
:math:`I` is the mean. Hence, there are always some spots within any given grid cell
that will generate runoff. This approach yields a smooth transition from
near-zero runoff (when :math:`I>>P`) to :math:`R \\approx P`
(when :math:`P>>I`), without a "hard threshold."
The following at-node fields must be specified in the grid:
- ``topographic__elevation``
"""
_required_fields = ["topographic__elevation"]
def __init__(self,
clock,
grid,
random_seed=0,
record_rain=False,
opt_stochastic_duration=False,
mean_storm_duration=1,
mean_interstorm_duration=1,
mean_storm_depth=1,
rainfall__shape_factor=1,
number_of_sub_time_steps=1,
rainfall_intermittency_factor=1,
rainfall__mean_rate=1,
storm_sequence_filename="storm_sequence.txt",
frequency_filename="exceedance_summary.txt",
**kwargs):
"""
Parameters
----------
clock : terrainbento Clock instance
grid : landlab model grid instance
The grid must have all required fields.
random_seed, int, optional
Random seed. Default is 0.
opt_stochastic_duration : bool, optional
Flag indicating if timestep is stochastic or constant. Default is
False.
mean_storm_duration : float, optional
Average duration of a precipitation event. Default is 1.
mean_interstorm_duration : float, optional
Average duration between precipitation events. Default is 1.
mean_storm_depth : float, optional
Average depth of precipitation events. Default is 1.
number_of_sub_time_steps : int, optional
Number of sub-timesteps. Default is 1.
rainfall_intermittency_factor : float, optional
Value between zero and one that indicates the proportion of time
rain occurs. A value of 0 means it never rains and a value of 1
means that rain never ceases. Default is 1.
rainfall__mean_rate : float, optional
Mean of the precipitation distribution. Default is 1.
rainfall__shape_factor : float, optional
Shape factor of the precipitation distribution. Default is 1.
record_rain : boolean
Flag to indicate if a sequence of storms should be saved. Default
is False.
storm_sequence_filename : str
Storm sequence filename. Default is "storm_sequence.txt"
frequency_filename : str
Filename for precipitation exceedance frequency summary. Default
value is "exceedance_summary.txt"
**kwargs :
Keyword arguments to pass to
:py:class:`ErosionModel`
Returns
-------
StochasticErosionModel : object
Examples
--------
This model is a base class and is not designed to be run on its own. We
recommend that you look at the terrainbento tutorials for examples of
usage.
"""
# Call StochasticErosionModel init
super(StochasticErosionModel, self).__init__(clock, grid, **kwargs)
# ensure Precipitator and RunoffGenerator are vanilla
self._ensure_precip_runoff_are_vanilla()
self.opt_stochastic_duration = opt_stochastic_duration
# verify that opt_stochastic_duration and PrecipChanger are consistent
if self.opt_stochastic_duration and ("PrecipChanger"
in self.boundary_handlers):
msg = ("terrainbento StochasticErosionModel: setting "
"opt_stochastic_duration=True and using the PrecipChanger "
"boundary condition handler are not compatible.")
raise ValueError(msg)
self.seed = int(random_seed)
self.random_seed = random_seed
self.frequency_filename = frequency_filename
self.storm_sequence_filename = storm_sequence_filename
self.mean_storm_duration = mean_storm_duration
self.mean_interstorm_duration = mean_interstorm_duration
self.mean_storm_depth = mean_storm_depth
self.shape_factor = rainfall__shape_factor
self.number_of_sub_time_steps = number_of_sub_time_steps
self.rainfall_intermittency_factor = rainfall_intermittency_factor
self.rainfall__mean_rate = rainfall__mean_rate
# initialize record for storms. Depending on how this model is run
# (stochastic time, number_time_steps>1, more manually) the step may
# change. Thus, rather than writing routines to reconstruct the time
# series of precipitation from the step could change based on users use,
# we"ll record this with the model run instead of re-running.
# make this the non-default option.
# Second, test that
if record_rain:
self.record_rain = True
self.rain_record = {
"event_start_time": [],
"event_duration": [],
"rainfall_rate": [],
"runoff_rate": [],
}
else:
self.record_rain = False
self.rain_record = None
def calc_runoff_and_discharge(self):
"""Calculate runoff rate and discharge; return runoff."""
if self.rain_rate > 0.0 and self.infilt > 0.0:
runoff = self.rain_rate - (
self.infilt * (1.0 - np.exp(-self.rain_rate / self.infilt)))
if runoff <= 0:
runoff = 0 # pragma: no cover
else:
runoff = self.rain_rate
self.grid.at_node["surface_water__discharge"][:] = (
runoff * self.grid.at_node["drainage_area"])
return runoff
def run_for_stochastic(self, step, runtime):
"""**Run_for** with stochastic duration.
Run model without interruption for a specified time period, using
random storm/interstorm sequence.
**run_for_stochastic** runs the model for the duration ``runtime`` with
model time steps given by the PrecipitationDistribution component.
Model run steps will not exceed the duration given by ``step``.
Parameters
----------
step : float
Model run timestep,
runtime : float
Total duration for which to run model.
"""
self.rain_generator.delta_t = step
self.rain_generator.run_time = runtime
for (
tr,
p,
) in self.rain_generator.yield_storm_interstorm_duration_intensity():
self.rain_rate = p
self.run_one_step(tr)
def instantiate_rain_generator(self):
"""Instantiate component used to generate storm sequence."""
# Handle option for duration.
if self.opt_stochastic_duration:
self.rain_generator = PrecipitationDistribution(
mean_storm_duration=self.mean_storm_duration,
mean_interstorm_duration=self.mean_interstorm_duration,
mean_storm_depth=self.mean_storm_depth,
total_t=self.clock.stop,
delta_t=self.clock.step,
random_seed=self.seed,
)
self.run_for = self.run_for_stochastic # override base method
else:
from scipy.special import gamma
self.rain_generator = PrecipitationDistribution(
mean_storm_duration=1.0,
mean_interstorm_duration=1.0,
mean_storm_depth=1.0,
random_seed=self.seed,
)
self.scale_factor = self.rainfall__mean_rate / gamma(
1.0 + (1.0 / self.shape_factor))
if (isinstance(self.number_of_sub_time_steps,
(int, np.integer)) is False):
raise ValueError(
("number_of_sub_time_steps must be of type integer."))
self.n_sub_steps = self.number_of_sub_time_steps
def reset_random_seed(self):
"""Reset the random number generation sequence."""
self.rain_generator.seed_generator(seedval=self.seed)
def _pre_water_erosion_steps(self):
"""Convenience function for pre-water erosion steps.
If a model needs to do anything before each erosion step is run,
e.g. recalculate a threshold value, that model should overwrite
this function.
"""
pass
def handle_water_erosion(self, step, flooded):
"""Handle water erosion for stochastic models.
If we are running stochastic duration, then self.rain_rate will
have been calculated already. It might be zero, in which case we
are between storms, so we don't do water erosion.
If we're NOT doing stochastic duration, then we'll run water
erosion for one or more sub-time steps, each with its own
randomly drawn precipitation intensity.
This routine assumes that a model-specific method:
**calc_runoff_and_discharge()**
will have been defined. Additionally a model eroder must also have been
defined.
For example, BasicStVs calculated runoff and discharge in a different
way than the other models.
If the model has a function **update_threshold_field**, this
function will test for it and run it. This is presently done in
BasicDdSt.
Parameters
----------
step : float
Model run timestep.
flooded_nodes : ndarray of int (optional)
IDs of nodes that are flooded and should have no erosion.
"""
# (if we're varying precipitation parameters through time, update them)
if "PrecipChanger" in self.boundary_handlers:
self.rainfall_intermittency_factor, self.rainfall__mean_rate = self.boundary_handlers[
"PrecipChanger"].get_current_precip_params()
if self.opt_stochastic_duration and self.rain_rate > 0.0:
self._pre_water_erosion_steps()
runoff = self.calc_runoff_and_discharge()
self.eroder.run_one_step(step, flooded_nodes=flooded)
if self.record_rain:
# save record into the rain record
self.record_rain_event(self.model_time, step, self.rain_rate,
runoff)
elif self.opt_stochastic_duration and self.rain_rate <= 0.0:
# calculate and record the time with no rain:
if self.record_rain:
self.record_rain_event(self.model_time, step, 0, 0)
elif not self.opt_stochastic_duration:
dt_water = (step * self.rainfall_intermittency_factor) / float(
self.n_sub_steps)
for i in range(self.n_sub_steps):
self.rain_rate = self.rain_generator.generate_from_stretched_exponential(
self.scale_factor, self.shape_factor)
self._pre_water_erosion_steps()
runoff = self.calc_runoff_and_discharge()
self.eroder.run_one_step(dt_water, flooded_nodes=flooded)
# save record into the rain record
if self.record_rain:
event_start_time = self.model_time + (i * dt_water)
self.record_rain_event(event_start_time, dt_water,
self.rain_rate, runoff)
# once all the rain time_steps are complete,
# calculate and record the time with no rain:
if self.record_rain:
# calculate dry time
dt_dry = step * (1 - self.rainfall_intermittency_factor)
# if dry time is greater than zero, record.
if dt_dry > 0:
event_start_time = self.model_time + (self.n_sub_steps *
dt_water)
self.record_rain_event(event_start_time, dt_dry, 0.0, 0.0)
def finalize(self):
"""Finalize stochastic erosion models.
The finalization step of stochastic erosion models in
terrainbento results in writing out the storm sequence file and
the precipitation exceedence statistics summary if
``record_rain`` was set to ``True``.
"""
# if rain was recorded, write it out.
if self.record_rain:
self.write_storm_sequence_to_file(
filename=self.storm_sequence_filename)
if self.opt_stochastic_duration is False:
# if opt_stochastic_duration is False, calculate exceedance
# frequencies and write out.
try:
self.write_exceedance_frequency_file(
filename=self.frequency_filename)
except IndexError:
msg = (
"terrainbento stochastic model: the rain record was "
"too short to calculate exceedance frequency statistics."
)
os.remove(self.frequency_filename)
raise RuntimeError(msg)
def record_rain_event(self, event_start_time, event_duration,
rainfall_rate, runoff_rate):
"""Record rain events.
Create a record of event start time, event duration, rainfall rate, and
runoff rate.
Parameters
----------
event_start_time : float
event_duration : float
rainfall_rate : float
runoff_rate : float
"""
self.rain_record["event_start_time"].append(event_start_time)
self.rain_record["event_duration"].append(event_duration)
self.rain_record["rainfall_rate"].append(rainfall_rate)
self.rain_record["runoff_rate"].append(runoff_rate)
def write_storm_sequence_to_file(self, filename="storm_sequence.txt"):
"""Write event duration and intensity to a formatted text file.
Parameters
----------
filename : str
Default value is "storm_sequence.txt"
"""
# Open a file for writing
if self.record_rain is False:
raise ValueError("Rain was not recorded when the model run. To "
"record rain, set the parameter 'record_rain'"
"to True.")
with open(filename, "w") as stormfile:
stormfile.write("event_start_time" + "," + "event_duration" + "," +
"rainfall_rate" + "," + "runoff_rate" + "\n")
n_events = len(self.rain_record["event_start_time"])
for i in range(n_events):
stormfile.write(
str(
np.around(self.rain_record["event_start_time"][i],
decimals=5)) + "," +
str(
np.around(self.rain_record["event_duration"][i],
decimals=5)) + "," +
str(
np.around(self.rain_record["rainfall_rate"][i],
decimals=5)) + "," +
str(
np.around(self.rain_record["runoff_rate"][i],
decimals=5)) + "\n")
def write_exceedance_frequency_file(self,
filename="exceedance_summary.txt"):
"""Write summary of rainfall exceedance statistics to file.
Parameters
----------
filename : str
Default value is "exceedance_summary.txt"
"""
if self.record_rain is False:
raise ValueError("Rain was not recorded when the model run. To "
"record rain, set the parameter 'record_rain'"
"to True.")
# calculate the number of wet days per year.
number_of_days_per_year = 365
nwet = int(
np.ceil(self.rainfall_intermittency_factor *
number_of_days_per_year))
if nwet == 0:
raise ValueError(
"No rain fell, which makes calculating exceedance "
"frequencies problematic. We recommend that you "
"check the valude of rainfall_intermittency_factor.")
with open(filename, "w") as exceedance_file:
# ndry = int(number_of_days_per_year - nwet)
# Write some basic information about the distribution to the file.
exceedance_file.write("Section 1: Distribution Description\n")
exceedance_file.write("Scale Factor: " + str(self.scale_factor) +
"\n")
exceedance_file.write("Shape Factor: " + str(self.shape_factor) +
"\n")
exceedance_file.write(
("Intermittency Factor: " +
str(self.rainfall_intermittency_factor) + "\n"))
exceedance_file.write(
("Number of wet days per year: " + str(nwet) + "\n\n"))
message_text = (
"The scale factor that describes this distribution is " +
"calculated based on a provided value for the mean wet day rainfall."
)
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
exceedance_file.write(("This provided value was:\n" +
str(self.rainfall__mean_rate) + "\n"))
# calculate the predictions for 10, 25, and 100 year event based on
# the analytical form of the exceedance function.
event_intervals = np.array([10.0, 25, 100.0])
# calculate the probability of each event based on the number of years
# and the number of wet days per year.
daily_distribution_exceedance_probabilities = 1.0 / (
nwet * event_intervals)
# exceedance probability is given as
# Probability of daily rainfall of p exceeding a value of po is given as:
#
# P(p>po) = e^(-(po/P)^c)
# P = scale
# c = shape
#
# this can be re-arranged to
#
# po = P * (- ln (P(p>po))) ^ (1 / c)
expected_rainfall = self.scale_factor * (
-1.0 * np.log(daily_distribution_exceedance_probabilities))**(
1.0 / self.shape_factor)
exceedance_file.write("\n\nSection 2: Theoretical Predictions\n")
message_text = (
"Based on the analytical form of the wet day rainfall " +
"distribution, we can calculate theoretical predictions " +
"of the daily rainfall amounts associated with N-year events.")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
for i in range(len(daily_distribution_exceedance_probabilities)):
exceedance_file.write(
("Expected value for the wet day total of the " +
str(event_intervals[i]) + " year event is: " +
str(np.round(expected_rainfall[i], decimals=3)) + "\n"))
# get rainfall record and filter out time without any rain
all_precipitation = np.array(self.rain_record["rainfall_rate"])
rainy_day_inds = np.where(all_precipitation > 0)
wet_day_totals = all_precipitation[rainy_day_inds]
num_days = len(wet_day_totals)
# construct the distribution of yearly maxima.
# here an effective year is represented by the number of draws implied
# by the intermittency factor
# first calculate the number of effective years.
num_effective_years = int(np.floor(wet_day_totals.size / nwet))
# write out the calculated event only if the duration
exceedance_file.write("\n\n")
message_text = (
"Section 3: Predicted 95% confidence bounds on the " +
"exceedance values based on number of samples drawn.")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
message_text = (
"The ability to empirically estimate the rainfall " +
"associated with an N-year event depends on the " +
"probability of that event occurring and the number of " +
"draws from the probability distribution. The ability " +
"to estimate increases with an increasing number of samples " +
"and decreases with decreasing probability of event " +
"occurrence.")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
message_text = (
"Exceedance values calculated from " +
str(len(wet_day_totals)) +
" draws from the daily-rainfall probability distribution. " +
"This corresponds to " + str(num_effective_years) +
" effective years.")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
# For a general probability distribution, f, with a continuous not zero
# quantile function at F-1(p), the order statistic associated with the
# p percentile given n draws from the distribution is given as:
# X[np] ~ AN ( F-1(p), (p * (p - 1 ))/ (n * [f (F-1 (p)) ]**2))
# where AN is the asymptotic normal. The value for the variance is more
# intuitive once you consider that [f (F-1 (p)) ] is the probability
# that an event of percentile p will occur. Thus the variance increases
# non-linearly with decreasing event probability and decreases linearly
# with increaseing observations.
# we"ve already calculated F-1(p) for our events, and it is represented
# by the variable expected_rainfall
daily_distribution_event_percentile = (
1.0 - daily_distribution_exceedance_probabilities)
event_probability = (
(self.shape_factor / self.scale_factor) *
((expected_rainfall / self.scale_factor)
**(self.shape_factor - 1.0)) *
(np.exp(-1.0 * (expected_rainfall / self.scale_factor)**
self.shape_factor)))
event_variance = (daily_distribution_event_percentile *
(1.0 - daily_distribution_event_percentile)) / (
num_days * (event_probability**2))
event_std = event_variance**0.5
t_statistic = stats.t.ppf(0.975,
num_effective_years,
loc=0,
scale=1)
exceedance_file.write("\n")
message_text = ("For the given number of samples, the 95% " +
"confidence bounds for the following event " +
"return intervals are as follows: ")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
for i in range(len(event_intervals)):
min_expected_val = (expected_rainfall[i] -
t_statistic * event_std[i])
max_expected_val = (expected_rainfall[i] +
t_statistic * event_std[i])
exceedance_file.write(
("Expected range for the wet day total of the " +
str(event_intervals[i]) + " year event is: (" +
str(np.round(min_expected_val, decimals=3)) + ", " +
str(np.round(max_expected_val, decimals=3)) + ")\n"))
# next, calculate the emperical exceedance values, if a sufficient record
# exists.
# inititialize a container for the maximum yearly precipitation.
maximum_yearly_precipitation = np.nan * np.zeros(
(num_effective_years))
for yi in range(num_effective_years):
# identify the starting and ending index coorisponding to the
# year
starting_index = yi * nwet
ending_index = starting_index + nwet
# select the years portion of the wet_day_totals
selected_wet_day_totals = wet_day_totals[
starting_index:ending_index]
# record the yearly maximum precipitation
maximum_yearly_precipitation[yi] = selected_wet_day_totals.max(
)
# calculate the distribution percentiles associated with each interval
event_percentiles = (1.0 - (1.0 / event_intervals)) * 100.0
# calculated the event magnitudes associated with the percentiles.
event_magnitudes = np.percentile(maximum_yearly_precipitation,
event_percentiles)
# write out the calculated event only if the duration
exceedance_file.write("\n\nSection 4: Empirical Values\n")
message_text = (
"These empirical values should be interpreted in the " +
"context of the expected ranges printed in Section 3. " +
"If the expected range is large, consider using a longer " +
"record of rainfall. The empirical values should fall " +
"within the expected range at a 95% confidence level.")
exceedance_file.write("\n".join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write("\n")
for i in range(len(event_percentiles)):
exceedance_file.write(
("Estimated value for the wet day total of the " +
str(np.round(event_intervals[i], decimals=3)) +
" year event is: " +
str(np.round(event_magnitudes[i], decimals=3)) + "\n"))
else:
raise TypeError("grid should be Raster or Hex")
gdp = GroundwaterDupuitPercolator(
mg,
porosity=n,
hydraulic_conductivity=Ks,
regularization_f=0.01,
recharge_rate=0.0,
courant_coefficient=0.05,
vn_coefficient=0.05,
callback_fun=write_SQ,
)
pdr = PrecipitationDistribution(mg,
mean_storm_duration=tr,
mean_interstorm_duration=tb,
mean_storm_depth=ds,
total_t=T_h)
pdr.seed_generator(seedval=2)
hm = HydrologyEventStreamPower(
mg,
routing_method=method,
precip_generator=pdr,
groundwater_model=gdp,
)
#run model
hm.run_step_record_state()
f.close()
rmg5.set_closed_boundaries_at_grid_edges(True, True, True, True)
rmg5.status_at_node[1] = 1
rmg10.set_closed_boundaries_at_grid_edges(True, True, True, True)
rmg10.status_at_node[1] = 1
link_to_sample = 200
sample_das = [201, 24048, 14834, 14268, 12097, 8035, 5022] # sample drainage areas
## GENERATE AND SET PRECIPITATION TIME SERIES
total_t = 1000.* 365.25 * 24 # 1,000 years of rainfall generated.
thresh = 0.5 # Intensity threshold (lower limit, mm/hr)
PD = PrecipitationDistribution(mean_storm_duration = 10.75,
mean_interstorm_duration=433.58,
mean_storm_depth = 9.62,
total_t=total_t)
storm_arr = np.array(PD.get_storm_time_series())
intensity_threshold, = np.where(storm_arr[:, 2] > thresh)
interstorm_durs = (storm_arr[intensity_threshold][:,0][np.arange(1,
len(storm_arr[intensity_threshold]) - 1)] -
storm_arr[intensity_threshold][:, 1][np.arange(0,
len(storm_arr[intensity_threshold]) - 2)])
durations = (storm_arr[intensity_threshold][:,1] -
storm_arr[intensity_threshold][:,0])
durations_s = [x * 3600. for x in durations]
def initialize(data, grid, grid1, grid2, elevation):
"""Initialize random plant type field.
Plant types are defined as the following:
* GRASS = 0
* SHRUB = 1
* TREE = 2
* BARE = 3
* SHRUBSEEDLING = 4
* TREESEEDLING = 5
"""
grid['cell']['vegetation__plant_functional_type'] = compose_veg_grid(
grid,
percent_bare=data['percent_bare_initial'],
percent_grass=data['percent_grass_initial'],
percent_shrub=data['percent_shrub_initial'],
percent_tree=data['percent_tree_initial'])
# Assign plant type for representative ecohydrologic simulations
grid1.at_cell['vegetation__plant_functional_type'] = np.arange(6)
grid1.at_node['topographic__elevation'] = np.full(grid1.number_of_nodes,
1700.)
grid.at_node['topographic__elevation'] = elevation
grid2.at_node['topographic__elevation'] = elevation
if data['runon_switch']:
(ordered_cells, grid2) = get_ordered_cells_for_soil_moisture(
grid2, outlet_id=4877) # hugo10mws: 1331 # 36704
grid.at_node['flow__receiver_node'] = (
grid2.at_node['flow__receiver_node'])
else:
ordered_cells = None
precip_dry = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_dry'],
mean_interstorm_duration=data['mean_interstorm_dry'],
mean_storm_depth=data['mean_storm_depth_dry'],
random_seed=None)
precip_wet = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_wet'],
mean_interstorm_duration=data['mean_interstorm_wet'],
mean_storm_depth=data['mean_storm_depth_wet'],
random_seed=None)
radiation = Radiation(grid)
rad_pet = Radiation(grid1)
pet_tree = PotentialEvapotranspiration(grid1,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_tree'],
delta_d=data['DeltaD'])
pet_shrub = PotentialEvapotranspiration(grid1,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_shrub'],
delta_d=data['DeltaD'])
pet_grass = PotentialEvapotranspiration(grid1,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_grass'],
delta_d=data['DeltaD'])
soil_moisture = SoilMoisture(grid, ordered_cells=ordered_cells,
**data) # Soil Moisture object
vegetation = Vegetation(grid, **data) # Vegetation object
vegca = VegCA(grid, **data) # Cellular automaton object
# # Initializing inputs for Soil Moisture object
grid['cell']['vegetation__live_leaf_area_index'] = (
1.6 * np.ones(grid.number_of_cells))
grid['cell']['soil_moisture__initial_saturation_fraction'] = (
0.59 * np.ones(grid.number_of_cells))
# Initializing Soil Moisture
return (precip_dry, precip_wet, radiation, rad_pet, pet_tree, pet_shrub,
pet_grass, soil_moisture, vegetation, vegca, ordered_cells)
def initialize_components(data, grid_veg=None, grid=None, pet_method='Cosine'):
# Plant types are defined as following:
# GRASS = 0; SHRUB = 1; TREE = 2; BARE = 3;
# SHRUBSEEDLING = 4; TREESEEDLING = 5
# Initialize random plant type field
grid.at_cell['vegetation__plant_functional_type'] = compose_veg_grid(
grid,
percent_bare=data['percent_bare_initial'],
percent_grass=data['percent_grass_initial'],
percent_shrub=data['percent_shrub_initial'],
percent_tree=data['percent_tree_initial'])
# Assign plant type for representative ecohydrologic simulations
grid_veg.at_cell['vegetation__plant_functional_type'] = np.arange(0, 6)
grid.at_node['topographic__elevation'] = np.full(grid.number_of_nodes,
1700.)
grid_veg.at_node['topographic__elevation'] = np.full(
grid_veg.number_of_nodes, 1700.)
precip_dry = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_dry'],
mean_interstorm_duration=data['mean_interstorm_dry'],
mean_storm_depth=data['mean_storm_depth_dry'],
random_seed=None)
precip_wet = PrecipitationDistribution(
mean_storm_duration=data['mean_storm_wet'],
mean_interstorm_duration=data['mean_interstorm_wet'],
mean_storm_depth=data['mean_storm_depth_wet'],
random_seed=None)
radiation = Radiation(grid_veg)
if pet_method == 'Cosine':
pet_tree = PotentialEvapotranspiration(
grid_veg,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_tree'],
delta_d=data['DeltaD'])
pet_shrub = PotentialEvapotranspiration(
grid_veg,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_shrub'],
delta_d=data['DeltaD'])
pet_grass = PotentialEvapotranspiration(
grid_veg,
method=data['PET_method'],
MeanTmaxF=data['MeanTmaxF_grass'],
delta_d=data['DeltaD'])
elif pet_method == 'PriestleyTaylor':
pet_met = PotentialEvapotranspiration(grid_veg,
method='PriestleyTaylor')
soil_moisture = SoilMoisture(grid_veg, **data) # Soil Moisture object
vegetation = Vegetation(grid_veg, **data) # Vegetation object
vegca = VegCA(grid, **data) # Cellular automaton object
# # Initializing inputs for Soil Moisture object
grid_veg.at_cell['vegetation__live_leaf_area_index'] = (
1.6 * np.ones(grid_veg.number_of_cells))
grid_veg.at_cell['soil_moisture__initial_saturation_fraction'] = (
0.59 * np.ones(grid_veg.number_of_cells))
# Initializing Soil Moisture
if pet_method == 'Cosine':
return (precip_dry, precip_wet, radiation, pet_tree, pet_shrub,
pet_grass, soil_moisture, vegetation, vegca)
elif pet_method == 'PriestleyTaylor':
return (precip_dry, precip_wet, radiation, pet_met, soil_moisture,
vegetation, vegca)
if hard_layer_on_or_off == 1:
k[:] = k_unerodible #for hard rock
mg['node']['K_values'] = k
elif hard_layer_on_or_off == 0:
k[:] = k_erodible #soft rock
else:
print 'WARNING: MUST SELECT 0 OR 1 IN LAYERED PARAM'
#set up its boundary conditions (left, top, right, bottom is inactive)
mg.set_closed_boundaries_at_grid_edges(False, True, False, True)
# Display initialization message
print('Running ...')
#instantiate the components:
pr = PrecipitationDistribution(input_file)
fr = Flow(mg)
sp = Fsc(mg, input_file)
hd = Diff(mg, input_file)
####################RUN
track_uplift = 0 #track cumulative uplift to know top of hard layer
last_trunc = runtime
for (interval_duration, rainfall_rate) in pr.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.:
# note diffusion also only happens when it's raining...
_ = fr.route_flow()
_ = sp.erode(mg, interval_duration, K_if_used='K_values')
_ = hd.diffuse(interval_duration)
track_uplift += uplift_rate * interval_duration #top of beginning surface
mg.at_node['topographic__elevation'][mg.core_nodes] += uplift_rate * interval_duration
rmg5.set_closed_boundaries_at_grid_edges(True, True, True, True)
rmg5.status_at_node[1] = 1
rmg10.set_closed_boundaries_at_grid_edges(True, True, True, True)
rmg10.status_at_node[1] = 1
link_to_sample = 200
sample_das = [201, 24048, 14834, 14268, 12097, 8035,
5022] # sample drainage areas
## GENERATE AND SET PRECIPITATION TIME SERIES
total_t = 1000. * 365.25 * 24 # 1,000 years of rainfall generated.
thresh = 0.5 # Intensity threshold (lower limit, mm/hr)
PD = PrecipitationDistribution(mean_storm_duration=11.75,
mean_interstorm_duration=146.25,
mean_storm_depth=4.775,
total_t=total_t)
storm_arr = np.array(PD.get_storm_time_series())
intensity_threshold, = np.where(storm_arr[:, 2] > thresh)
interstorm_durs = (storm_arr[intensity_threshold][:, 0][np.arange(
1,
len(storm_arr[intensity_threshold]) - 1)] -
storm_arr[intensity_threshold][:, 1][np.arange(
0,
len(storm_arr[intensity_threshold]) - 2)])
durations = (storm_arr[intensity_threshold][:, 1] -
storm_arr[intensity_threshold][:, 0])
durations_s = [x * 3600. for x in durations]
mg["node"]["topographic__elevation"] = z + numpy.random.rand(len(z)) / 1000.
mg.add_zeros("node", "water__unit_flux_in")
# make some K values in a field to test
# mg.at_node['K_values'] = 0.1+numpy.random.rand(nrows*ncols)/10.
mg.at_node["K_values"] = numpy.empty(nrows * ncols, dtype=float)
# mg.at_node['K_values'].fill(0.1+numpy.random.rand()/10.)
mg.at_node["K_values"].fill(0.001)
print("Running ...")
# instantiate the components:
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(mg, input_file_string)
# fsp = FastscapeEroder(mg, input_file_string)
precip = PrecipitationDistribution(input_file=input_file_string)
# load the Fastscape module too, to allow direct comparison
fsp = FastscapeEroder(mg, input_file_string)
try:
# raise NameError
mg = copy.deepcopy(mg_mature)
except NameError:
print("building a new grid...")
out_interval = 50000.
last_trunc = time_to_run # we use this to trigger taking an output plot
# run to a steady state:
# We're going to cheat by running Fastscape SP for the first part of the solution
for (
interval_duration,
mg['node'][ 'topographic__elevation'] = z + numpy.random.rand(len(z))/1000.
mg.add_zeros('node', 'water__unit_flux_in')
#make some K values in a field to test
#mg.at_node['K_values'] = 0.1+numpy.random.rand(nrows*ncols)/10.
mg.at_node['K_values'] = numpy.empty(nrows*ncols, dtype=float)
#mg.at_node['K_values'].fill(0.1+numpy.random.rand()/10.)
mg.at_node['K_values'].fill(0.001)
print( 'Running ...' )
#instantiate the components:
fr = FlowAccumulator(mg, flow_director='D8')
sp = StreamPowerEroder(mg, input_file_string)
#fsp = FastscapeEroder(mg, input_file_string)
precip = PrecipitationDistribution(input_file=input_file_string)
#load the Fastscape module too, to allow direct comparison
fsp = FastscapeEroder(mg, input_file_string)
try:
#raise NameError
mg = copy.deepcopy(mg_mature)
except NameError:
print('building a new grid...')
out_interval = 50000.
last_trunc = time_to_run #we use this to trigger taking an output plot
#run to a steady state:
#We're going to cheat by running Fastscape SP for the first part of the solution
for (interval_duration, rainfall_rate) in precip.yield_storm_interstorm_duration_intensity():
if rainfall_rate != 0.:
class _StochasticErosionModel(_ErosionModel):
"""
An StochasticErosionModel is a basic model for erosion and landscape
evolution in a watershed, as represented by an input DEM.
This is a base class that handles only processes used by all Stochastic
Hydrology based modeles.
"""
def __init__(self,
input_file=None,
params=None,
BaselevelHandlerClass=None):
"""Initialize the _BaseSt base class."""
# Call _StochasticErosionModel init
super(_StochasticErosionModel,
self).__init__(input_file=input_file,
params=params,
BaselevelHandlerClass=BaselevelHandlerClass)
self.opt_stochastic_duration = (self.params['opt_stochastic_duration'])
# initialize record for storms. Depending on how this model is run
# (stochastic time, number_time_steps>1, more manually) the dt may
# change. Thus, rather than writing routines to reconstruct the time
# series of precipitation from the dt could change based on users use,
# we'll record this with the model run instead of re-running.
# make this the non-default option.
# First test for consistency between filenames and boolean parameters
if (((self.params.get('storm_sequence_filename') is not None) or
(self.params.get('frequency_filename') is not None))
and (self.params.get('record_rain') != True)):
print('A storm sequence or frequency filename was specified but '
'record_rain was not set or set to False. Overriding '
'record_rain and recording rain so that the file can be '
'written')
self.params['record_rain'] = True
# Second, test that
if self.params.get('record_rain'):
self.record_rain = True
self.rain_record = {
'event_start_time': [],
'event_duration': [],
'rainfall_rate': [],
'runoff_rate': []
}
else:
self.record_rain = False
self.rain_record = None
# check that if (self.opt_stochastic_duration==True) that
# frequency_filename does not exist. For stochastic time, computing
# exceedance frequencies is not super sensible. So make a warning that
# it won't be done.
if ((self.opt_stochastic_duration == True)
and (self.params.get('frequency_filename'))):
print('opt_stochastic_duration is set to True and a '
'frequency_filename was specified. Frequency calculations '
'are not done with stochastic time so the filename is being '
'ignored.')
def run_for_stochastic(self, dt, runtime):
"""
Run model without interruption for a specified time period, using
random storm/interstorm sequence.
"""
self.rain_generator.delta_t = dt
self.rain_generator.run_time = runtime
for (
tr, p
) in self.rain_generator.yield_storm_interstorm_duration_intensity():
self.rain_rate = p
self.run_one_step(tr)
def instantiate_rain_generator(self):
"""Instantiate RainGenerator."""
# Handle option for duration.
self.opt_stochastic_duration = (self.params['opt_stochastic_duration'])
if self.opt_stochastic_duration:
self.rain_generator = \
PrecipitationDistribution(mean_storm_duration=self.params['mean_storm_duration'],
mean_interstorm_duration=self.params['mean_interstorm_duration'],
mean_storm_depth=self.params['mean_storm_depth'],
total_t=self.params['run_duration'],
delta_t=self.params['dt'],
random_seed=int(self.params['random_seed']))
self.run_for = self.run_for_stochastic # override base method
else:
from scipy.special import gamma
mean_storm__intensity = (self._length_factor) * self.params[
'mean_storm__intensity'] # has units length per time
intermittency_factor = self.params['intermittency_factor']
self.rain_generator = \
PrecipitationDistribution(mean_storm_duration=1.0,
mean_interstorm_duration=1.0,
mean_storm_depth=1.0,
random_seed=int(self.params['random_seed']))
self.intermittency_factor = intermittency_factor
self.mean_storm__intensity = mean_storm__intensity
self.shape_factor = self.params['precip_shape_factor']
self.scale_factor = (self.mean_storm__intensity /
gamma(1.0 + (1.0 / self.shape_factor)))
self.n_sub_steps = int(self.params['number_of_sub_time_steps'])
def reset_random_seed(self):
"""Re-set the random number generation sequence."""
try:
seed = int(self.params['random_seed'])
except KeyError:
seed = 0
self.rain_generator.seed_generator(seedval=seed)
def handle_water_erosion(self, dt, flooded):
"""Handle water erosion.
If we are running stochastic duration, then self.rain_rate will
have been calculated already. It might be zero, in which case we
are between storms, so we don't do water erosion.
If we're NOT doing stochastic duration, then we'll run water
erosion for one or more sub-time steps, each with its own
randomly drawn precipitation intensity.
This routine assumes that a model-specific method
**calc_runoff_and_discharge()**
will have been defined.
For example, BasicStVs calculated runoff and discharge in a different
way than the other models.
If the model has a function **update_threshold_field**, this
function will test for it and run it. This is presently done in
BasicDdSt.
"""
# (if we're varying precipitation parameters through time, update them)
if self.opt_var_precip:
self.intermittency_factor, self.mean_storm__intensity = self.pc.get_current_precip_params(
self.model_time)
if self.opt_stochastic_duration and self.rain_rate > 0.0:
runoff = self.calc_runoff_and_discharge()
self.eroder.run_one_step(dt,
flooded_nodes=flooded,
rainfall_intensity_if_used=runoff)
if self.record_rain:
#save record into the rain record
self.record_rain_event(self.model_time, dt, self.rain_rate,
runoff)
elif self.opt_stochastic_duration and self.rain_rate <= 0.0:
# calculate and record the time with no rain:
if self.record_rain:
self.record_rain_event(self.model_time, dt, 0, 0)
elif not self.opt_stochastic_duration:
dt_water = ((dt * self.intermittency_factor) /
float(self.n_sub_steps))
for i in range(self.n_sub_steps):
self.rain_rate = \
self.rain_generator.generate_from_stretched_exponential(
self.scale_factor, self.shape_factor)
runoff = self.calc_runoff_and_discharge()
self.eroder.run_one_step(dt_water,
flooded_nodes=flooded,
rainfall_intensity_if_used=runoff)
#save record into the rain record
if self.record_rain:
event_start_time = self.model_time + (i * dt_water)
self.record_rain_event(event_start_time, dt_water,
self.rain_rate, runoff)
# once all the rain time_steps are complete,
# calculate and record the time with no rain:
if self.record_rain:
# calculate dry time
dt_dry = dt * (1 - self.intermittency_factor)
# if dry time is greater than zero, record.
if dt_dry > 0:
event_start_time = self.model_time + ((i + 1) * dt_water)
self.record_rain_event(event_start_time, dt_dry, 0.0, 0.0)
def finalize(self):
# if rain was recorded, write it out.
if self.record_rain:
filename = self.params.get('storm_sequence_filename')
self.write_storm_sequence_to_file(filename)
if self.record_rain and (self.opt_stochastic_duration == False):
# if opt_stochastic_duration = False, calculate exceedance
# frequencies and write out.
frequency_filename = self.params.get('frequency_filename')
self.write_exceedance_frequency_file(frequency_filename)
def record_rain_event(self, event_start_time, event_duration,
rainfall_rate, runoff_rate):
"""Record rain events.
Create a record of event start time, event duration, rainfall rate, and
runoff rate.
"""
self.rain_record['event_start_time'].append(event_start_time)
self.rain_record['event_duration'].append(event_duration)
self.rain_record['rainfall_rate'].append(rainfall_rate)
self.rain_record['runoff_rate'].append(runoff_rate)
def write_storm_sequence_to_file(self, filename=None):
"""
Write event duration and intensity to a formatted text file.
"""
# Open a file for writing
if self.record_rain == False:
raise ValueError('Rain was not recorded when the model run. To '
'record rain, set the parameter "record_rain"'
'to True.')
if filename is None:
filename = 'event_sequence.txt'
stormfile = open(filename, 'w')
stormfile.write('event_start_time' + ',' + 'event_duration' + ',' +
'rainfall_rate' + ',' + 'runoff_rate' + '\n')
n_events = len(self.rain_record['event_start_time'])
for i in range(n_events):
stormfile.write(
str(self.rain_record['event_start_time'][i]) + ',' +
str(self.rain_record['event_duration'][i]) + ',' +
str(self.rain_record['rainfall_rate'][i]) + ',' +
str(self.rain_record['runoff_rate'][i]) + '\n')
# Close the file
stormfile.close()
def write_exceedance_frequency_file(self, filename=None):
"""
"""
if filename is None:
filename = 'exceedance_summary.txt'
exceedance_file = open(filename, 'w')
# calculate the number of wet days per year.
number_of_days_per_year = 365
nwet = int(np.ceil(self.intermittency_factor *
number_of_days_per_year))
#ndry = int(number_of_days_per_year - nwet)
# Write some basic information about the distribution to the file.
exceedance_file.write('Section 1: Distribution Description\n')
exceedance_file.write('Scale Factor: ' + str(self.scale_factor) + '\n')
exceedance_file.write('Shape Factor: ' + str(self.shape_factor) + '\n')
exceedance_file.write(
('Intermittency Factor: ' + str(self.intermittency_factor) + '\n'))
exceedance_file.write(
('Number of wet days per year: ' + str(nwet) + '\n\n'))
message_text = (
'The scale factor that describes this distribution is ' +
'calculated based on a provided value for the mean wet day rainfall.'
)
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
exceedance_file.write(('This provided value was:\n' +
str(self.mean_storm__intensity) + '\n'))
# calculate the predictions for 10, 25, and 100 year event based on
# the analytical form of the exceedance function.
event_intervals = np.array([10., 25, 100.])
# calculate the probability of each event based on the number of years
# and the number of wet days per year.
daily_distribution_exceedance_probabilities = (
1. / (nwet * event_intervals))
# exceedance probability is given as
# Probability of daily rainfall of p exceeding a value of po is given as:
#
# P(p>po) = e^(-(po/P)^c)
# P = scale
# c = shape
#
# this can be re-arranged to
#
# po = P * (- ln (P(p>po))) ^ (1 / c)
expected_rainfall = self.scale_factor * (
-1. * np.log(daily_distribution_exceedance_probabilities))**(
1. / self.shape_factor)
exceedance_file.write('\n\nSection 2: Theoretical Predictions\n')
message_text = (
'Based on the analytical form of the wet day rainfall ' +
'distribution, we can calculate theoretical predictions ' +
'of the daily rainfall amounts associated with N-year events.')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
for i in range(len(daily_distribution_exceedance_probabilities)):
exceedance_file.write(
('Expected value for the wet day total of the ' +
str(event_intervals[i]) + ' year event is: ' +
str(np.round(expected_rainfall[i], decimals=3)) + '\n'))
# get rainfall record and filter out time without any rain
all_precipitation = np.array(self.rain_record['rainfall_rate'])
rainy_day_inds = np.where(all_precipitation > 0)
if len(rainy_day_inds[0]) > 0:
wet_day_totals = all_precipitation[rainy_day_inds]
else:
raise ValueError(
'No rain fell, which makes calculating exceedance '
'frequencies problematic. We recommend that you '
'check the valude of intermittency_factor.')
# construct the distribution of yearly maxima.
# here an effective year is represented by the number of draws implied
# by the intermittency factor
# first calculate the number of effective years.
num_days = len(wet_day_totals)
num_effective_years = int(np.floor(wet_day_totals.size / nwet))
# write out the calculated event only if the duration
exceedance_file.write('\n\n')
message_text = ('Section 3: Predicted 95% confidence bounds on the ' +
'exceedance values based on number of samples drawn.')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
message_text = (
'The ability to empirically estimate the rainfall ' +
'associated with an N-year event depends on the ' +
'probability of that event occurring and the number of ' +
'draws from the probability distribution. The ability ' +
'to estimate increases with an increasing number of samples ' +
'and decreases with decreasing probability of event ' +
'occurrence.')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
message_text = (
'Exceedance values calculated from ' + str(len(wet_day_totals)) +
' draws from the daily-rainfall probability distribution. ' +
'This corresponds to ' + str(num_effective_years) +
' effective years.')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
# For a general probability distribution, f, with a continuous not zero
# quantile function at F-1(p), the order statistic associated with the
# p percentile given n draws from the distribution is given as:
# X[np] ~ AN ( F-1(p), (p * (p - 1 ))/ (n * [f (F-1 (p)) ]**2))
# where AN is the asymptotic normal. The value for the variance is more
# intuitive once you consider that [f (F-1 (p)) ] is the probability
# that an event of percentile p will occur. Thus the variance increases
# non-linearly with decreasing event probability and decreases linearly
# with increaseing observations.
# we've already calculated F-1(p) for our events, and it is represented
# by the variable expected_rainfall
daily_distribution_event_percentile = 1.0 - daily_distribution_exceedance_probabilities
event_probability = ((self.shape_factor / self.scale_factor) * (
(expected_rainfall / self.scale_factor)
**(self.shape_factor - 1.0)) * (np.exp(
-1. *
(expected_rainfall / self.scale_factor)**self.shape_factor)))
event_variance = ((daily_distribution_event_percentile *
(1.0 - daily_distribution_event_percentile)) /
(num_days * (event_probability**2)))
event_std = event_variance**0.5
t_statistic = stats.t.ppf(0.975, num_effective_years, loc=0, scale=1)
exceedance_file.write('\n')
message_text = ('For the given number of samples, the 95% ' +
'confidence bounds for the following event ' +
'return intervals are as follows: ')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
for i in range(len(event_intervals)):
min_expected_val = expected_rainfall[i] - t_statistic * event_std[i]
max_expected_val = expected_rainfall[i] + t_statistic * event_std[i]
exceedance_file.write(
('Expected range for the wet day total of the ' +
str(event_intervals[i]) + ' year event is: (' +
str(np.round(min_expected_val, decimals=3)) + ', ' +
str(np.round(max_expected_val, decimals=3)) + ')\n'))
# next, calculate the emperical exceedance values, if a sufficient record
# exists.
# inititialize a container for the maximum yearly precipitation.
maximum_yearly_precipitation = np.nan * np.zeros((num_effective_years))
for yi in range(num_effective_years):
# identify the starting and ending index coorisponding to the
# year
starting_index = yi * nwet
ending_index = starting_index + nwet
# select the years portion of the wet_day_totals
selected_wet_day_totals = wet_day_totals[
starting_index:ending_index]
# record the yearly maximum precipitation
maximum_yearly_precipitation[yi] = selected_wet_day_totals.max()
# calculate the distribution percentiles associated with each interval
event_percentiles = (1. - (1. / event_intervals)) * 100.
# calculated the event magnitudes associated with the percentiles.
event_magnitudes = np.percentile(maximum_yearly_precipitation,
event_percentiles)
# write out the calculated event only if the duration
exceedance_file.write('\n\nSection 4: Empirical Values\n')
message_text = (
'These empirical values should be interpreted in the ' +
'context of the expected ranges printed in Section 3. ' +
'If the expected range is large, consider using a longer ' +
'record of rainfall. The empirical values should fall ' +
'within the expected range at a 95% confidence level.')
exceedance_file.write('\n'.join(
textwrap.wrap(message_text, _STRING_LENGTH)))
exceedance_file.write('\n')
for i in range(len(event_percentiles)):
exceedance_file.write(
('Estimated value for the wet day total of the ' +
str(np.round(event_intervals[i], decimals=3)) +
' year event is: ' +
str(np.round(event_magnitudes[i], decimals=3)) + '\n'))
exceedance_file.close()
mg["node"]["topographic__elevation"] = z + numpy.random.rand(len(z)) / 1000.
mg.add_zeros("node", "water__unit_flux_in")
# make some K values in a field to test
# mg.at_node['K_values'] = 0.1+numpy.random.rand(nrows*ncols)/10.
mg.at_node["K_values"] = numpy.empty(nrows * ncols, dtype=float)
# mg.at_node['K_values'].fill(0.1+numpy.random.rand()/10.)
mg.at_node["K_values"].fill(0.001)
print("Running ...")
# instantiate the components:
fr = FlowAccumulator(mg, flow_director="D8")
sp = StreamPowerEroder(mg, input_file_string)
# fsp = FastscapeEroder(mg, input_file_string)
precip = PrecipitationDistribution(input_file=input_file_string)
# load the Fastscape module too, to allow direct comparison
fsp = FastscapeEroder(mg, input_file_string)
try:
# raise NameError
mg = copy.deepcopy(mg_mature)
except NameError:
print("building a new grid...")
out_interval = 50000.
last_trunc = time_to_run # we use this to trigger taking an output plot
# run to a steady state:
# We're going to cheat by running Fastscape SP for the first part of the solution
for (
interval_duration,
class StochasticDischargeHortonianModel(_ErosionModel):
"""
A StochasticDischargeHortonianModel generates a random sequency of
runoff events across a topographic surface, calculating the resulting
water discharge at each node.
"""
def __init__(self, input_file=None, params=None):
"""Initialize the StochasticDischargeHortonianModel."""
# Call ErosionModel's init
super(StochasticDischargeHortonianModel,
self).__init__(input_file=input_file, params=params)
# Instantiate components
self.flow_router = FlowRouter(self.grid, **self.params)
self.lake_filler = DepressionFinderAndRouter(self.grid, **self.params)
self.rain_generator = \
PrecipitationDistribution(delta_t=self.params['dt'],
total_time=self.params['run_duration'],
**self.params)
# Add a field for discharge
if 'surface_water__discharge' not in self.grid.at_node:
self.grid.add_zeros('node', 'surface_water__discharge')
self.discharge = self.grid.at_node['surface_water__discharge']
# Get the infiltration-capacity parameter
self.infilt = self.params['infiltration_capacity']
# Run flow routing and lake filler (only once, because we are not
# not changing topography)
self.flow_router.run_one_step()
self.lake_filler.map_depressions()
def reset_random_seed(self):
"""Re-set the random number generation sequence."""
try:
seed = self.params['random_seed']
except KeyError:
seed = 0
self.rain_generator.seed_generator(seedval=seed)
def run_one_step(self, dt):
"""
Advance model for one time-step of duration dt.
"""
# Calculate discharge field
area = self.grid.at_node['drainage_area']
if self.infilt > 0.0:
runoff = self.rain_rate - (self.infilt *
(1.0 -
np.exp(-self.rain_rate / self.infilt)))
else:
runoff = self.rain_rate
self.discharge[:] = runoff * area
def run_for(self, dt, runtime):
"""
Run model without interruption for a specified time period.
"""
self.rain_generator.delta_t = dt
self.rain_generator.run_time = runtime
for (tr, p) in self.rain_generator.yield_storm_interstorm_duration_intensity():
self.rain_rate = p
self.run_one_step(tr)
def write_storm_sequence_to_file(self, filename=None):
"""
Write event duration and intensity to a formatted text file.
"""
# Re-seed the random number generator, so we get the same sequence.
self.reset_random_seed()
# Generate a set of event parameters. This is needed because the
# PrecipitationDistribution component automatically generates a
# parameter set on initialization. Therefore, to get to the same
# starting point that we used in the sequence-through-time, we need
# to regenerate these.
self.rain_generator.get_precipitation_event_duration()
self.rain_generator.get_interstorm_event_duration()
self.rain_generator.get_storm_depth()
self.rain_generator.get_storm_intensity()
# Open a file for writing
if filename is None:
filename = 'event_sequence.txt'
stormfile = open(filename, 'w')
# Set the generator's time step and run time to the full duration of
# the run. This ensures that we get a sequence that is as long as the
# model run, and does not split events by time step (unlike the actual
# model run)
self.rain_generator.delta_t = self.params['run_duration']
self.rain_generator.run_time = self.params['run_duration']
tt = 0.0
for (tr, p) in self.rain_generator.yield_storm_interstorm_duration_intensity():
runoff = p - self.infilt * (1.0 - np.exp(-p / self.infilt))
stormfile.write(str(tt) + ',' + str(p) + ',' + str(runoff) + '\n')
tt += tr
stormfile.write(str(tt) + ',' + str(p) + ',' + str(runoff) + '\n')
# Close the file
stormfile.close()
grid = RasterModelGrid((125, 125), xy_spacing=v0)
grid.set_status_at_node_on_edges(right=grid.BC_NODE_IS_CLOSED, top=grid.BC_NODE_IS_CLOSED, \
left=grid.BC_NODE_IS_FIXED_VALUE, bottom=grid.BC_NODE_IS_CLOSED)
elev = grid.add_zeros('node', 'topographic__elevation')
elev[:] = b + 0.1 * hg * np.random.rand(len(elev))
base = grid.add_zeros('node', 'aquifer_base__elevation')
wt = grid.add_zeros('node', 'water_table__elevation')
wt[:] = elev.copy()
#initialize landlab components
gdp = GroundwaterDupuitPercolator(grid, porosity=n, hydraulic_conductivity=ksat, \
regularization_f=0.01, recharge_rate=0.0, \
courant_coefficient=0.9, vn_coefficient = 0.9)
pd = PrecipitationDistribution(grid,
mean_storm_duration=tr,
mean_interstorm_duration=tb,
mean_storm_depth=ds,
total_t=Th)
pd.seed_generator(seedval=1235)
ld = LinearDiffuser(grid, linear_diffusivity=D)
#initialize other models
hm = HydrologyEventStreamPower(
grid,
precip_generator=pd,
groundwater_model=gdp,
)
#use surface_water_area_norm__discharge (Q/sqrt(A)) for Theodoratos definitions
sp = FastscapeEroder(grid,
K_sp=Ksp,
import numpy as np
from landlab.utils.depth_dependent_roughness import depth_dependent_mannings_n
from matplotlib import pyplot as plt
from landlab.plot import imshow_grid
from scipy.stats import norm
import time
import matplotlib.pyplot as plt
plt.rcParams["font.family"] = "Helvetica"
plt.rcParams['font.size'] = 10
## Generate 1000 year precipitation time series
total_t = 1000.*365.25*24
lowRvar_PrecipDist = PrecipitationDistribution(mean_storm_duration = 11.75,
mean_interstorm_duration = 146.25,
mean_storm_depth = 4.775,
total_t=total_t)
highRvar_PrecipDist = PrecipitationDistribution(mean_storm_duration = 10.75,
mean_interstorm_duration = 433.58,
mean_storm_depth = 9.62,
total_t=total_t)
thresh = 0.5
# Get actual time series for the high Rvar case
highRvar_storm_arr = np.array(highRvar_PrecipDist.get_storm_time_series())
highRvar_intensity_threshold, = np.where(highRvar_storm_arr[:, 2] > thresh)
highRvar_durations = highRvar_storm_arr[highRvar_intensity_threshold][:,1]-highRvar_storm_arr[highRvar_intensity_threshold][:,0]
highRvar_durations_s = [x *3600. for x in highRvar_durations]
