Parallel kslice

src/postw90/comms.F90

@@ -99,6 +99,7 @@ module w90_comms
   interface comms_gatherv
 !     module procedure comms_gatherv_int    ! to be done
      module procedure comms_gatherv_real
+     module procedure comms_gatherv_logical
 !     module procedure comms_gatherv_cmplx
   end interface comms_gatherv
 
@@ -846,6 +847,36 @@ subroutine comms_gatherv_real(array,localcount,rootglobalarray,counts,displs)
 
   end subroutine comms_gatherv_real
 
+  subroutine comms_gatherv_logical(array,localcount,rootglobalarray,counts,displs)
+    !! Gather real data to root node
+    implicit none
+
+    logical, intent(inout)           :: array
+    !! local array for sending data
+    integer, intent(in)                       :: localcount
+    !! localcount elements will be sent to the root node
+    logical, intent(inout)           :: rootglobalarray
+    !! array on the root node to which data will be sent
+    integer, dimension(num_nodes), intent(in) :: counts
+    !! how data should be partitioned, see MPI documentation or
+    !! function comms_array_split
+    integer, dimension(num_nodes), intent(in) :: displs
+
+#ifdef MPI
+    integer :: error
+
+    call MPI_gatherv(array,localcount,MPI_logical,rootglobalarray,counts,&
+         displs,MPI_logical,root_id,mpi_comm_world,error)
+
+    if(error.ne.MPI_success) then
+       call io_error('Error in comms_gatherv_logical')
+    end if
+#else
+!    rootglobalarray(1:localcount)=array(1:localcount)
+#endif
+
+  end subroutine comms_gatherv_logical
+
   subroutine comms_scatterv_real(array,localcount,rootglobalarray,counts,displs)
     !! Scatter data from root node
     implicit none