getVariable.f

!!!@PROCESS NOEXTCHK
subroutine getvariable(fileName,DateStr,dh,VarName,VarBuff,IM,JSTA_2L,JEND_2U,LM,IM1,JS,JE,LM1)


!     SUBPROGRAM DOCUMENTATION BLOCK
!                .      .    .
! SUBPROGRAM:    getVariable    Read data from WRF output
!   PRGRMMR: MIKE BALDWIN    ORG: NSSL/SPC   DATE: 2002-04-08
!
! ABSTRACT:  THIS ROUTINE READS DATA FROM A WRF OUTPUT FILE
!   USING WRF I/O API.
!   .
!
! PROGRAM HISTORY LOG:
!
! USAGE:    CALL getVariable(fileName,DateStr,dh,VarName,VarBuff,IM,JSTA_2L,JEND_2U,LM,IM1,JS,JE,LM1)
!
!   INPUT ARGUMENT LIST:
!     fileName : Character(len=256) : name of WRF output file
!     DateStr  : Character(len=19)  : date/time of requested variable
!     dh :  integer                 : data handle
!     VarName :  Character(len=31)  : variable name
!     IM :  integer  : X dimension of data array
!     JSTA_2L :  integer  : start Y dimension of data array
!     JEND_2U :  integer  : end Y dimension of data array
!     LM :  integer  : Z dimension of data array
!     IM1 :  integer  : amount of data pulled in X dimension 
!     JS :  integer  : start Y dimension of amount of data array pulled
!     JE :  integer  : end Y dimension of amount of data array pulled
!     LM1 :  integer  : amount of data pulled in Z dimension
!
!   data is flipped in the Z dimension from what is originally given
!   the code requires the Z dimension to increase with pressure
!
!   OUTPUT ARGUMENT LIST:
!     VarBuff : real(IM,JSTA_2L:JEND_2U,LM) : requested data array
!
!   OUTPUT FILES:
!     NONE
!
!   SUBPROGRAMS CALLED:
!     UTILITIES:
!       NONE
!     LIBRARY:
!       WRF I/O API
!       NETCDF

 ! This subroutine reads the values of the variable named VarName into the buffer
 ! VarBuff. VarBuff is filled with data only for I=1,IM1 and for J=JS,JE
 ! and for L=1,Lm1, presumably this will be
 ! the portion of VarBuff that is needed for this task.
 !  use mpi
   use wrf_io_flags_mod, only: wrf_real, wrf_real8
!- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
   implicit none

   include "mpif.h"

!
   character(len=256) ,intent(in) :: filename
   character(len=19) ,intent(in) :: datestr
   integer ,intent(in) :: dh
   character(*) ,intent(in) :: varname
   real,intent(out) :: varbuff(im,jsta_2l:jend_2u,lm)
   integer,intent(in) :: im,lm,jsta_2l,jend_2u
   integer,intent(in) :: im1,lm1,js,je
   integer :: ndim
   integer :: wrftype,i,j,l,ll
   integer, dimension(4) :: start_index, end_index
   character (len= 4) :: staggering
   character (len= 3) :: ordering
   character (len=80), dimension(3) :: dimnames
   real, allocatable, dimension(:,:,:,:) :: data
   integer :: ierr,size,mype,idsize,ier
   character(len=132) :: stagger
   real spval

!    call set_wrf_debug_level ( 1 )
   call mpi_comm_rank(mpi_comm_world,mype,ier)
   start_index = 1
   end_index = 1
!   print*,'SPVAL in getVariable = ',SPVAL
   call ext_ncd_get_var_info(dh,trim(varname),ndim,ordering,stagger,start_index,end_index,wrftype,ierr)
   IF ( ierr /= 0 ) THEN
     write(*,*)'Error: ',ierr,trim(varname),' not found in ',filename
     varbuff=0.  
     return
   ENDIF
   allocate(data (end_index(1), end_index(2), end_index(3), 1))
   if( wrftype /= wrf_real .AND. wrftype /= wrf_real8 ) then !Ignore if not a real variable
     write(*,*) 'Error: Not a real variable',wrftype
     return
   endif
!  write(*,'(A9,1x,I1,3(1x,I3),1x,A,1x,A)')&
!           trim(VarName), ndim, end_index(1), end_index(2), end_index(3), &
!           trim(ordering), trim(DateStr)
!   allocate(data (end_index(1), end_index(2), end_index(3), 1))
!   call ext_ncd_read_field(dh,DateStr,TRIM(VarName),data,WrfType,0,0,0,ordering,&
! CHANGE WrfType to WRF_REAL BECAUSE THIS TELLS WRF IO API TO CONVERT TO REAL
          print  *,' GWVX XT_NCD GET FIELD',size(data), size(varbuff),mype
     idsize=size(data)
   if(mype == 0) then
   call ext_ncd_read_field(dh,datestr,trim(varname),data,wrftype,0,0,0,ordering,&
                             staggering, dimnames , &
                             start_index,end_index, & !dom 
                             start_index,end_index, & !mem
                             start_index,end_index, & !pat
                             ierr)
    endif
     call mpi_bcast(data,idsize,mpi_real,0,mpi_comm_world,ierr) 
   IF ( ierr /= 0 ) THEN
     write(*,*)'Error reading ',varname,' from ',filename
     write(*,*)' ndim = ', ndim
     write(*,*)' end_index(1) ',end_index(1)
     write(*,*)' end_index(2) ',end_index(2)
     write(*,*)' end_index(3) ',end_index(3)
     varbuff = 0.0
     return
   ENDIF
   if (im1>end_index(1)) write(*,*) 'Err:',varname,' IM1=',im1,&
                ' but data dim=',end_index(1)
   if (je>end_index(2)) write(*,*) 'Err:',varname,' JE=',je,&
                ' but data dim=',end_index(2)
   if (lm1>end_index(3)) write(*,*) 'Err:',varname,' LM1=',lm1,&
                ' but data dim=',end_index(3)
   if (ndim>3) then
     write(*,*) 'Error: ndim = ',ndim
   endif 
   do l=1,lm1
     ll=lm1-l+1  ! flip the z axis not sure about soil
     do i=1,im1
      do j=js,je
       varbuff(i,j,l)=data(i,j,ll,1)
      enddo
     enddo
!     write(*,*) Varname,' L ',l,': = ',data(1,1,ll,1)
   enddo
   deallocate(data)
   return

end subroutine getvariable