gobangAI/main.cpp at master · xiaohaoxiang/gobangAI · 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
#include "ai.h"
#include "board.h"
#include "defs.h"
#include "keys.h"
#include <iostream>

const char *const posi[] = {" 1  ", " 2  ", " 3  ", " 4  ", " 5  ", " 6  ", " 7  ", " 8  ",
                            " 9  ", " 10 ", " 11 ", " 12 ", " 13 ", " 14 ", " 15 "};

int main()
{
    ChessBoard brd;

    for (int x = 7, y = 7, chcnt = 0;;)
    {
        if (brd.state() == BoardState::WinWhite)
        {
            std::cout << "you win\n";
            return 0;
        }
        auto ai = AI::solve(brd);
        brd.put(ai);
        x = ai.x;
        y = ai.y;

        auto showBoard = [&]() {
            std::cout << "      ";
            for (int i = 0; i < BoardSize; i++)
            {
                std::cout << posi[i];
            }
            std::cout << "\n\n";

            for (int i = 0; i < BoardSize; i++)
            {
                if (i == y)
                {
                    std::cout << "      ";
                    for (int k = 0; k < BoardSize; k++)
                    {
                        std::cout << (k == x ? "|--|" : "----");
                    }
                }
                std::cout << '\n' << posi[i] << "  ";
                for (int j = 0; j < BoardSize; j++)
                {
                    switch (brd[j][i])
                    {
                    case PieceState::Black:
                        std::cout << (j == x ? "|■■|" : " ■■ ");
                        break;
                    case PieceState::White:
                        std::cout << (j == x ? "|□□|" : " □□ ");
                        break;
                    case PieceState::Empty:
                        std::cout << (j == x ? "|  |" : "    ");
                        break;
                    }
                }
                std::cout << '\n';
                if (i == y)
                {
                    std::cout << "      ";
                    for (int k = 0; k < BoardSize; k++)
                    {
                        std::cout << (k == x ? "|--|" : "----");
                    }
                }
                else if (i != y - 1)
                {
                    std::cout << "      ";
                    for (int j = 0; j < BoardSize; j++)
                    {
                        std::cout << (j == x ? "|  |" : "    ");
                    }
                }
            }
            std::cout << "\n\n";
            std::cout.flush();
        };

        if (brd.state() == BoardState::WinBlack)
        {
            showBoard();
            std::cout << "computer win\n";
            break;
        }

        showBoard();

        for (int ch, change = 0;; change = 0)
        {
            ch = keyBoradHit();
            if (ch == 'w' || ch == 'W')
            {
                if (y > 0)
                    y--;
                change = 1;
            }
            else if (ch == 'a' || ch == 'A')
            {
                if (x > 0)
                    x--;
                change = 1;
            }
            else if (ch == 's' || ch == 'S')
            {
                if (y < BoardSize - 1)
                    y++;
                change = 1;
            }
            else if (ch == 'd' || ch == 'D')
            {
                if (x < BoardSize - 1)
                    x++;
                change = 1;
            }
            else if (ch == ' ')
            {
                if (brd[x][y] == PieceState::Empty)
                {
                    break;
                }
                change = 1;
            }
            if (change)
            {
                showBoard();
            }
        }

        brd.put(x, y);
        showBoard();
    }

    return 0;
}