Skip to content

Explicit domain decomposition of horizontal axes #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Hallberg-NOAA merged 1 commit into from
Mar 26, 2021
Merged

Explicit domain decomposition of horizontal axes #5

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 15 additions & 1 deletion config_src/infra/FMS2/MOM_io_infra.F90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ module MOM_io_infra
use fms2_io_mod, only : variable_att_exists, get_variable_attribute, get_variable_num_dimensions
use fms2_io_mod, only : get_variable_dimension_names, is_dimension_registered, get_dimension_size
use fms2_io_mod, only : is_dimension_unlimited, register_axis, unlimited
use fms2_io_mod, only : get_global_io_domain_indices

use fms_mod, only : write_version_number, open_namelist_file, check_nml_error
use fms_io_mod, only : file_exist, field_exist, field_size, read_data
Expand Down Expand Up @@ -132,6 +133,7 @@ module MOM_io_infra
character(len=256) :: name !< The name of this axis in the files.
type(mpp_axistype) :: AT !< The FMS1 axis-type that this type wraps
real, allocatable, dimension(:) :: ax_data !< The values of the data on the axis.
logical :: domain_decomposed = .false. !< True if axis is domain-decomposed
end type axistype

!> For now, these module-variables are hard-coded to exercise the new FMS2 interfaces.
Expand Down Expand Up @@ -1726,8 +1728,16 @@ subroutine MOM_write_axis(IO_handle, axis)
type(file_type), intent(in) :: IO_handle !< Handle for a file that is open for writing
type(axistype), intent(in) :: axis !< An axis type variable with information to write

integer :: is, ie

if (IO_handle%FMS2_file) then
call write_data(IO_handle%fileobj, trim(axis%name), axis%ax_data)
if (axis%domain_decomposed) then
! FMS2 does not domain-decompose 1d arrays, so we explicitly slice it
call get_global_io_domain_indices(IO_handle%fileobj, trim(axis%name), is, ie)
call write_data(IO_handle%fileobj, trim(axis%name), axis%ax_data(is:ie))
else
call write_data(IO_handle%fileobj, trim(axis%name), axis%ax_data)
endif
else
call mpp_write(IO_handle%unit, axis%AT)
endif
Expand Down Expand Up @@ -1781,6 +1791,10 @@ subroutine write_metadata_axis(IO_handle, axis, name, units, longname, cartesian
if ((index(cart, "T") == 1) .or. (index(cart, "t") == 1)) is_t = .true.
endif

! For now, we assume that all horizontal axes are domain-decomposed.
if (is_x .or. is_y) &
axis%domain_decomposed = .true.

if (is_x) then
if (present(edge_axis)) then ; if (edge_axis) position = EAST_FACE ; endif
call register_axis(IO_handle%fileobj, trim(name), 'x', domain_position=position)
Expand Down