Adding a AOIScene copy method with exclude parameter.

1 files changed, 13 insertions, 0 deletions

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."""