sll_m_preconditioner_jacobi.F90

Go to the documentation of this file.

 
 
module sll_m_preconditioner_jacobi
  !+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#include "sll_assert.h"
#include "sll_memory.h"
#include "sll_working_precision.h"
 
  use sll_m_linear_operator_abstract, only : &
       sll_t_linear_operator_abstract
 
  use sll_m_linear_solver_abstract, only : &
       sll_t_linear_solver_abstract
 
  implicit none
 
  public :: &
       sll_t_preconditioner_jacobi
 
  private
  !+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
  type, extends(sll_t_linear_solver_abstract) :: sll_t_preconditioner_jacobi
     sll_real64, allocatable :: diag(:) 
 
   contains
 
     procedure :: create => create_3d_trafo
     procedure :: free => free_3d_trafo
     procedure :: solve_real => solve_3d_trafo
     procedure :: set_verbose
     procedure :: print_info
     procedure :: read_from_file
 
  end type sll_t_preconditioner_jacobi
 
contains
 
 
  subroutine create_3d_trafo( self, linop)
    class(sll_t_preconditioner_jacobi), intent( inout ) :: self
    class(sll_t_linear_operator_abstract)    :: linop
    !local variables
    sll_int32 :: i
    sll_real64, allocatable :: scratch1(:), scratch2(:)
            
    allocate( self%diag(1:linop%n_rows) )
    allocate( scratch1(1:linop%n_rows) )
    allocate( scratch2(1:linop%n_rows) )
    
    do i = 1, linop%n_rows
       scratch1 = 0._f64
       scratch1(i) = 1._f64
       call linop%dot( scratch1, scratch2 )
       self%diag(i) = scratch2(i)
    end do
    
  end subroutine create_3d_trafo
 
  subroutine solve_3d_trafo(self, rhs, unknown)
    class(sll_t_preconditioner_jacobi), intent( inout ) :: self
    sll_real64, intent(in   ) :: rhs(:) 
    sll_real64, intent(  out) :: unknown(:) 
     
    unknown = rhs/self%diag
 
  end subroutine solve_3d_trafo
 
  subroutine set_verbose(self, verbose)
    class(sll_t_preconditioner_jacobi), intent( inout ) :: self
    logical, intent( in ) :: verbose
 
    self%verbose = verbose
  end subroutine set_verbose
 
  subroutine print_info(self)
    class(sll_t_preconditioner_jacobi), intent( in ) :: self
 
  end subroutine print_info
 
  subroutine read_from_file(self, filename)
    class(sll_t_preconditioner_jacobi), intent( inout ) :: self
    character(len=*), intent( in ) :: filename
    
  end subroutine read_from_file
 
 
  subroutine free_3d_trafo( self )
    class(sll_t_preconditioner_jacobi), intent(inout) :: self
 
  end subroutine free_3d_trafo
 
 
end module sll_m_preconditioner_jacobi