Implement mutable addition

Author: blegat

Project.toml

@@ -17,6 +17,7 @@ ExproniconLite = "55351af7-c7e9-48d6-89ff-24e801d99491"
 LabelledArrays = "2ee39098-c373-598a-b85f-a56591580800"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MultivariatePolynomials = "102ac46a-7ee4-5c85-9060-abc95bfdeaa3"
+MutableArithmetics = "d8a4904e-b15c-11e9-3269-09a3773c0cb0"
 NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
@@ -50,6 +51,7 @@ DynamicPolynomials = "0.5, 0.6"
 ExproniconLite = "0.10.14"
 LabelledArrays = "1.5"
 MultivariatePolynomials = "0.5"
+MutableArithmetics = "1.6.4"
 NaNMath = "0.3, 1.1.2"
 OhMyThreads = "0.7"
 ReverseDiff = "1"

src/SymbolicUtils.jl

@@ -82,5 +82,7 @@ include("code.jl")
 # Adjoints
 include("adjoints.jl")
 
+# Mutable Arithmetics
+include("mutable_arithmetics.jl")
 
 end # module

src/mutable_arithmetics.jl

@@ -0,0 +1,17 @@
+import MutableArithmetics as MA
+
+function MA.operate!!(::typeof(+), a::BasicSymbolic, b::BasicSymbolic)
+    if SymbolicUtils.isadd(a)
+        if SymbolicUtils.isadd(b)
+            a.coeff = MA.add!!(a.coeff, b.coeff)
+            for (k, v) in b.dict
+                a.dict[k] = MA.add!!(get(a.dict, k, 0)::Int, v)
+            end
+        else
+            a.dict[b] = MA.add!!(get(a.dict, b, 0), 1)
+        end
+        return a
+    else
+        return a + b
+    end
+end