NAFPack_meshgrid_complex_3D.f90 Source File

submodule(NAFPack_meshgrid) NAFPack_meshgrid_complex_3D

    implicit none(type, external)

    interface
        module subroutine compute_meshgrid_cmplx_3D_sp( &
            x_vector, y_vector, z_vector, &
            X, Y, Z, &
            Nx, Ny, Nz, &
            loop_method)
            complex(sp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
            complex(sp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
            integer(isp), intent(in) :: Nx, Ny, Nz
            type(LoopMethod), intent(in) :: loop_method
        end subroutine compute_meshgrid_cmplx_3D_sp
    end interface

    interface
        module subroutine compute_meshgrid_cmplx_3D_dp( &
            x_vector, y_vector, z_vector, &
            X, Y, Z, &
            Nx, Ny, Nz, &
            loop_method)
            complex(dp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
            complex(dp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
            integer(isp), intent(in) :: Nx, Ny, Nz
            type(LoopMethod), intent(in) :: loop_method
        end subroutine compute_meshgrid_cmplx_3D_dp
    end interface

    interface
        module subroutine compute_meshgrid_cmplx_3D_qp( &
            x_vector, y_vector, z_vector, &
            X, Y, Z, &
            Nx, Ny, Nz, &
            loop_method)
            complex(qp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
            complex(qp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
            integer(isp), intent(in) :: Nx, Ny, Nz
            type(LoopMethod), intent(in) :: loop_method
        end subroutine compute_meshgrid_cmplx_3D_qp
    end interface

contains

    module subroutine meshgrid_cmplx_sp_3D( &
        x_vector, y_vector, z_vector, &
        X, Y, Z, &
        indexing, &
        strict_mode, &
        loop_method)
        complex(sp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
        complex(sp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
        type(meshgrid_indexing), optional, intent(in) :: indexing
        logical, optional, intent(in) :: strict_mode
        type(LoopMethod), optional, intent(in) :: loop_method
        type(LoopMethod) :: loop_method_used
        logical :: use_ij_indexing, use_xy_indexing
        integer :: Nx, Ny, Nz

        call check_indexing(indexing, strict_mode, use_ij_indexing, use_xy_indexing)

        if (present(loop_method)) then
            loop_method_used = check_loop_method(loop_method)
        else
            loop_method_used = default_loop_method
        end if

        Nx = size(x_vector, 1)
        Ny = size(y_vector, 1)
        Nz = size(z_vector, 1)
        if (use_ij_indexing) then
            call compute_meshgrid_cmplx_3D_sp(x_vector, y_vector, z_vector, X, Y, Z, Nx, Ny, Nz, loop_method_used)
        else if (use_xy_indexing) then
            call compute_meshgrid_cmplx_3D_sp(y_vector, x_vector, z_vector, Y, X, Z, Ny, Nx, Nz, loop_method_used)
        end if

    end subroutine meshgrid_cmplx_sp_3D

    module subroutine meshgrid_cmplx_dp_3D( &
        x_vector, y_vector, z_vector, &
        X, Y, Z, &
        indexing, &
        strict_mode, &
        loop_method)
        complex(dp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
        complex(dp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
        type(meshgrid_indexing), optional, intent(in) :: indexing
        logical, optional, intent(in) :: strict_mode
        type(LoopMethod), optional, intent(in) :: loop_method
        type(LoopMethod) :: loop_method_used
        logical :: use_ij_indexing, use_xy_indexing
        integer :: Nx, Ny, Nz

        call check_indexing(indexing, strict_mode, use_ij_indexing, use_xy_indexing)

        if (present(loop_method)) then
            loop_method_used = check_loop_method(loop_method)
        else
            loop_method_used = default_loop_method
        end if

        Nx = size(x_vector, 1)
        Ny = size(y_vector, 1)
        Nz = size(z_vector, 1)
       if (use_ij_indexing) then
            call compute_meshgrid_cmplx_3D_dp(x_vector, y_vector, z_vector, X, Y, Z, Nx, Ny, Nz, loop_method_used)
        else if (use_xy_indexing) then
            call compute_meshgrid_cmplx_3D_dp(y_vector, x_vector, z_vector, Y, X, Z, Ny, Nx, Nz, loop_method_used)
        end if

    end subroutine meshgrid_cmplx_dp_3D

    module subroutine meshgrid_cmplx_qp_3D( &
        x_vector, y_vector, z_vector, &
        X, Y, Z, &
        indexing, &
        strict_mode, &
        loop_method)
        complex(qp), dimension(:), intent(in) :: x_vector, y_vector, z_vector
        complex(qp), dimension(:, :, :), allocatable, intent(out) :: X, Y, Z
        type(meshgrid_indexing), optional, intent(in) :: indexing
        logical, optional, intent(in) :: strict_mode
        type(LoopMethod), optional, intent(in) :: loop_method
        type(LoopMethod) :: loop_method_used
        logical :: use_ij_indexing, use_xy_indexing
        integer :: Nx, Ny, Nz

        call check_indexing(indexing, strict_mode, use_ij_indexing, use_xy_indexing)

        if (present(loop_method)) then
            loop_method_used = check_loop_method(loop_method)
        else
            loop_method_used = default_loop_method
        end if

        Nx = size(x_vector, 1)
        Ny = size(y_vector, 1)
        Nz = size(z_vector, 1)
        if (use_ij_indexing) then
            call compute_meshgrid_cmplx_3D_qp(x_vector, y_vector, z_vector, X, Y, Z, Nx, Ny, Nz, loop_method_used)
        else if (use_xy_indexing) then
            call compute_meshgrid_cmplx_3D_qp(y_vector, x_vector, z_vector, Y, X, Z, Ny, Nx, Nz, loop_method_used)
        end if

    end subroutine meshgrid_cmplx_qp_3D

end submodule NAFPack_meshgrid_complex_3D