Implemented product of NBodyMatrixElements

JuliaAtoms · Nov 27, 2023 · 2aa72f7 · 2aa72f7
1 parent 232504c
commit 2aa72f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/nbody_matrix_elements.jl b/src/nbody_matrix_elements.jl
@@ -353,6 +353,9 @@ Base.:(*)(a::NBodyMatrixElement, b::Union{Number,NBodyTerm,NBodyTermFactor}) =
 Base.:(*)(a::Union{Number,NBodyTerm,NBodyTermFactor}, b::NBodyMatrixElement) =
     NBodyMatrixElement([a*t for t in b.terms])
 
+Base.:(*)(a::NBodyMatrixElement, b::NBodyMatrixElement) =
+    NBodyMatrixElement(vec(map(prod, Iterators.product(a.terms, b.terms))))
+
 Base.:(-)(f::NBodyMatrixElement) = (-1)*f
 
 Base.:(-)(a::Union{NBodyTerm,NBodyTermFactor,NBodyMatrixElement},