submesh.F90

Go to the documentation of this file.

!! Copyright (C) 2007 X. Andrade
!!
!! This program is free software; you can redistribute it and/or modify
!! it under the terms of the GNU General Public License as published by
!! the Free Software Foundation; either version 2, or (at your option)
!! any later version.
!!
!! This program is distributed in the hope that it will be useful,
!! but WITHOUT ANY WARRANTY; without even the implied warranty of
!! MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
!! GNU General Public License for more details.
!!
!! You should have received a copy of the GNU General Public License
!! along with this program; if not, write to the Free Software
!! Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
!! 02110-1301, USA.
!!
 
#include "global.h"
 
module submesh_oct_m
  use accel_oct_m
  use batch_oct_m
  use boundaries_oct_m
  use box_oct_m
  use debug_oct_m
  use global_oct_m
  use index_oct_m
  use, intrinsic :: iso_fortran_env
  use lalg_basic_oct_m
  use lattice_vectors_oct_m
  use messages_oct_m
  use sort_oct_m
  use mesh_oct_m
  use mesh_cube_map_oct_m
  use mesh_function_oct_m
  use mpi_oct_m
  use par_vec_oct_m
  use profiling_oct_m
  use space_oct_m
  use types_oct_m
 
  implicit none
  private
 
  public ::                      &
    submesh_t,                   &
    submesh_init,                &
    submesh_merge,               &
    submesh_shift_center,        &
    submesh_broadcast,           &
    submesh_build_global,        &
    submesh_end_global,          &
    dsm_integrate,               &
    zsm_integrate,               &
    dsm_integrate_frommesh,      &
    zsm_integrate_frommesh,      &
    dsm_nrm2,                    &
    zsm_nrm2,                    &
    submesh_add_to_mesh,         &
    dsubmesh_batch_add,          &
    zsubmesh_batch_add,          &
    dzsubmesh_batch_add,         &
    submesh_to_mesh_dotp,        &
    dsubmesh_dotp,               &
    zsubmesh_dotp,               &
    dsubmesh_batch_add_matrix,   &
    zsubmesh_batch_add_matrix,   &
    dsubmesh_batch_dotp_matrix,  &
    zsubmesh_batch_dotp_matrix,  &
    submesh_overlap,             &
    dsubmesh_copy_from_mesh,     &
    zsubmesh_copy_from_mesh,     &
    submesh_compatible,          &
    submesh_end,                 &
    submesh_get_cube_dim,        &
    submesh_init_cube_map,       &
    submesh_end_cube_map
 
  type submesh_t
    ! Components are public by default
    real(real64), allocatable    :: center(:)
    real(real64)          :: radius = m_zero
    class(box_t), pointer :: box => null()  
    integer               :: np = -1
    integer,  allocatable :: map(:)
    integer               :: num_regions
    integer,  allocatable :: regions(:)
    type(accel_mem_t)     :: buff_map
    real(real64),    allocatable :: rel_x(:,:)
    real(real64),    allocatable :: r(:)
    real(real64),    allocatable :: vol_pp(:)
    type(mesh_t), pointer :: mesh => null() 
    logical               :: overlap
    integer               :: np_global = -1
    real(real64),    allocatable :: rel_x_global(:,:)
    integer,  allocatable :: part_v(:)
    integer,  allocatable :: global2local(:)
 
    type(mesh_cube_map_t) :: cube_map
  end type submesh_t
 
  interface submesh_add_to_mesh
    module procedure ddsubmesh_add_to_mesh, zdsubmesh_add_to_mesh, zzsubmesh_add_to_mesh
  end interface submesh_add_to_mesh
 
  interface submesh_to_mesh_dotp
    module procedure ddsubmesh_to_mesh_dotp, zdsubmesh_to_mesh_dotp, zzsubmesh_to_mesh_dotp
  end interface submesh_to_mesh_dotp
 
