!  
!   
!                             AMG4PSBLAS version 1.0
!    Algebraic Multigrid Package
!               based on PSBLAS (Parallel Sparse BLAS version 3.7)
!    
!    (C) Copyright 2021 
!  
!        Salvatore Filippone  
!        Pasqua D'Ambra   
!        Fabio Durastante        
!   
!    Redistribution and use in source and binary forms, with or without
!    modification, are permitted provided that the following conditions
!    are met:
!      1. Redistributions of source code must retain the above copyright
!         notice, this list of conditions and the following disclaimer.
!      2. Redistributions in binary form must reproduce the above copyright
!         notice, this list of conditions, and the following disclaimer in the
!         documentation and/or other materials provided with the distribution.
!      3. The name of the AMG4PSBLAS group or the names of its contributors may
!         not be used to endorse or promote products derived from this
!         software without specific prior written permission.
!   
!    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
!    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
!    TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
!    PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AMG4PSBLAS GROUP OR ITS CONTRIBUTORS
!    BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
!    CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
!    SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
!    INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
!    CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
!    ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
!    POSSIBILITY OF SUCH DAMAGE.
!   
!  
subroutine amg_d_base_onelev_map_rstr_v(lv,alpha,vect_u,beta,vect_v,info,&
     &  work,vtx,vty)
  use psb_base_mod
  use amg_d_onelev_mod, amg_protect_name =>  amg_d_base_onelev_map_rstr_v
  implicit none
  class(amg_d_onelev_type), target, intent(inout) :: lv
  real(psb_dpk_), intent(in)           :: alpha, beta
  type(psb_d_vect_type), intent(inout) :: vect_u, vect_v
  integer(psb_ipk_), intent(out)       :: info
  real(psb_dpk_), optional          :: work(:)
  type(psb_d_vect_type), optional, target, intent(inout)  :: vtx,vty
  type(psb_d_vect_type), pointer   :: vty_
      integer(psb_mpk_) :: me, np

  write(0,*) 'New map_rstr ',lv%remap_data%ac_pre_remap%is_asb()
  if (present(vty)) then
    vty_ => vty
  else
    vty_ => lv%wrk%wv(1)
  end if
  
  if (lv%remap_data%ac_pre_remap%is_asb()) then
    !
    ! Remap has happened, deal with it
    !
    write(0,*) 'Level map_rstr with remapping '
    block
      type(psb_ctxt_type) :: ctxt, rctxt
      integer(psb_mpk_) :: i,j,ip, idest, nsrc, nrl, kp
      integer(psb_mpk_) :: rme, rnp
      real(psb_dpk_), allocatable :: rsnd(:), rrcv(:)
      type(psb_d_vect_type) :: tv
      
      ctxt = lv%remap_data%desc_ac_pre_remap%get_ctxt()
      call psb_info(ctxt,me,np)
      rctxt = lv%desc_ac%get_ctxt()
      call psb_info(rctxt,rme,rnp)
      write(0,*) 'New context map rstr',rme,rnp,me,np
      idest = lv%remap_data%idest
      associate(isrc => lv%remap_data%isrc, nrsrc => lv%remap_data%nrsrc)
        write(0,*) 'Should apply maps, then send data from ',me,' to ',idest
        if (rme >= 0) write(0,*) rme, '  Receiving      data from ',isrc(:)
        nsrc = size(isrc)
        nrl  = lv%remap_data%desc_ac_pre_remap%get_local_rows()
        call psb_geall(tv,lv%remap_data%desc_ac_pre_remap,info)
        call psb_geasb(tv,lv%remap_data%desc_ac_pre_remap,info,mold=vect_u%v) 
        write(0,*) me,' remap map_rstr calling U2V: ',me,np,rme,rnp,tv%get_nrows(),&
             & psb_errstatus_fatal()
        flush(0)
        call psb_barrier(ctxt)
        call lv%linmap%map_U2V(alpha,vect_u,beta,tv,info,&
             & work=work,vtx=vtx,vty=vty_)
        !call tv%sync()
        !rsnd = tv%get_vect()
        !call psb_snd(ctxt,rsnd(1:nrl),idest) 
        write(0,*) me,' map_rstr sending ',me,idest,psb_errstatus_fatal()
        call psb_snd(ctxt,tv%v%v(1:nrl),idest)
        if (rme >=0) then
          allocate(rrcv(sum(nrsrc)))
!!$          write(0,*) me,rme,' Size check ',size(rrcv)!,lv%desc_ac%get_local_rows()
          kp = 0
          do i = 1,size(isrc)
            ip = isrc(i)
            nrl = nrsrc(i)
            write(0,*) me,' map_rstr receiving',rme,ip,psb_errstatus_fatal()           
            call psb_rcv(ctxt,rrcv(kp+1:kp+nrl),ip)
            kp = kp + nrl
          end do
          call vect_v%set_vect(rrcv)
        end if
      end associate
      write(0,*) me, ' Restrictor with remap done ',psb_errstatus_fatal()     
    end block
 
  else
    ! Default transfer
    block
      type(psb_ctxt_type) :: ctxt, rctxt
      ctxt = lv%linmap%p_desc_U%get_ctxt()
      call psb_info(ctxt,me,np)
      write(0,*) me,' map_rstr calling U2V: ',me,np
      call lv%linmap%map_U2V(alpha,vect_u,beta,vect_v,info,&
           & work=work,vtx=vtx,vty=vty_)
    end block
  end if
  write(0,*) me, 'End of restriction ',info,psb_errstatus_fatal()
end subroutine amg_d_base_onelev_map_rstr_v

subroutine amg_d_base_onelev_map_rstr_a(lv,alpha,u,beta,v,info,work)
  use psb_base_mod
  use amg_d_onelev_mod, amg_protect_name =>  amg_d_base_onelev_map_rstr_a
  implicit none
  class(amg_d_onelev_type), target, intent(inout) :: lv
  real(psb_dpk_), intent(in)     :: alpha, beta
  real(psb_dpk_), intent(inout)   :: u(:)
  real(psb_dpk_), intent(out)     :: v(:)
  integer(psb_ipk_), intent(out) :: info
  real(psb_dpk_), optional       :: work(:)

  if (lv%remap_data%ac_pre_remap%is_asb()) then
    !
    ! Remap has happened, deal with it
    !
    write(0,*) 'Remap R handling not implemented yet for A'
  else
    ! Default transfer
    call lv%linmap%map_U2V(alpha,u,beta,v,info,&
         & work=work)
  end if
  
end subroutine amg_d_base_onelev_map_rstr_a
