电脑时间 发表于 2022-12-1 15:56:01

Python 五子棋

本帖最后由 电脑时间 于 2022-12-2 09:16 编辑

知道最近我怎么没发帖么?
我***,因为人机程序太难编写了


所以我今天给的是五子棋
人机版(高级版)
源码
人人对战
"""五子棋之人人对战"""

import sys
import pygame
from pygame.locals import *
import pygame.gfxdraw
from checkerboard import Checkerboard, BLACK_CHESSMAN, WHITE_CHESSMAN, Point


SIZE = 30
Line_Points = 19
Outer_Width = 20
Border_Width = 4
Inside_Width = 4
Border_Length = SIZE * (Line_Points - 1) + Inside_Width * 2 + Border_Width
Start_X = Start_Y = Outer_Width + int(Border_Width / 2) + Inside_Width
SCREEN_HEIGHT = SIZE * (Line_Points - 1) + Outer_Width * 2 + Border_Width + Inside_Width * 2
SCREEN_WIDTH = SCREEN_HEIGHT + 200

Stone_Radius = SIZE // 2 - 3
Stone_Radius2 = SIZE // 2 + 3
Checkerboard_Color = (0xE3, 0x92, 0x65)
BLACK_COLOR = (0, 0, 0)
WHITE_COLOR = (255, 255, 255)
RED_COLOR = (200, 30, 30)
BLUE_COLOR = (30, 30, 200)
BLACK_STONE_COLOR = (45, 45, 45)
WHITE_STONE_COLOR = (219, 219, 219)

RIGHT_INFO_POS_X = SCREEN_HEIGHT + Stone_Radius2 * 2 + 10


def print_text(screen, font, x, y, text, fcolor=(255, 255, 255)):
    imgText = font.render(text, True, fcolor)
    screen.blit(imgText, (x, y))


