!---------------------------------------------------------------------- ! Copyright (c) 2016 Shin-ichi Takehiro. All rights reserved. !---------------------------------------------------------------------- ! !表題 ua_mpi_module_svpack テストプログラム :: スペクトル関数のテスト ! !履歴 2016/08/15 竹広真一 wa_mpi_spectrum_mpi_module_supack_test.f90 より改造 ! program ua_mpi_module_svpack_spectrum_test use dc_message, only : MessageNotify use dc_test, only : AssertEqual use ua_mpi_module_svpack use mpi implicit none integer, parameter :: im=64, jm=32, nm=21, km=2 real(8), allocatable :: ua_data(:,:) real(8), allocatable :: xyb_data(:,:,:) real(8), dimension(0:nm,-nm:nm,km) :: nma_data real(8), dimension(0:nm,km) :: na_data real(8), dimension(km) :: a_data ! 判定誤差設定 integer, parameter :: check_digits = 10 integer, parameter :: ignore = -11 integer :: n,m integer :: iproc, np, ierr !---------------- MPI スタート --------------------- call MPI_INIT(IERR) call MPI_COMM_RANK(MPI_COMM_WORLD,IPROC,IERR) call MPI_COMM_SIZE(MPI_COMM_WORLD,NP,IERR) call ua_mpi_Initial( nm, im, jm, km ) allocate(ua_data(nc,km)) allocate(xyb_data(0:im-1,1:jm,kc)) call MessageNotify('M','ua_spectrum_mpi_module_svpack_test',& 'ua_mpi_module_svpack spectrum functions tests.') ! 2Y_1^0+3Y_2^1 if ( kp_kl(1) /= 0 ) & xyb_data(:,:,kp_kl(1)) = 2*sqrt(3.0D0)*sin(xy_Lat) & + 3*sqrt(5.0D0/6)*3.0*sin(xy_Lat)*cos(xy_Lat) * cos(xy_Lon)! ! 3*Y_1^0 + 2Y_1^1 + 2Y_2^0 +Y_2^-2 if ( kp_kl(2) /= 0 ) & xyb_data(:,:,kp_kl(2)) = 3*sqrt(3.0D0)*sin(xy_Lat) & + 2* sqrt(3.0D0/2)*cos(xy_Lat)*cos(xy_Lon) & + 2*sqrt(5.0D0)*(3.0/2*sin(xy_Lat)**2-1/2.0) & - sqrt(5.0D0/24)*3.0*cos(xy_Lat)**2*sin(2*xy_Lon) ua_data = ua_xyb(xyb_data) !--------- Energy spectrum --------- nma_data=ua_spectrum_VMiss do n=0,nm do m=-n,n nma_data(n,m,:) = 0.0D0 enddo enddo nma_data(1,0,1) = 4.0D0 nma_data(2,1,1) = 2*3*(3.0D0/2)**2/2 nma_data(2,-1,1)= nma_data(2,1,1) nma_data(1,0,2) = 9.0D0 nma_data(1,1,2) = 1*2*(2/2)**2/2 nma_data(1,-1,2)= nma_data(1,1,2) nma_data(2,0,2) = 12.0D0 nma_data(2,-2,2)= 2*3*(1.0D0/2)**2/2 nma_data(2,2,2) = nma_data(2,-2,2) call AssertEqual(& message='nma_EnergyFromStreamfunc_ua', & answer = nma_EnergyFromStreamfunc_ua(ua_data), & check = nma_data, & significant_digits = check_digits, ignore_digits = ignore & ) na_data = 0.0D0 na_data(1,1) = 4.0D0 na_data(2,1) = 2 * 3 *(3.0D0/2)**2 na_data(1,2) = 9.0D0 + 1*2*(2/2)**2 na_data(2,2) = 12.0D0 + 2*3*(1.0D0/2)**2 call AssertEqual(& message='na_EnergyFromStreamfunc_ua', & answer = na_EnergyFromStreamfunc_ua(ua_data), & check = na_data, & significant_digits = check_digits, ignore_digits = ignore & ) nma_Data = nma_EnergyFromStreamfunc_ua(ua_data) a_data = 0.0D0 do n=0,nm do m=-n,n a_data = a_data + nma_Data(n,m,:) enddo enddo call AssertEqual(& message='total energy by nm-spectrum', & answer = a_AvrLonLat_xya( & 0.5*(xyb_GradLon_ua(ua_data)**2+xyb_GradLat_ua(ua_data)**2) ),& check = a_data(k1:k2), & significant_digits = check_digits, ignore_digits = ignore & ) !!$ a_data = a_AvrLonLat_xya( 0.5*(xyb_GradLon_ua(ua_data)**2+xyb_GradLat_ua(ua_data)**2) ) !!$ call AssertEqual(& !!$ message='total energy with n-spectrum', & !!$ answer = sum(na_EnergyFromStreamfunc_ua(ua_data),1), & !!$ check = a_data, & !!$ significant_digits = check_digits, ignore_digits = ignore & !!$ ) !--------- Enstrophy spectrum --------- nma_data=ua_spectrum_VMiss do n=0,nm do m=-n,n nma_data(n,m,:) = 0.0D0 enddo enddo nma_data(1,0,1) = 8.0D0 nma_data(2,1,1) = 2**2*3**2*(3.0D0/2)**2/2 nma_data(2,-1,1) = nma_data(2,1,1) nma_data(1,0,2) = 18.0D0 nma_data(1,1,2) = 1**2 * 2**2 * (2/2)**2 / 2 nma_data(1,-1,2) = nma_data(1,1,2) nma_data(2,0,2) = 72.0D0 nma_data(2,-2,2) = 2**2 * 3**2 * (1.0D0/2)**2 /2 nma_data(2,2,2) = nma_data(2,-2,2) call AssertEqual(& message='nma_EnstrophyFromStreamfunc_ua', & answer = nma_EnstrophyFromStreamfunc_ua(ua_data), & check = nma_data, & significant_digits = check_digits, ignore_digits = ignore & ) na_data = 0.0 na_data(1,1) = 8.0D0 ; na_data(2,1) = 2**2*3**2*(3.0D0/2)**2 na_data(1,2) = 18.0D0 + 1**2 * 2**2 * (2/2)**2 na_data(2,2) = 72.0D0 + 2**2 * 3**2 * (1.0D0/2)**2 call AssertEqual(& message='na_EnstrophyFromStreamfunc_ua', & answer = na_EnstrophyFromStreamfunc_ua(ua_data), & check = na_data, & significant_digits = check_digits, ignore_digits = ignore & ) nma_Data = nma_EnstrophyFromStreamfunc_ua(ua_data) a_data = 0.0D0 do n=0,nm do m=-n,n a_data = a_data + nma_Data(n,m,:) enddo enddo call AssertEqual(& message='total enstrophy by nm-spectrum', & answer = a_AvrLonLat_xya( 0.5*(xyb_ua(ua_Lapla_ua(ua_data))**2)),& check = a_data(k1:k2), & significant_digits = check_digits, ignore_digits = ignore & ) !!$ a_data = a_AvrLonLat_xya( 0.5*(xyb_ua(ua_Lapla_ua(ua_data))**2)) !!$ !!$ call AssertEqual(& !!$ message='total energy with n-spectrum', & !!$ answer = sum(na_EnstrophyFromStreamfunc_ua(ua_data),1), & !!$ check = a_data, & !!$ significant_digits = check_digits, ignore_digits = ignore & !!$ ) call MessageNotify('M','ua_spectrum_mpi_module_svpack_test', & 'ua_mpimodule_svpack spectrum function tests succeeded!') !------ MPIの終了 ------ call MPI_FINALIZE(IERR) end program ua_mpi_module_svpack_spectrum_test