XMWAI 0.4.5__tar.gz → 0.4.7__tar.gz

{xmwai-0.4.5 → xmwai-0.4.7}/MANIFEST.in

@@ -2,4 +2,4 @@ recursive-include XMWAI/gif *
 recursive-include XMWAI/file *.json
 recursive-include XMWAI/templates *.html
 recursive-include XMWAI/static *.js *.css *.png *.jpg *.gif *.jpeg
-recursive-include XMWAI/assets *
+recursive-include XMWAI/assets *.png *.ttf

{xmwai-0.4.5/XMWAI.egg-info → xmwai-0.4.7}/PKG-INFO

@@ -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 → xmwai-0.4.7}/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 → xmwai-0.4.7/XMWAI.egg-info}/PKG-INFO

@@ -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 → xmwai-0.4.7}/XMWAI.egg-info/SOURCES.txt

@@ -16,7 +16,6 @@ XMWAI.egg-info/dependency_links.txt
 XMWAI.egg-info/requires.txt
 XMWAI.egg-info/top_level.txt
 XMWAI/assets/1.png
-XMWAI/assets/__init__.py
 XMWAI/assets/g.png
 XMWAI/assets/h.png
 XMWAI/assets/l.png

{xmwai-0.4.5 → xmwai-0.4.7}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="XMWAI",  # 包名（pip install XMWAI）
-    version="0.4.5",  # 每次上传记得升级版本号
+    version="0.4.7",  # 每次上传记得升级版本号
     author="pydevelopment",  # 作者
     author_email="hekai@xiaoma.cn",  # 邮箱
     description="Small code King AI related library",  # 简短描述