refactor: implement new hasname function from SII

JuliaSymbolics · Nov 30, 2023 · aa50d32 · aa50d32
1 parent 3afe412
commit aa50d32
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/variable.jl b/src/variable.jl
@@ -409,6 +409,12 @@ getsource(x, val=_fail) = getmetadata(unwrap(x), VariableSource, val)
 SymbolicIndexingInterface.symbolic_type(::Type{<:Symbolics.Num}) = ScalarSymbolic()
 SymbolicIndexingInterface.symbolic_type(::Type{<:Symbolics.Arr}) = ArraySymbolic()
 
+SymbolicIndexingInterface.hasname(x::Union{Num,Arr}) = hasname(unwrap(x))
+
+function SymbolicIndexingInterface.hasname(x::Symbolic)
+    issym(x) || !istree(x) || istree(x) && (issym(operation(x)) || operation(x) == getindex)
+end
+
 SymbolicIndexingInterface.getname(x, val=_fail) = _getname(unwrap(x), val)
 
 function getparent(x, val=_fail)