More refined pretty printing

Author: fverdugo

src/debug_array.jl

@@ -90,6 +90,9 @@ function Base.show(io::IO,k::MIME"text/plain",data::DebugArray)
         println(io,"$index = $(data.items[i])")
     end
 end
+function Base.show(io::IO,data::DebugArray)
+    print(io,"DebugArray(…)")
+end
 getany(a::DebugArray) = getany(a.items)
 
 function Base.similar(a::DebugArray,::Type{T},dims::Dims) where T

src/mpi_array.jl

@@ -196,6 +196,11 @@ function Base.show(io::IO,k::MIME"text/plain",data::MPIArray)
         MPI.Barrier(data.comm)
     end
 end
+function Base.show(io::IO,data::MPIArray)
+    if MPI.Comm_rank(data.comm) == 0
+        print(io,"MPIArray(…)")
+    end
+end
 
 getany(a::MPIArray) = a.item
 i_am_main(a::MPIArray) = MPI.Comm_rank(a.comm)+1 == MAIN

src/p_sparse_matrix.jl

@@ -885,6 +885,9 @@ function Base.show(io::IO,k::MIME"text/plain",data::PSparseMatrix)
         println(io,"$(m)×$(n) PSparseMatrix partitioned into $np parts of type $T")
     end
 end
+function Base.show(io::IO,data::PSparseMatrix)
+    print(io,"PSparseMatrix(…)")
+end
 
 """
     local_values(a::PSparseMatrix)

src/p_vector.jl

@@ -230,6 +230,9 @@ function Base.show(io::IO,k::MIME"text/plain",data::PVector)
         println(io,"$n-element PVector partitioned into $np parts of type $T")
     end
 end
+function Base.show(io::IO,data::PVector)
+    print(io,"PVector(…)")
+end
 
 function p_vector_cache(vector_partition,index_partition)
     p_vector_cache_impl(eltype(vector_partition),vector_partition,index_partition)

test/fem_example.jl

@@ -293,6 +293,7 @@ function fem_example(distribute)
     b,cacheb = pvector(II,VV,dof_partition;reuse=true) |> fetch
     psparse!(A,V,cacheA) |> wait
     pvector!(b,VV,cacheb) |> wait
+    display((A,cacheA))
 
     x = IterativeSolvers.cg(A,b,verbose=i_am_main(rank))
     @test norm(x-x̂) < 1.0e-5

test/p_sparse_matrix_tests.jl

@@ -296,6 +296,9 @@ function p_sparse_matrix_tests(distribute)
     A,b,cache = psystem(I,J,V,I2,V2,rows,cols,reuse=true) |> fetch
     psystem!(A,b,V,V2,cache) |> wait
 
+    display((A,A))
+    display((b,b))
+
     #Ar = renumber(A)
     #br,cache = renumber(b,partition(axes(Ar,1)),reuse=true)
     #cr = Ar\br