1

1

src/engine/hand.py

@@ -1,21 +1,24 @@
-from collections import defaultdict
-
+from src.engine.mahjong_tile import MahjongTile
 from collections import defaultdict
 
 class Hand:
     def __init__(self):
-        # 存储所有的牌
+        # 存储所有的 MahjongTile 对象
         self.tiles = []
-        # 存储每种牌的数量，默认值为 0
+        # 存储每种牌的数量，键为 MahjongTile 对象，值为数量
         self.tile_count = defaultdict(int)
 
     def add_tile(self, tile):
         """ 向手牌中添加一张牌 """
+        if not isinstance(tile, MahjongTile):
+            raise ValueError("必须添加 MahjongTile 类型的牌")
         self.tiles.append(tile)  # 将牌添加到手牌中
         self.tile_count[tile] += 1  # 增加牌的数量
 
     def remove_tile(self, tile):
         """ 从手牌中移除一张牌 """
+        if not isinstance(tile, MahjongTile):
+            raise ValueError("必须移除 MahjongTile 类型的牌")
         if self.tile_count[tile] > 0:
             self.tiles.remove(tile)
             self.tile_count[tile] -= 1
@@ -24,16 +27,23 @@ class Hand:
 
     def get_tile_count(self, tile):
         """ 获取手牌中某张牌的数量 """
+        if not isinstance(tile, MahjongTile):
+            raise ValueError("必须是 MahjongTile 类型的牌")
         return self.tile_count[tile]
 
     def can_peng(self, tile):
         """ 判断是否可以碰（即是否已经有2张相同的牌，摸一张牌后可以碰） """
+        if not isinstance(tile, MahjongTile):
+            raise ValueError("必须是 MahjongTile 类型的牌")
         return self.tile_count[tile] == 2  # 摸一张牌后总数为 3 张，才可以碰
 
     def can_gang(self, tile):
         """ 判断是否可以杠（即是否已经有3张相同的牌，摸一张牌后可以杠） """
-        return self.tile_count[tile] == 3  # 摸一张牌后总数为 4 张，才可以杠
+        if not isinstance(tile, MahjongTile):
+            raise ValueError("必须是 MahjongTile 类型的牌")
+        return self.tile_count[tile] == 4  # 摸一张牌后总数为 4 张，才可以杠
 
     def __repr__(self):
         """ 返回手牌的字符串表示 """
-        return f"手牌: {self.tiles}, 牌的数量: {dict(self.tile_count)}"
+        tiles_str = ", ".join(str(tile) for tile in self.tiles)
+        return f"手牌: [{tiles_str}], 牌的数量: {dict(self.tile_count)}"