use old tosymbol but disable escaping

Author: isaacsas

src/systems/diffeqs/abstractodesystem.jl

@@ -158,12 +158,13 @@ function DiffEqBase.ODEFunction{iip}(sys::AbstractODESystem, dvs = states(sys),
 
     _M = (u0 === nothing || M == I) ? M : ArrayInterface.restructure(u0 .* u0',M)
 
+    sts = states(sys)
     ODEFunction{iip}(f,
                      jac = _jac === nothing ? nothing : _jac,
                      tgrad = _tgrad === nothing ? nothing : _tgrad,
                      mass_matrix = _M,
                      jac_prototype = sparse ? similar(sys.jac[],Float64) : nothing,
-                     syms = tosymbol.(states(sys)))
+                     syms = tosymbol.(sts, states=sts, escape=false))
 end
 
 """
@@ -219,13 +220,13 @@ function ODEFunctionExpr{iip}(sys::AbstractODESystem, dvs = states(sys),
         tgrad = $_tgrad
         jac = $_jac
         M = $_M
-
+        sts = $(states(sys))
         ODEFunction{$iip}(f,
                          jac = jac,
                          tgrad = tgrad,
                          mass_matrix = M,
                          jac_prototype = $jp_expr,
-                         syms = $(tosymbol.(states(sys))))
+                         syms = $(tosymbol.(sts, states=sts, escape=false)))
     end
     !linenumbers ? striplines(ex) : ex
 end

src/systems/diffeqs/sdesystem.jl

@@ -190,13 +190,14 @@ function DiffEqBase.SDEFunction{iip}(sys::SDESystem, dvs = sys.states, ps = sys.
     M = calculate_massmatrix(sys)
     _M = (u0 === nothing || M == I) ? M : ArrayInterface.restructure(u0 .* u0',M)
 
+    sts = states(sys)
     SDEFunction{iip}(f,g,
                      jac = _jac === nothing ? nothing : _jac,
                      tgrad = _tgrad === nothing ? nothing : _tgrad,
                      Wfact = _Wfact === nothing ? nothing : _Wfact,
                      Wfact_t = _Wfact_t === nothing ? nothing : _Wfact_t,
                      mass_matrix = _M,
-                     syms = tosymbol.(states(sys)))
+                     syms = tosymbol.(sts, states=sts, escape=false))
 end
 
 function DiffEqBase.SDEFunction(sys::SDESystem, args...; kwargs...)
@@ -262,14 +263,14 @@ function SDEFunctionExpr{iip}(sys::SDESystem, dvs = states(sys),
         Wfact = $_Wfact
         Wfact_t = $_Wfact_t
         M = $_M
-
+        sts = $(states(sys))
         SDEFunction{$iip}(f,g,
                          jac = jac,
                          tgrad = tgrad,
                          Wfact = Wfact,
                          Wfact_t = Wfact_t,
                          mass_matrix = M,
-                         syms = $(tosymbol.(states(sys))),kwargs...)
+                         syms = $(tosymbol.(sts, states=sts, escape=false)),kwargs...)
     end
     !linenumbers ? striplines(ex) : ex
 end

src/systems/jumps/jumpsystem.jl

@@ -197,7 +197,9 @@ function DiffEqBase.DiscreteProblem(sys::JumpSystem, u0map, tspan::Tuple,
         p = parammap
     end
     f  = DiffEqBase.DISCRETE_INPLACE_DEFAULT
-    df = DiscreteFunction{true,true}(f, syms=tosymbol.(states(sys)))
+    sts = states(sys)
+    df = DiscreteFunction{true,true}(f, syms=tosymbol.(sts, states=sts, escape=false))
+    @show df.syms
     DiscreteProblem(df, u0, tspan, p; kwargs...)
 end
 
@@ -230,7 +232,8 @@ function DiscreteProblemExpr(sys::JumpSystem, u0map, tspan::Tuple,
         u0 = $u0
         p = $p
         tspan = $tspan
-        df = DiscreteFunction{true,true}(f, syms=$(tosymbol.(states(sys))))
+        sts = $(states(sys))
+        df = DiscreteFunction{true,true}(f, syms=$(tosymbol.(sts, states=sts, escape=false)))
         DiscreteProblem(df, u0, tspan, p; kwargs...)
     end
 end

src/utils.jl

@@ -159,7 +159,6 @@ tovar(s::Sym) = s
 Base.Symbol(x::Union{Num,Symbolic}) = tosymbol(x)
 tosymbol(x; kwargs...) = x
 tosymbol(x::Sym; kwargs...) = nameof(x)
-tosymbol(x::Term; kwargs...) = nameof(operation(x))
 tosymbol(t::Num; kwargs...) = tosymbol(value(t); kwargs...)
 
 """