modify show for AbstractSpace and CompoundDimensions

Author: ytdHuang

src/qobj/arithmetic_and_attributes.jl

@@ -53,7 +53,7 @@ end
 function check_mul_dims(from::SVector{NA,AbstractSpace}, to::SVector{NB,AbstractSpace}) where {NA,NB}
     (from != to) && throw(
         DimensionMismatch(
-            "The quantum object with dims = $from can not multiply a quantum object with dims = $to on the right-hand side.",
+            "The quantum object with (right) dims = $(dims_to_list(from)) can not multiply a quantum object with (left) dims = $(dims_to_list(to)) on the right-hand side.",
         ),
     )
     return nothing

src/qobj/dimensions.jl

@@ -46,7 +46,7 @@ function CompoundDimensions(dims::Union{AbstractVector{T},NTuple{N,T}}) where {T
     )
 end
 
-Base.show(io::IO, D::CompoundDimensions) = print(io, "[", D.to, ", ", D.from, "]")
+#Base.show(io::IO, D::CompoundDimensions) = print(io, "[", D.to, ", ", D.from, "]")
 
 _gen_dims(dims::AbstractDimensions) = dims
 _gen_dims(dims::Union{AbstractVector{T},NTuple{N,T}}) where {T<:Integer,N} = Dimensions(dims)
@@ -59,6 +59,13 @@ dims_to_list(dimsvec::SVector{N,AbstractSpace}) where {N} = vcat(map(dims_to_lis
 dims_to_list(dims::Dimensions) = dims_to_list(dims.to)
 dims_to_list(dims::CompoundDimensions) = SVector{2}(dims_to_list(dims.to), dims_to_list(dims.from))
 
+# for printing `dims_to_list(CompoundDimensions)` 
+function Base.show(io::IO, svec::SVector{2,SVector{N,T}}) where {N,T<:Int}
+    print(io, "[")
+    join(io, svec, ", ")
+    print(io, "]")
+end
+
 Base.length(::AbstractDimensions{N}) where {N} = N
 
 # need to specify return type `Int` for `_get_space_size`

src/qobj/space.jl

@@ -2,6 +2,13 @@ export AbstractSpace, Field, Space
 
 abstract type AbstractSpace end
 
+# for printing `AbstractDimensions` 
+function Base.show(io::IO, svec::SVector{N,AbstractSpace}) where {N}
+    print(io, "[")
+    join(io, svec, ", ")
+    print(io, "]")
+end
+
 # this replaces Space(1), so that we don't need to store the value `1`
 struct Field <: AbstractSpace end
 Base.getproperty(s::Field, key::Symbol) = getproperty(s, Val{key}())