Update function about veros HOT 17 CLOSED

[image: image.png] This is what I wrote. I was trying to introduce a density perturbation at the surface but, the changes don't show up. I had to setup a for loop instead. I'm new to python so I might just be writing something wrong but can you help? thank you.

Yeah, sure! Here it is. I'm just modifying the ACC Model code on the veros website to familiarise myself with it. from veros import VerosSetup, veros_routine from veros.variables import allocate, Variable from veros.distributed import global_min, global_max from veros.core.operators import numpy as npx, update, at class EQ(VerosSetup): @veros_routine def set_parameter(self, state): settings = state.settings settings.identifier = "r11" settings.nx, settings.ny, settings.nz = 20, 20, 15 settings.dt_mom = 4800 settings.dt_tracer = 86400 / 2.0 settings.runlen = 86400 * 365 settings.x_origin = 20.0 settings.y_origin = -21.0 settings.coord_degree = True settings.enable_cyclic_x = True settings.enable_neutral_diffusion = True settings.K_iso_0 = 1000.0 settings.K_iso_steep = 500.0 settings.iso_dslope = 0.005 settings.iso_slopec = 0.01 settings.enable_skew_diffusion = True settings.enable_hor_friction = True settings.A_h = (2 * settings.degtom) ** 3 * 2e-11 settings.enable_hor_friction_cos_scaling = True settings.hor_friction_cosPower = 1 settings.enable_bottom_friction = True settings.r_bot = 1e-5 settings.enable_implicit_vert_friction = True settings.enable_tke = True settings.c_k = 0.1 settings.c_eps = 0.7 settings.alpha_tke = 30.0 settings.mxl_min = 1e-8 settings.tke_mxl_choice = 2 settings.kappaM_min = 2e-4 settings.kappaH_min = 2e-5 settings.enable_kappaH_profile = True settings.K_gm_0 = 1000.0 settings.enable_eke = True settings.eke_k_max = 1e4 settings.eke_c_k = 0.4 settings.eke_c_eps = 0.5 settings.eke_cross = 2.0 settings.eke_crhin = 1.0 settings.eke_lmin = 100.0 settings.enable_eke_superbee_advection = True settings.enable_eke_isopycnal_diffusion = True settings.enable_idemix = False settings.eq_of_state_type = 3 var_meta = state.var_meta var_meta.update( t_star=Variable("t_star", ("yt",), "deg C", "Reference surface temperature"), t_rest=Variable("t_rest", ("xt", "yt"), "1/s", "Surface temperature restoring time scale"), ) @veros_routine def set_grid(self, state): vs = state.variables ddz = npx.array( [50.0, 70.0, 100.0, 140.0, 190.0, 240.0, 290.0, 340.0, 390.0, 440.0, 490.0, 540.0, 590.0, 640.0, 690.0] ) vs.dxt = update(vs.dxt, at[...], 2.0) vs.dyt = update(vs.dyt, at[...], 2.0) vs.dzt = update(vs.dzt, at[...], ddz[::-1] / 2.5) @veros_routine def set_coriolis(self, state): vs = state.variables settings = state.settings vs.coriolis_t = update( vs.coriolis_t, at[...], 2 * settings.omega * npx.sin(vs.yt[None, :] / 180.0 * settings.pi) ) @veros_routine def set_topography(self, state): vs = state.variables x, y = npx.meshgrid(vs.xt, vs.yt, indexing="ij") vs.kbot = npx.logical_or(x > 1.0, y < -20).astype("int") @veros_routine def set_initial_conditions(self, state): vs = state.variables settings = state.settings # initial conditions vs.temp = update(vs.temp, at[...], ((1 - vs.zt[None, None, :] / vs .zw[0]) * 15 * vs.maskT)[..., None]) vs.salt = update(vs.salt, at[...], 35.0 * vs.maskT[..., None]) print(vs.salt.flags) vs.salt.setflags(write=1) # this is the for loop i was talking about y=6 for x in range(-2,0): for z in range(-2,0): vs.salt[x][y][z] = 35.3 # this is my attempt at using the update function vs.salt = update(vs.salt, at[-2:0, 6:7, -2,0], 35.0) # wind stress forcing yt_min = global_min(vs.yt.min()) yu_min = global_min(vs.yu.min()) yt_max = global_max(vs.yt.max()) yu_max = global_max(vs.yu.max()) # surface heatflux forcing vs.t_star = allocate(state.dimensions, ("yt",), fill=15) vs.t_star = npx.where(vs.yt < -20, 15 * (vs.yt - yt_min) / (-20 - yt_min), vs.t_star) vs.t_star = npx.where(vs.yt > 20, 15 * (1 - (vs.yt - 20) / (yt_max - 20)), vs.t_star) vs.t_rest = vs.dzt[npx.newaxis, -1] / (30.0 * 86400.0) * vs.maskT[:, :, -1] if settings.enable_tke: vs.forc_tke_surface = update( vs.forc_tke_surface, at[2:-2, 2:-2], npx.sqrt( (0.5 * (vs.surface_taux[2:-2, 2:-2] + vs.surface_taux[1: -3, 2:-2]) / settings.rho_0) ** 2 + (0.5 * (vs.surface_tauy[2:-2, 2:-2] + vs.surface_tauy[ 2:-2, 1:-3]) / settings.rho_0) ** 2 ) ** (1.5), ) if settings.enable_idemix: vs.forc_iw_bottom = 1e-6 * vs.maskW[:, :, -1] vs.forc_iw_surface = 1e-7 * vs.maskW[:, :, -1] @veros_routine def set_forcing(self, state): vs = state.variables vs.forc_temp_surface = vs.t_rest * (vs.t_star - vs.temp[:, :, -1, vs .tau]) @veros_routine def set_diagnostics(self, state): settings = state.settings diagnostics = state.diagnostics diagnostics["snapshot"].output_frequency = 86400 * 10 diagnostics["averages"].output_variables = ( "salt", "temp", "u", "v", "w", "psi", "surface_taux", "surface_tauy" ) diagnostics["averages"].output_frequency = 365 * 86400.0 diagnostics["averages"].sampling_frequency = settings.dt_tracer * 10 diagnostics["overturning"].output_frequency = 365 * 86400.0 / 48.0 diagnostics["overturning"].sampling_frequency = settings.dt_tracer * 10 diagnostics["tracer_monitor"].output_frequency = 365 * 86400.0 / 12.0 diagnostics["energy"].output_frequency = 365 * 86400.0 / 48 diagnostics["energy"].sampling_frequency = settings.dt_tracer * 10 @veros_routine def after_timestep(self, state): pass

Oh okay. Thank you so much!

Hi! I'm still not seeing a change after doing what you suggested.

Okay cool. Figured that out. I have another question though. I was trying to increase the grid size of the model from 40*40 to 100*100. And I got a latitudnal filtering error. How do I fix that?

Oh cool! I’ll try that. Also, is there any way to make ssh a time varying function like salinity and temperature.

Okay I’ll do that. Thank you so much.

Hi Dion, Just wanted to confirm that ssh will eveolve like other paramters like salinity and temperature with time if I do this. I wanted to put a ssh anomaly in the middle of a square basin and see it evolve in time. On Fri, 27 Oct 2023 at 16:27, Kashika Singh < ***@***.***> wrote:

I tried this, and while the anomaly does show up, it is not time-varying. I just get an average value. Is there a way to include psi or ssh in the snapshots folder? [image: image.png]