hecmw_precond_ML_nn.f90

Go to the documentation of this file.

!-------------------------------------------------------------------------------
! Copyright (c) 2019 FrontISTR Commons
! This software is released under the MIT License, see LICENSE.txt
!-------------------------------------------------------------------------------
 
!C
!C***
!C*** module hecmw_precond_ML_nn
!C***
!C
module hecmw_precond_ml_nn
  use  hecmw_util
 
  private
 
  public:: hecmw_precond_ml_nn_setup
  public:: hecmw_precond_ml_nn_apply
  public:: hecmw_precond_ml_nn_clear
 
  integer(kind=kint), save :: id
 
  logical, save :: INITIALIZED = .false.
 
contains
 
  subroutine hecmw_precond_ml_nn_setup(hecMAT, hecMESH, sym)
    use hecmw_matrix_misc
    use hecmw_mat_id
    implicit none
    type(hecmwst_matrix), intent(inout) :: hecmat
    type(hecmwst_local_mesh), intent(in) :: hecmesh
    integer(kind=kint), intent(in) :: sym
    integer(kind=kint) :: ierr
    integer(kind=kint), save :: n_recycle = 0
    if (initialized) then
      if (hecmat%Iarray(98) == 1) then      ! need symbolic and numerical setup
        call hecmw_precond_ml_nn_clear()
      else if (hecmat%Iarray(97) == 1) then ! need numerical setup only
        call hecmw_precond_ml_nn_clear()
      else                                  ! no need to setup or skip setup
        call hecmw_mat_id_clear(id)
        call hecmw_mat_id_set(hecmat, hecmesh, id)
        return
      endif
    endif
    call hecmw_mat_id_set(hecmat, hecmesh, id)
    call hecmw_ml_wrapper_setup(id, sym, hecmat%NDOF, ierr)
    initialized = .true.
    hecmat%Iarray(98) = 0 ! symbolic setup done
    hecmat%Iarray(97) = 0 ! numerical setup done
    n_recycle = 0
  end subroutine hecmw_precond_ml_nn_setup
 
  subroutine hecmw_precond_ml_nn_apply(WW)
    implicit none
    real(kind=kreal), intent(inout) :: ww(:)
    integer(kind=kint) :: ierr
    call hecmw_ml_wrapper_apply(id, ww, ierr)
  end subroutine hecmw_precond_ml_nn_apply
 
  subroutine hecmw_precond_ml_nn_clear()
    use hecmw_mat_id
    implicit none
    integer(kind=kint) :: ierr
    call hecmw_ml_wrapper_clear(id, ierr)
    call hecmw_mat_id_clear(id)
    initialized = .false.
  end subroutine hecmw_precond_ml_nn_clear
 
end module     hecmw_precond_ml_nn