XMWAI 0.4.2__py3-none-any.whl → 0.4.4__py3-none-any.whl

XMWAI/__init__.py

@@ -2,10 +2,12 @@ from .core import story, photo, reply, poem, crack, show  # 从子模块导入
 from .magic_core import birthday  # 从子模块导入函数到顶层
 from .bomb_core import bomb  # 从子模块导入函数到顶层
 from .idiom_core import idiom, searchIdiom, get_json_path  # 从子模块导入函数到顶层
-from .trial_class import make, get_file_content_as_base64, save_pic, detect_windows_scale, get_file_content_as_base64_2, cartoon  # 从子模块导入函数到顶层
+from .trial_class import make, get_file_content_as_base64, save_pic, detect_scale, open_image, get_file_content_as_base64_2, cartoon  # 从子模块导入函数到顶层
 from .web_core import burger, cookbook  # 从子模块导入函数到顶层
+from .snake_core import SnakeGame
+
 
 __all__ = ["story", "photo", "reply", "poem", 'birthday', 'bomb',
            "idiom", "searchIdiom", "get_json_path", "crack", "show",
-           "make", "get_file_content_as_base64", "save_pic", "detect_windows_scale",
-           "get_file_content_as_base64_2", "cartoon", "burger", "cookbook"]  # 可选：明确导出的内容
+           "make", "get_file_content_as_base64", "save_pic", "detect_scale", "open_image",
+           "get_file_content_as_base64_2", "cartoon", "burger", "cookbook", "SnakeGame"]  # 可选：明确导出的内容

XMWAI/snake_core.py

