From 2e909d67876c47d497653102a07b7dc913b83f3d Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Wed, 14 Sep 2022 16:29:42 +0200 Subject: Adding a AOIScene copy method with exclude parameter. --- src/argaze/AreaOfInterest/AOIFeatures.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py index f74920e..087b3f3 100644 --- a/src/argaze/AreaOfInterest/AOIFeatures.py +++ b/src/argaze/AreaOfInterest/AOIFeatures.py @@ -186,6 +186,19 @@ class AOIScene(): return max_bounds - min_bounds + def copy(self, exclude=[]): + """Copy scene partly excluding aoi by name.""" + + scene_copy = type(self)() + + for name, area in self.areas.items(): + + if name not in exclude: + + scene_copy[name] = numpy.array(area).astype(numpy.float32).view(AreaOfInterest) + + return scene_copy + class TimeStampedAOIScenes(DataStructures.TimeStampedBuffer): """Define timestamped buffer to store AOI scenes in time.""" -- cgit v1.1