def test_sed_dep_new():
"""
This tests only the power_law version of the SDE.
It uses a landscape run to 5000 100y iterations, then having experienced
a 20-fold uplift acceleration for a further 30000 y. It tests the outcome
of the next 1000 y of erosion.
"""
mg = RasterModelGrid((25, 50), 200.)
for edge in (mg.nodes_at_left_edge, mg.nodes_at_top_edge,
mg.nodes_at_right_edge):
mg.status_at_node[edge] = CLOSED_BOUNDARY
z = mg.add_zeros('node', 'topographic__elevation')
fr = FlowAccumulator(mg, flow_director='D8')
sde = SedDepEroder(mg, K_sp=1.e-4, sed_dependency_type='almost_parabolic',
Qc='power_law', K_t=1.e-4)
initconds = os.path.join(os.path.dirname(__file__),
'perturbedcondst300.txt')
finalconds = os.path.join(os.path.dirname(__file__),
'tenmorestepsfrom300.txt')
z[:] = np.loadtxt(initconds)
dt = 100.
up = 0.05
for i in range(10):
fr.run_one_step()
sde.run_one_step(dt)
z[mg.core_nodes] += 20.*up
assert_array_almost_equal(z, np.loadtxt(finalconds))
def test_sed_dep_new():
"""
This tests only the power_law version of the SDE.
It uses a landscape run to 5000 100y iterations, then having experienced
a 20-fold uplift acceleration for a further 30000 y. It tests the outcome
of the next 1000 y of erosion.
"""
mg = RasterModelGrid((25, 50), xy_spacing=200.)
for edge in (mg.nodes_at_left_edge, mg.nodes_at_top_edge, mg.nodes_at_right_edge):
mg.status_at_node[edge] = CLOSED_BOUNDARY
z = mg.add_zeros("node", "topographic__elevation")
fr = FlowAccumulator(mg, flow_director="D8")
sde = SedDepEroder(
mg,
K_sp=1.e-4,
sed_dependency_type="almost_parabolic",
Qc="power_law",
K_t=1.e-4,
)
initconds = os.path.join(os.path.dirname(__file__), "perturbedcondst300.txt")
finalconds = os.path.join(os.path.dirname(__file__), "tenmorestepsfrom300.txt")
z[:] = np.loadtxt(initconds)
dt = 100.
up = 0.05
for i in range(10):
fr.run_one_step()
sde.run_one_step(dt)
z[mg.core_nodes] += 20. * up
assert_array_almost_equal(z, np.loadtxt(finalconds))
def test_route_to_multiple_error_raised_run_SedDepEroder():
mg = RasterModelGrid((10, 10))
z = mg.add_zeros('node', 'topographic__elevation')
z += mg.x_of_node + mg.y_of_node
sp = SedDepEroder(mg)
fa = FlowAccumulator(mg, flow_director='MFD')
fa.run_one_step()
with pytest.raises(NotImplementedError):
sp.run_one_step(10)
def test_route_to_multiple_error_raised_run_SedDepEroder():
mg = RasterModelGrid((10, 10))
z = mg.add_zeros("node", "topographic__elevation")
z += mg.x_of_node + mg.y_of_node
sp = SedDepEroder(mg)
fa = FlowAccumulator(mg, flow_director="MFD")
fa.run_one_step()
with pytest.raises(NotImplementedError):
sp.run_one_step(10)
def test_sed_dep():
uplift_rate = 0.0001
runtime = 20000.0
dt = 5000.0
nt = int(runtime // dt)
uplift_per_step = uplift_rate * dt
mg = RasterModelGrid((50, 50), xy_spacing=1000.0)
mg.add_zeros("topographic__elevation", at="node")
z = np.loadtxt(os.path.join(_THIS_DIR, "seddepinit.txt"))
mg["node"]["topographic__elevation"] = z
mg.set_closed_boundaries_at_grid_edges(True, False, True, False)
fr = FlowAccumulator(mg, flow_director="D8")
sde = SedDepEroder(
mg,
K_sp=1.0e-6,
b_sp=0.5,
c_sp=1.0,
Qc="MPM",
k_Q=2.5e-7,
k_w=2.5,
mannings_n=0.05,
Dchar=0.05,
threshold_Shields=0.05,
sed_dependency_type="almost_parabolic",
set_threshold_from_Dchar=True,
g=9.81,
)
for i in range(nt):
mg.at_node["topographic__elevation"][mg.core_nodes] += uplift_per_step
mg = fr.run_one_step()
mg, _ = sde.run_one_step(dt)
z_tg = np.loadtxt(os.path.join(_THIS_DIR, "seddepz_tg.txt"))
assert_array_almost_equal(
mg.at_node["topographic__elevation"][mg.core_nodes], z_tg[mg.core_nodes]
)
from landlab.components import FlowAccumulator, SedDepEroder
from landlab import RasterModelGrid
import landlab
import numpy as np
from pylab import show, figure
from landlab.io import read_esri_ascii
(mg, z) = read_esri_ascii("bacia_piratini_90m.asc", name="topographic__elevation")
fda = FlowAccumulator(mg, 'topographic__elevation')
sde1 = SedDepEroder(mg, Qc="MPM")
fda.run_one_step()
sde1.run_one_step(dt=30) #dt em anos
print(sde1.characteristic_grainsize)
def extend_perturbed_runs(total_iters_to_reach=0):
"""Load all perturbed runs in current folder, and extend them.
Function should be called from within an experiment folder
(extend all perturbations for all starting uplift rates), an
'uplift_rate_XXXX' folder (extend all perturbations for this rate) or an
'accel_XXX' folder (extend this accel only).
Does NOT create a new expt or run ID, just extends the old ones. Adds a
text file annotating what has happened.
"""
# look for the params to use. Also tells us where we are in the hierarchy
level = 0 # 0: top, 1: uplift, 2: accel:
cwd = os.getcwd()
while True:
try:
paramdict = np.load('expt_ID_paramdict.npy').item()
except IOError:
os.chdir('..')
level += 1
else:
break
# now back to where we started in the dir str:
os.chdir(cwd)
if level == 2: # in accel_ folder
# get the accel that this is:
accel_factors = [
get_float_of_folder_name(),
]
# get the U of the host folder:
uplift_rates = [
get_float_of_folder_name(directory=(cwd + '/..')),
]
wd_stub = os.path.abspath(os.getcwd() + '/../..')
elif level == 1: # in uplift_ folder
accel_fnames = [
filename for filename in os.listdir('.')
if filename.startswith('accel_')
]
accel_factors = [
get_float_of_folder_name(directory=(cwd + '/' + filename))
for filename in accel_fnames
]
uplift_rates = [
get_float_of_folder_name(),
]
wd_stub = os.path.abspath(os.getcwd() + '/..')
elif level == 0: # in top folder
uplift_fnames = [
filename for filename in os.listdir('.')
if filename.startswith('uplift_rate_')
]
uplift_rates = [
get_float_of_folder_name(directory=(cwd + '/' + filename))
for filename in uplift_fnames
]
accel_factors = paramdict['accel_factors']
wd_stub = os.path.abspath(os.getcwd())
for uplift_rate in uplift_rates:
for accel_factor in accel_factors:
wd = (wd_stub + '/uplift_rate_' + str(uplift_rate) + '/accel_' +
str(accel_factor))
# get the saved filenames that already exist in this folder:
runnames = [
filename for filename in os.listdir(wd)
if filename.startswith('topographic__elevation')
]
seddepthnames = [
filename for filename in os.listdir(wd)
if filename.startswith('channel_sediment__depth')
]
# as elsewhere, the final entry is the last run, so --
# establish the loop number of that run:
run_ID = runnames[-1][-14:-4] # is a str
_format = 0
while True:
char = runnames[-1][-16 - _format]
try:
num = int(char)
except ValueError: # was a str
break
else:
_format += 1
finaliter = int(runnames[-1][(-15 - _format):-15])
finalsediter = int(seddepthnames[-1][(-15 - _format):-15])
assert finaliter == finalsediter # ...just in case
# test we need to actually do more runs:
if total_iters_to_reach < finaliter + paramdict['out_interval']:
continue
# check we aren't going to have a "zero problem"; correct if we do
max_zeros = len(str(total_iters_to_reach))
if max_zeros + 1 > _format: # less won't be possible from continue
extra_zeros = max_zeros + 1 - _format
for allfile in os.listdir(wd):
if allfile[-14:-4] == run_ID:
os.rename(
wd + '/' + allfile,
(wd + '/' + allfile[:(-15 - _format)] +
'0' * extra_zeros + allfile[(-15 - _format):]))
runnames = [
filename for filename in os.listdir(wd)
if filename.startswith('topographic__elevation')
]
seddepthnames = [
filename for filename in os.listdir(wd)
if filename.startswith('channel_sediment__depth')
]
if max_zeros + 1 < _format:
max_zeros = _format - 1 # in case of any bonus 0s from old run
# build the structures:
mg = RasterModelGrid(paramdict['shape'], paramdict['dx'])
for edge in (mg.nodes_at_left_edge, mg.nodes_at_top_edge,
mg.nodes_at_right_edge):
mg.status_at_node[edge] = CLOSED_BOUNDARY
z = mg.add_zeros('node', 'topographic__elevation')
seddepth = mg.add_zeros('node', 'channel_sediment__depth')
fr = FlowRouter(mg)
eroder = SedDepEroder(mg, **paramdict)
ld = LinearDiffuser(mg, **paramdict)
# load the last available elev data:
z[:] = np.loadtxt(wd + '/' + runnames[-1])
seddepth[:] = np.loadtxt(wd + '/' + seddepthnames[-1])
# save a note
try:
appendfile = open(wd + '/appended_run_readme.txt', 'a')
except IOError:
appendfile = open(wd + '/appended_run_readme.txt', 'w')
appendfile.write('This run was appended at timestamp ' +
str(int(time.time())) + '.\n')
appendfile.write('New loops were added from iteration ' +
str(finaliter) + ' and terminated at iteration ' +
str(total_iters_to_reach) + '.\n\n')
appendfile.close()
# get runnin'
print('Extending uplift ' + str(uplift_rate) + ' accel ' +
str(accel_factor) + ' from iter number ' + str(finaliter))
dt = paramdict['dt']
for i in xrange(finaliter + 1, total_iters_to_reach):
fr.route_flow()
eroder.run_one_step(dt)
ld.run_one_step(dt)
z[mg.core_nodes] += accel_factor * uplift_rate * dt
print(i)
if i % out_interval == 0:
zeros_to_add = max_zeros - len(str(i)) + 1
# note an OoM buffer! Just to be safe
if zeros_to_add < 0:
# ...just in case, though should never happen
print('Problem allocating zeros on savefiles')
ilabel = '0' * zeros_to_add + str(i)
identifier = ilabel + '_' + str(run_ID)
for field in out_fields:
np.savetxt(
wd + '/' + field + '_' + identifier + '.txt',
mg.at_node[field])