def set_initial_conditions(self, vs):
# initial conditions
vs.temp[:, :, :, 0:2] = ((1 - vs.zt[None, None, :] / vs.zw[0]) * 15 * vs.maskT)[..., None]
vs.salt[:, :, :, 0:2] = 35.0 * vs.maskT[..., None]
# wind stress forcing
yt_min = global_min(vs, vs.yt.min())
yu_min = global_min(vs, vs.yu.min())
yt_max = global_max(vs, vs.yt.max())
yu_max = global_max(vs, vs.yu.max())
taux = allocate(vs, ('yt',))
taux[vs.yt < -20] = 0.1 * np.sin(vs.pi * (vs.yu[vs.yt < -20] - yu_min) / (-20.0 - yt_min))
taux[vs.yt > 10] = 0.1 * (1 - np.cos(2 * vs.pi * (vs.yu[vs.yt > 10] - 10.0) / (yu_max - 10.0)))
vs.surface_taux[:, :] = taux * vs.maskU[:, :, -1]
# surface heatflux forcing
vs._t_star = allocate(vs, ('yt',), fill=15)
vs._t_star[vs.yt < -20] = 15 * (vs.yt[vs.yt < -20] - yt_min) / (-20 - yt_min)
vs._t_star[vs.yt > 20] = 15 * (1 - (vs.yt[vs.yt > 20] - 20) / (yt_max - 20))
vs._t_rest = vs.dzt[None, -1] / (30. * 86400.) * vs.maskT[:, :, -1]
if vs.enable_tke:
vs.forc_tke_surface[2:-2, 2:-2] = np.sqrt((0.5 * (vs.surface_taux[2:-2, 2:-2] + vs.surface_taux[1:-3, 2:-2]) / vs.rho_0)**2
+ (0.5 * (vs.surface_tauy[2:-2, 2:-2] + vs.surface_tauy[2:-2, 1:-3]) / vs.rho_0)**2)**(1.5)
if vs.enable_idemix:
vs.forc_iw_bottom[...] = 1e-6 * vs.maskW[:, :, -1]
vs.forc_iw_surface[...] = 1e-7 * vs.maskW[:, :, -1]
def set_initial_conditions(self, vs):
# initial conditions
vs.temp[:, :, :, 0:2] = ((1 - vs.zt[None, None, :] / vs.zw[0]) * 15 *
vs.maskT)[..., None]
vs.salt[:, :, :, 0:2] = 35.0 * vs.maskT[..., None]
# wind stress forcing
yt_min = global_min(vs, vs.yt.min())
yu_min = global_min(vs, vs.yu.min())
yt_max = global_max(vs, vs.yt.max())
yu_max = global_max(vs, vs.yu.max())
taux = allocate(vs, ('yt', ))
north = vs.yt > 30
subequatorial_north_n = (vs.yt >= 15) & (vs.yt < 30)
subequatorial_north_s = (vs.yt > 0) & (vs.yt < 15)
equator = (vs.yt > -5) & (vs.yt < 5)
subequatorial_south_n = (vs.yt > -15) & (vs.yt < 0)
subequatorial_south_s = (vs.yt <= -15) & (vs.yt > -30)
south = vs.yt < -30
taux[north] = -5e-2 * np.sin(np.pi * (vs.yu[north] - yu_max) /
(yt_max - 30.))
taux[subequatorial_north_s] = 5e-2 * np.sin(
np.pi * (vs.yu[subequatorial_north_s] - 30.) / 30.)
taux[subequatorial_north_n] = 5e-2 * np.sin(
np.pi * (vs.yt[subequatorial_north_n] - 30.) / 30.)
taux[subequatorial_south_n] = -5e-2 * np.sin(
np.pi * (vs.yu[subequatorial_south_n] - 30.) / 30.)
taux[subequatorial_south_s] = -5e-2 * np.sin(
np.pi * (vs.yt[subequatorial_south_s] - 30.) / 30.)
taux[equator] = -1.5e-2 * np.cos(np.pi *
(vs.yu[equator] - 10.) / 10.) - 2.5e-2
taux[south] = 15e-2 * np.sin(np.pi * (vs.yu[south] - yu_min) /
(-30. - yt_min))
vs.surface_taux[:, :] = taux * vs.maskU[:, :, -1]
# surface heatflux forcing
DELTA_T, TS, TN = 25., 0., 5.
vs._t_star = allocate(vs, ('yt', ), fill=DELTA_T)
vs._t_star[vs.yt < 0] = TS + DELTA_T * np.sin(
np.pi * (vs.yt[vs.yt < 0] + 60.) / np.abs(2 * vs.y_origin))
vs._t_star[vs.yt > 0] = TN + (DELTA_T + TS - TN) * np.sin(
np.pi * (vs.yt[vs.yt > 0] + 60.) / np.abs(2 * vs.y_origin))
vs._t_rest = vs.dzt[None, -1] / (10. * 86400.) * vs.maskT[:, :, -1]
if vs.enable_tke:
vs.forc_tke_surface[2:-2, 2:-2] = np.sqrt(
(0.5 *
(vs.surface_taux[2:-2, 2:-2] + vs.surface_taux[1:-3, 2:-2]) /
vs.rho_0)**2 +
(0.5 *
(vs.surface_tauy[2:-2, 2:-2] + vs.surface_tauy[2:-2, 1:-3]) /
vs.rho_0)**2)**(1.5)
if vs.enable_idemix:
vs.forc_iw_bottom[...] = 1e-6 * vs.maskW[:, :, -1]
vs.forc_iw_surface[...] = 1e-7 * vs.maskW[:, :, -1]
def set_initial_conditions(self, vs):
vs.u[:, :, :, vs.tau] = np.random.rand(54, 104, 10)
# wind stress forcing
yt_min = global_min(vs, vs.yt.min())
yu_min = global_min(vs, vs.yu.min())
yt_max = global_max(vs, vs.yt.max())
yu_max = global_max(vs, vs.yu.max())
# surface heatflux forcing
vs._t_star = allocate(vs, ('yt',), fill=15)
vs._t_star[vs.yt < -20] = 15 * (vs.yt[vs.yt < -20] - yt_min) / (-20 - yt_min)
vs._t_star[vs.yt > 20] = 15 * (1 - (vs.yt[vs.yt > 20] - 20) / (yt_max - 20))
vs._t_rest = vs.dzt[None, -1] / (30. * 86400.) * vs.maskT[:, :, -1]
pass