def main():
    pygame.init()
    screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
    pygame.display.set_caption('五子棋')

    font1 = pygame.font.SysFont('SimHei', 36)
    font2 = pygame.font.SysFont('SimHei', 72)
    fwidth, fheight = font2.size('黑方获胜')

    checkerboard = Checkerboard(Line_Points)
    cur_runner = BLACK_CHESSMAN
    winner = None

    while True:
      for event in pygame.event.get():
            if event.type == QUIT:
                sys.exit()
            elif event.type == KEYDOWN:
                if event.key == K_RETURN:
                  if winner is not None:
                        winner = None
                        cur_runner = BLACK_CHESSMAN
                        checkerboard = Checkerboard(Line_Points)
            elif event.type == MOUSEBUTTONDOWN:
                if winner is None:
                  pressed_array = pygame.mouse.get_pressed()
                  if pressed_array:
                        mouse_pos = pygame.mouse.get_pos()
                        click_point = _get_clickpoint(mouse_pos)
                        if click_point is not None:
                            if checkerboard.can_drop(click_point):
                              winner = checkerboard.drop(cur_runner, click_point)
                              if cur_runner == BLACK_CHESSMAN:
                                    cur_runner = WHITE_CHESSMAN
                              else:
                                    cur_runner = BLACK_CHESSMAN
                        else:
                            print('超出棋盘区域')

      _draw_checkerboard(screen)

      for i, row in enumerate(checkerboard.checkerboard):
            for j, cell in enumerate(row):
                if cell == BLACK_CHESSMAN.Value:
                  _draw_chessman(screen, Point(j, i), BLACK_CHESSMAN.Color)
                elif cell == WHITE_CHESSMAN.Value:
                  _draw_chessman(screen, Point(j, i), WHITE_CHESSMAN.Color)

      _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, Start_X + 20), BLACK_STONE_COLOR)
      _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, Start_X + 20 + Stone_Radius2 * 3), WHITE_STONE_COLOR)

      if winner:
            print_text(screen, font2, (SCREEN_WIDTH - fwidth)//2, (SCREEN_HEIGHT - fheight)//2, winner.Name + '获胜', RED_COLOR)

      if cur_runner == BLACK_CHESSMAN:
            print_text(screen, font1, RIGHT_INFO_POS_X, Start_X, '失败' if winner else '落子中', BLUE_COLOR)
      else:
            print_text(screen, font1, RIGHT_INFO_POS_X, Start_X + Stone_Radius2 * 3, '失败' if winner else '落子中', BLUE_COLOR)

      pygame.display.flip()


def _draw_checkerboard(screen):
    screen.fill(Checkerboard_Color)
    pygame.draw.rect(screen, BLACK_COLOR, (Outer_Width, Outer_Width, Border_Length, Border_Length), Border_Width)
    for i in range(Line_Points):
      pygame.draw.line(screen, BLACK_COLOR,
                         (Start_Y, Start_Y + SIZE * i),
                         (Start_Y + SIZE * (Line_Points - 1), Start_Y + SIZE * i),
                         1)
    for j in range(Line_Points):
      pygame.draw.line(screen, BLACK_COLOR,
                         (Start_X + SIZE * j, Start_X),
                         (Start_X + SIZE * j, Start_X + SIZE * (Line_Points - 1)),
                         1)
    for i in (3, 9, 15):
      for j in (3, 9, 15):
            if i == j == 9:
                radius = 5
            else:
                radius = 3
            # pygame.draw.circle(screen, BLACK, (Start_X + SIZE * i, Start_Y + SIZE * j), radius)
            pygame.gfxdraw.aacircle(screen, Start_X + SIZE * i, Start_Y + SIZE * j, radius, BLACK_COLOR)
            pygame.gfxdraw.filled_circle(screen, Start_X + SIZE * i, Start_Y + SIZE * j, radius, BLACK_COLOR)


def _draw_chessman(screen, point, stone_color):
    # pygame.draw.circle(screen, stone_color, (Start_X + SIZE * point.X, Start_Y + SIZE * point.Y), Stone_Radius)
    pygame.gfxdraw.aacircle(screen, Start_X + SIZE * point.X, Start_Y + SIZE * point.Y, Stone_Radius, stone_color)
    pygame.gfxdraw.filled_circle(screen, Start_X + SIZE * point.X, Start_Y + SIZE * point.Y, Stone_Radius, stone_color)


def _draw_chessman_pos(screen, pos, stone_color):
    pygame.gfxdraw.aacircle(screen, pos, pos, Stone_Radius2, stone_color)
    pygame.gfxdraw.filled_circle(screen, pos, pos, Stone_Radius2, stone_color)


def _get_clickpoint(click_pos):
    pos_x = click_pos - Start_X
    pos_y = click_pos - Start_Y
    if pos_x < -Inside_Width or pos_y < -Inside_Width:
      return None
    x = pos_x // SIZE
    y = pos_y // SIZE
    if pos_x % SIZE > Stone_Radius:
      x += 1
    if pos_y % SIZE > Stone_Radius:
      y += 1
    if x >= Line_Points or y >= Line_Points:
      return None

    return Point(x, y)


if __name__ == '__main__':
    main()
人机版(不开大一点你们看不见)
"""五子棋之人机对战"""

import sys
import random
import pygame
from pygame.locals import *
import pygame.gfxdraw
from checkerboard import Checkerboard, BLACK_CHESSMAN, WHITE_CHESSMAN, offset, Point

SIZE = 30
Line_Points = 19
Outer_Width = 20
Border_Width = 4
Inside_Width = 4
Border_Length = SIZE * (Line_Points - 1) + Inside_Width * 2 + Border_Width
Start_X = Start_Y = Outer_Width + int(Border_Width / 2) + Inside_Width
SCREEN_HEIGHT = SIZE * (Line_Points - 1) + Outer_Width * 2 + Border_Width + Inside_Width * 2
SCREEN_WIDTH = SCREEN_HEIGHT + 200

Stone_Radius = SIZE // 2 - 3
Stone_Radius2 = SIZE // 2 + 3
Checkerboard_Color = (0xE3, 0x92, 0x65)
BLACK_COLOR = (0, 0, 0)
WHITE_COLOR = (255, 255, 255)
RED_COLOR = (200, 30, 30)
BLUE_COLOR = (30, 30, 200)

RIGHT_INFO_POS_X = SCREEN_HEIGHT + Stone_Radius2 * 2 + 10


def print_text(screen, font, x, y, text, fcolor=(255, 255, 255)):
    imgText = font.render(text, True, fcolor)
    screen.blit(imgText, (x, y))


def main():
    pygame.init()
    screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
    pygame.display.set_caption('五子棋')

    font1 = pygame.font.SysFont('SimHei', 32)
    font2 = pygame.font.SysFont('SimHei', 72)
    fwidth, fheight = font2.size('黑方获胜')

    checkerboard = Checkerboard(Line_Points)
    cur_runner = BLACK_CHESSMAN
    winner = None
    computer = AI(Line_Points, WHITE_CHESSMAN)

    black_win_count = 0
    white_win_count = 0

    while True:
      for event in pygame.event.get():
            if event.type == QUIT:
                sys.exit()
            elif event.type == KEYDOWN:
                if event.key == K_RETURN:
                  if winner is not None:
                        winner = None
                        cur_runner = BLACK_CHESSMAN
                        checkerboard = Checkerboard(Line_Points)
                        computer = AI(Line_Points, WHITE_CHESSMAN)
            elif event.type == MOUSEBUTTONDOWN:
                if winner is None:
                  pressed_array = pygame.mouse.get_pressed()
                  if pressed_array:
                        mouse_pos = pygame.mouse.get_pos()
                        click_point = _get_clickpoint(mouse_pos)
                        if click_point is not None:
                            if checkerboard.can_drop(click_point):
                              winner = checkerboard.drop(cur_runner, click_point)
                              if winner is None:
                                    cur_runner = _get_next(cur_runner)
                                    computer.get_opponent_drop(click_point)
                                    AI_point = computer.AI_drop()
                                    winner = checkerboard.drop(cur_runner, AI_point)
                                    if winner is not None:
                                        white_win_count += 1
                                    cur_runner = _get_next(cur_runner)
                              else:
                                    black_win_count += 1
                        else:
                            print('超出棋盘区域')

      _draw_checkerboard(screen)

      for i, row in enumerate(checkerboard.checkerboard):
            for j, cell in enumerate(row):
                if cell == BLACK_CHESSMAN.Value:
                  _draw_chessman(screen, Point(j, i), BLACK_CHESSMAN.Color)
                elif cell == WHITE_CHESSMAN.Value:
                  _draw_chessman(screen, Point(j, i), WHITE_CHESSMAN.Color)

      _draw_left_info(screen, font1, cur_runner, black_win_count, white_win_count)

      if winner:
            print_text(screen, font2, (SCREEN_WIDTH - fwidth)//2, (SCREEN_HEIGHT - fheight)//2, winner.Name + '获胜', RED_COLOR)

      pygame.display.flip()


def _get_next(cur_runner):
    if cur_runner == BLACK_CHESSMAN:
      return WHITE_CHESSMAN
    else:
      return BLACK_CHESSMAN


def _draw_checkerboard(screen):
    screen.fill(Checkerboard_Color)
    pygame.draw.rect(screen, BLACK_COLOR, (Outer_Width, Outer_Width, Border_Length, Border_Length), Border_Width)
    for i in range(Line_Points):
      pygame.draw.line(screen, BLACK_COLOR,
                         (Start_Y, Start_Y + SIZE * i),
                         (Start_Y + SIZE * (Line_Points - 1), Start_Y + SIZE * i),
                         1)
    for j in range(Line_Points):
      pygame.draw.line(screen, BLACK_COLOR,
                         (Start_X + SIZE * j, Start_X),
                         (Start_X + SIZE * j, Start_X + SIZE * (Line_Points - 1)),
                         1)
    for i in (3, 9, 15):
      for j in (3, 9, 15):
            if i == j == 9:
                radius = 5
            else:
                radius = 3
            pygame.gfxdraw.aacircle(screen, Start_X + SIZE * i, Start_Y + SIZE * j, radius, BLACK_COLOR)
            pygame.gfxdraw.filled_circle(screen, Start_X + SIZE * i, Start_Y + SIZE * j, radius, BLACK_COLOR)

def _draw_chessman(screen, point, stone_color):
    pygame.gfxdraw.aacircle(screen, Start_X + SIZE * point.X, Start_Y + SIZE * point.Y, Stone_Radius, stone_color)
    pygame.gfxdraw.filled_circle(screen, Start_X + SIZE * point.X, Start_Y + SIZE * point.Y, Stone_Radius, stone_color)


def _draw_left_info(screen, font, cur_runner, black_win_count, white_win_count):
    _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, Start_X + Stone_Radius2), BLACK_CHESSMAN.Color)
    _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, Start_X + Stone_Radius2 * 4), WHITE_CHESSMAN.Color)

    print_text(screen, font, RIGHT_INFO_POS_X, Start_X + 3, '玩家', BLUE_COLOR)
    print_text(screen, font, RIGHT_INFO_POS_X, Start_X + Stone_Radius2 * 3 + 3, '电脑', BLUE_COLOR)

    print_text(screen, font, SCREEN_HEIGHT, SCREEN_HEIGHT - Stone_Radius2 * 8, '战况:', BLUE_COLOR)
    _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, SCREEN_HEIGHT - int(Stone_Radius2 * 4.5)), BLACK_CHESSMAN.Color)
    _draw_chessman_pos(screen, (SCREEN_HEIGHT + Stone_Radius2, SCREEN_HEIGHT - Stone_Radius2 * 2), WHITE_CHESSMAN.Color)
    print_text(screen, font, RIGHT_INFO_POS_X, SCREEN_HEIGHT - int(Stone_Radius2 * 5.5) + 3, f'{black_win_count} 胜', BLUE_COLOR)
    print_text(screen, font, RIGHT_INFO_POS_X, SCREEN_HEIGHT - Stone_Radius2 * 3 + 3, f'{white_win_count} 胜', BLUE_COLOR)


