Merge pull request #3035 from SciML/revert-3027-double_ss_error

Throw an error if structural simplification is applied twice (take 2)

Author: YingboMa

.github/workflows/Tests.yml

@@ -28,8 +28,6 @@ jobs:
         group:
           - InterfaceI
           - InterfaceII
-          - SymbolicIndexingInterface
-          - Extended
           - Extensions
           - Downstream
           - RegressionI

src/structural_transformation/StructuralTransformations.jl

@@ -24,7 +24,7 @@ using ModelingToolkit: ODESystem, AbstractSystem, var_from_nested_derivative, Di
                        invalidate_cache!, Substitutions, get_or_construct_tearing_state,
                        filter_kwargs, lower_varname_with_unit, setio, SparseMatrixCLIL,
                        get_fullvars, has_equations, observed,
-                       Schedule
+                       Schedule, schedule
 
 using ModelingToolkit.BipartiteGraphs
 import .BipartiteGraphs: invview, complete

src/structural_transformation/symbolics_tearing.jl

@@ -125,6 +125,7 @@ function tearing_substitution(sys::AbstractSystem; kwargs...)
     neweqs = full_equations(sys::AbstractSystem; kwargs...)
     @set! sys.eqs = neweqs
     @set! sys.substitutions = nothing
+    @set! sys.schedule = nothing
 end
 
 function tearing_assignments(sys::AbstractSystem)
@@ -615,6 +616,7 @@ function tearing_reassemble(state::TearingState, var_eq_matching,
     if sys isa ODESystem
         @set! sys.schedule = Schedule(var_eq_matching, dummy_sub)
     end
+    sys = schedule(sys)
     @set! state.sys = sys
     @set! sys.tearing_state = state
     return invalidate_cache!(sys)

src/systems/abstractsystem.jl

@@ -849,6 +849,33 @@ iscomplete(sys::AbstractSystem) = isdefined(sys, :complete) && getfield(sys, :co
 """
 $(TYPEDSIGNATURES)
 
+Mark a system as scheduled. It is only intended in compiler internals. A system
+is scheduled after tearing based simplifications where equations are converted
+into assignments.
+"""
+function schedule(sys::AbstractSystem)
+    has_schedule(sys) ? sys : (@set! sys.isscheduled = true)
+end
+
+"""
+$(TYPEDSIGNATURES)
+
+If a system is scheduled, then changing its equations, variables, and
+parameters is no longer legal.
+"""
+function isscheduled(sys::AbstractSystem)
+    if has_schedule(sys)
+        get_schedule(sys) !== nothing
+    elseif has_isscheduled(sys)
+        get_isscheduled(sys)
+    else
+        false
+    end
+end
+
+"""
+$(TYPEDSIGNATURES)
+
 Mark a system as completed. If a system is complete, the system will no longer
 namespace its subsystems or variables, i.e. `isequal(complete(sys).v.i, v.i)`.
 """
@@ -907,7 +934,8 @@ for prop in [:eqs
              :split_idxs
              :parent
              :index_cache
-             :is_scalar_noise]
+             :is_scalar_noise
+             :isscheduled]
     fname_get = Symbol(:get_, prop)
     fname_has = Symbol(:has_, prop)
     @eval begin

src/systems/diffeqs/sdesystem.jl

@@ -133,13 +133,15 @@ struct SDESystem <: AbstractODESystem
     be `true` when `noiseeqs isa Vector`. 
     """
     is_scalar_noise::Bool
+    isscheduled::Bool
 
     function SDESystem(tag, deqs, neqs, iv, dvs, ps, tspan, var_to_name, ctrls, observed,
             tgrad,
             jac,
             ctrl_jac, Wfact, Wfact_t, name, systems, defaults, connector_type,
             cevents, devents, parameter_dependencies, metadata = nothing, gui_metadata = nothing,
-            complete = false, index_cache = nothing, parent = nothing, is_scalar_noise = false;
+            complete = false, index_cache = nothing, parent = nothing, is_scalar_noise = false,
+            isscheduled = false;
             checks::Union{Bool, Int} = true)
         if checks == true || (checks & CheckComponents) > 0
             check_independent_variables([iv])
@@ -162,7 +164,8 @@ struct SDESystem <: AbstractODESystem
         new(tag, deqs, neqs, iv, dvs, ps, tspan, var_to_name, ctrls, observed, tgrad, jac,
             ctrl_jac,
             Wfact, Wfact_t, name, systems, defaults, connector_type, cevents, devents,
-            parameter_dependencies, metadata, gui_metadata, complete, index_cache, parent, is_scalar_noise)
+            parameter_dependencies, metadata, gui_metadata, complete, index_cache, parent, is_scalar_noise,
+            isscheduled)
     end
 end
 

src/systems/discrete_system/discrete_system.jl

@@ -91,14 +91,16 @@ struct DiscreteSystem <: AbstractTimeDependentSystem
     The hierarchical parent system before simplification.
     """
     parent::Any
+    isscheduled::Bool
 
     function DiscreteSystem(tag, discreteEqs, iv, dvs, ps, tspan, var_to_name,
             observed,
             name,
             systems, defaults, preface, connector_type, parameter_dependencies = Equation[],
             metadata = nothing, gui_metadata = nothing,
             tearing_state = nothing, substitutions = nothing,
-            complete = false, index_cache = nothing, parent = nothing;
+            complete = false, index_cache = nothing, parent = nothing,
+            isscheduled = false;
             checks::Union{Bool, Int} = true)
         if checks == true || (checks & CheckComponents) > 0
             check_independent_variables([iv])
@@ -113,7 +115,7 @@ struct DiscreteSystem <: AbstractTimeDependentSystem
             systems,
             defaults,
             preface, connector_type, parameter_dependencies, metadata, gui_metadata,
-            tearing_state, substitutions, complete, index_cache, parent)
+            tearing_state, substitutions, complete, index_cache, parent, isscheduled)
     end
 end
 

