Merge pull request #3201 from BenChung/additional-passes

Add a mechanism to add additional passes to the end of structural simplification before completion.

Author: ChrisRackauckas

src/systems/systems.jl

@@ -26,7 +26,7 @@ topological sort of the observed equations in `sys`.
 + `fully_determined=true` controls whether or not an error will be thrown if the number of equations don't match the number of inputs, outputs, and equations.
 """
 function structural_simplify(
-        sys::AbstractSystem, io = nothing; simplify = false, split = true,
+        sys::AbstractSystem, io = nothing; additional_passes = [], simplify = false, split = true,
         allow_symbolic = false, allow_parameter = true, conservative = false, fully_determined = true,
         kwargs...)
     isscheduled(sys) && throw(RepeatedStructuralSimplificationError())
@@ -46,6 +46,9 @@ function structural_simplify(
             not yet supported.
         """)
     end
+    for pass in additional_passes
+        newsys = pass(newsys)
+    end
     if newsys isa ODESystem || has_parent(newsys)
         @set! newsys.parent = complete(sys; split, flatten = false)
     end

test/structural_transformation/utils.jl

@@ -152,3 +152,12 @@ end
         end
     end
 end
+
+@testset "additional passes" begin
+    @variables x(t) y(t)
+    @named sys = ODESystem([D(x) ~ x, y ~ x + t], t)
+    value = Ref(0)
+    pass(sys; kwargs...) = (value[] += 1; return sys)
+    structural_simplify(sys; additional_passes = [pass])
+    @test value[] == 1
+end