From 6b751757dd176e5246820b8fb28c0f877b54241a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurentfainsin@protonmail.com>
Date: Sat, 26 Nov 2022 19:57:41 +0100
Subject: [PATCH] feat: draft eyes

---
 src/environment.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/src/environment.py b/src/environment.py
index ef429d9..16cb381 100644
--- a/src/environment.py
+++ b/src/environment.py
@@ -8,7 +8,16 @@ from parts.eye import Eye
 
 
 class Environment:
+    """The environment is the main class of the application.
+
+    It is responsible for the following:
+        - detecting the keypoints
+        - drawing the keypoints
+        - drawing the avatar
+    """
+
     def __init__(self, camera: cv2.VideoCapture) -> None:
+        """Initialize the environment."""
         # store reference to webcam
         self.cam = camera
 
@@ -33,11 +42,12 @@ class Environment:
 
         # create body parts
         self.body_parts = {
-            "left_eye": Eye(False),
-            "right_eye": Eye(True),
+            "left_eye": Eye(False, self),
+            "right_eye": Eye(True, self),
         }
 
     def start(self) -> None:
+        """Start the environment."""
         while self.cam.isOpened():
             # read webcam
             success, self.frame = self.cam.read()
@@ -65,6 +75,7 @@ class Environment:
             self.draw_avatar()
 
     def detect_keypoints(self) -> None:
+        """Detect the keypoints on the frame."""
         with self.mp_face_mesh.FaceMesh(
             max_num_faces=1,
             refine_landmarks=True,
@@ -72,7 +83,7 @@ class Environment:
             min_tracking_confidence=0.5,
         ) as face_mesh:
 
-            # perf, mark image as not writeable to pass by reference
+            # perf, mark image as not writeable, to pass by reference
             self.frame.flags.writeable = False
 
             # convert the BGR image to RGB before processing
@@ -88,6 +99,7 @@ class Environment:
             self.frame = cv2.cvtColor(self.frame, cv2.COLOR_RGB2BGR)
 
     def draw_keypoints(self) -> None:
+        """Draw the keypoints on the screen."""
         # draw the face mesh annotations on the image.
         if self.results.multi_face_landmarks:
             for face_landmarks in self.results.multi_face_landmarks:
@@ -117,6 +129,7 @@ class Environment:
         cv2.imshow("MediaPipe Face Mesh", cv2.flip(self.frame, 1))
 
     def draw_avatar(self) -> None:
+        """Draw the avatar on the screen."""
         self.screen.fill(pg.Color("green"))
 
         for part in self.body_parts.values():