Fix structural simplification of non-1-indexed variable arrays

Author: hersle

Project.toml

@@ -32,6 +32,7 @@ Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MLStyle = "d8e11817-5142-5d16-987a-aa16d5891078"
 NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
+OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"

src/structural_transformation/symbolics_tearing.jl

@@ -1,3 +1,5 @@
+using OffsetArrays: Origin
+
 # N.B. assumes `slist` and `dlist` are unique
 function substitution_graph(graph, slist, dlist, var_eq_matching)
     ns = length(slist)
@@ -573,7 +575,9 @@ function tearing_reassemble(state::TearingState, var_eq_matching,
         Symbolics.shape(lhs) !== Symbolics.Unknown() || continue
         arg1 = arguments(lhs)[1]
         haskey(obs_arr_subs, arg1) && continue
-        obs_arr_subs[arg1] = [arg1[i] for i in eachindex(arg1)]
+        obs_arr_subs[arg1] = [arg1[i] for i in eachindex(arg1)] # e.g. p => [p[1], p[2]]
+        index_first = eachindex(arg1)[1] 
+        obs_arr_subs[arg1] = Origin(index_first)(obs_arr_subs[arg1]) # respect non-1-indexed arrays
     end
     for i in eachindex(neweqs)
         neweqs[i] = fast_substitute(neweqs[i], obs_arr_subs; operator = Symbolics.Operator)

test/odesystem.jl

@@ -1159,3 +1159,13 @@ for sys in [sys1, sys2]
         @test variable_index(sys, x[i]) == variable_index(sys, x)[i]
     end
 end
+
+@testset "Non-1-indexed variable array (issue #2670)" begin
+    @variables x(t)[0:1] # 0-indexed variable array
+    sys = ODESystem([
+        x[0] ~ 0.0
+        D(x[1]) ~ x[0]
+    ], t, [x], []; name=:sys)
+    @test_nowarn sys = structural_simplify(sys)
+    @test equations(sys) == [D(x[1]) ~ 0.0]
+end