|
|
# -*- coding: utf-8 -*-
|
|
|
"""
|
|
|
Created on Wed May 25 10:04:02 2022
|
|
|
|
|
|
@author: Lenovo
|
|
|
"""
|
|
|
|
|
|
#调用pygame库
|
|
|
import pygame
|
|
|
import sys
|
|
|
#调用常用关键字常量
|
|
|
from pygame.locals import QUIT,KEYDOWN
|
|
|
import numpy as np
|
|
|
#初始化pygame
|
|
|
pygame.init()
|
|
|
#获取对显示系统的访问，并创建一个窗口screen
|
|
|
#窗口大小为670x670
|
|
|
screen = pygame.display.set_mode((670,670))
|
|
|
screen_color=[238,154,73]#设置画布颜色,[238,154,73]对应为棕黄色
|
|
|
line_color = [0,0,0]#设置线条颜色，[0,0,0]对应黑色
|
|
|
|
|
|
def check_win(over_pos):#判断五子连心
|
|
|
mp=np.zeros([15,15],dtype=int)
|
|
|
for val in over_pos:
|
|
|
x=int((val[0][0]-27)/44)
|
|
|
y=int((val[0][1]-27)/44)
|
|
|
if val[1]==white_color:
|
|
|
mp[x][y]=2#表示白子
|
|
|
else:
|
|
|
mp[x][y]=1#表示黑子
|
|
|
|
|
|
for i in range(15):
|
|
|
pos1=[]
|
|
|
pos2=[]
|
|
|
for j in range(15):
|
|
|
if mp[i][j]==1:
|
|
|
pos1.append([i,j])
|
|
|
else:
|
|
|
pos1=[]
|
|
|
if mp[i][j]==2:
|
|
|
pos2.append([i,j])
|
|
|
else:
|
|
|
pos2=[]
|
|
|
if len(pos1)>=5:#五子连心
|
|
|
return [1,pos1]
|
|
|
if len(pos2)>=5:
|
|
|
return [2,pos2]
|
|
|
|
|
|
for j in range(15):
|
|
|
pos1=[]
|
|
|
pos2=[]
|
|
|
for i in range(15):
|
|
|
if mp[i][j]==1:
|
|
|
pos1.append([i,j])
|
|
|
else:
|
|
|
pos1=[]
|
|
|
if mp[i][j]==2:
|
|
|
pos2.append([i,j])
|
|
|
else:
|
|
|
pos2=[]
|
|
|
if len(pos1)>=5:
|
|
|
return [1,pos1]
|
|
|
if len(pos2)>=5:
|
|
|
return [2,pos2]
|
|
|
for i in range(15):
|
|
|
for j in range(15):
|
|
|
pos1=[]
|
|
|
pos2=[]
|
|
|
for k in range(15):
|
|
|
if i+k>=15 or j+k>=15:
|
|
|
break
|
|
|
if mp[i+k][j+k]==1:
|
|
|
pos1.append([i+k,j+k])
|
|
|
else:
|
|
|
pos1=[]
|
|
|
if mp[i+k][j+k]==2:
|
|
|
pos2.append([i+k,j+k])
|
|
|
else:
|
|
|
pos2=[]
|
|
|
if len(pos1)>=5:
|
|
|
return [1,pos1]
|
|
|
if len(pos2)>=5:
|
|
|
return [2,pos2]
|
|
|
for i in range(15):
|
|
|
for j in range(15):
|
|
|
pos1=[]
|
|
|
pos2=[]
|
|
|
for k in range(15):
|
|
|
if i+k>=15 or j-k<0:
|
|
|
break
|
|
|
if mp[i+k][j-k]==1:
|
|
|
pos1.append([i+k,j-k])
|
|
|
else:
|
|
|
pos1=[]
|
|
|
if mp[i+k][j-k]==2:
|
|
|
pos2.append([i+k,j-k])
|
|
|
else:
|
|
|
pos2=[]
|
|
|
if len(pos1)>=5:
|
|
|
return [1,pos1]
|
|
|
if len(pos2)>=5:
|
|
|
return [2,pos2]
|
|
|
return [0,[]]
|
|
|
|
|
|
def find_pos(x,y):#找到显示的可以落子的位置
|
|
|
for i in range(27,670,44):
|
|
|
for j in range(27,670,44):
|
|
|
L1=i-22
|
|
|
L2=i+22
|
|
|
R1=j-22
|
|
|
R2=j+22
|
|
|
if x>=L1 and x<=L2 and y>=R1 and y<=R2:
|
|
|
return i,j
|
|
|
return x,y
|
|
|
|
|
|
def check_over_pos(x,y,over_pos):#检查当前的位置是否已经落子
|
|
|
for val in over_pos:
|
|
|
if val[0][0]==x and val[0][1]==y:
|
|
|
return False
|
|
|
return True#表示没有落子
|
|
|
flag=False
|
|
|
tim=0
|
|
|
|
|
|
over_pos=[]#表示已经落子的位置
|
|
|
white_color=[255,255,255]#白棋颜色
|
|
|
black_color=[0,0,0]#黑棋颜色
|
|
|
|
|
|
while True:#不断训练刷新画布
|
|
|
|
|
|
for event in pygame.event.get():#获取事件，如果鼠标点击右上角关闭按钮，关闭
|
|
|
if event.type in (QUIT,KEYDOWN):
|
|
|
sys.exit()
|
|
|
|
|
|
screen.fill(screen_color)#清屏
|
|
|
for i in range(27,670,44):
|
|
|
#先画竖线
|
|
|
if i==27 or i==670-27:#边缘线稍微粗一些
|
|
|
pygame.draw.line(screen,line_color,[i,27],[i,670-27],4)
|
|
|
else:
|
|
|
pygame.draw.line(screen,line_color,[i,27],[i,670-27],2)
|
|
|
#再画横线
|
|
|
if i==27 or i==670-27:#边缘线稍微粗一些
|
|
|
pygame.draw.line(screen,line_color,[27,i],[670-27,i],4)
|
|
|
else:
|
|
|
pygame.draw.line(screen,line_color,[27,i],[670-27,i],2)
|
|
|
|
|
|
#在棋盘中心画个小圆表示正中心位置
|
|
|
pygame.draw.circle(screen, line_color,[27+44*7,27+44*7], 8,0)
|
|
|
|
|
|
for val in over_pos:#显示所有落下的棋子
|
|
|
pygame.draw.circle(screen, val[1],val[0], 20,0)
|
|
|
|
|
|
#判断是否存在五子连心
|
|
|
res=check_win(over_pos)
|
|
|
if res[0]!=0:
|
|
|
for pos in res[1]:
|
|
|
pygame.draw.rect(screen,[238,48,167],[pos[0]*44+27-22,pos[1]*44+27-22,44,44],2,1)
|
|
|
pygame.display.update()#刷新显示
|
|
|
continue#游戏结束，停止下面的操作
|
|
|
#获取鼠标坐标信息
|
|
|
x,y = pygame.mouse.get_pos()
|
|
|
|
|
|
x,y=find_pos(x,y)
|
|
|
if check_over_pos(x,y,over_pos):#判断是否可以落子，再显示
|
|
|
pygame.draw.rect(screen,[0 ,229 ,238 ],[x-22,y-22,44,44],2,1)
|
|
|
|
|
|
keys_pressed = pygame.mouse.get_pressed()#获取鼠标按键信息
|
|
|
#鼠标左键表示落子,tim用来延时的，因为每次循环时间间隔很断，容易导致明明只按了一次左键，却被多次获取，认为我按了多次
|
|
|
if keys_pressed[0] and tim==0:
|
|
|
flag=True
|
|
|
if check_over_pos(x,y,over_pos):#判断是否可以落子，再落子
|
|
|
if len(over_pos)%2==0:#黑子
|
|
|
over_pos.append([[x,y],black_color])
|
|
|
else:
|
|
|
over_pos.append([[x,y],white_color])
|
|
|
|
|
|
#鼠标左键延时作用
|
|
|
if flag:
|
|
|
tim+=1
|
|
|
if tim%50==0:#延时200ms
|
|
|
flag=False
|
|
|
tim=0
|
|
|
|
|
|
pygame.display.update()#刷新显示 |
