diff --git a/src/ReachSets/DiscretePost/DiscretePost.jl b/src/ReachSets/DiscretePost/DiscretePost.jl index fdc7a57e..310f93d7 100644 --- a/src/ReachSets/DiscretePost/DiscretePost.jl +++ b/src/ReachSets/DiscretePost/DiscretePost.jl @@ -116,9 +116,18 @@ end # --- default methods for handling assignments --- +# default implementation: use 'apply' from MathematicalSystems function apply_assignment(𝒫::DiscretePost, constrained_map::AbstractMap, R⋂G::LazySet; kwargs...) return apply(constrained_map, R⋂G) end + +# for reset maps: return a lazy ResetMap from LazySets +function apply_assignment(𝒫::DiscretePost, + constrained_map::ConstrainedResetMap, + R⋂G::LazySet; + kwargs...) + return LazySets.ResetMap(R⋂G, constrained_map.dict) +end