add pretty printing for SumLieAlgebraAction

Author: vitktor

src/interpolation_bases/monomials.jl

@@ -221,6 +221,11 @@ prodpow(
     e::AbstractSparseVector
 ) = prod(view(M,e.nzind,:).^e.nzval; dims=1)
 
+HC.evaluate(
+    mons::MonomialBasis,
+    sample::AbstractVector
+) = [prodpow(sample, mexp) for mexp in mons.mexps]
+
 function HC.evaluate(
     mons::MonomialBasis,
     samples::AbstractMatrix{T}
@@ -233,9 +238,9 @@ function HC.evaluate(
 end
 
 Base.getindex(
-    mons::MonomialBasis{Tv,Ti} where {Tv<:Integer,Ti<:Integer},
+    mons::MonomialBasis{Tv,Ti},
     inds...
-) = MonomialBasis{Tv,Ti}(getindex(mons.mexps, inds...), mons.vars)
+) where {Tv<:Integer,Ti<:Integer} = MonomialBasis{Tv,Ti}(getindex(mons.mexps, inds...), mons.vars)
 Base.getindex(mons::MonomialBasis, ind::Int) = Monomial(mons.mexps[ind], mons.vars)
 
 Base.findfirst(

src/on_varieties/lie-symmetries/lie-algebras/actions.jl

@@ -31,9 +31,11 @@ algebra(a::ScalingLieAction) = a.alg
 variables(a::ScalingLieAction) = a.vars
 exponents(a::ScalingLieAction) = a.alg.exps
 
-function Base.show(io::IO, a::ScalingLieAction)
-    println(io, "ScalingLieAction of $(name(algebra(a)))")
-    print(io, " action:")
+function show_action(io::IO, a::ScalingLieAction; offset::Int=0, show_name::Bool=false)
+    if show_name 
+        print(io, " "^offset, "scaling algebra $(name(algebra(a))): ")
+        offset += 1
+    end
     U = exponents(a)
     if size(U, 1) == 1
         @var λ
@@ -49,14 +51,20 @@ function Base.show(io::IO, a::ScalingLieAction)
         action[j] = collect(zip(vars[nzind], exprs))
     end
     for free_action in action
-        print(io, "\n  ")
+        print(io, "\n", " "^offset)
         for (j, (var, expr)) in enumerate(free_action)
             print(io, var, " ↦ ", expr)
             j < length(free_action) && print(io, ", ")
         end
     end
 end
 
+function Base.show(io::IO, a::ScalingLieAction)
+    println(io, "ScalingLieAction of $(name(algebra(a)))")
+    print(io, " action:")
+    show_action(io, a; offset=2)
+end
+
 function act(elem::LieAlgebraElem{ScalingLieAlgebra}, f::Union{Expression, Monomial}, action::ScalingLieAction)
     X = as_matrix(elem)
     return expand(dot(differentiate(f, variables(action)), -X*variables(action)))
@@ -113,9 +121,19 @@ weights(a::LieAlgebraAction) = weights(a.alg)
 weights(a::LieAlgebraAction, inds...) = weights(a.alg, inds...)
 nweights(a::LieAlgebraAction) = nweights(a.alg)
 
+function show_action(io::IO, a::LieAlgebraAction; offset::Int=0, show_name::Bool=false)
+    if show_name
+        print(io, " "^offset, "$(name(algebra(a))): ")
+        print(io, "[", join([join(vars, ", ") for vars in var_groups(a)], "], ["), "]")
+    else
+        print(io, " "^offset, "[", join([join(vars, ", ") for vars in var_groups(a)], "], ["), "]")
+    end
+end
+
 function Base.show(io::IO, a::LieAlgebraAction)
     println(io, "LieAlgebraAction of $(name(algebra(a)))")
-    print(io, " action: [", join([join(vars, ", ") for vars in var_groups(a)], "], ["), "]")
+    print(io, " action: ")
+    show_action(io, a; offset=0)
 end
 
 function act(elem::LieAlgebraElem{LieAlgebra}, f::Union{Expression, Monomial}, action::LieAlgebraAction)
@@ -157,12 +175,11 @@ Base.getindex(a::SumLieAlgebraAction, i::Int) = actions(a)[i]
 
 function Base.show(io::IO, a::SumLieAlgebraAction)
     println(io, "SumLieAlgebraAction of $(name(algebra(a)))")
-    # for (i, a) in enumerate(actions(g))
-    #     print(io, " action of $(name(algebra(a))): [")
-    #     print(io, join([join(vars, ", ") for vars in var_groups(a)], "], ["), "]")
-    #     i < nsummands(g) && print(io, "\n")
-    # end
-    print(io, " action: ")
+    print(io, " action:")
+    for action in actions(a)
+        println(io)
+        show_action(io, action; offset=2, show_name=true)
+    end
 end
 
 # TODO

src/on_varieties/lie-symmetries/lie-reprs/representation.jl

@@ -46,7 +46,8 @@ function Base.show(io::IO, ::MIME"text/plain", ρ::LieAlgebraRepresentation)
     )
     println(io, " Lie algebra: ", name(algebra(ρ)))
     println(io, " number of isotypic components: ", nisotypic(ρ))
-    print(io, " dimensions of isotypic components: ", join([dim(ic) for ic in isotypic_components(ρ)], ", "))
+    println(io, " dimensions of isotypic components: ", join([dim(ic) for ic in isotypic_components(ρ)], ", "))
+    print(io, " multiplicities of irreducible representations: ", join([nirreducible(ic) for ic in isotypic_components(ρ)], ", "))
 end
 
 # called by print and inside vectors/matrices

src/on_varieties/lie-symmetries/weights.jl

@@ -15,6 +15,7 @@ export WeightStructure,
     sum_weight_structure
 
 
+# TODO: change to SparseVector
 struct WeightVector
     weight::Vector{Int}
     vector::Vector{ComplexF64}
@@ -24,6 +25,7 @@ weight(wv::WeightVector) = wv.weight
 vector(wv::WeightVector) = wv.vector
 Base.setindex!(wv::WeightVector, n::Number, i::Integer) = wv.vector[i] = n
 
+# TODO: change to SparseMatrix
 mutable struct WeightSpace
     weight::Vector{Int}
     matrix::Matrix{ComplexF64}
@@ -172,16 +174,6 @@ end
 
 Base.:∩(ws::Vector{WeightStructure}) = reduce(∩, ws)
 
-# function sum_weight_structure(ws::WeightStructure, d::Integer)
-#     new_weight_spaces = [zeros(ComplexF64, space_dim(ws)*d, size(M, 2)*d) for M in weight_spaces(ws)]
-#     for (M, new_M) in zip(weight_spaces(ws), new_weight_spaces)
-#         for j in 1:d
-#             new_M[(1:space_dim(ws)) .+ (j-1)*space_dim(ws), (1:size(M,2)) .+ (j-1)*size(M,2)] = M
-#         end
-#     end
-#     return WeightStructure(weights(ws), new_weight_spaces)
-# end
-
 function sym_weight_structure(ws::WeightStructure, d::Integer, mexps::Vector{<:SparseVector})
     d = Int(d)
     d == 0 && return WeightStructure([zeros(Int, weight_size(ws))], [[1;;]])