rework tensor show again

Author: lkdvos

src/tensors/abstracttensor.jl

@@ -503,3 +503,33 @@ function Base.convert(::Type{Array}, t::AbstractTensorMap)
         return A
     end
 end
+
+# Show and friends
+# ----------------
+Base.dims2string(V::HomSpace) = join(dim.(codomain(V)), '×') * "←" * join(dim.(domain(V)), '×')
+
+Base.summary(io::IO, t::AbstractTensorMap) = tensor_summary(io, t, space(t))
+
+function tensor_summary(io::IO, t, V)
+    print(io, Base.dims2string(V), " ")
+    Base.showarg(io, t, true)
+    return nothing
+end
+
+# Human-readable:
+function Base.show(io::IO, ::MIME"text/plain", t::AbstractTensorMap)
+    # 1) show summary: typically d₁×d₂×… ← d₃×d₄×… $(typeof(t)):
+    summary(io, t)
+    println(io, ":")
+
+    # 2) show spaces
+    # println(io, " space(t):")
+    print(io, "  ", space(t))
+
+    # 3) [optional]: show data
+    get(io, :compact, true) && return nothing
+    ioc = IOContext(io, :typeinfo => sectortype(t))
+    println(io, "\n\n blocks(t):")
+    show_blocks(io, MIME"text/plain"(), blocks(t))
+    return nothing
+end

src/tensors/blockiterator.jl

@@ -44,3 +44,21 @@ function foreachblock(f, t::AbstractTensorMap; scheduler = nothing)
     end
     return nothing
 end
+
+function show_blocks(io, mime::MIME"text/plain", iter)
+    first = true
+    for (c, b) in iter
+        first || print(io, "\n\n")
+        print(io, " * ", c, " => ")
+        show(io, mime, b)
+        first = false
+    end
+    return nothing
+end
+
+function show_blocks(io, iter)
+    print(io, "(")
+    join(io, iter, ", ")
+    print(io, ")")
+    return nothing
+end

src/tensors/tensor.jl

@@ -566,42 +566,19 @@ end
 
 # Show
 #------
-function Base.summary(io::IO, t::TensorMap)
-    return print(io, "TensorMap(", space(t), ")")
+function type_repr(::Type{TensorMap{T, S, N₁, N₂, A}}) where {T, S, N₁, N₂, A}
+    return "TensorMap{$T, $(type_repr(S)), $N₁, $N₂, $A}"
 end
-Base.show(io::IO, t::TensorMap) = summary(io, t)
 
-function Base.show(io::IO, ::MIME"text/plain", t::TensorMap)
-    if get(io, :compact, false)
-        print(io, "TensorMap(", space(t), ")")
-        return
-    end
-    println(io, "TensorMap(", space(t), "):")
-
-    for (c, b) in blocks(t)
-        print(io, "* block for charge ", c, ":")
-        summary(io, b)
-        println(io)
-    end
-    # if sectortype(t) == Trivial
-    #     Base.print_array(io, t[])
-    #     println(io)
-    # elseif FusionStyle(sectortype(t)) isa UniqueFusion
-    #     for (f₁, f₂) in fusiontrees(t)
-    #         println(io, "* Data for sector ", f₁.uncoupled, " ← ", f₂.uncoupled, ":")
-    #         Base.print_array(io, t[f₁, f₂])
-    #         println(io)
-    #     end
-    # else
-    #     for (f₁, f₂) in fusiontrees(t)
-    #         println(io, "* Data for fusiontree ", f₁, " ← ", f₂, ":")
-    #         Base.print_array(io, t[f₁, f₂])
-    #         println(io)
-    #     end
-    # end
+function Base.showarg(io::IO, t::TensorMap, toplevel::Bool)
+    !toplevel && print(io, "::")
+    print(io, type_repr(typeof(t)))
     return nothing
 end
 
+Base.show(io::IO, t::TensorMap) =
+    print(io, type_repr(typeof(t)), "(", t.data, ", ", space(t), ")")
+
 # Complex, real and imaginary parts
 #-----------------------------------
 for f in (:real, :imag, :complex)