上期视频，我们分享了一个AI来玩贪吃蛇的视频，本期我们讲解一下其基础代码，利用本代码自己也可以写游戏了。
游戏界面
贪吃蛇小游戏是一代人的梦，本期文章我们简单制作一个贪吃蛇小游戏import cv2import numpy as npfrom random import randintfrom random import choice
代码截图
首先导入我们需要的第三方库，然后定义一个贪吃蛇的类来定义贪吃蛇的运动，头部等信息
class SnakePart: def __init__(self, front, x, y): self.front = front self.x = x self.y = y def move(self): self.x = self.front.x self.y = self.front.y
代码截图
此类我们定义贪吃蛇的身体以及运动等
class Head: def __init__(self, direction, x, y): self.direction = direction self.x = x self.y = y def move(self): if self.direction == 0: self.x += 1 elif self.direction == 1: self.y += 1 elif self.direction == 2: self.x -= 1 elif self.direction == 3: self.y -= 1
代码截图
然后我们定义贪吃蛇的头部以及运动信息，这里我们控制键盘的W\S\A\D四个键来控制贪吃蛇头部的运动方向
def display(): board = np.zeros([BOARD_SIZE, BOARD_SIZE, 3]) for part in snake: board[part.y, part.x] = [0, 255, 0] board[appley, applex] = [0, 0, 255] cv2.imshow("Snake Game", np.uint8(board.repeat(CELL_SIZE, 0).repeat(CELL_SIZE, 1))) key = cv2.waitKey(int(1000/SPEED)) return key
代码截图
Display类来显示贪吃蛇的颜色以及要吃的苹果的颜色
def win_focus(): cv2.namedWindow("Snake Game", cv2.WINDOW_AUTOSIZE); board = np.zeros([BOARD_SIZE * CELL_SIZE, BOARD_SIZE * CELL_SIZE, 3]) cv2.imshow("Snake Game", board); cv2.setWindowProperty("Snake Game", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN); cv2.waitKey(2000) cv2.setWindowProperty("Snake Game", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_AUTOSIZE)
代码截图
Win_focus 类函数来显示游戏画面
1. cv2.VideoCapture(0) #构建视频抓捕器
参数说明：0表示需要启动的摄像头，这里也可以写视频的路径
2. cv2.nameWindow(name, cv2.WINDOW_NORMAL) # 构建视频的窗口
参数说明: 表示窗口的名字, cv2.WINDOW_NORMAL表示窗口的大小，这里窗口的大小是正常,
3.cv2.setWindowProperty(name, cv2.WND_PROP_FULLSCREEN, cv2.WND_PROP_FULLSCREEN)
参数说明: name表示需要更改像素的窗口名字, cv2.WND_PROP_FULLSCREEN表示全屏
4. video_capture.set(cv2.CAP_PROP_FRAME_WIDTH, 1920) 对窗口像素进行设置
参数说明: cv2.CAP_PROP_FRAME_WIDTH 表示设置其宽的大小, 1920表示设置的像素
以上参数类设置后，我们开始主函数的部分
if __name__ == '__main__' : # 每个cell的大小 CELL_SIZE = 20 # cell的数量 BOARD_SIZE = 45 # 游戏中贪吃蛇的运行速度 SPEED = 6 # 贪吃蛇成长的速度 GROWTH = 3 eaten = True quit = False grow = 0 snake = [] #游戏中从屏幕中间开始执行 head = Head(0, int((BOARD_SIZE - 1)/2), int((BOARD_SIZE - 1)/2)) snake.append(head) # 设置键盘 print('w = top, a = left, s = down, d = right') win_focus() while True: if eaten: s = list(range(0, BOARD_SIZE ** 2)) for part in snake: s.remove(part.x * BOARD_SIZE + part.y) a = choice(s) applex = int(a/BOARD_SIZE) appley = a % BOARD_SIZE eaten = False
代码截图
接下来，我们设置一下贪吃蛇的键盘移动设置，W/A/S/D，分别代表贪吃蛇的移动方向，这个跟普通游戏设置的一样，由于opencv不能设置箭头，这里只能设置为W/A/S/D
key = display()if key == 8 or key == 27: breakelif key == ord("d") : head.direction = 0elif key == ord("s") : head.direction = 1elif key == ord("a") : head.direction = 2elif key == ord("w") : head.direction = 3
代码截图
设置完成后，就可以进行贪吃蛇的移动控制，以及贪吃蛇的游戏规则，当贪吃蛇吃过一个苹果后，其身体要增加一个大小尺寸
# 移动贪吃蛇for part in snake[::-1]: part.move()# 设置游戏规则if head.x < 0 or head.x > BOARD_SIZE - 1 or head.y < 0 or head.y > BOARD_SIZE - 1: breakfor part in snake[1:]: if head.x == part.x and head.y == part.y: quit = True breakif quit: break# 设置贪吃蛇if grow > 0: snake.append(SnakePart(snake[-1], subx, suby)) grow -= 1# 当吃过一个苹果后，增加贪吃蛇的长度if applex == head.x and appley == head.y: subx = snake[-1].x suby = snake[-1].y eaten = True grow += GROWTH
Ok,所有设置完成后，便可以进行游戏了，当然这里只是设置了基本的贪吃蛇规格，你也可以在代码里面增加分数设置，看看你能跑多少分吧
游戏截图
