BingGo/beach.py at master · windbell0711/BingGo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
"""
-*- coding: utf-8 -*-
@Time    : 2025-01-19
@Github  : windbell0711/BingGo
@Author  : windbell07
@License : Apache 2.0
@File    : beach.py
"""
from __future__ import annotations

from typing import List

import config
from qizi import *


class Beach:
    def __init__(self):
        self.beach: List[Qizi | None]  = [None] * 90  # 沙场，无子为None，每行末尾无子

    def __str__(self):
        res = "  0 1 2 3 4 5 6 7 8\n0 "
        for i in range(len(self.beach)):
            if i % 10 == 9:
                res += "\n%d " % (i // 10 + 1)
            elif self.beach[i] is None:
                res += "  "
            else:
                res += config.typ_num2str[self.beach[i].typ]
        return res

    def initiate(self, li):
        for p in range(89):
            i = li[p]
            if i is None:
                self.set_son(None, p=p)
            else:
                self.set_son(Qizi(p=p, typ=li[p], beach=self), p=p)

    def to_int(self):
        return [None if q is None else q.typ for q in self.beach]

    def clone(self):
        b = Beach()
        b.initiate(self.to_int())
        return b

    def __getitem__(self, item: int):
        """返回该位置棋子或None"""
        if item is None:  # 新增None检查
            raise ValueError("Position cannot be None")
        if not 0 <= item < 90:  # 确保在棋盘范围内
            raise IndexError("Position %d out of range" % item)
        return self.beach[item]

    def set_son(self, qizi, p: int):
        """将指定位置设定为棋子或None"""
        self.beach[p] = qizi
        # if qizi is None:
        #     return None
        # self.pieces.append(qizi)
        # qizi.idt = len(self.pieces) - 1
        # return qizi.idt

    # def quick_set(self, qizis: dict):
    #     for key, value in qizis.items():
    #         if isinstance(value, str):
    #             value = config.typ_dict[value]
    #         qizi = Qizi(p=key, typ=value, beach=self)
    #         self.set_son(qizi, key)
    #     return True

    def move_son(self, pfrom: int, pto: int) -> int:
        """移动棋子，会覆盖"""
        if self.beach[pto] is not None:
            self.beach[pto].alive = False
            # raise Exception("已弃用")
        self.beach[pto] = self.beach[pfrom]  # 移动到新位置
        self.beach[pfrom] = None  # 从原位置移除
        self.beach[pto].p = pto
        return self.beach[pto].idt

    def place_son(self, typ: int, p: int, idt=None):
        """新增棋子，会覆盖"""
        if self.beach[p] is not None:
            self.beach[p].alive = False
        self.beach[p] = Qizi(p=p, typ=typ, beach=self, idt=idt)

    def kill_son(self, p: int):
        """删除棋子"""
        self.beach[p].alive = False
        self.beach[p] = None

    def virtual_move(self, qizi, p: int):
        """虚拟移动，用于将死判断"""
        self.beach[p] = qizi
        if qizi is None:
            return None

    def valid(self, x: int) -> bool:  # 合法
        """检测当前位置合法"""
        if x % 10 == 9:
            return False
        if not 0 <= x <= 89:
            return False
        return True

    def occupied(self, x: int) -> bool:  # 非空
        """检测当前位置合法并且有子"""
        if self.valid(x) and self.beach[x] is not None:
            return True
        return False

    def not_occupied(self, x: int) -> bool:  # 非空
        """检测当前位置合法并且有子"""
        if self.valid(x) and self.beach[x] is None:
            return True
        return False

    def ch_occupied(self, x: int) -> bool:  # 中
        if self.occupied(x) and not self.beach[x].camp_intl:
            return True
        return False

    def in_occupied(self, x: int) -> bool:  # 国
        if self.occupied(x) and self.beach[x].camp_intl:
            return True
        return False


if __name__ == '__main__':
    beach = Beach()
    pao = Qizi(p=60, typ=5, beach=beach)  # 炮
    bingo = Qizi(p=50, typ=7, beach=beach)  # 兵
    pawn = Qizi(p=10, typ=13, beach=beach)  # Pawn
    beach.set_son(qizi=pao, p=60)
    beach.set_son(qizi=bingo, p=50)
    beach.set_son(qizi=pawn, p=10)
    pao.get_ma()
    pao.move(10)