From f5eaba8165fff2a915bcffc8cf6230fd12082a69 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 31 Jan 2023 16:12:00 +0100
Subject: Allowing to define AOI as child of another AOI.

---
 src/argaze/ArScene.py | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

(limited to 'src')

diff --git a/src/argaze/ArScene.py b/src/argaze/ArScene.py
index c8500fc..5e38c1e 100644
--- a/src/argaze/ArScene.py
+++ b/src/argaze/ArScene.py
@@ -262,14 +262,25 @@ class ArScene():
 
 		aoi_scene = {}
 
-		for name, marker_corners in self.aruco_aoi.items():
+		for name, data in self.aruco_aoi.items():
 
-			aoi_points = []
-			for marker_id, corner_id in marker_corners:
+			if type(data) == list:
 
-				aoi_points.append(self.aruco_tracker.tracked_markers[marker_id].corners[0][corner_id])
+				marker_corners = data
 
-			aoi_scene[name] = AOIFeatures.AreaOfInterest(aoi_points)
+				aoi_points = []
+				for marker_id, corner_id in marker_corners:
+
+					aoi_points.append(self.aruco_tracker.tracked_markers[marker_id].corners[0][corner_id])
+
+				aoi_scene[name] = AOIFeatures.AreaOfInterest(aoi_points)
+
+			elif type(data) == dict:
+
+				parent_aoi = aoi_scene[data['parent']]
+				aoi_points = [numpy.array(parent_aoi.outter_axis(inner)) for inner in data['inner_points']]
+
+				aoi_scene[name] = AOIFeatures.AreaOfInterest(aoi_points)
 
 		return AOI2DScene.AOI2DScene(aoi_scene)
 
-- 
cgit v1.1