simplify MPIsafe write and read for InfiniteMPS

Author: AFeuerpfeil

src/SharedMPS/mpipropertywrapper.jl

@@ -0,0 +1,21 @@
+struct MPIPropertyWrapper{A<:AbstractVector}
+    property::A
+end
+
+Base.parent(wrapper::MPIPropertyWrapper) = wrapper.property
+function Base.setindex!(wrapper::MPIPropertyWrapper{B}, A::C, i::Int) where {B<:AbstractVector, C}
+    A_shared = MPI.bcast(A, MPI.COMM_WORLD) ## TODO: Write own chunked version
+    wrapper.property[i] = A_shared
+    return parent(wrapper)
+end
+
+Base.getindex(wrapper::MPIPropertyWrapper, i::Int) = parent(wrapper)[i]
+Base.getindex(wrapper::MPIPropertyWrapper, I) = parent(wrapper)[I]
+
+function Base.show(io::IO, ::MIME"text/plain", op::MPIPropertyWrapper)
+    show(io, MIME"text/plain"(), parent(op))
+end
+Base.show(io::IO, op::MPIOperator) = show(convert(IOContext, io), op)
+function Base.show(io::IOContext, op::MPIOperator)
+    show(io, parent(op))
+end

src/SharedMPS/sharedmps.jl

@@ -0,0 +1,5 @@
+function Base.getproperty(ψ::InfiniteMPS, prop::Symbol)
+    return MPIPropertyWrapper(getfield(ψ, prop))
+end
+
+## Issue: Base.getproperty is already overloaded for MultilineMPS, QP, MultilineQP, WindowMPS, FiniteMPS

src/SharedState/finitemps.jl

@@ -7,5 +7,4 @@ include("MPIOperator/environments.jl")
 include("MPIOperator/ortho.jl")
 include("MPIOperator/transfermatrix.jl")
 
-include("multiprocessing/mpi/helper.jl")
-include("multiprocessing/mpi/mailbox.jl")
+include("multiprocessing/mpi/helper.jl")

src/SharedState/infinitemps.jl

@@ -22,8 +22,8 @@ function mpi_rank()
         return 0
     end
 end
-function mpi_is_root()
-    return mpi_rank() == 0
+function mpi_is_root(root=Cint(0))
+    return mpi_rank() == root
 end
 
 function mpi_execute_on_root(F::Function, args...; blocking::Bool, kwargs...)