Skip to content

Array paramter scoping #2352

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 4 commits into from
Closed

Array paramter scoping #2352

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
5bc895f
Add test for scoping of array paramters
contradict Nov 12, 2023
2f13d13
Allow attaching scope metadata to arrays
contradict Nov 12, 2023
560dbc0
Don't drop metadata during array indexing operation
contradict Nov 12, 2023
be22156
Formatting
contradict Nov 12, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 13 additions & 6 deletions src/systems/abstractsystem.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -394,7 +394,7 @@ end
abstract type SymScope end

struct LocalScope <: SymScope end
function LocalScope(sym::Union{Num, Symbolic})
function LocalScope(sym::Union{Num, Symbolic, Symbolics.Arr{Num, N}}) where {N}
apply_to_variables(sym) do sym
setmetadata(sym, SymScope, LocalScope())
end
Expand All @@ -403,7 +403,7 @@ end
struct ParentScope <: SymScope
parent::SymScope
end
function ParentScope(sym::Union{Num, Symbolic})
function ParentScope(sym::Union{Num, Symbolic, Symbolics.Arr{Num, N}}) where {N}
apply_to_variables(sym) do sym
setmetadata(sym, SymScope,
ParentScope(getmetadata(value(sym), SymScope, LocalScope())))
Expand All @@ -414,16 +414,18 @@ struct DelayParentScope <: SymScope
parent::SymScope
N::Int
end
function DelayParentScope(sym::Union{Num, Symbolic}, N)
function DelayParentScope(sym::Union{Num, Symbolic, Symbolics.Arr{Num, M}}, N) where {M}
apply_to_variables(sym) do sym
setmetadata(sym, SymScope,
DelayParentScope(getmetadata(value(sym), SymScope, LocalScope()), N))
end
end
DelayParentScope(sym::Union{Num, Symbolic}) = DelayParentScope(sym, 1)
function DelayParentScope(sym::Union{Num, Symbolic, Symbolics.Arr{Num, N}}) where {N}
DelayParentScope(sym, 1)
end

struct GlobalScope <: SymScope end
function GlobalScope(sym::Union{Num, Symbolic})
function GlobalScope(sym::Union{Num, Symbolic, Symbolics.Arr{Num, N}}) where {N}
apply_to_variables(sym) do sym
setmetadata(sym, SymScope, GlobalScope())
end
Expand Down Expand Up @@ -503,8 +505,13 @@ function namespace_expr(O, sys, n = nameof(sys); ivs = independent_variables(sys
return O
elseif istree(O)
T = typeof(O)
op = operation(O)
args = arguments(O)
if op == getindex && hasmetadata(O, SymScope)
args[1] = setmetadata(args[1], SymScope, getmetadata(O, SymScope))
end
renamed = let sys = sys, n = n, T = T
map(a -> namespace_expr(a, sys, n; ivs)::Any, arguments(O))
map(a -> namespace_expr(a, sys, n; ivs)::Any, args)
end
if isvariable(O)
# Use renamespace so the scope is correct, and make sure to use the
Expand Down
8 changes: 8 additions & 0 deletions test/variable_scope.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,3 +72,11 @@ ps = ModelingToolkit.getname.(parameters(level3))
@test isequal(ps[4], :level2₊level0₊d)
@test isequal(ps[5], :level1₊level0₊e)
@test isequal(ps[6], :f)

@parameters xx[1:2]
arr_p = [ParentScope(xx[1]), xx[2]]
arr0 = ODESystem(Equation[], t, [], arr_p; name = :arr0)
arr1 = ODESystem(Equation[], t, [], []; name = :arr1) ∘ arr0
arr_ps = ModelingToolkit.getname.(parameters(arr1))
@test isequal(arr_ps[1], Symbol("xx"))
@test isequal(arr_ps[2], Symbol("arr0₊xx"))