79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
import pygame
|
|
|
|
EMPTY = 0
|
|
BLACK = 1
|
|
WHITE = 2
|
|
|
|
black_color = [0, 0, 0]
|
|
white_color = [255, 255, 255]
|
|
|
|
|
|
class RenjuBoard(object):
|
|
|
|
def __init__(self):
|
|
self._board = [[]] * 15
|
|
self.reset()
|
|
|
|
def reset(self):
|
|
for row in range(len(self._board)):
|
|
self._board[row] = [EMPTY] * 15
|
|
|
|
def move(self, row, col, is_black):
|
|
if self._board[row][col] == EMPTY:
|
|
self._board[row][col] = BLACK if is_black else WHITE
|
|
return True
|
|
return False
|
|
|
|
def draw(self, screen):
|
|
for index in range(1, 16):
|
|
pygame.draw.line(screen, black_color,
|
|
[40, 40 * index], [600, 40 * index], 1)
|
|
pygame.draw.line(screen, black_color,
|
|
[40 * index, 40], [40 * index, 600], 1)
|
|
pygame.draw.rect(screen, black_color, [36, 36, 568, 568], 4)
|
|
pygame.draw.circle(screen, black_color, [320, 320], 5, 0)
|
|
pygame.draw.circle(screen, black_color, [160, 160], 5, 0)
|
|
pygame.draw.circle(screen, black_color, [480, 480], 5, 0)
|
|
pygame.draw.circle(screen, black_color, [480, 160], 5, 0)
|
|
pygame.draw.circle(screen, black_color, [160, 480], 5, 0)
|
|
for row in range(len(self._board)):
|
|
for col in range(len(self._board[row])):
|
|
if self._board[row][col] != EMPTY:
|
|
ccolor = black_color \
|
|
if self._board[row][col] == BLACK else white_color
|
|
pos = [40 * (col + 1), 40 * (row + 1)]
|
|
pygame.draw.circle(screen, ccolor, pos, 20, 0)
|
|
|
|
|
|
def main():
|
|
board = RenjuBoard()
|
|
is_black = True
|
|
pygame.init()
|
|
pygame.display.set_caption('五子棋')
|
|
screen = pygame.display.set_mode([640, 640])
|
|
screen.fill([255, 255, 0])
|
|
board.draw(screen)
|
|
pygame.display.flip()
|
|
running = True
|
|
while running:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
running = False
|
|
elif event.type == pygame.KEYUP:
|
|
pass
|
|
elif event.type == pygame.MOUSEBUTTONDOWN\
|
|
and event.button == 1:
|
|
x, y = event.pos
|
|
row = round((y - 40) / 40)
|
|
col = round((x - 40) / 40)
|
|
if board.move(row, col, is_black):
|
|
is_black = not is_black
|
|
screen.fill([255, 255, 0])
|
|
board.draw(screen)
|
|
pygame.display.flip()
|
|
pygame.quit()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|