feat: add new all-symbols methods from SII

SciML · Dec 18, 2023 · 71c23ce · 71c23ce
1 parent ab2c452
commit 71c23ce
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -98,7 +98,7 @@ SimpleNonlinearSolve = "0.1.0, 1"
 SparseArrays = "1"
 SpecialFunctions = "0.7, 0.8, 0.9, 0.10, 1.0, 2"
 StaticArrays = "0.10, 0.11, 0.12, 1.0"
-SymbolicIndexingInterface = "0.3"
+SymbolicIndexingInterface = "0.3.1"
 SymbolicUtils = "1.0"
 Symbolics = "5.7"
 URIs = "1"

diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
@@ -278,6 +278,14 @@ SymbolicIndexingInterface.is_time_dependent(::AbstractTimeIndependentSystem) = f
 
 SymbolicIndexingInterface.constant_structure(::AbstractSystem) = true
 
+function SymbolicIndexingInterface.all_solvable_symbols(sys::AbstractSystem)
+    vcat(variable_symbols(sys), getproperty.(observed(sys), :lhs))
+end
+
+function SymbolicIndexingInterface.all_symbols(sys::AbstractSystem)
+    vcat(all_solvable_symbols(sys), parameter_symbols(sys), independent_variable_symbols(sys))
+end
+
 iscomplete(sys::AbstractSystem) = isdefined(sys, :complete) && getfield(sys, :complete)
 
 """