Rough fix

Author: YingboMa

Project.toml

@@ -1,7 +1,7 @@
 name = "DiffEqOperators"
 uuid = "9fdde737-9c7f-55bf-ade8-46b3f136cc48"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "4.14.0"
+version = "4.15.0"
 
 [deps]
 BandedMatrices = "aae01518-5342-5314-be14-df237901396f"
@@ -24,7 +24,7 @@ DiffEqBase = "6.4.1"
 ForwardDiff = "0.10"
 LazyArrays = "0.17, 0.18, 0.19"
 LazyBandedMatrices = "0.3"
-ModelingToolkit = "0.10.0, 1.0, 2.0, 3.0"
+ModelingToolkit = "4"
 NNlib = "0.6, 0.7"
 StaticArrays = "0.10, 0.11, 0.12"
 julia = "1"

src/MOL_discretization.jl

@@ -10,8 +10,8 @@ function get_bcs(bcs,tdomain,domain)
     lhs_deriv_depvars_bcs = Dict()
     no_bcs = size(bcs,1)
     for i = 1:no_bcs
-        var = bcs[i].lhs.op
-        if var isa Variable
+        var = ModelingToolkit.operation(bcs[i].lhs)
+        if var isa Sym
             var = var.name
             if !haskey(lhs_deriv_depvars_bcs,var)
                 lhs_deriv_depvars_bcs[var] = Array{Expr}(undef,3)
@@ -24,10 +24,10 @@ function get_bcs(bcs,tdomain,domain)
             elseif isequal(bcs[i].lhs.args[2],domain.upper) # u(t,x=x_final)
                 j = 3
             end
-            if bcs[i].rhs isa ModelingToolkit.Constant
-                lhs_deriv_depvars_bcs[var][j] = :(var=$(bcs[i].rhs.value))
+            if !(bcs[i].rhs isa ModelingToolkit.Symbolic)
+                lhs_deriv_depvars_bcs[var][j] = :(var=$(bcs[i].rhs))
             else
-                lhs_deriv_depvars_bcs[var][j] = Expr(bcs[i].rhs)
+                lhs_deriv_depvars_bcs[var][j] = toexpr(bcs[i].rhs)
             end
         end
     end
@@ -44,12 +44,12 @@ end
 
 function discretize_2(input,deriv_order,approx_order,dx,X,len,
                       deriv_var,dep_var_idx,indep_var_idx)
-    if input isa ModelingToolkit.Constant
+    if !(input isa ModelingToolkit.Symbolic)
         return :($(input.value))
-    elseif input isa Operation
-        if input.op isa Variable
+    else
+        if input isa Sym
             expr = :(0.0)
-            var = input.op.name
+            var = nameof(input)
             if haskey(indep_var_idx,var) # ind. var.
                 if var != :(t)
                     i = indep_var_idx[var]
@@ -76,22 +76,24 @@ function discretize_2(input,deriv_order,approx_order,dx,X,len,
             end
             return expr
         elseif input.op isa Differential
-            var = input.op.x.op.name
+            var = nameof(input.op.x)
             push!(deriv_var,var)
             return discretize_2(input.args[1],deriv_order+1,approx_order,dx,X,
                                 len,deriv_var,dep_var_idx,indep_var_idx)
             pop!(deriv_var,var)
         else
+            name = ModelingToolkit.operation(input).name
             if size(input.args,1) == 1
                 aux = discretize_2(input.args[1],deriv_order,approx_order,dx,X,
                                    len,deriv_var,dep_var_idx,indep_var_idx)
-                return :(broadcast($(input.op), $aux))
+                @show input.op
+                return :(broadcast($name, $aux))
             else
                 aux_1 = discretize_2(input.args[1],deriv_order,approx_order,dx,X,
                                      len,deriv_var,dep_var_idx,indep_var_idx)
                 aux_2 = discretize_2(input.args[2],deriv_order,approx_order,dx,X,
                                      len,deriv_var,dep_var_idx,indep_var_idx)
-                return :(broadcast($(input.op), $aux_1, $aux_2))
+                return :(broadcast($name, $aux_1, $aux_2))
             end
         end
     end
@@ -132,7 +134,7 @@ function DiffEqBase.discretize(pdesys::PDESystem,discretization::MOLFiniteDiffer
     len = Array{Any}(undef,no_indep_vars)
     k = 0
     for i = 1:no_indep_vars
-        var = pdesys.domain[i].variables.op.name
+        var = nameof(pdesys.domain[i].variables)
         indep_var_idx[var] = i
         domain[i] = pdesys.domain[i].domain
         if var != :(t)
@@ -167,10 +169,10 @@ function DiffEqBase.discretize(pdesys::PDESystem,discretization::MOLFiniteDiffer
     no_dep_vars = size(eqs,1)
     for j = 1:no_dep_vars
         input = eqs[j].lhs
-        if input.op isa Variable
-            var = input.op.name
+        if input.op isa Sym
+            var = nameof(input)
         else #var isa Differential
-            var = input.args[1].op.name
+            var = nameof(ModelingToolkit.operation(input.args[1]))
             lhs_deriv_depvars[var] = j
         end
         dep_var_idx[var] = j
@@ -179,6 +181,7 @@ function DiffEqBase.discretize(pdesys::PDESystem,discretization::MOLFiniteDiffer
         aux = discretize_2( eqs[j].rhs,0,approx_order,dx,X,len,
                             [],dep_var_idx,indep_var_idx)
         # TODO: is there a better way to convert an Expr into a Function?
+        @show aux
         dep_var_disc[var] = @eval (Q,u,t) -> $aux
     end