src/systems/jumps/jumpsystem.jl

@@ -114,12 +114,13 @@ struct JumpSystem{U <: ArrayPartition} <: AbstractTimeDependentSystem
     Cached data for fast symbolic indexing.
     """
     index_cache::Union{Nothing, IndexCache}
+    isscheduled::Bool
 
     function JumpSystem{U}(tag, ap::U, iv, unknowns, ps, var_to_name, observed, name,
             systems,
             defaults, connector_type, devents, parameter_dependencies,
             metadata = nothing, gui_metadata = nothing,
-            complete = false, index_cache = nothing;
+            complete = false, index_cache = nothing, isscheduled = false;
             checks::Union{Bool, Int} = true) where {U <: ArrayPartition}
         if checks == true || (checks & CheckComponents) > 0
             check_independent_variables([iv])
@@ -132,7 +133,7 @@ struct JumpSystem{U <: ArrayPartition} <: AbstractTimeDependentSystem
         end
         new{U}(tag, ap, iv, unknowns, ps, var_to_name, observed, name, systems, defaults,
             connector_type, devents, parameter_dependencies, metadata, gui_metadata,
-            complete, index_cache)
+            complete, index_cache, isscheduled)
     end
 end
 function JumpSystem(tag, ap, iv, states, ps, var_to_name, args...; kwargs...)

src/systems/nonlinear/nonlinearsystem.jl

@@ -89,21 +89,22 @@ struct NonlinearSystem <: AbstractTimeIndependentSystem
     The hierarchical parent system before simplification.
     """
     parent::Any
+    isscheduled::Bool
 
     function NonlinearSystem(tag, eqs, unknowns, ps, var_to_name, observed, jac, name,
             systems,
             defaults, connector_type, parameter_dependencies = Equation[], metadata = nothing,
             gui_metadata = nothing,
             tearing_state = nothing, substitutions = nothing,
-            complete = false, index_cache = nothing, parent = nothing; checks::Union{
-                Bool, Int} = true)
+            complete = false, index_cache = nothing, parent = nothing,
+            isscheduled = false; checks::Union{Bool, Int} = true)
         if checks == true || (checks & CheckUnits) > 0
             u = __get_unit_type(unknowns, ps)
             check_units(u, eqs)
         end
         new(tag, eqs, unknowns, ps, var_to_name, observed, jac, name, systems, defaults,
             connector_type, parameter_dependencies, metadata, gui_metadata, tearing_state,
-            substitutions, complete, index_cache, parent)
+            substitutions, complete, index_cache, parent, isscheduled)
     end
 end
 

src/systems/optimization/optimizationsystem.jl

@@ -64,10 +64,12 @@ struct OptimizationSystem <: AbstractOptimizationSystem
     The hierarchical parent system before simplification.
     """
     parent::Any
+    isscheduled::Bool
 
     function OptimizationSystem(tag, op, unknowns, ps, var_to_name, observed,
             constraints, name, systems, defaults, metadata = nothing,
-            gui_metadata = nothing, complete = false, index_cache = nothing, parent = nothing;
+            gui_metadata = nothing, complete = false, index_cache = nothing, parent = nothing,
+            isscheduled = false;
             checks::Union{Bool, Int} = true)
         if checks == true || (checks & CheckUnits) > 0
             u = __get_unit_type(unknowns, ps)
@@ -77,7 +79,7 @@ struct OptimizationSystem <: AbstractOptimizationSystem
         end
         new(tag, op, unknowns, ps, var_to_name, observed,
             constraints, name, systems, defaults, metadata, gui_metadata, complete,
-            index_cache, parent)
+            index_cache, parent, isscheduled)
     end
 end
 

src/systems/systems.jl

@@ -29,7 +29,7 @@ function structural_simplify(
         sys::AbstractSystem, io = nothing; simplify = false, split = true,
         allow_symbolic = false, allow_parameter = true, conservative = false, fully_determined = true,
         kwargs...)
-    iscomplete(sys) && throw(RepeatedStructuralSimplificationError())
+    isscheduled(sys) && throw(RepeatedStructuralSimplificationError())
     newsys′ = __structural_simplify(sys, io; simplify,
         allow_symbolic, allow_parameter, conservative, fully_determined,
         kwargs...)