Adding new reframe method.

1 files changed, 37 insertions, 4 deletions

diff --git a/src/argaze/AreaOfInterest/AOI2DScene.py b/src/argaze/AreaOfInterest/AOI2DScene.py
index 8301e23..60f17ef 100644
--- a/src/argaze/AreaOfInterest/AOI2DScene.py
+++ b/src/argaze/AreaOfInterest/AOI2DScene.py
@@ -7,15 +7,18 @@ __credits__ = []
 __copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)"
 __license__ = "BSD"
 
-from typing import Tuple
+from typing import TypeVar, Tuple
 
 from argaze import DataStructures
 from argaze.AreaOfInterest import AOIFeatures
 from argaze import GazeFeatures
 
-import cv2 as cv
+import cv2
 import numpy
 
+AOI2DSceneType = TypeVar('AOI2DScene', bound="AOI2DScene")
+# Type definition for type annotation convenience
+
 class AOI2DScene(AOIFeatures.AOIScene):
 	"""Define AOI 2D scene."""
 
@@ -60,7 +63,7 @@ class AOI2DScene(AOIFeatures.AOIScene):
 			if matching:
 
 				top_left_corner_pixel = numpy.rint(aoi.clockwise()[0]).astype(int)
-				cv.putText(frame, name, top_left_corner_pixel, cv.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv.LINE_AA)
+				cv2.putText(frame, name, top_left_corner_pixel, cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv2.LINE_AA)
 
 			# Draw form
 			aoi.draw(frame, color)
@@ -101,10 +104,40 @@ class AOI2DScene(AOIFeatures.AOIScene):
 			if matching:
 
 				top_left_corner_pixel = numpy.rint(aoi.clockwise()[0]).astype(int)
-				cv.putText(frame, name, top_left_corner_pixel, cv.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv.LINE_AA)
+				cv2.putText(frame, name, top_left_corner_pixel, cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv2.LINE_AA)
 
 				# Draw matching region
 				matching_region.draw(frame, matching_color, 4)
 
 			# Draw form
 			aoi.draw(frame, color)
+
+	def reframe(self, aoi_name: str, size: tuple) -> AOI2DSceneType:
+		"""
+		Project whole scene into an AOI.
+
+		Parameters:
+			aoi: name of AOI where to project scene
+
+		Returns:
+			reframed AOI 2D scene
+		"""
+		
+		assert(self[aoi_name].points_number == 4)
+
+		# Edit perspective (P) allowing to transform source axis (Src) into destination axis (Dst)
+		Src = self[aoi_name].clockwise().astype(numpy.float32)
+		Src_origin = Src[0]
+		Src = Src - Src_origin
+		Dst = numpy.float32([[0, 0], [size[0], 0], [size[0], size[1]], [0, size[1]]])
+
+		P = cv2.getAffineTransform(Src[:3], Dst[:3])[:, :2]
+
+		# Apply perspective matrix to each AOI
+		aoi2D_scene = AOI2DScene()
+		
+		for name, aoi2D in self.items():
+
+			aoi2D_scene[name] = numpy.matmul(aoi2D - Src_origin, P.T)
+
+		return aoi2D_scene
\ No newline at end of file