-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Some initial infrastructure for online coarse-graining (#26)
* Add basic example of a coarse-grained diagnostic * Slight refactor; added error checking * Minor cleanup * Remove placeholder implementation of weighted_block_average; add static diagnostic fields * Make sure fine array bounds are properly initialized * Reorder subroutines * Clean up static diagnostics * Remove unneeded block_sum_3d * Reduce things down to a minimal example * Simplify interface some * Shorten names somewhat * Address Jeremy's commemnts * Add more comments * Address Oli's comment; a few other minor cleanups * Move do_coarse_graining flag up a level
- Loading branch information
1 parent
31fc2ee
commit 0e240ab
Showing
6 changed files
with
462 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
135 changes: 135 additions & 0 deletions
135
FV3/atmos_cubed_sphere/tools/coarse_grained_diagnostics.F90
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,135 @@ | ||
| module coarse_grained_diagnostics_mod | ||
|
|
||
| use diag_manager_mod, only: diag_axis_init, register_diag_field, register_static_field, send_data | ||
| use fv_arrays_mod, only: fv_atmos_type, fv_coarse_diag_type, fv_coarse_graining_type, fv_grid_bounds_type | ||
| use mpp_domains_mod, only: domain2d | ||
| use mpp_mod, only: FATAL, mpp_error | ||
| use coarse_graining_mod, only: block_sum, get_fine_array_bounds, get_coarse_array_bounds, MODEL_LEVEL, weighted_block_average | ||
| use time_manager_mod, only: time_type | ||
|
|
||
| implicit none | ||
| private | ||
|
|
||
| public :: fv_coarse_diag_init, fv_coarse_diag | ||
|
|
||
| integer :: tile_count = 1 ! Following fv_diagnostics.F90 | ||
|
|
||
| contains | ||
|
|
||
| subroutine fv_coarse_diag_init(bd, Time, id_pfull, id_phalf, coarse_graining) | ||
| type(fv_grid_bounds_type), intent(in) :: bd | ||
| type(time_type), intent(in) :: Time | ||
| integer, intent(in) :: id_pfull, id_phalf | ||
| type(fv_coarse_graining_type), intent(inout) :: coarse_graining | ||
|
|
||
| integer :: is, ie, js, je, is_coarse, ie_coarse, js_coarse, je_coarse | ||
|
|
||
| call get_fine_array_bounds(bd, is, ie, js, je) | ||
| call get_coarse_array_bounds(coarse_graining%bd, is_coarse, ie_coarse, js_coarse, je_coarse) | ||
| call initialize_coarse_diagnostic_axes(coarse_graining%domain, coarse_graining%nx_coarse, & | ||
| coarse_graining%id_x_coarse, coarse_graining%id_y_coarse, coarse_graining%id_xt_coarse, & | ||
| coarse_graining%id_yt_coarse) | ||
|
|
||
| coarse_graining%id_pfull = id_pfull | ||
| coarse_graining%id_phalf = id_phalf | ||
|
|
||
| call register_coarse_diagnostics(coarse_graining%idiag, Time, & | ||
| coarse_graining%id_xt_coarse, coarse_graining%id_yt_coarse, id_pfull) | ||
| end subroutine fv_coarse_diag_init | ||
|
|
||
| subroutine initialize_coarse_diagnostic_axes(coarse_domain, & | ||
| nx_coarse, id_x_coarse, id_y_coarse, id_xt_coarse, id_yt_coarse) | ||
| type(domain2d), intent(in) :: coarse_domain | ||
| integer, intent(in) :: nx_coarse | ||
| integer, intent(inout) :: id_x_coarse, id_y_coarse, id_xt_coarse, id_yt_coarse | ||
|
|
||
| integer :: i, j | ||
| real, allocatable :: grid_x_coarse(:), grid_y_coarse(:), grid_xt_coarse(:), grid_yt_coarse(:) | ||
|
|
||
| allocate(grid_x_coarse(nx_coarse + 1)) | ||
| allocate(grid_y_coarse(nx_coarse + 1)) | ||
| allocate(grid_xt_coarse(nx_coarse)) | ||
| allocate(grid_yt_coarse(nx_coarse)) | ||
|
|
||
| grid_x_coarse = (/ (i, i=1, nx_coarse + 1) /) | ||
| grid_y_coarse = (/ (j, j=1, nx_coarse + 1) /) | ||
| grid_xt_coarse = (/ (i, i=1, nx_coarse) /) | ||
| grid_yt_coarse = (/ (j, j=1, nx_coarse) /) | ||
|
|
||
| id_x_coarse = diag_axis_init('grid_x_coarse', grid_x_coarse, & | ||
| 'index', 'x', 'x-index of cell corner points', set_name='coarse_grid', & | ||
| Domain2=coarse_domain, tile_count=tile_count) | ||
| id_y_coarse = diag_axis_init('grid_y_coarse', grid_y_coarse, & | ||
| 'index', 'y', 'y-index of cell corner points', set_name='coarse_grid', & | ||
| Domain2=coarse_domain, tile_count=tile_count) | ||
|
|
||
| id_xt_coarse = diag_axis_init('grid_xt_coarse', grid_xt_coarse, & | ||
| 'index', 'x', 'x-index of cell center points', set_name='coarse_grid', & | ||
| Domain2=coarse_domain, tile_count=tile_count) | ||
| id_yt_coarse = diag_axis_init('grid_yt_coarse', grid_yt_coarse, & | ||
| 'index', 'y', 'y-index of cell center points', set_name='coarse_grid', & | ||
| Domain2=coarse_domain, tile_count=tile_count) | ||
| end subroutine initialize_coarse_diagnostic_axes | ||
|
|
||
| subroutine register_coarse_diagnostics(idiag_coarse, Time, id_xt_coarse,& | ||
| id_yt_coarse, id_pfull) | ||
| type(fv_coarse_diag_type), intent(inout) :: idiag_coarse | ||
| type(time_type), intent(in) :: Time | ||
| integer, intent(in) :: id_xt_coarse, id_yt_coarse, id_pfull | ||
|
|
||
| integer :: coarse_axes_t(3) | ||
| real :: missing_value = -1.0e10 ! Following fv_diagnostics.F90 | ||
|
|
||
| coarse_axes_t = (/ id_xt_coarse, id_yt_coarse, id_pfull /) | ||
| idiag_coarse%id_omega_coarse = register_diag_field('dynamics', & | ||
| 'omega_coarse', coarse_axes_t(1:3), Time, & | ||
| 'coarse-grained omega', & | ||
| 'Pa/s', missing_value=missing_value) | ||
| end subroutine register_coarse_diagnostics | ||
|
|
||
| subroutine fv_coarse_diag(Atm, Time) | ||
| type(fv_atmos_type), intent(in), target :: Atm(:) | ||
| type(time_type), intent(in) :: Time | ||
|
|
||
| character(len=256) :: error_message | ||
|
|
||
| if (trim(Atm(tile_count)%coarse_graining%strategy) .eq. MODEL_LEVEL) then | ||
| call fv_coarse_diag_model_levels(Atm, Time) | ||
| endif | ||
| end subroutine fv_coarse_diag | ||
|
|
||
| subroutine fv_coarse_diag_model_levels(Atm, Time) | ||
| type(fv_atmos_type), intent(in), target :: Atm(:) | ||
| type(time_type), intent(in) :: Time | ||
|
|
||
| real, allocatable :: work_3d_coarse(:,:,:) | ||
| integer :: diagnostic_ids_3d(Atm(tile_count)%coarse_graining%idiag%n_3d_diagnostics) | ||
| integer :: is, ie, js, je, is_coarse, ie_coarse, js_coarse, je_coarse, npz | ||
| logical :: used | ||
|
|
||
| call get_diagnostic_ids_3d(Atm(tile_count)%coarse_graining%idiag, diagnostic_ids_3d) | ||
| call get_fine_array_bounds(Atm(tile_count)%bd, is, ie, js, je) | ||
| call get_coarse_array_bounds(Atm(tile_count)%coarse_graining%bd, is_coarse, ie_coarse, js_coarse, je_coarse) | ||
| npz = Atm(tile_count)%npz | ||
|
|
||
| if (any(diagnostic_ids_3d > 0)) then | ||
| allocate(work_3d_coarse(is_coarse:ie_coarse,js_coarse:je_coarse,1:npz)) | ||
| endif | ||
|
|
||
| if (Atm(tile_count)%coarse_graining%idiag%id_omega_coarse > 0) then | ||
| call weighted_block_average( & | ||
| Atm(tile_count)%gridstruct%area(is:ie,js:je), & | ||
| Atm(tile_count)%omga(is:ie,js:je,1:npz), & | ||
| work_3d_coarse) | ||
| used = send_data(Atm(tile_count)%coarse_graining%idiag%id_omega_coarse, work_3d_coarse, Time) | ||
| endif | ||
| end subroutine fv_coarse_diag_model_levels | ||
|
|
||
| subroutine get_diagnostic_ids_3d(idiag_coarse, diagnostic_ids_3d) | ||
| type(fv_coarse_diag_type), intent(in) :: idiag_coarse | ||
| integer, intent(out) :: diagnostic_ids_3d(idiag_coarse%n_3d_diagnostics) | ||
|
|
||
| diagnostic_ids_3d = (/ idiag_coarse%id_omega_coarse /) | ||
| end subroutine | ||
|
|
||
| end module coarse_grained_diagnostics_mod |
Oops, something went wrong.