From 0ed01b9172fddceda411b681ae0522ecd313a256 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurentfainsin@protonmail.com>
Date: Fri, 25 Nov 2022 11:20:04 +0100
Subject: [PATCH] chore: move code to main function

---
 src/main.py | 98 ++++++++++++++++++++++++++++++++---------------------
 1 file changed, 59 insertions(+), 39 deletions(-)

diff --git a/src/main.py b/src/main.py
index 3a5ee8e..388e623 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,48 +1,68 @@
+import logging
+
 import cv2
 import mediapipe as mp
 
-mp_drawing = mp.solutions.drawing_utils  # type: ignore
-mp_drawing_styles = mp.solutions.drawing_styles  # type: ignore
-mp_holistic = mp.solutions.holistic  # type: ignore
 
-# open webcam
-cap = cv2.VideoCapture(0)
+def main() -> None:
+    """Main function."""
+    mp_drawing = mp.solutions.drawing_utils  # type: ignore
+    mp_drawing_styles = mp.solutions.drawing_styles  # type: ignore
+    mp_holistic = mp.solutions.holistic  # type: ignore
 
-with mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5) as holistic:
-    while cap.isOpened():
-        # read webcam
-        success, image = cap.read()
-        if not success:
-            print("Ignoring empty camera frame.")
-            # If loading a video, use 'break' instead of 'continue'
-            continue
+    # open webcam
+    cap = cv2.VideoCapture(0)
 
-        # to improve performance, optionally mark the image as not writeable to pass by reference
-        image.flags.writeable = False
-        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        results = holistic.process(image)
+    with mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5) as holistic:
+        while cap.isOpened():
+            # read webcam
+            success, image = cap.read()
+            if not success:
+                print("Ignoring empty camera frame.")
+                # If loading a video, use 'break' instead of 'continue'
+                continue
 
-        # draw landmark annotation on the image
-        image.flags.writeable = True
-        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
-        mp_drawing.draw_landmarks(
-            image,
-            results.face_landmarks,
-            mp_holistic.FACEMESH_CONTOURS,
-            landmark_drawing_spec=None,
-            connection_drawing_spec=mp_drawing_styles.get_default_face_mesh_contours_style(),
-        )
-        mp_drawing.draw_landmarks(
-            image,
-            results.pose_landmarks,
-            mp_holistic.POSE_CONNECTIONS,
-            landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
-        )
+            # to improve performance, optionally mark the image as not writeable to pass by reference
+            image.flags.writeable = False
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+            results = holistic.process(image)
 
-        # flip the image horizontally for a selfie-view display.
-        cv2.imshow("MediaPipe Holistic", cv2.flip(image, 1))
-        if cv2.waitKey(5) & 0xFF == 27:
-            break
+            # log landmarks
+            logging.debug(results.face_landmarks)
+            logging.debug(results.pose_landmarks)
 
-# close webcam
-cap.release()
+            # draw landmark annotation on the image
+            image.flags.writeable = True
+            image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+            mp_drawing.draw_landmarks(
+                image,
+                results.face_landmarks,
+                mp_holistic.FACEMESH_CONTOURS,
+                landmark_drawing_spec=None,
+                connection_drawing_spec=mp_drawing_styles.get_default_face_mesh_contours_style(),
+            )
+            mp_drawing.draw_landmarks(
+                image,
+                results.pose_landmarks,
+                mp_holistic.POSE_CONNECTIONS,
+                landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
+            )
+
+            # flip the image horizontally for a selfie-view display.
+            cv2.imshow("MediaPipe Holistic", cv2.flip(image, 1))
+            if cv2.waitKey(5) & 0xFF == 27:
+                break
+
+    # close webcam
+    cap.release()
+
+
+if __name__ == "__main__":
+    # setup logging format
+    logging.basicConfig(
+        level=logging.DEBUG,
+        format="%(asctime)s %(name)s %(levelname)-8s  %(message)s",
+        datefmt="(%F %T)",
+    )
+
+    main()