def _draw_chessman_pos(screen, pos, stone_color):
    pygame.gfxdraw.aacircle(screen, pos, pos, Stone_Radius2, stone_color)
    pygame.gfxdraw.filled_circle(screen, pos, pos, Stone_Radius2, stone_color)


def _get_clickpoint(click_pos):
    pos_x = click_pos - Start_X
    pos_y = click_pos - Start_Y
    if pos_x < -Inside_Width or pos_y < -Inside_Width:
      return None
    x = pos_x // SIZE
    y = pos_y // SIZE
    if pos_x % SIZE > Stone_Radius:
      x += 1
    if pos_y % SIZE > Stone_Radius:
      y += 1
    if x >= Line_Points or y >= Line_Points:
      return None

    return Point(x, y)


class AI:
    def __init__(self, line_points, chessman):
      self._line_points = line_points
      self._my = chessman
      self._opponent = BLACK_CHESSMAN if chessman == WHITE_CHESSMAN else WHITE_CHESSMAN
      self._checkerboard = [ * line_points for _ in range(line_points)]

    def get_opponent_drop(self, point):
      self._checkerboard = self._opponent.Value

    def AI_drop(self):
      point = None
      score = 0
      for i in range(self._line_points):
            for j in range(self._line_points):
                if self._checkerboard == 0:
                  _score = self._get_point_score(Point(i, j))
                  if _score > score:
                        score = _score
                        point = Point(i, j)
                  elif _score == score and _score > 0:
                        r = random.randint(0, 100)
                        if r % 2 == 0:
                            point = Point(i, j)
      self._checkerboard = self._my.Value
      return point

    def _get_point_score(self, point):
      score = 0
      for os in offset:
            score += self._get_direction_score(point, os, os)
      return score

    def _get_direction_score(self, point, x_offset, y_offset):
      count = 0
      _count = 0
      space = None   
      _space = None
      both = 0   
      _both = 0   

      flag = self._get_stone_color(point, x_offset, y_offset, True)
      if flag != 0:
            for step in range(1, 6):
                x = point.X + step * x_offset
                y = point.Y + step * y_offset
                if 0 <= x < self._line_points and 0 <= y < self._line_points:
                  if flag == 1:
                        if self._checkerboard == self._my.Value:
                            count += 1
                            if space is False:
                              space = True
                        elif self._checkerboard == self._opponent.Value:
                            _both += 1
                            break
                        else:
                            if space is None:
                              space = False
                            else:
                              break
                  elif flag == 2:
                        if self._checkerboard == self._my.Value:
                            _both += 1
                            break
                        elif self._checkerboard == self._opponent.Value:
                            _count += 1
                            if _space is False:
                              _space = True
                        else:
                            if _space is None:
                              _space = False
                            else:
                              break
                else:
                  if flag == 1:
                        both += 1
                  elif flag == 2:
                        _both += 1

      if space is False:
            space = None
      if _space is False:
            _space = None

      _flag = self._get_stone_color(point, -x_offset, -y_offset, True)
      if _flag != 0:
            for step in range(1, 6):
                x = point.X - step * x_offset
                y = point.Y - step * y_offset
                if 0 <= x < self._line_points and 0 <= y < self._line_points:
                  if _flag == 1:
                        if self._checkerboard == self._my.Value:
                            count += 1
                            if space is False:
                              space = True
                        elif self._checkerboard == self._opponent.Value:
                            _both += 1
                            break
                        else:
                            if space is None:
                              space = False
                            else:
                              break   
                  elif _flag == 2:
                        if self._checkerboard == self._my.Value:
                            _both += 1
                            break
                        elif self._checkerboard == self._opponent.Value:
                            _count += 1
                            if _space is False:
                              _space = True
                        else:
                            if _space is None:
                              _space = False
                            else:
                              break
                else:
                  if _flag == 1:
                        both += 1
                  elif _flag == 2:
                        _both += 1

      score = 0
      if count == 4:
            score = 10000
      elif _count == 4:
            score = 9000
      elif count == 3:
            if both == 0:
                score = 1000
            elif both == 1:
                score = 100
            else:
                score = 0
      elif _count == 3:
            if _both == 0:
                score = 900
            elif _both == 1:
                score = 90
            else:
                score = 0
      elif count == 2:
            if both == 0:
                score = 100
            elif both == 1:
                score = 10
            else:
                score = 0
      elif _count == 2:
            if _both == 0:
                score = 90
            elif _both == 1:
                score = 9
            else:
                score = 0
      elif count == 1:
            score = 10
      elif _count == 1:
            score = 9
      else:
            score = 0

      if space or _space:
            score /= 2

      return score

    def _get_stone_color(self, point, x_offset, y_offset, next):
      x = point.X + x_offset
      y = point.Y + y_offset
      if 0 <= x < self._line_points and 0 <= y < self._line_points:
            if self._checkerboard == self._my.Value:
                return 1
            elif self._checkerboard == self._opponent.Value:
                return 2
            else:
                if next:
                  return self._get_stone_color(Point(x, y), x_offset, y_offset, False)
                else:
                  return 0
      else:
            return 0


if __name__ == '__main__':
    main()

对了还要有一个chekerboard
from collections import namedtuple

Chessman = namedtuple('Chessman', 'Name Value Color')
Point = namedtuple('Point', 'X Y')

BLACK_CHESSMAN = Chessman('黑子', 1, (45, 45, 45))
WHITE_CHESSMAN = Chessman('白子', 2, (219, 219, 219))

offset = [(1, 0), (0, 1), (1, 1), (1, -1)]


class Checkerboard:
    def __init__(self, line_points):
      self._line_points = line_points
      self._checkerboard = [ * line_points for _ in range(line_points)]

    def _get_checkerboard(self):
      return self._checkerboard

    checkerboard = property(_get_checkerboard)

    def can_drop(self, point):
      return self._checkerboard == 0

    def drop(self, chessman, point):
      print(f'{chessman.Name} ({point.X}, {point.Y})')
      self._checkerboard = chessman.Value

      if self._win(point):
            print(f'{chessman.Name}获胜')
            return chessman

    def _win(self, point):
      cur_value = self._checkerboard
      for os in offset:
            if self._get_count_on_direction(point, cur_value, os, os):
                return True

    def _get_count_on_direction(self, point, value, x_offset, y_offset):
      count = 1
      for step in range(1, 5):
            x = point.X + step * x_offset
            y = point.Y + step * y_offset
            if 0 <= x < self._line_points and 0 <= y < self._line_points and self._checkerboard == value:
                count += 1
            else:
                break
      for step in range(1, 5):
            x = point.X - step * x_offset
            y = point.Y - step * y_offset
            if 0 <= x < self._line_points and 0 <= y < self._line_points and self._checkerboard == value:
                count += 1
            else:
                break

      return count >= 5


敲了我5个小时{:16:}

0Percent 发表于 2022-12-1 16:09:02

可以可以 我要下回来试试

jyssysz 发表于 2022-12-1 16:45:35

这个有兴趣的人可以试一下,谢谢楼主的分享。

0Percent 发表于 2022-12-1 16:54:29

我服了没装pygame pip都没有 国内curl pip速度感人

爱电脑的昕宇 发表于 2022-12-1 20:08:21

我用Python3.11环境安装pygame报错

0Percent 发表于 2022-12-2 09:20:30

爱电脑的昕宇 发表于 2022-12-1 20:08
我用Python3.11环境安装pygame报错

我macOS是pip都没装上...

wdscxsj 发表于 2022-12-2 15:45:53

Ace_qingfeng 发表于 2022-12-2 09:20
我macOS是pip都没装上...

参见这里的回复:https://bbs.luobotou.org/thread-53765-1-1.html

追光者 发表于 2022-12-3 10:36:05

厉害,毅力可真强,如果是我早就废了                        
页: [1] 2
查看完整版本: Python 五子棋