@@ -0,0 +1,426 @@
+import cv2
+import math
+import random
+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  # 引入资源目录
+
+
+# ---------- 工具函数：从 assets 读取资源 ----------
+def get_asset_path(filename: str) -> str:
+    """
+    获取包内资源的绝对路径 (兼容 pip 安装后的情况)
+    """
+    with pkg_resources.path(XMWAI.assets, filename) as p:
+        return str(p)
+
+
+class SnakeGame:
+    def __init__(self, width=720, height=720, snakeInitLength=150, snakeGrowth=50,
+                 snakeLineWidth=10, snakeHeadSize=15, foodPaths=None, foodNames=None, foodScores=None,
+                 obstaclePaths=None, fontPath=None):
+
+        self.resolution = (width, height)
+        self.snakeInitLength = snakeInitLength
+        self._snakeGrowth = snakeGrowth
+        self._snakeHeadSize = snakeHeadSize
+        self._foodScores = foodScores if foodScores is not None else [3, 2, 1]
+        self.snakeLineWidth = snakeLineWidth
+
+        # 默认资源路径
+        if foodPaths is None:
+            foodPaths = [get_asset_path("h.png"),
+                         get_asset_path("s.png"),
+                         get_asset_path("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")]
+        if fontPath is None:
+            fontPath = get_asset_path("微软雅黑.ttf")
+
+        self.foodPaths = foodPaths
+        self.foodNames = foodNames
+        self.obstaclePaths = obstaclePaths
+        self.fontPath = fontPath
+
+        self.cap = None
+        self.detector = None
+        self.snake = None
+        self.foodManager = None
+        self.obstacleManager = None
+        self.img = None
+        self.overlayTexts = []
+
+        self.timer = 30
+        self.start_time = None
+
+        self._init_game_objects()
+        self.open_window()
+
+    # ----------- 属性自动同步 -----------
+    @property
+    def snakeHeadSize(self):
+        return self._snakeHeadSize
+
+    @snakeHeadSize.setter
+    def snakeHeadSize(self, value):
+        self._snakeHeadSize = value
+        if self.snake:
+            self.snake.headSize = value
+
+    @property
+    def foodScores(self):
+        return self._foodScores
+
+    @foodScores.setter
+    def foodScores(self, scores):
+        self._foodScores = scores
+        if self.foodManager:
+            self.foodManager.foodScores = scores
+
+    @property
+    def snakeGrowth(self):
+        return self._snakeGrowth
+
+    @snakeGrowth.setter
+    def snakeGrowth(self, value):
+        self._snakeGrowth = value
+
+    # ---------------- 工具函数 ----------------
+    def _putChineseText(self, img, text, pos, fontSize=40, color=(0, 0, 255)):
+        img_pil = Image.fromarray(img)
+        draw = ImageDraw.Draw(img_pil)
+        try:
+            font = ImageFont.truetype(self.fontPath, fontSize)
+        except OSError:
+            font = ImageFont.load_default()
+        draw.text(pos, text, font=font, fill=color)
+        return np.array(img_pil)
+
+    def _init_game_objects(self):
+        self.snake = self.Snake(color=(0, 0, 255), initLength=self.snakeInitLength,
+                                lineWidth=self.snakeLineWidth, headSize=self._snakeHeadSize)
+        self.foodManager = self.FoodManager(
+            self.foodPaths, self.foodNames, self._foodScores)
+        self.obstacleManager = self.ObstacleManager(self.obstaclePaths)
+        self.obstacleManager.randomObstacles()
+
+    def _render_frame(self, show_food=True, show_obstacle=True):
+        success, self.img = self.cap.read()
+        if not success:
+            return
+        self.img = cv2.flip(self.img, 1)
+        hands, self.img = self.detector.findHands(self.img, flipType=False)
+        player_head = tuple(hands[0]['lmList'][8][0:2]) if hands else None
+
+        if not self.snake.gameOver and player_head:
+            self.snake.update(self.img, player_head, self.obstacleManager)
+
+        if not self.snake.gameOver and player_head and show_food:
+            cx, cy = player_head
+            rx, ry = self.foodManager.foodPoint
+            w, h = self.foodManager.wFood, self.foodManager.hFood
+            if rx - w//2 <= cx <= rx + w//2 and ry - h//2 <= cy <= ry + h//2:
+                self.snake.score += self.foodManager.foodScores[self.foodManager.foodIndex]
+                self.snake.allowedLength += self._snakeGrowth
+                self.foodManager.randomFoodLocation(self.obstacleManager)
+
+        if show_obstacle:
+            self.img = self.obstacleManager.draw(self.img)
+            self.obstacleManager.moveObstacles(
+                self.resolution[0], self.resolution[1])
+
+        if show_food:
+            self.img = self.foodManager.draw(self.img)
+
+    # ---------------- 对外接口 ----------------
+    def open_window(self):
+        self.cap = cv2.VideoCapture(0)
+        self.cap.set(3, self.resolution[0])
+        self.cap.set(4, self.resolution[1])
+        self.detector = HandDetector(detectionCon=0.7, maxHands=1)
+        success, self.img = self.cap.read()
+        if not success:
+            print("摄像头打开失败")
+            return
+        self.img = cv2.flip(self.img, 1)
+        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
+        self.obstacleManager.randomObstacles()
+        self.foodManager.foodScores = self._foodScores
+        self.foodManager.randomFoodLocation(self.obstacleManager)
+        self.start_time = time.time()
+        self.timer = 30
+        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")
+
+        if os.path.exists(path):
+            gameover_img = cv2.imread(path)
+            gameover_img = cv2.resize(gameover_img, self.resolution)
+        else:
+            gameover_img = np.zeros(
+                (self.resolution[1], self.resolution[0], 3), np.uint8)
+            cv2.putText(gameover_img, "Game Over Image Missing!", (50, 100),
+                        cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
+
+        gameover_img = self._putChineseText(
+            gameover_img, f"最终分数：{self.snake.score}", size, 40, (68, 84, 106))
+
+        while True:
+            cv2.imshow("AI Snake", gameover_img)
+            key = cv2.waitKey(0) & 0xFF
+            if key == ord('r'):  # 重启游戏
+                self.reset_game()
+                self.start()
+                break
+            elif key == ord('q'):  # 退出游戏
+                if self.cap is not None:
+                    self.cap.release()
+                cv2.destroyAllWindows()
+                break
+
+    # ---------------- 游戏主循环 ----------------
+    def start(self):
+        if self.cap is None or not self.cap.isOpened():
+            self.open_window()
+        self.start_time = time.time()
+        self.timer = 30
+
+        while True:
+            if self.snake.gameOver or self.timer == 0:
+                # 游戏结束，进入 gameover 界面
+                self.gameover()
+                break
+
+            self._render_frame(show_food=True, show_obstacle=True)
+            elapsed = int(time.time() - self.start_time)
+            self.timer = max(0, 30 - elapsed)
+
+            self.overlayTexts = [
+                (f'玩家分数：{self.snake.score}', (50, 50), 30, (255, 0, 255)),
+                (f'倒计时：{self.timer} 秒', (50, 120), 30, (255, 0, 255))
+            ]
+
+            if self.img is not None:
+                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)
+
+            key = cv2.waitKey(1)
+            if key == ord('r'):  # 游戏中途重置
+                self.reset_game()
+            elif key == ord('q'):
+                if self.cap is not None:
+                    self.cap.release()
+                cv2.destroyAllWindows()
+                break
+
+    # ---------------- 内部类：蛇 ----------------
+    class Snake:
+        def __init__(self, color, initLength=150, lineWidth=10, headSize=15):
+            self.points = []
+            self.currentLength = 0
+            self.allowedLength = initLength
+            self.previousHead = None
+            self.score = 0
+            self.color = color
+            self.gameOver = False
+            self.lineWidth = lineWidth
+            self.headSize = headSize
+
+        def reset(self):
+            self.points = []
+            self.currentLength = 0
+            self.allowedLength = 150
+            self.previousHead = None
+            self.score = 0
+            self.gameOver = False
+
+        def update(self, imgMain, currentHead, obstacleManager=None):
+            if self.gameOver:
+                return
+            cx, cy = currentHead
+            if cx is None or cy is None:
+                return
+            if self.previousHead is None:
+                self.previousHead = (cx, cy)
+            px, py = self.previousHead
+
+            alpha = 0.7
+            cx = int(px * (1 - alpha) + cx * alpha)
+            cy = int(py * (1 - alpha) + cy * alpha)
+
+            maxStep = 50
+            dx = cx - px
+            dy = cy - py
+            distance = math.hypot(dx, dy)
+            if distance > maxStep:
+                steps = int(distance / maxStep)
+                for i in range(1, steps + 1):
+                    ix = int(px + dx * i / steps)
+                    iy = int(py + dy * i / steps)
+                    self.points.append((ix, iy))
+                    self.currentLength += maxStep
+            else:
+                self.points.append((cx, cy))
+                self.currentLength += distance
+
+            self.previousHead = (cx, cy)
+
+            while self.currentLength > self.allowedLength:
+                if len(self.points) > 1:
+                    removed_dx = self.points[1][0] - self.points[0][0]
+                    removed_dy = self.points[1][1] - self.points[0][1]
+                    removed_dist = math.hypot(removed_dx, removed_dy)
+                    self.currentLength -= removed_dist
+                self.points.pop(0)
+
+            for i in range(1, len(self.points)):
+                cv2.line(
+                    imgMain, self.points[i-1], self.points[i], self.color, self.lineWidth)
+
+            snakeHeadColor = (random.randint(0, 255), random.randint(
+                0, 255), random.randint(0, 255))
+            cv2.circle(imgMain, (cx, cy), self.headSize,
+                       snakeHeadColor, cv2.FILLED)
+
+            h, w, _ = imgMain.shape
+            margin = 5
+            if cx <= margin or cx >= w - margin or cy <= margin or cy >= h - margin:
+                self.gameOver = True
+
+            if obstacleManager:
+                for ox, oy, ow, oh, *_ in obstacleManager.obstacles:
+                    if ox <= cx <= ox + ow and oy <= cy <= oy + oh:
+                        self.gameOver = True
+
+    # ---------------- 内部类：食物 ----------------
+    class FoodManager:
+        def __init__(self, foodPaths, foodNames, foodScores):
+            self.foodImages = []
+            for path in foodPaths:
+                if os.path.exists(path):
+                    self.foodImages.append(
+                        cv2.imread(path, cv2.IMREAD_UNCHANGED))
+                else:
+                    self.foodImages.append(np.zeros((50, 50, 4), np.uint8))
+            self.foodNames = foodNames
+            self.foodScores = foodScores
+            self.foodIndex = 0
+            self.hFood, self.wFood = 0, 0
+            self.foodPoint = 0, 0
+            self.randomFoodLocation()
+
+        def randomFoodLocation(self, obstacleManager=None):
+            max_attempts = 100
+            for _ in range(max_attempts):
+                self.foodPoint = random.randint(
+                    50, 1230), random.randint(50, 670)
+                self.foodIndex = random.randint(0, len(self.foodImages)-1)
+                self.hFood, self.wFood, _ = self.foodImages[self.foodIndex].shape
+                if obstacleManager:
+                    overlap = False
+                    for ox, oy, ow, oh, *_ in obstacleManager.obstacles:
+                        if ox < self.foodPoint[0] < ox + ow and oy < self.foodPoint[1] < oy + oh:
+                            overlap = True
+                            break
+                    if not overlap:
+                        return
+            return
+
+        def draw(self, imgMain):
+            rx, ry = self.foodPoint
+            imgMain = cvzone.overlayPNG(imgMain, self.foodImages[self.foodIndex],
+                                        (rx - self.wFood//2, ry - self.hFood//2))
+            return imgMain
+
+    # ---------------- 内部类：障碍 ----------------
+    class ObstacleManager:
+        def __init__(self, obstaclePaths):
+            self.obstacleImages = []
+            for path in obstaclePaths:
+                if os.path.exists(path):
+                    self.obstacleImages.append(
+                        cv2.imread(path, cv2.IMREAD_UNCHANGED))
+                else:
+                    self.obstacleImages.append(np.zeros((50, 50, 4), np.uint8))
+            self.obstacles = []
+
+        def randomObstacles(self):
+            self.obstacles.clear()
+            for img in self.obstacleImages:
+                h, w, _ = img.shape
+                x = random.randint(150, 1230)
+                y = random.randint(50, 670)
+                dx = random.choice([-5, 5])
+                dy = random.choice([-5, 5])
+                self.obstacles.append([x, y, w, h, dx, dy, img])
+
+        def moveObstacles(self, wMax, hMax):
+            for obs in self.obstacles:
+                x, y, ow, oh, dx, dy, img = obs
+                x += dx
+                y += dy
+                if x <= 0 or x + ow >= wMax:
+                    dx *= -1
+                if y <= 0 or y + oh >= hMax:
+                    dy *= -1
+                obs[0], obs[1], obs[4], obs[5] = x, y, dx, dy
+
+        def draw(self, imgMain):
+            for obs in self.obstacles:
+                x, y, w, h, dx, dy, img = obs
+                imgMain = cvzone.overlayPNG(imgMain, img, (int(x), int(y)))
+            return imgMain

{xmwai-0.4.2.dist-info → xmwai-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.4.2
+Version: 0.4.4
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment
@@ -17,6 +17,7 @@ Requires-Dist: opencv-python>=3.4.18.65
 Requires-Dist: numpy>=1.26.0
 Requires-Dist: flask>=3.1.0
 Requires-Dist: pyecharts>=2.0.8
+Requires-Dist: cvzone>=1.6.1
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{xmwai-0.4.2.dist-info → xmwai-0.4.4.dist-info}/RECORD

@@ -1,10 +1,20 @@
-XMWAI/__init__.py,sha256=-0GLZK0X3vEuTbu5n2AX3cWw6SXtJTERtBIAssZq41o,908
+XMWAI/__init__.py,sha256=RFHMM1PF3ELvHSMeKe2G2E0astGPum2ezuf8Hjw4D6g,968
 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=oa0pGvcomna_oXknVa5PasbRw0ILEdThfNExJjRbz8w,16418
 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
+XMWAI/assets/m.png,sha256=4tl9Rb2JoMD7XLMj3w8jg-92y6D7O-1u0sZCYEoeUtk,10303
+XMWAI/assets/s.png,sha256=v_qmHGmSPhG838vXQdUR2ZX_Z5KLI8T46kaR4P_ktUg,15120
+XMWAI/assets/t.png,sha256=GebzA2UWhXn4u62UeuUjitdpwnJvnxfZ2z_4MFlxvm8,12838
+XMWAI/assets/微软雅黑.ttf,sha256=dpc4EmGE1ojdwHjfIwTdr3olyEDAk3FwyreQSC9AdQ8,15043584
 XMWAI/file/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 XMWAI/file/idiom.json,sha256=HUtPRUzhxBbWoasjadbmbA_5ngQ5AXLu9weQSZ4hzhk,10319857
 XMWAI/gif/0.gif,sha256=LGpAReVTyZEb1J-bWYTpxxHbIxmLJ-3wA0lw4cBqdsY,303
@@ -107,8 +117,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.2.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
-xmwai-0.4.2.dist-info/METADATA,sha256=6GJQaqzQytoahOWk1ssi_sezcfONReHCN5h2rj9rtJQ,1197
-xmwai-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xmwai-0.4.2.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
-xmwai-0.4.2.dist-info/RECORD,,
+xmwai-0.4.4.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
+xmwai-0.4.4.dist-info/METADATA,sha256=Qsm0Z6Kkd35aNmUsTNNp4AmIrlIAPYVxnJP8VPiPLs0,1227
+xmwai-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xmwai-0.4.4.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
+xmwai-0.4.4.dist-info/RECORD,,