contains
 
  ! -------------------------------------------------------------
  ! Multipliers for recursive formulation of n-ellipsoid volume
  ! simplifying the Gamma function
  ! f(n) = 2f(n-2)/n, f(0)=1, f(1)=2
  recursive real(real64) function f_n(dims) result(fn)
    integer :: dims
 
    if (dims == 0) then
      fn = m_one
    else if (dims == 1) then
      fn = m_two
    else
      fn = m_two * f_n(dims - 2) / dims
    end if
 
  end function f_n
 
  ! -------------------------------------------------------------
  subroutine submesh_init(this, space, mesh, latt, center, rc)
    type(submesh_t),         intent(inout)  :: this
    class(space_t),          intent(in)     :: space
    class(mesh_t), target,   intent(in)     :: mesh
    type(lattice_vectors_t), intent(in)     :: latt
    real(real64),            intent(in)     :: center(1:space%dim)
    real(real64),            intent(in)     :: rc
 
    real(real64) :: r2, rc2, xx(space%dim), rc_norm_n
    real(real64), allocatable :: center_copies(:,:), xtmp(:, :), rtmp(:)
    integer :: icell, is, ip, ix, iy, iz
    integer(int64) :: max_elements_count
    type(lattice_iterator_t) :: latt_iter
    integer, allocatable :: map_inv(:), map_temp(:)
    integer :: nmax(3), nmin(3)
    real(real64), parameter :: tol = 1e-13_real64
 
 
    push_sub(submesh_init)
    call profiling_in("SUBMESH_INIT")
 
    assert(space%dim <= 3)
 
    this%mesh => mesh
 
    safe_allocate(this%center(1:space%dim))
    this%center(:) = center(:)
 
    this%radius = rc
    ! We add a small number of avoid instabilities due to rounding errors
    rc2 = rc**2 + tol
 
    ! The spheres are generated differently for periodic coordinates,
    ! mainly for performance reasons.
    if (.not. space%is_periodic()) then
 
      call profiling_in("SUBMESH_INIT_MAP_INV")
      safe_allocate(map_inv(0:this%mesh%np))
      map_inv(0:this%mesh%np) = 0
 
      nmin = 0
      nmax = 0
 
      ! get a cube of points that contains the sphere
      nmin(1:space%dim) = int((center(1:space%dim) - abs(rc))/mesh%spacing(1:space%dim)) - 1
      nmax(1:space%dim) = int((center(1:space%dim) + abs(rc))/mesh%spacing(1:space%dim)) + 1
 
      ! make sure that the cube is inside the grid
      ! parts of the cube which would fall outside the simulation box are chopped off.
      nmin(1:space%dim) = max(mesh%idx%nr(1, 1:space%dim), nmin(1:space%dim))
      nmax(1:space%dim) = min(mesh%idx%nr(2, 1:space%dim), nmax(1:space%dim))
 
      ! Get the total number of points inside the sphere
      is = 0   ! this index counts inner points
      do iz = nmin(3), nmax(3)
        do iy = nmin(2), nmax(2)
          do ix = nmin(1), nmax(1)
            ip = mesh_local_index_from_coords(mesh, [ix, iy, iz])
            if (ip == 0 .or. ip > mesh%np) cycle
            r2 = sum((mesh%x(:, ip) - center)**2)
            if (r2 <= rc2) then
              is = is + 1
              map_inv(ip) = is
            end if
          end do
        end do
      end do
      this%np = is
      call profiling_out("SUBMESH_INIT_MAP_INV")
 
      call profiling_in("SUBMESH_INIT_RTMP")
      safe_allocate(this%map(1:this%np))
      safe_allocate(xtmp(1:space%dim, 1:this%np))
      safe_allocate(rtmp(1:this%np))
 
      ! Generate the table and the positions
      do iz = nmin(3), nmax(3)
        do iy = nmin(2), nmax(2)
          do ix = nmin(1), nmax(1)
            ip = mesh_local_index_from_coords(mesh, [ix, iy, iz])
            if (ip == 0 .or. ip > mesh%np) cycle
            is = map_inv(ip)
            if (is == 0) cycle
            this%map(is) = ip
            xtmp(:, is) = mesh%x(:, ip) - center
            rtmp(is) = norm2(xtmp(:,is))
          end do
        end do
      end do
 
      safe_deallocate_a(map_inv)
      call profiling_out("SUBMESH_INIT_RTMP")
 
      ! This is the case for a periodic system
    else
 
      ! Get the total number of points inside the sphere considering
      ! replicas along PBCs
 
      ! this requires some optimization
 
      ! We use non-zero centered periodic copies to minimize the number of cells to test.
      ! By passing the center (in reduced coordinates), the iterator only keeps the cells that are actually relevant
      latt_iter = lattice_iterator_t(latt, rc, center=latt%cart_to_red(center) - mesh%red_min(space))
 
      ! We get the shifted centers for the cells we care about
      safe_allocate(center_copies(1:space%dim, 1:latt_iter%n_cells))
      do icell = 1, latt_iter%n_cells
        center_copies(:, icell) = center + latt_iter%get(icell)
      end do
 
      !Recursive formulation for the volume of n-ellipsoid
      !Garry Tee, NZ J. Mathematics Vol. 34 (2005) p. 165 eqs. 53,55
      rc_norm_n = product(ceiling(rc / mesh%spacing(1:space%dim), int64) + m_one)
      if (mesh%use_vol_pp) rc_norm_n = rc_norm_n / mesh%coord_system%min_mesh_scaling_product
      max_elements_count = 3**space%dim * int(m_pi**floor(0.5 * space%dim) * rc_norm_n * f_n(space%dim), int64)
 
      call profiling_in("SUBMESH_INIT_PERIODIC_R")
      safe_allocate(map_temp(1:max_elements_count))
      safe_allocate(xtmp(1:space%dim, 1:max_elements_count))
      safe_allocate(rtmp(1:max_elements_count))
 
      is = 0
      do ip = 1, mesh%np
        do icell = 1, latt_iter%n_cells
          xx = mesh%x(:,ip) - center_copies(:, icell)
          if(any(abs(xx)>rc+tol)) cycle
          r2 = sum(xx**2)
          if (r2 > rc2) cycle
          is = is + 1
          map_temp(is) = ip
          rtmp(is) = sqrt(r2)
          xtmp(:, is) = xx
          ! Note that xx can be outside the unit cell
        end do
      end do
      assert(is < huge(is))
      this%np = is
 
      safe_allocate(this%map(1:this%np))
      this%map(1:this%np) = map_temp(1:this%np)
      call profiling_out("SUBMESH_INIT_PERIODIC_R")
 
      safe_deallocate_a(map_temp)
      safe_deallocate_a(center_copies)
 
    end if
 
    call submesh_reorder_points(this, space, xtmp, rtmp)
    call profiling_out("SUBMESH_INIT")
 
    safe_deallocate_a(xtmp)
    safe_deallocate_a(rtmp)
 
    call submesh_get_vol_pp(this)
 
    pop_sub(submesh_init)
  end subroutine submesh_init
 
  ! --------------------------------------------------------------
  subroutine submesh_get_vol_pp(this)
    type(submesh_t), intent(inout) :: this
 
    integer :: is
 
    if (.not. this%mesh%use_vol_pp) return
 
    push_sub(submesh_get_vol_pp)
 
    ! submesh_broadcast calls this again on the root rank
    safe_deallocate_a(this%vol_pp)
    safe_allocate(this%vol_pp(1:this%np))
    do is = 1, this%np
      this%vol_pp(is) = this%mesh%vol_pp(this%map(is))
    end do
 
    pop_sub(submesh_get_vol_pp)
  end subroutine submesh_get_vol_pp
 
  subroutine submesh_reorder_points(this, space, xtmp, rtmp)
    type(submesh_t), intent(inout) :: this
    class(space_t), intent(in) :: space
    real(real64), contiguous, intent(in) :: xtmp(:, :), rtmp(:)
 
    integer :: ip, i_region, offset
    integer, allocatable :: order(:), order_new(:)
    integer, allocatable :: map_new(:)
    integer, allocatable :: np_region(:), tmp_array(:)
 
    push_sub(submesh_reorder_points)
 
    ! now order points for better locality
 
    call profiling_in("SUBMESH_INIT_ORDER")
    safe_allocate(order(1:this%np))
    safe_allocate(this%rel_x(1:space%dim, 1:this%np))
    safe_allocate(this%r(1:this%np))
 
    do ip = 1, this%np
      order(ip) = ip
    end do
 
    ! First we just reorder in order to determine overlap:
    call sort(this%map, order)
 
    !check whether points overlap (i.e. whether a submesh contains the same point more than once)
    this%overlap = .false.
    do ip = 1, this%np - 1
      if (this%map(ip) == this%map(ip + 1)) then
        ! this simplified test works, as the points are ordered.
        this%overlap = .true.
        exit
      end if
    end do
 
    this%num_regions = 1
    call profiling_out("SUBMESH_INIT_ORDER")
 
    if(this%overlap) then
      call profiling_in("SUBMESH_INIT_OVERLAP")
      !disentangle the map into injective regions
 
      safe_allocate(tmp_array(1:this%np))
      safe_allocate(order_new(1:this%np))
      safe_allocate(np_region(1:this%np))
      safe_allocate(map_new(  1:this%np))
 
      np_region(1) = 1
      tmp_array(1) = 1
      i_region = 1
 
      do ip = 2, this%np
        if (this%map(ip) == this%map(ip - 1)) then
          i_region = i_region + 1
          if (i_region > this%num_regions) then
            this%num_regions = i_region
            np_region(i_region) = 0
          end if
        else
          i_region = 1
        end if
        tmp_array(ip) = i_region                        ! which region does ip belong to
        np_region(i_region) = np_region(i_region) + 1   ! increase number of points in i_region
      end do
 
      assert( .not. allocated(this%regions))
 
      ! construct array of offsets
      safe_allocate(this%regions(1:this%num_regions+1))
 
      this%regions(1) = 1
 
      if(this%num_regions > 1) then
        do i_region = 1, this%num_regions
          this%regions(i_region + 1) = this%regions(i_region) + np_region(i_region)
        end do
      else
        this%regions(2) = this%np + 1
      end if
 
      np_region(1:this%np) = 0
      order_new(1:this%np) = -1
      map_new(1:this%np) = -1
 
 
      !reassemble regions into global map array
      do ip = 1, this%np
        i_region = tmp_array(ip)
        np_region(i_region) = np_region(i_region) + 1
        offset = this%regions(i_region) - 1
        map_new(   offset + np_region(i_region) ) = this%map(ip)
        order_new( offset + np_region(i_region) ) = order(ip)
      end do
 
      order(1:this%np) = order_new(1:this%np)
      this%map(1:this%np) = map_new(1:this%np)
 
      safe_deallocate_a(tmp_array)
      safe_deallocate_a(order_new)
      safe_deallocate_a(np_region)
      safe_deallocate_a(map_new)
      call profiling_out("SUBMESH_INIT_OVERLAP")
 
    else
      this%num_regions = 1
      safe_allocate(this%regions(1:2))
      this%regions(1) = 1
      this%regions(2) = this%np + 1
    end if
 
    ! Lastly, reorder the points according to the new scheme
    do ip = 1, this%np
      this%rel_x(:, ip) = xtmp(:, order(ip))
      this%r(ip) = rtmp(order(ip))
    end do
 
    safe_deallocate_a(order)
 
    pop_sub(submesh_reorder_points)
  end subroutine submesh_reorder_points
 
 
  ! --------------------------------------------------------------
  !This routine takes two submeshes and merge them into a bigger submesh
  !The grid is centered on the first center
  subroutine submesh_merge(this, space, mesh, sm1, sm2, shift)
    type(submesh_t),       intent(inout)  :: this
    class(space_t),        intent(in)     :: space
    class(mesh_t), target, intent(in)     :: mesh
    type(submesh_t),       intent(in)     :: sm1
    type(submesh_t),       intent(in)     :: sm2
    real(real64), optional,       intent(in)     :: shift(:)
 
    real(real64) :: r2
    integer :: ip, is
    real(real64) :: xx(space%dim), diff_centers(space%dim)
 
    push_sub(submesh_merge)
    call profiling_in("SUBMESH_MERGE")
 
    this%mesh => mesh
 
    safe_allocate(this%center(1:space%dim))
    this%center(:)  = sm1%center(:)
    this%radius = sm1%radius
 
    ! This is a quick fix to prevent uninitialized variables. To properly check the self-overlap,
    ! a similar approach as in submesh_init should be taken with respect to the merged map.
    this%overlap = sm1%overlap .or. sm2%overlap
 
    diff_centers = sm1%center - sm2%center
    if (present(shift)) diff_centers = diff_centers - shift
 
    !As we take the union of the two submeshes, we know that we have all the points from the first one included.
    !The extra points from the second submesh are those which are not included in the first one
    is = sm1%np
    do ip = 1, sm2%np
      !sm2%x contains points coordinates defined with respect to sm2%center
      xx = sm2%rel_x(:, ip) - diff_centers
      !If the point is not in sm1, we add it
      if (sum(xx**2) > sm1%radius**2) is = is + 1
    end do
 
    this%np = is
 
    safe_allocate(this%map(1:this%np))
    safe_allocate(this%rel_x(1:space%dim, 1:this%np))
    safe_allocate(this%r(1:this%np))
    this%map(1:sm1%np) = sm1%map(1:sm1%np)
    this%rel_x(:, 1:sm1%np) = sm1%rel_x(:, 1:sm1%np)
    this%r(1:sm1%np) = sm1%r(1:sm1%np)
 
    !iterate again to fill the tables
    is = sm1%np
    do ip = 1, sm2%np
      xx = sm2%rel_x(:, ip) - diff_centers
      r2 = sum(xx**2)
      if (r2 > sm1%radius**2) then
        is = is + 1
        this%map(is) = sm2%map(ip)
        this%r(is) = sqrt(r2)
        this%rel_x(:, is) = xx
      end if
    end do
 
    call submesh_get_vol_pp(this)
 
    call profiling_out("SUBMESH_MERGE")
    pop_sub(submesh_merge)
  end subroutine submesh_merge
 
  ! --------------------------------------------------------------
  !This routine shifts the center of a submesh, without changing the grid points
  subroutine submesh_shift_center(this, space, newcenter)
    type(submesh_t),      intent(inout)  :: this
    class(space_t),       intent(in)     :: space
    real(real64),         intent(in)     :: newcenter(:)
 
    real(real64) :: xx(space%dim), diff_centers(space%dim), oldcenter(space%dim)
    integer :: ip
 
    push_sub(submesh_shift_center)
    call profiling_in("SUBMESH_SHIFT")
 
    oldcenter = this%center
    this%center(:)  = newcenter(:)
 
    diff_centers = newcenter - oldcenter
 
    do ip = 1, this%np
      xx = this%rel_x(:, ip) - diff_centers
      this%r(ip) = norm2(xx)
      this%rel_x(:, ip) = xx
    end do
 
    call profiling_out("SUBMESH_SHIFT")
    pop_sub(submesh_shift_center)
  end subroutine submesh_shift_center
 
  ! --------------------------------------------------------------
  subroutine submesh_broadcast(this, space, mesh, center, radius, root, mpi_grp)
    type(submesh_t),      intent(inout)  :: this
    class(space_t),       intent(in)     :: space
    type(mesh_t), target, intent(in)     :: mesh
    real(real64),         intent(in)     :: center(1:space%dim)
    real(real64),         intent(in)     :: radius
    integer,              intent(in)     :: root
    type(mpi_grp_t),      intent(in)     :: mpi_grp
 
    integer :: nparray(1:3)
 
    push_sub(submesh_broadcast)
    call profiling_in('SUBMESH_BCAST')
 
    if (root /= mpi_grp%rank) then
      this%mesh => mesh
      safe_allocate(this%center(1:space%dim))
      this%center(:) = center(:)
      this%radius = radius
    end if
 
    if (mpi_grp%size > 1) then
 
      if (root == mpi_grp%rank) then
        nparray(1) = this%np
        nparray(2) = this%num_regions
        if (this%overlap) then
          nparray(3) = 1
        else
          nparray(3) = 0
        end if
      end if
 
      call mpi_grp%bcast(nparray, 3, mpi_integer, root)
      this%np = nparray(1)
      this%num_regions = nparray(2)
      this%overlap = (nparray(3) == 1)
 
      if (root /= mpi_grp%rank) then
        safe_allocate(this%map(1:this%np))
        safe_allocate(this%rel_x(1:space%dim, 1:this%np))
        safe_allocate(this%r(1:this%np))
        safe_allocate(this%regions(1:this%num_regions+1))
      end if
 
      call mpi_grp%bcast(this%regions(1), this%num_regions+1, mpi_integer, root)
 
      if (this%np > 0) then
        call mpi_grp%bcast(this%map(1), this%np, mpi_integer, root)
        call mpi_grp%bcast(this%rel_x(1, 1), this%np*space%dim, mpi_double_precision, root)
        call mpi_grp%bcast(this%r(1), this%np, mpi_double_precision, root)
      end if
 
    end if
 
    call submesh_get_vol_pp(this)
 
    call profiling_out('SUBMESH_BCAST')
    pop_sub(submesh_broadcast)
  end subroutine submesh_broadcast
 
  ! --------------------------------------------------------------
  logical function submesh_compatible(this, radius, center, dx) result(compatible)
    type(submesh_t),   intent(in) :: this
    real(real64),      intent(in) :: radius
    real(real64),      intent(in) :: center(:)
    real(real64),      intent(in) :: dx
 
    compatible =.false.
    if (allocated(this%center)) then
      if (radius <= this%radius+dx*1e-6_real64 .and. all(abs(this%center - center) < 0.25*dx)) then
        compatible = .true.
      end if
    end if
 
  end function submesh_compatible
 
  ! --------------------------------------------------------------
  subroutine submesh_end(this)
    type(submesh_t),   intent(inout)  :: this
 
    push_sub(submesh_end)
 
    nullify(this%mesh)
    nullify(this%box)
    if (this%np /= -1) then
      this%np = -1
      safe_deallocate_a(this%center)
      safe_deallocate_a(this%map)
      safe_deallocate_a(this%rel_x)
      safe_deallocate_a(this%r)
      safe_deallocate_a(this%vol_pp)
      safe_deallocate_a(this%regions)
    end if
 
    if (accel_is_enabled()) then
      call accel_free_buffer(this%buff_map)
    end if
 
    pop_sub(submesh_end)
  end subroutine submesh_end
 
 
  ! --------------------------------------------------------------
  logical function submesh_overlap(sm1, sm2, space) result(overlap)
    type(submesh_t),      intent(in)   :: sm1
    type(submesh_t),      intent(in)   :: sm2
    class(space_t),       intent(in)   :: space
 
    integer :: ii, jj, dd
    real(real64) :: distance
 
    !no PUSH_SUB, called too often
 
    if (.not. space%is_periodic()) then
      !first check the distance
      distance = sum((sm1%center - sm2%center)**2)
      overlap = distance <= (1.5_real64*(sm1%radius + sm2%radius))**2
 
      ! if they are very far, no need to check in detail
      if (.not. overlap) return
    end if
 
    ! Otherwise check whether they have the some point in common. We
    ! can make the comparison faster using that the arrays are sorted.
    overlap = .false.
    ii = 1
    jj = 1
    do while(ii <= sm1%np .and. jj <= sm2%np)
      dd = sm1%map(ii) - sm2%map(jj)
      if (dd < 0) then
        ii = ii + 1
      else if (dd > 0) then
        jj = jj + 1
      else
        overlap = .true.
        exit
      end if
    end do
 
    if (sm1%mesh%parallel_in_domains) then
      call sm1%mesh%mpi_grp%allreduce_inplace(overlap, 1, mpi_logical, mpi_lor)
    end if
 
  end function submesh_overlap
 
  ! -------------------------------------------------------------
  subroutine submesh_build_global(this, space)
    type(submesh_t),      intent(inout) :: this
    class(space_t),       intent(in)    :: space
 
    integer, allocatable :: part_np(:)
    integer :: ipart, ind, ip
 
    push_sub(submesh_build_global)
 
    if (.not. this%mesh%parallel_in_domains) then
      this%np_global = this%np
      pop_sub(submesh_build_global)
      return
    end if
 
    safe_allocate(part_np(this%mesh%pv%npart))
    part_np = 0
    part_np(this%mesh%pv%partno) = this%np
 
    call this%mesh%allreduce(part_np)
    this%np_global = sum(part_np)
 
    safe_allocate(this%rel_x_global(1:space%dim, 1:this%np_global))
    safe_allocate(this%part_v(1:this%np_global))
    safe_allocate(this%global2local(1:this%np_global))
    this%rel_x_global(1:space%dim, 1:this%np_global) = m_zero
    this%part_v(1:this%np_global) = 0
    this%global2local(1:this%np_global) = 0
 
    ind = 0
    do ipart = 1, this%mesh%pv%npart
      if (ipart == this%mesh%pv%partno) then
        do ip = 1, this%np
          this%rel_x_global(:, ind + ip) = this%rel_x(:, ip)
          this%part_v(ind + ip) = this%mesh%pv%partno
          this%global2local(ind + ip) = ip
        end do
      end if
      ind = ind + part_np(ipart)
    end do
 
    call this%mesh%allreduce(this%rel_x_global)
    call this%mesh%allreduce(this%part_v)
    call this%mesh%allreduce(this%global2local)
 
    safe_deallocate_a(part_np)
 
    pop_sub(submesh_build_global)
  end subroutine submesh_build_global
 
  ! -----------------------------------------------------------
  subroutine submesh_end_global(this)
    type(submesh_t),      intent(inout)   :: this
 
    push_sub(submesh_end_global)
 
    safe_deallocate_a(this%rel_x_global)
    this%np_global = -1
    safe_deallocate_a(this%part_v)
    safe_deallocate_a(this%global2local)
 
    pop_sub(submesh_end_global)
  end subroutine submesh_end_global
 
 
  ! -----------------------------------------------------------
  subroutine zzsubmesh_add_to_mesh(this, sphi, phi, factor)
    type(submesh_t), intent(in) :: this
    complex(real64), contiguous, intent(in) :: sphi(:)
    complex(real64), contiguous, intent(inout) :: phi(:)
    complex(real64), optional, intent(in) :: factor
 
    integer :: ip, m
 
    push_sub(zzsubmesh_add_to_mesh)
 
    if (present(factor)) then
      !Loop unrolling inspired by BLAS axpy routine
      m = mod(this%np,4)
      do ip = 1, m
        phi(this%map(ip)) = phi(this%map(ip)) + factor*sphi(ip)
      end do
      if (this%np.ge.4) then
        do ip = m+1, this%np, 4
          phi(this%map(ip))   = phi(this%map(ip))   + factor*sphi(ip)
          phi(this%map(ip+1)) = phi(this%map(ip+1)) + factor*sphi(ip+1)
          phi(this%map(ip+2)) = phi(this%map(ip+2)) + factor*sphi(ip+2)
          phi(this%map(ip+3)) = phi(this%map(ip+3)) + factor*sphi(ip+3)
        end do
      end if
    else
      m = mod(this%np,4)
      do ip = 1, m
        phi(this%map(ip)) = phi(this%map(ip)) + sphi(ip)
      end do
      if (this%np.ge.4) then
        do ip = m+1, this%np, 4
          phi(this%map(ip))   = phi(this%map(ip))   + sphi(ip)
          phi(this%map(ip+1)) = phi(this%map(ip+1)) + sphi(ip+1)
          phi(this%map(ip+2)) = phi(this%map(ip+2)) + sphi(ip+2)
          phi(this%map(ip+3)) = phi(this%map(ip+3)) + sphi(ip+3)
        end do
      end if
    end if
 
    pop_sub(zzsubmesh_add_to_mesh)
  end subroutine zzsubmesh_add_to_mesh
 
  !------------------------------------------------------------
  complex(real64) function zzsubmesh_to_mesh_dotp(this, sphi, phi, reduce) result(dotp)
    type(submesh_t), intent(in) :: this
    complex(real64), contiguous, intent(in) :: sphi(:)
    complex(real64), contiguous, intent(in) :: phi(:)
    logical, optional, intent(in) :: reduce
 
    integer :: is, m, ip
 
    push_sub(zzsubmesh_to_mesh_dotp)
 
    dotp = m_z0
 
    if (this%mesh%use_vol_pp) then
      do is = 1, this%np
        dotp = dotp + this%vol_pp(is)*phi(this%map(is))*conjg(sphi(is))
      end do
    else
      m = mod(this%np,4)
      do ip = 1, m
        dotp = dotp + phi(this%map(ip))*conjg(sphi(ip))
      end do
      if (this%np.ge.4) then
        do ip = m+1, this%np, 4
          dotp = dotp + phi(this%map(ip))*conjg(sphi(ip))     &
            + phi(this%map(ip+1))*conjg(sphi(ip+1)) &
            + phi(this%map(ip+2))*conjg(sphi(ip+2)) &
            + phi(this%map(ip+3))*conjg(sphi(ip+3))
        end do
      end if
      dotp = dotp*this%mesh%vol_pp(1)
    end if
 
    if (optional_default(reduce, .true.)) then
      call profiling_in("SM_REDUCE_DOTP")
      call this%mesh%allreduce(dotp)
      call profiling_out("SM_REDUCE_DOTP")
    end if
 
    pop_sub(zzsubmesh_to_mesh_dotp)
  end function zzsubmesh_to_mesh_dotp
 
  !------------------------------------------------------------
  subroutine submesh_get_cube_dim(sm, space, db)
    type(submesh_t), target,  intent(in)  :: sm
    class(space_t),           intent(in)  :: space
    integer,                  intent(out) :: db(1:space%dim)
 
    integer :: ip, idir
    real(real64) :: chi(space%dim), max_chi(space%dim)
    integer :: db_red(1:space%dim)
    real(real64), parameter :: tol=1.0e-10_real64
 
    push_sub(submesh_get_cube_dim)
 
    max_chi = m_zero
    do ip = 1, sm%np
      chi = sm%mesh%coord_system%from_cartesian(sm%rel_x(:, ip))
      do idir = 1, space%dim
        max_chi(idir) = max(max_chi(idir), abs(chi(idir))/sm%mesh%spacing(idir))
      end do
    end do
 
    do idir = 1, space%dim
      db(idir) = nint(max_chi(idir)-tol)
    end do
 
    if(sm%mesh%parallel_in_domains) then
      call sm%mesh%mpi_grp%allreduce(db(1), db_red(1), space%dim, mpi_integer, mpi_max)
      db(1:space%dim) = db_red(1:space%dim)
    end if
 
    db = 2 * db + 1
 
    pop_sub(submesh_get_cube_dim)
  end subroutine submesh_get_cube_dim
 
  !------------------------------------------------------------
  subroutine submesh_init_cube_map(sm, space)
    type(submesh_t), target,  intent(inout) :: sm
    class(space_t),           intent(in)    :: space
 
    integer(int64) :: ip
    integer ::idir
    real(real64) :: chi(space%dim), shift(space%dim)
 
    push_sub(submesh_init_cube_map)
 
    sm%cube_map%nmap = sm%np
 
    safe_allocate(sm%cube_map%map(1:space%dim, 1:sm%cube_map%nmap))
 
    !The center of the submesh does not belong to the mesh
    !So we first need to find the closest grid point, and center the cube to it
    chi = sm%mesh%coord_system%from_cartesian(sm%center)
    do idir = 1, space%dim
      shift(idir) = nint(chi(idir)/sm%mesh%spacing(idir))*sm%mesh%spacing(idir)
    end do
    shift = sm%mesh%coord_system%to_cartesian(shift)
    shift = shift - sm%center
 
    do ip = 1, sm%cube_map%nmap
      chi = sm%mesh%coord_system%from_cartesian(sm%rel_x(:,ip) - shift)
      do idir = 1, space%dim
        sm%cube_map%map(idir, ip) = nint(chi(idir)/sm%mesh%spacing(idir))
      end do
    end do
 
    if (accel_is_enabled()) then
      call accel_create_buffer(sm%cube_map%map_buffer, accel_mem_read_only, type_integer, sm%cube_map%nmap*space%dim)
      call accel_write_buffer(sm%cube_map%map_buffer, space%dim, sm%cube_map%nmap, sm%cube_map%map)
    end if
 
    pop_sub(submesh_init_cube_map)
  end subroutine submesh_init_cube_map
 
  !------------------------------------------------------------
  subroutine submesh_end_cube_map(sm)
    type(submesh_t),   intent(inout)  :: sm
 
    push_sub(submesh_end_cube_map)
 
    call mesh_cube_map_end(sm%cube_map)
 
    pop_sub(submesh_end_cube_map)
  end subroutine submesh_end_cube_map
 
  subroutine dzsubmesh_batch_add(this, ss, mm)
    type(submesh_t),  intent(in)    :: this
    class(batch_t),   intent(in)    :: ss
    class(batch_t),   intent(inout) :: mm
 
    integer :: ist, idim, jdim, is
 
    push_sub(dzsubmesh_batch_add)
 
    assert(.not. mm%is_packed())
    assert(ss%type() == type_float)
    assert(mm%type() == type_cmplx)
    assert(ss%nst_linear == mm%nst_linear)
    assert(ss%status() == mm%status())
    assert(ss%dim == mm%dim)
 
    assert(mm%nst == ss%nst)
 
    !$omp parallel do private(ist, idim, jdim, is) if(.not. this%overlap)
    do ist =  1, mm%nst
      do idim = 1, mm%dim
        jdim = min(idim, ss%dim)
 
        do is = 1, this%np
          mm%zff(this%map(is), idim, ist) = &
            mm%zff(this%map(is), idim, ist) + ss%dff(is, jdim, ist)
        end do
 
      end do
    end do
    !$omp end parallel do
 
    pop_sub(dzsubmesh_batch_add)
  end subroutine dzsubmesh_batch_add
 
 
#include "undef.F90"
#include "real.F90"
#include "submesh_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "submesh_inc.F90"
 
end module submesh_oct_m
 
!! Local Variables:
!! mode: f90
!! coding: utf-8
!! End: