a0 = 1
n0 = 1
n1 = 0.5
a1 = -(a0 * n0) / n1
v[0] = a0 * sin(pi * n0 * x[0]) * cos(pi * n1 * x[1])
v[1] = a1 * cos(pi * n0 * x[0]) * sin(pi * n1 * x[1])
else:
a0 = 1
a1 = 1
n0 = 2
n1 = 2
n2 = 0.5
a2 = -(a0 * n0 + a1 * n1) / n2
v[0] = a0 * sin(pi * n0 * x[0]) * cos(pi * n1 * x[1]) * cos(
pi * n2 * x[2])
v[1] = a1 * cos(pi * n0 * x[0]) * sin(pi * n1 * x[1]) * cos(
pi * n2 * x[2])
v[2] = a2 * cos(pi * n0 * x[0]) * cos(pi * n1 * x[1]) * sin(
pi * n2 * x[2])
mts = Mountains(mydomain, eps=EPS)
while t < T_END:
print("STEP ", t)
mts.setVelocity(v * cos(OMEGA * t))
Z = mts.update()
saveVTK("state.%d.vtu" % n, sol=Z, v=mts.getVelocity())
print("Integral(Z)=", integrate(Z), Lsup(mts.getVelocity()[DIM - 1]))
n += 1
t += mts.getSafeTimeStepSize()
flow.setDruckerPragerLaw(tau_Y=TAU_Y / P_REF + BETA *
(1. - Function(dom).getX()[DIM - 1]))
flow.setElasticShearModulus(MUE)
flow.setTolerance(FLOW_TOL)
flow.setEtaTolerance(FLOW_TOL)
flow.setExternals(fixed_v_mask=fixed_v_mask)
print(("<%s> Flow solver has been set up." % time.asctime()))
#
# topography setup
#
boundary = FunctionOnBoundary(dom).getX()[DIM - 1]
top_boundary_mask = whereZero(boundary - sup(boundary))
surface_area = integrate(top_boundary_mask)
if CREATE_TOPO:
mts = Mountains(dom, eps=TOPO_SMOOTH)
mts.setTopography(topography)
print(("<%s> topography has been set up." % time.asctime()))
#
# let the show begin:
#
t1 = time.time()
print(("<%s> Start time step %d (t=%s)." % (time.asctime(), n, t)))
while t < T_END:
if CREATE_TOPO: topography_old = topography
v_old, p_old, stress_old = v, p, stress
T_old = T
#========================= solve for velocity ============================
eta_N = exp(
Ar *
