将蛇吸引到我的pygame游戏中似乎不起作用

时间:2019-05-08 13:28:30

标签: python pygame

在绘制我在pygame中设置的蛇时,我目前遇到问题。

我想画一条黑盒子,把水果当成红色盒子来做一条蛇,但是当我画它们时,我会看到不同类型的盒子和形状。

我仅画蛇的身体和果实,这使任务更加容易。到目前为止,我还没有找到解决问题的方法。

import pygame
import numpy
import random
import numpy as np
import time
import sys

class ENV:
    def __init__(self):       
        #AGENT
        self.agent = [[10,10], ]
        self.headAgent = self.agent[0]
        self.run = False

        #Move
        self.primMove = "up"
        self.directions = {
            "left" : [-1, 0],
            "right": [1, 0],
            "up": [0, 1],
            "down": [0, -1],
        }

        #Reward
        self.rewardLocationX = random.randint(1,18)
        self.rewardLocationY = random.randint(1,18)

        #BOARD
        self.boardSizeX = 500
        self.boardSizeY = 500

        self.boardDimX = 20
        self.boardDimY = 20

        self.textRep = True
        self.board = [
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
        ]

        #Game
        pygame.init()
        self.screen = pygame.display.set_mode((self.boardSizeX, self.boardSizeY))
        self.clock = pygame.time.Clock()

    def startGame(self):
        if self.run == False:
            self.run = True

        self.colorRed = (255,0,0)
        self.colorGreen = (0,255,0)
        self.colorBlue = (0,0,255)
        self.colorDarkBlue = (0,0,128)
        self.colorWhite = (255,255,255)
        self.colorYellow = (255,204,0)
        self.colorOrange = (255,100,71)
        self.colorBlack = (0,0,0)
        self.colorPink = (255,200,200)
        self.colorBgColor_black = (255,255,255)
        self.colorBgColor_white = (0,0,0)

        self.screen.fill(self.colorWhite)

    def newAction(self, action):
        reward = 0

        print(self.agent)
        if action != None:
            self.primMove = action

        direction = self.directions[self.primMove]
        a = self.agent[0][0] + direction[0]
        b = self.agent[0][1] + direction[1]    

        if action == None:
            if (a, b) == (self.rewardLocationX, self.rewardLocationY):
                self.agent.insert(0, [a,b])
                self.newReward()
                reward = 1
            else:
                self.agent.insert(0, [a,b])
                del self.agent[-1]

            self.check()

            return self.board, [self.rewardLocationX, self.rewardLocationY], self.agent, reward

        direction = self.directions[action]

        if (self.agent[0] + self.directions[action]) not in self.agent:
            if (self.agent[0] + self.directions[action]) == (self.rewardLocationX, self.rewardLocationY):
                self.agent.insert(0, [a,b])
                self.newReward()
                reward = 1
            else:
                print("DIRECTION", self.directions[action])
                self.agent.insert(0, [a,b])
                del self.agent[-1]

        else:
            print("DIRECTION", self.directions[action])
            self.agent.insert(0, [a,b])
            del self.agent[-1]

        if (a, b) == (self.rewardLocationX, self.rewardLocationY):
            self.agent.insert(0, [a,b])
            self.newReward()
            reward = 1 

        a = self.agent[0][0] + direction[0]
        b = self.agent[0][1] + direction[1]
        if [a, b] in self.agent:
            print("TOUCH OWN BODY")
            self.reset()

        print("HEAD : ", self.agent[0])
        print("REWARD : ", [self.rewardLocationX, self.rewardLocationY])

        self.check()

        self.updateBoard()
        self.printBoard()    

        return self.board, [self.rewardLocationX, self.rewardLocationY], self.agent, reward

    def reset(self):
        self.agent = [[10,10], ]
        self.headAgent = self.agent[0]
        self.run = True

        self.board = [
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
            [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
        ]

        self.updateBoard()

    def check(self):
        if self.agent[0][0] == 20 or self.agent[0][0] == -1 or self.agent[0][1] == -1 or self.agent[0][1] == 20:
            self.reset()

    def newReward(self):
        self.rewardLocationX = random.randint(1,18)
        self.rewardLocationY = random.randint(1,18)    

        self.rewardLocationX = random.randint(1,18)
        self.rewardLocationY = random.randint(1,18)

    def updateBoard(self):
        for x in range(20):
            for y in range(20):
                self.board[x][y] = " "

        for x in range(20):
            for y in range(20):
                if [x, y] in self.agent:
                    self.board[x][y] = "S"

        self.board[self.rewardLocationX][self.rewardLocationY] = "R"

    def printBoard(self):
        self.screen.fill(self.colorWhite)
        if self.textRep == True:
            for x in range(len(self.board)):
                print(self.board[x])

        for i in range(len(self.agent)):
            x1 = (self.agent[i][0] * 25) + 3
            y1 = (self.agent[i][1] * 25) + 3
            x2 = x1 + 22
            y2 = y1 + 22
            print("X1 : ", x1, "Y1 : ", y1, "X2 : ", x2, "Y2 : ", y2)
            pygame.draw.rect(self.screen, self.colorBlack, (x1, y1, x2, y2))

        print(self.rewardLocationX, self.rewardLocationY)
        x1 = (self.rewardLocationX * 25) + 3
        y1 = (self.rewardLocationY * 25) + 3
        x2 = x1 + 22
        y2 = y1 + 22
        print("REWARD   X1 : ", x1, "Y1 : ", y1, "X2 : ", x2, "Y2 : ", y2)
        pygame.draw.rect(self.screen, self.colorRed, (x1, y1, x2, y2))

        pygame.display.update()

    def makeBox(self, x, y, color):
        pygame.draw.rect(self.screen, color, ((y + 3), (x + 3), (x+22), (y+22)))

    def runGame(self):
        self.startGame()
        while self.run:
            pygame.display.update()
            self.updateBoard()
            self.printBoard()
            time.sleep(0.3)
            for event in pygame.event.get():
                if event.type == pygame.KEYDOWN:
                    if (event.key == pygame.K_RIGHT):
                        print("RIGHT")
                        board, rewardLocation, agentLocation, reward = self.newAction("up")

                    if (event.key == pygame.K_LEFT):
                        print("LEFT")
                        board, rewardLocation, agentLocation, reward = self.newAction("down")

                    if (event.key == pygame.K_UP):
                        print("UP")
                        board, rewardLocation, agentLocation, reward = self.newAction("left")

                    if (event.key == pygame.K_DOWN):
                        print("DOWN")
                        board, rewardLocation, agentLocation, reward = self.newAction("right")

                    if event.key == pygame.K_ESCAPE:
                        sys.exit()
            print("PRIM")
            board, rewardLocation, agentLocation, reward = self.newAction(None)     

env = ENV()
env.runGame()      

2 个答案:

答案 0 :(得分:1)

在这里我不会回答您的问题,您有太多代码无法合理筛选以查找错误。我要做的是建议您更改此行,这让我很难过:

self.board = [
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
    [" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "],
]

像这样写这行并不是最好的方法,您可以用这样的行更快地完成此操作:

self.board = [[" " for i in range(self.boardDimY)]for j in range(self.boardDimY)]

此行使用称为list comprehension(实际上是两个),它具有许多优点,例如长度,可读性,对self.boardDimXself.boardDimY的依赖...

答案 1 :(得分:1)

问题似乎出在以下类型的行中:

pygame.draw.rect(self.screen, self.colorRed, (x1, y1, x2, y2))

您似乎误解了pygame矩形的基本概念,正如Documentation中所说的那样,它们的参数是:

  

Rect(左侧,顶部,宽度,高度)->矩形

通过将x2 / y2放入最后两个参数,可以使尺寸与位置值成比例,这意味着荒谬的尺寸。

我建议将这些对x2 / y2的引用更改为静态值,例如:

pygame.draw.rect(self.screen, self.colorRed, (x1, y1, 10, 10))

这将创建一个不会根据坐标变化的简单正方形。