XMWAI 0.4.5__py3-none-any.whl → 0.4.7__py3-none-any.whl

XMWAI/snake_core.py

@@ -5,18 +5,15 @@ import cvzone
 import numpy as np
 from cvzone.HandTrackingModule import HandDetector
 from PIL import ImageFont, ImageDraw, Image
-import os
 import time
-import importlib.resources as pkg_resources
-
-# 引入包内资源
-import XMWAI.assets as assets
+from importlib.resources import files
+from pathlib import Path
+import os
 
 
-def get_asset_path(filename: str) -> str:
-    """获取包内资源的路径"""
-    with pkg_resources.path(assets, filename) as p:
-        return str(p)
+def get_resource_path(filename: str) -> str:
+    """返回 snake_core 模块内资源的绝对路径"""
+    return str(files("XMWAI.assets").joinpath(filename))
 
 
 class SnakeGame:
@@ -30,21 +27,16 @@ class SnakeGame:
         self._snakeHeadSize = snakeHeadSize
         self._foodScores = foodScores if foodScores is not None else [3, 2, 1]
         self.snakeLineWidth = snakeLineWidth
+        self.fontPath = fontPath or get_resource_path("微软雅黑.ttf")
 
-        # 字体路径（默认取包内字体）
-        self.fontPath = fontPath or get_asset_path("微软雅黑.ttf")
-
-        # 默认资源路径
         if foodPaths is None:
-            foodPaths = [get_asset_path("h.png"),
-                         get_asset_path("s.png"),
-                         get_asset_path("t.png")]
+            foodPaths = [get_resource_path(f)
+                         for f in ["h.png", "s.png", "t.png"]]
         if foodNames is None:
             foodNames = ["汉堡", "薯条", "甜甜圈"]
         if obstaclePaths is None:
-            obstaclePaths = [get_asset_path("g.png"),
-                             get_asset_path("l.png"),
-                             get_asset_path("m.png")]
+            obstaclePaths = [get_resource_path(f)
+                             for f in ["g.png", "l.png", "m.png"]]
 
         self.foodPaths = foodPaths
         self.foodNames = foodNames
@@ -64,7 +56,7 @@ class SnakeGame:
         self._init_game_objects()
         self.open_window()
 
-    # ----------- 属性自动同步 -----------
+    # ---------------- 属性自动同步 ----------------
     @property
     def snakeHeadSize(self):
         return self._snakeHeadSize
@@ -113,6 +105,8 @@ class SnakeGame:
         self.obstacleManager.randomObstacles()
 
     def _render_frame(self, show_food=True, show_obstacle=True):
+        if self.cap is None:
+            return
         success, self.img = self.cap.read()
         if not success:
             return
@@ -134,8 +128,7 @@ class SnakeGame:
 
         if show_obstacle:
             self.img = self.obstacleManager.draw(self.img)
-            self.obstacleManager.moveObstacles(
-                self.resolution[0], self.resolution[1])
+            self.obstacleManager.moveObstacles(*self.resolution)
 
         if show_food:
             self.img = self.foodManager.draw(self.img)
@@ -154,6 +147,39 @@ class SnakeGame:
         cv2.imshow("AI Snake", self.img)
         cv2.waitKey(1)
 
+    def hand(self):
+        if self.cap is None:
+            print("请先调用 open_window()")
+            return
+        if self.detector is None:
+            self.detector = HandDetector(detectionCon=0.8, maxHands=1)
+        while True:
+            success, self.img = self.cap.read()
+            if not success:
+                break
+            self.img = cv2.flip(self.img, 1)
+            hands, self.img = self.detector.findHands(self.img, flipType=False)
+            cv2.imshow("AI Snake", self.img)
+            key = cv2.waitKey(1) & 0xFF
+            if hands or key == ord('q'):
+                break
+
+    def display(self):
+        if self.img is None:
+            self._render_frame(show_food=False)
+        self.foodManager.randomFoodLocation(self.obstacleManager)
+        self._render_frame(show_food=True)
+        self.overlayTexts = [
+            (f'玩家分数：{self.snake.score}', (50, 50), 30, (255, 0, 255)),
+            (f'倒计时：{self.timer} 秒', (50, 120), 30, (255, 0, 255))
+        ]
+        img_copy = self.img.copy()
+        for txt, pos, size, color in self.overlayTexts:
+            img_copy = self._putChineseText(img_copy, txt, pos, size, color)
+        cv2.imshow("AI Snake", img_copy)
+        cv2.waitKey(1)
+
+    # ---------------- 重置游戏 ----------------
     def reset_game(self):
         self.snake.reset()
         self.snake.headSize = self._snakeHeadSize
@@ -165,9 +191,10 @@ class SnakeGame:
         if self.cap is None or not self.cap.isOpened():
             self.open_window()
 
+    # ---------------- 游戏结束 ----------------
     def gameover(self, path=None, size=(100, 100)):
         if path is None:
-            path = get_asset_path("1.png")
+            path = get_resource_path("1.png")
 
         if os.path.exists(path):
             gameover_img = cv2.imread(path)
@@ -194,6 +221,7 @@ class SnakeGame:
                 cv2.destroyAllWindows()
                 break
 
+    # ---------------- 游戏主循环 ----------------
     def start(self):
         if self.cap is None or not self.cap.isOpened():
             self.open_window()
