Adding looked_region method the AreaOfInterest class to get intersection region with gaze circle.

2 files changed, 34 insertions, 10 deletions

diff --git a/setup.py b/setup.py
index d8fb29c..a85b54d 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@ setup(
     packages=find_packages(where='src'),
 
     python_requires='>=3.6, <4',
-    install_requires=['opencv-python==4.6.0.66', 'opencv-contrib-python==4.6.0.66', 'numpy', 'av', 'rtsp', 'pandas', 'matplotlib'],
+    install_requires=['opencv-python==4.6.0.66', 'opencv-contrib-python==4.6.0.66', 'numpy', 'av', 'rtsp', 'pandas', 'matplotlib', 'shapely'],
 
     project_urls={
         'Bug Reports': 'https://git.recherche.enac.fr/projects/argaze/issues',
diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py
index 8e4965c..987a25f 100644
--- a/src/argaze/AreaOfInterest/AOIFeatures.py
+++ b/src/argaze/AreaOfInterest/AOIFeatures.py
@@ -7,6 +7,8 @@ from argaze import DataStructures, GazeFeatures
 import cv2 as cv
 import matplotlib.path as mpath
 import numpy
+from shapely.geometry import Polygon
+from shapely.geometry.point import Point
 
 @dataclass
 class AreaOfInterest(numpy.ndarray):
@@ -88,17 +90,39 @@ class AreaOfInterest(numpy.ndarray):
 
         return numpy.rint(Lp).astype(int).tolist()
 
-    def draw(self, frame, color):
+    def looked_region(self, gaze_position, gaze_radius):
+        """Get intersection shape with gaze circle as the looked area, (looked area / AOI area) and (looked area / gaze circle area)."""
 
-        # Draw form
-        pixels = numpy.rint(self).astype(int)
-        cv.line(frame, pixels[-1], pixels[0], color, 1)
-        for A, B in zip(pixels, pixels[1:]):
-            cv.line(frame, A, B, color, 1)
+        self_polygon = Polygon(self)
+        gaze_circle = Point(gaze_position).buffer(gaze_radius)
 
-        # Draw center
-        center_pixel = numpy.rint(self.center()).astype(int)
-        cv.circle(frame, center_pixel, 1, color, -1)
+        if self_polygon.intersects(gaze_circle):
+
+            intersection = self_polygon.intersection(gaze_circle)
+
+            intersection_array = numpy.array([list(xy) for xy in intersection.exterior.coords[:]]).astype(numpy.float32).view(AreaOfInterest)
+
+            return intersection_array, intersection.area / self_polygon.area, intersection.area / gaze_circle.area
+
+        else:
+
+            empty_array = numpy.array([list([])]).astype(numpy.float32).view(AreaOfInterest)
+
+            return empty_array, 0., 0.
+
+    def draw(self, frame, color, border_size=1):
+
+        if len(self) > 1:
+
+            # Draw form
+            pixels = numpy.rint(self).astype(int)
+            cv.line(frame, pixels[-1], pixels[0], color, border_size)
+            for A, B in zip(pixels, pixels[1:]):
+                cv.line(frame, A, B, color, border_size)
+
+            # Draw center
+            center_pixel = numpy.rint(self.center()).astype(int)
+            cv.circle(frame, center_pixel, 1, color, -1)
 
 @dataclass
 class AOIScene():