Skip to content

Handle 4D variables in getsfcensmeanp.x utility. #4

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
aerorahul merged 2 commits into from
Jul 18, 2022
Merged

Handle 4D variables in getsfcensmeanp.x utility. #4

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
209b681
cldfra (a 4D variable [time, pfull, nlat, nlon]) is now in sfcfHHH file.
aerorahul Jul 16, 2022
f1dc784
Add code to process 4D variables in netcdf sfcfHHH
RussTreadon-NOAA Jul 18, 2022
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
35 changes: 26 additions & 9 deletions src/EnKF/gfs/src/getsfcensmeanp.fd/getsfcensmeanp.f90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,14 +42,15 @@ program getsfcensmeanp
logical:: nemsio, sfcio, ncio

type(Dataset) :: dset,dseto
type(Dimension) :: londim,latdim
type(Dimension) :: londim,latdim,levdim
real(4), allocatable, dimension(:,:) :: values_2d, values_2d_avg
real(4), allocatable, dimension(:,:,:):: values_3d, values_3d_avg

character*500 filenamein,filenameout,datapath,fileprefix
character*3 charnanal
integer lunin,lunout,iret,nanals,k
integer mype,mype1,npe,orig_group, new_group, new_comm
integer nrec, lonb, latb, n, npts, nvar
integer nrec, lonb, latb, levs, n, npts, nvar
integer,dimension(7):: idate
integer,dimension(:),allocatable:: new_group_members
real(8) rnanals
Expand Down Expand Up @@ -308,7 +309,9 @@ program getsfcensmeanp
elseif (ncio) then
londim = get_dim(dset,'grid_xt'); lonb = londim%len
latdim = get_dim(dset,'grid_yt'); latb = latdim%len
levdim = get_dim(dset,'pfull'); levs = levdim%len
allocate(values_2d_avg(lonb,latb))
allocate(values_3d_avg(lonb,latb,levs))
if (mype == 0) then
dseto = create_dataset(filenameout, dset, copy_vardata=.true.)
print *,'opened netcdf file ',trim(filenameout)
Expand All @@ -328,16 +331,30 @@ program getsfcensmeanp
trim(dset%variables(nvar)%name) == 'orog') then
cycle
endif
call read_vardata(dset,trim(dset%variables(nvar)%name),values_2d)
call mpi_allreduce(values_2d,values_2d_avg,lonb*latb,mpi_real4,mpi_sum,new_comm,iret)
values_2d_avg = values_2d_avg * rnanals
if (mype == 0) then
print *,'writing ens mean ',trim(dset%variables(nvar)%name)
call write_vardata(dseto,trim(dset%variables(nvar)%name),values_2d_avg)
if (dset%variables(nvar)%ndims == 3) then
call read_vardata(dset,trim(dset%variables(nvar)%name),values_2d)
call mpi_allreduce(values_2d,values_2d_avg,lonb*latb,mpi_real4,mpi_sum,new_comm,iret)
values_2d_avg = values_2d_avg * rnanals
if (mype == 0) then
print *,'writing ens mean ',trim(dset%variables(nvar)%name)
call write_vardata(dseto,trim(dset%variables(nvar)%name),values_2d_avg)
endif
elseif (dset%variables(nvar)%ndims == 4) then
call read_vardata(dset,trim(dset%variables(nvar)%name),values_3d)
call mpi_allreduce(values_3d,values_3d_avg,lonb*latb*levs,mpi_real4,mpi_sum,new_comm,iret)
values_3d_avg = values_3d_avg * rnanals
if (mype == 0) then
print *,'writing ens mean ',trim(dset%variables(nvar)%name)
call write_vardata(dseto,trim(dset%variables(nvar)%name),values_3d_avg)
endif
else
write(6,*)'***ERROR*** invalid ndims= ',dset%variables(nvar)%ndims,&
' for variable ',trim(dset%variables(nvar)%name)
call MPI_Abort(MPI_COMM_WORLD,98,iret)
endif
enddo
if (mype == 0) call close_dataset(dseto)
deallocate(values_2d,values_2d_avg)
deallocate(values_2d,values_2d_avg,values_3d,values_3d_avg)
endif
! Jump here if more mpi processors than files to process
else
Expand Down