@@ -222,7 +250,7 @@ class SnakeGame:
                 cv2.imshow("AI Snake", img_copy)
 
             key = cv2.waitKey(1)
-            if key == ord('r'):
+            if key == ord('r'):  # 游戏中途重置
                 self.reset_game()
             elif key == ord('q'):
                 if self.cap is not None:
@@ -230,7 +258,7 @@ class SnakeGame:
                 cv2.destroyAllWindows()
                 break
 
-    # ---------------- 内部类：蛇 ----------------
+    # ---------------- 内部类 ----------------
     class Snake:
         def __init__(self, color, initLength=150, lineWidth=10, headSize=15):
             self.points = []
@@ -309,7 +337,7 @@ class SnakeGame:
                     if ox <= cx <= ox + ow and oy <= cy <= oy + oh:
                         self.gameOver = True
 
-    # ---------------- 内部类：食物 ----------------
+    # ---------------- 内部类 ----------------
     class FoodManager:
         def __init__(self, foodPaths, foodNames, foodScores):
             self.foodImages = []
@@ -349,7 +377,7 @@ class SnakeGame:
                                         (rx - self.wFood//2, ry - self.hFood//2))
             return imgMain
 
-    # ---------------- 内部类：障碍 ----------------
+    # ---------------- 内部类 ----------------
     class ObstacleManager:
         def __init__(self, obstaclePaths):
             self.obstacleImages = []
@@ -365,8 +393,8 @@ class SnakeGame:
             self.obstacles.clear()
             for img in self.obstacleImages:
                 h, w, _ = img.shape
-                x = random.randint(150, 600)
-                y = random.randint(50, 600)
+                x = random.randint(150, 570)
+                y = random.randint(50, 570)
                 dx = random.choice([-5, 5])
                 dy = random.choice([-5, 5])
                 self.obstacles.append([x, y, w, h, dx, dy, img])

{xmwai-0.4.5.dist-info → xmwai-0.4.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.4.5
+Version: 0.4.7
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment

{xmwai-0.4.5.dist-info → xmwai-0.4.7.dist-info}/RECORD

@@ -3,11 +3,10 @@ XMWAI/bomb_core.py,sha256=h2ZPH3SuoG2L_XOf1dcK8p3lhw5QzhneWl2yMLj1RiU,1819
 XMWAI/core.py,sha256=rOXj7FnewSdnzBcFLjpnBtrOTCsvMfiycIcdPDagxho,10012
 XMWAI/idiom_core.py,sha256=yU-VHhqqoutVm6GVikcjL3m9yuB1hUsFBpPYvwY4n5g,1689
 XMWAI/magic_core.py,sha256=Ms4b12PJ8rjsmceg1VUqWCWx2ebvdhLp4sIF6K_Vaok,3491
-XMWAI/snake_core.py,sha256=oKcjGwEm2MB7Q8gfRo80NRqul20LGiEIe3--4k1-E74,14782
+XMWAI/snake_core.py,sha256=fMm65EUfEmJK9nat9EBKz-ysmo8luTDx8N3vZ320aZE,16055
 XMWAI/trial_class.py,sha256=fPsl7BZvhzch2FOIG4azr999kjtoly53Acm3LqL8f98,9724
 XMWAI/web_core.py,sha256=7awPg1kYW3lYrbgylqJvUF3g050bn6H21PgmQ7Kv1wA,10927
 XMWAI/assets/1.png,sha256=eEuKH_M_q3tc_O2bYnuLOsRP-NlJHIbNg0pgrKXEEjw,139720
-XMWAI/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 XMWAI/assets/g.png,sha256=hr9hlKJ7y95X-g6-tllrzDNgL1WQkbq5cA5L4jASEAM,11686
 XMWAI/assets/h.png,sha256=qO-kJJOPA8qUth5rqLeOVa_6_n7tU-ABQ14O0EW_YCE,14929
 XMWAI/assets/l.png,sha256=Urm6LxH33HID6ZZbs2oMViUk4GiZ3upLPdsrNU8FlP0,9921
@@ -117,8 +116,8 @@ XMWAI/static/images/tomato.png,sha256=FEOEAOdUhW_BDFgTpxOkYc0I5Iu29_gtHb3RIPEej0
 XMWAI/templates/burger.html,sha256=vDnxpSW8phetyScySsalScZnFKl3LNpy5lJjKxGXgbI,3320
 XMWAI/templates/nutrition_pie.html,sha256=yJVXI28i-UfvF0xOXGSNLMb8oCJNhh2J3zoRDr5_7DM,5567
 XMWAI/templates/创意菜谱.html,sha256=RcDgH58QLyUJ9A59wobu3wvchGBY1snVsXcZQZam5M0,4805
-xmwai-0.4.5.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
-xmwai-0.4.5.dist-info/METADATA,sha256=PBWm8ONw169ZRDJr5VSBWpGwb-XGsjGMrjzwaZs_umw,1227
-xmwai-0.4.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xmwai-0.4.5.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
-xmwai-0.4.5.dist-info/RECORD,,
+xmwai-0.4.7.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
+xmwai-0.4.7.dist-info/METADATA,sha256=TFk6r7_ikJKUfVXu41zaNzzL1LppOUdYJGplWcZHQeg,1227
+xmwai-0.4.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xmwai-0.4.7.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
+xmwai-0.4.7.dist-info/RECORD,,