save_state Subroutine

private pure subroutine save_state(ms)

Copy h_layer, u_face_x_layer, v_face_y_layer into the *_0 save buffers on the multilayer state. Per-tracer hTr is saved by the caller via copy_field_3d (the registry has to be walked on the host).

Arguments

Type IntentOptional Attributes Name
type(multilayer_state_t), intent(inout) :: ms

Called by

proc~~save_state~~CalledByGraph proc~save_state save_state proc~ocean_dyn_step ocean_dyn_step proc~ocean_dyn_step->proc~save_state proc~ocean_dyn_step_split ocean_dyn_step_split proc~ocean_dyn_step_split->proc~save_state proc~engine_step engine_step proc~engine_step->proc~ocean_dyn_step proc~engine_step->proc~ocean_dyn_step_split proc~driver_run_ocean driver_run_ocean proc~driver_run_ocean->proc~engine_step proc~rdb_ocean_step rdb_ocean_step proc~rdb_ocean_step->proc~engine_step proc~driver_run driver_run proc~driver_run->proc~driver_run_ocean

Variables

Type Visibility Attributes Name Initial
integer, private :: i
integer, private :: j
integer, private :: k
integer, private :: nx
integer, private :: nx_face
integer, private :: nx_vface
integer, private :: ny
integer, private :: ny_face
integer, private :: ny_uface
integer, private :: nz

Source Code

   pure subroutine save_state(ms)
      !! Copy h_layer, u_face_x_layer, v_face_y_layer into the *_0
      !! save buffers on the multilayer state.  Per-tracer hTr is
      !! saved by the caller via copy_field_3d (the registry has to
      !! be walked on the host).
      type(multilayer_state_t), intent(inout) :: ms
      integer :: i, j, k, nx, ny, nz, nx_face, ny_uface, nx_vface, ny_face

      nx = size(ms%h_layer, 1)
      ny = size(ms%h_layer, 2)
      nz = ms%nz_ml
      nx_face = size(ms%u_face_x_layer, 1)
      ny_uface = size(ms%u_face_x_layer, 2)
      nx_vface = size(ms%v_face_y_layer, 1)
      ny_face = size(ms%v_face_y_layer, 2)

      do concurrent(k=1:nz, j=1:ny, i=1:nx)
         ms%h_layer0(i, j, k) = ms%h_layer(i, j, k)
      end do
      do concurrent(k=1:nz, j=1:ny_uface, i=1:nx_face)
         ms%u_face_x_layer0(i, j, k) = ms%u_face_x_layer(i, j, k)
      end do
      do concurrent(k=1:nz, j=1:ny_face, i=1:nx_vface)
         ms%v_face_y_layer0(i, j, k) = ms%v_face_y_layer(i, j, k)
      end do
   end subroutine save_state