85 lines
2.1 KiB
Fortran
85 lines
2.1 KiB
Fortran
|
program test_sharp
|
||
|
|
use mpi
|
||
|
|
use sharp
|
||
|
|
use iso_c_binding, only : c_ptr, c_double
|
||
|
|
implicit none
|
||
|
|
|
||
|
|
integer, parameter :: lmax = 2, nside = 2
|
||
|
|
type(sharp_alm_info) :: alm_info
|
||
|
|
type(sharp_geom_info) :: geom_info
|
||
|
|
|
||
|
|
real(c_double), dimension(0:(lmax + 1)**2 - 1, 1:1) :: alm
|
||
|
|
real(c_double), dimension(0:12*nside**2 - 1, 1:1) :: map
|
||
|
|
|
||
|
|
integer(c_int), dimension(1:lmax + 1) :: ms
|
||
|
|
integer(c_int), dimension(1:4 * nside - 1) :: rings
|
||
|
|
integer(c_int) :: nm, m, nrings, iring
|
||
|
|
integer :: nodecount, rank, ierr
|
||
|
|
|
||
|
|
call MPI_Init(ierr)
|
||
|
|
call MPI_Comm_size(MPI_COMM_WORLD, nodecount, ierr)
|
||
|
|
call MPI_Comm_rank(MPI_COMM_WORLD, rank, ierr)
|
||
|
|
|
||
|
|
nm = 0
|
||
|
|
do m = rank, lmax, nodecount
|
||
|
|
nm = nm + 1
|
||
|
|
ms(nm) = m
|
||
|
|
end do
|
||
|
|
|
||
|
|
nrings = 0
|
||
|
|
do iring = rank + 1, 4 * nside - 1, nodecount
|
||
|
|
nrings = nrings + 1
|
||
|
|
rings(nrings) = iring
|
||
|
|
end do
|
||
|
|
|
||
|
|
alm = 0
|
||
|
|
map = 0
|
||
|
|
if (rank == 0) then
|
||
|
|
alm(0, 1) = 1
|
||
|
|
end if
|
||
|
|
|
||
|
|
print *, ms(1:nm)
|
||
|
|
call sharp_make_mmajor_real_packed_alm_info(lmax, ms=ms(1:nm), alm_info=alm_info)
|
||
|
|
print *, 'alm_info%n_local', alm_info%n_local
|
||
|
|
call sharp_make_healpix_geom_info(nside, rings=rings(1:nrings), geom_info=geom_info)
|
||
|
|
print *, 'geom_info%n_local', geom_info%n_local
|
||
|
|
print *, 'execute'
|
||
|
|
call sharp_execute(SHARP_Y, 0, 1, alm, alm_info, map, geom_info, comm=MPI_COMM_WORLD)
|
||
|
|
|
||
|
|
print *, alm
|
||
|
|
print *, map
|
||
|
|
|
||
|
|
call sharp_destroy_alm_info(alm_info)
|
||
|
|
call sharp_destroy_geom_info(geom_info)
|
||
|
|
print *, 'DONE'
|
||
|
|
call MPI_Finalize(ierr)
|
||
|
|
|
||
|
|
print *, 'LEGENDRE TRANSFORMS'
|
||
|
|
|
||
|
|
call test_legendre_transforms()
|
||
|
|
|
||
|
|
contains
|
||
|
|
subroutine test_legendre_transforms()
|
||
|
|
integer, parameter :: lmax = 20, nx=10
|
||
|
|
real(c_double) :: bl(0:lmax)
|
||
|
|
real(c_double) :: x(nx), out(nx)
|
||
|
|
real(c_float) :: out_s(nx)
|
||
|
|
!--
|
||
|
|
integer :: l, i
|
||
|
|
|
||
|
|
do l = 0, lmax
|
||
|
|
bl(l) = 1.0 / real(l + 1, c_double)
|
||
|
|
end do
|
||
|
|
do i = 1, nx
|
||
|
|
x(i) = 1 / real(i, c_double)
|
||
|
|
end do
|
||
|
|
out = 0
|
||
|
|
call sharp_legendre_transform(bl, x, out)
|
||
|
|
print *, out
|
||
|
|
call sharp_legendre_transform(real(bl, c_float), real(x, c_float), out_s)
|
||
|
|
print *, out_s
|
||
|
|
end subroutine test_legendre_transforms
|
||
|
|
|
||
|
|
|
||
|
|
end program test_sharp
|