update(#8): make enemies prioritize the face.

Author: Raven

main.py

@@ -146,12 +146,24 @@ def render(self, image):
         previous_spawn_time = current_time
         spawn_delay = random.normalvariate(4, 1)
 
+    face_bounding_box = None
 
     if face_results.detections:
         detection = next(iter(face_results.detections))
 
         mp_drawing.draw_detection(image, detection)
 
+        bbox = detection.location_data.relative_bounding_box
+        fbbx = int(bbox.xmin * w)
+        fbby = int(bbox.ymin * h)
+        fbbw = int(bbox.width * w)
+        fbbh = int(bbox.height * h)
+        face_center = np.array((fbbx + fbbw / 2, fbby + fbbh / 2))
+        face_bounding_box = (fbbx, fbby, fbbw, fbbh)
+
+        for rect in rect_list:
+            rect.chase(face_center, delta_time)
+
     if hand_results.multi_hand_landmarks:
         for hand_landmarks in hand_results.multi_hand_landmarks:
             mp_drawing.draw_landmarks(image, hand_landmarks, mp_hands.HAND_CONNECTIONS)
@@ -172,7 +184,8 @@ def render(self, image):
                 if len(hits) > 0:
                     hit_indices.append(i)
 
-                rect.chase(index_mcp, delta_time)
+                if not face_results.detections:
+                    rect.chase(index_mcp, delta_time)
 
                 if is_hit:
                     continue