Implement broadcasting for SparseCoefficients

JuliaAlgebra · Jul 3, 2024 · f87c542 · f87c542
1 parent 2d643d2
commit f87c542
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/sparse_coeffs.jl b/src/sparse_coeffs.jl
@@ -38,6 +38,17 @@ function Base.setindex!(sc::SparseCoefficients{K}, val, key::K) where {K}
     return sc
 end
 
+################
+# Broadcasting #
+################
+# Inspired from `JuMP.Containers.SparseAxisArray`
+
+struct BroadcastStyle{K} end
+
+Base.broadcastable(sc::SparseCoefficients) = sc
+Base.Broadcast.BroadcastStyle(::Type{<:SparseCoefficients{K}}) where {K} = BroadcastStyle{K}()
+Base.axes(sc::SparseCoefficients) = (sc.basis_elements,)
+
 function Base.zero(sc::SparseCoefficients)
     return SparseCoefficients(empty(keys(sc)), empty(values(sc)))
 end