Code for RF snapshots

Katetc · Dec 9, 2024 · a15f13a · a15f13a
1 parent 47dd341
commit a15f13a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/physics/cam7/physpkg.F90 b/src/physics/cam7/physpkg.F90
@@ -2164,14 +2164,26 @@ subroutine tphysac (ztodt,   cam_in,  &
     ! Rayleigh friction calculation
     !===================================================
     call t_startf('rayleigh_friction')
+    if (trim(cam_take_snapshot_before) == "user_set") then
+       call cam_snapshot_all_outfld_tphysac(cam_snapshot_before_num, state, tend, cam_in, cam_out, pbuf, &
+            fh2o, surfric, obklen, flx_heat, cmfmc, dlf, det_s, det_ice, net_flx)
+    end if
     call rayleigh_friction_tend( ztodt, state, ptend)
+    if ( (trim(cam_take_snapshot_after) == "user_set") .and.      &
+         (trim(cam_take_snapshot_before) == trim(cam_take_snapshot_after))) then
+       call cam_snapshot_ptend_outfld(ptend, lchnk)
+    end if
     if ( ptend%lu ) then
       call outfld( 'UTEND_RAYLEIGH', ptend%u, pcols, lchnk)
     end if
     if ( ptend%lv ) then
       call outfld( 'VTEND_RAYLEIGH', ptend%v, pcols, lchnk)
     end if
     call physics_update(state, ptend, ztodt, tend)
+    if (trim(cam_take_snapshot_after) == "user_set") then
+       call cam_snapshot_all_outfld_tphysac(cam_snapshot_after_num, state, tend, cam_in, cam_out, pbuf, &
+            fh2o, surfric, obklen, flx_heat, cmfmc, dlf, det_s, det_ice, net_flx)
+    end if
     call t_stopf('rayleigh_friction')
 
     if (do_clubb_sgs) then