refactor to PrettyTables

Author: vyudu

Project.toml

@@ -6,6 +6,7 @@ version = "0.3.37"
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
+PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 RuntimeGeneratedFunctions = "7e49a35a-f44a-4d26-94aa-eba1b4ca6b47"
 StaticArraysCore = "1e83bf80-4336-4d27-bf5d-d5a4f845583c"
 Term = "22787eb5-b846-44ae-b979-8e399b8463ab"
@@ -15,6 +16,7 @@ Accessors = "0.1.36"
 Aqua = "0.8"
 ArrayInterface = "7.9"
 Pkg = "1"
+PrettyTables = "2.4.0"
 RuntimeGeneratedFunctions = "0.5.12"
 SafeTestsets = "0.0.1"
 StaticArrays = "1.9"

src/SymbolicIndexingInterface.jl

@@ -4,7 +4,7 @@ using RuntimeGeneratedFunctions
 import StaticArraysCore: MArray, similar_type
 import ArrayInterface
 using Accessors: @reset
-import Term: Table # for pretty-printing
+using PrettyTables # for pretty printing
 
 RuntimeGeneratedFunctions.init(@__MODULE__)
 

src/parameter_indexing_proxy.jl

@@ -18,19 +18,52 @@ function Base.setindex!(p::ParameterIndexingProxy, val, idx)
     return setp(p.wrapped, idx)(p.wrapped, val)
 end
 
-function Base.show(io::IO, pip::ParameterIndexingProxy; kwargs...)
+function Base.show(io::IO, ::MIME"text/plain", pip::ParameterIndexingProxy)
+    showparams(io, pip; num_rows = 20, show_all = false, scalarize = true)
+end
+
+"""
+    showparams(io::IO, pip::ParameterIndexingProxy; num_rows = 20, show_all = false, scalarize = true, kwargs...)
+
+Method for customizing the table output. Keyword args:
+- num_rows
+- show_all: whether to show all parameters
+- scalarize: whether to scalarize array symbolics in the table output.
+- kwargs... are passed to the pretty_table call.
+"""
+function showparams(io::IO, pip::ParameterIndexingProxy; num_rows = 20, show_all = false, scalarize = true, kwargs...) 
     params = Any[]
     vals = Any[]
     for p in parameter_symbols(pip.wrapped)
-        push!(params, p)
-        val = getp(pip.wrapped, p)(pip.wrapped)
-        push!(vals, val)
+        if symbolic_type(p) === ArraySymbolic() && scalarize
+            val = getp(pip.wrapped, p)(pip.wrapped)
+            for (_p, _v) in zip(collect(p), val)
+                push!(params, _p)
+                push!(vals, _v)
+            end
+        else
+            push!(params, p)
+            val = getp(pip.wrapped, p)(pip.wrapped)
+            push!(vals, val)
+        end
     end
 
-    print(
-          Table([params vals]; 
-                box=:SIMPLE, 
-                header=["Parameter", "Value"], 
-                kwargs...)
-         )
+    num_shown = if show_all
+            length(params)
+        else
+            if num_rows > length(params) 
+                length(params) 
+            else
+                num_rows
+            end
+        end
+
+    pretty_table(io, [params[1:num_shown] vals[1:num_shown]];
+                  header=["Parameter", "Value"], 
+                  kwargs...)
+
+    if num_shown < length(params)
+        println(io, 
+                "$num_shown of $(length(params)) params shown. To show all the parameters, call `showparams(io, ps, show_all = true)`. Adjust the number of rows with the num_rows kwarg. Consult `showparams` docstring for more options.")
+    end
 end