module comspe

! Spectral space arrays

use shr_kind_mod, only: r8 => shr_kind_r8
use pmgrid,       only: plev, plat
use pspect,       only: pmmax, pspt

implicit none

real(r8), dimension(:,:), allocatable :: vz   ! Vorticity spectral coefficients
real(r8), dimension(:,:), allocatable :: d    ! Divergence spectral coefficients
real(r8), dimension(:,:), allocatable :: t    ! Temperature spectral coefficients
real(r8), dimension(:), allocatable :: alps ! Log-pressure spectral coefficients

#if ( defined SPMD )
integer :: maxm           = huge(1)  ! max number of Fourier wavenumbers per MPI task
integer :: lpspt          = huge(1)  ! number of local spectral coefficients
integer, dimension(:), allocatable   :: numm
                                    ! number of Fourier wavenumbers owned per task
integer, dimension(:,:), allocatable :: locm, locrm
                                    ! assignment of wavenumbers to MPI tasks
integer, dimension(:), allocatable   :: lnstart 
                                    ! Starting indices for local spectral arrays (real)
#else
integer :: numm(0:0)      = pmmax
integer :: maxm           = pmmax
integer :: lpspt          = pspt
integer :: locm(1:pmmax, 0:0) = huge(1)
integer :: locrm(1:2*pmmax, 0:0) = huge(1)
integer :: lnstart(1:pmmax) = huge(1)
#endif

integer :: nstart(pmmax) = huge(1)   ! Starting indices for spectral arrays (real)
integer :: nlen(pmmax)   = huge(1)   ! Length vectors for spectral arrays

real(r8), dimension(:,:), allocatable :: alp  ! Legendre polynomials (pspt,plat/2)
real(r8), dimension(:,:), allocatable :: dalp ! Legendre polynomial derivatives (pspt,plat/2)

real(r8), dimension(:,:), allocatable :: lalp  ! local Legendre polynomials
real(r8), dimension(:,:), allocatable :: ldalp ! local Legendre polynomial derivatives

end module comspe