Support FunctionWrappersWrappers in modelingtoolkitize

Author: ChrisRackauckas

Project.toml

@@ -17,6 +17,7 @@ Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 DomainSets = "5b8099bc-c8ec-5219-889f-1d9e522a28bf"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+FunctionWrappersWrappers = "77dc65aa-8811-40c2-897b-53d922fa7daf"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"

src/ModelingToolkit.jl

@@ -26,6 +26,7 @@ using Base: RefValue
 using Combinatorics
 import IfElse
 import Distributions
+import FunctionWrappersWrappers
 
 RuntimeGeneratedFunctions.init(@__MODULE__)
 

src/systems/diffeqs/modelingtoolkitize.jl

@@ -45,7 +45,12 @@ function modelingtoolkitize(prob::DiffEqBase.ODEProblem; kwargs...)
     if DiffEqBase.isinplace(prob)
         rhs = ArrayInterfaceCore.restructure(prob.u0, similar(vars, Num))
         fill!(rhs, 0)
-        prob.f(rhs, vars, params, t)
+        if prob.f isa ODEFunction &&
+           prob.f.f isa FunctionWrappersWrappers.FunctionWrappersWrapper
+            prob.f.f.fw[1].obj[](rhs, vars, params, t)
+        else
+            prob.f(rhs, vars, params, t)
+        end
     else
         rhs = prob.f(vars, params, t)
     end