class GameScreen():
def __init__(self):
self.snake = Snake()
self.controls = Controls()
self.textFont = pygame.font.Font(config.FONT, 13)
self.game_over = False
self.textOffset = (3, 3)
self.scoreText = Text(
self.textFont,
"Your Score: {:}".format(self.snake.snake_length - 1),
config.BLACK,
tX=self.textOffset[0],
tY=self.textOffset[0])
self.spawnTime = 50 # In frames
self.foodSmall = Food(
(config.DIS_WIDTH,
config.DIS_HEIGHT - self.scoreText.tH - self.textOffset[1]), 2)
self.foodBig = Food(
(config.DIS_WIDTH,
config.DIS_HEIGHT - self.scoreText.tH - self.textOffset[1]),
3,
self.spawnTime,
0.5,
config.GREEN,
blink_rate=2)
self.foodBig.turn_idle()
self.cursorPos = 0
self.eatSound = pygame.mixer.Sound("sounds/eat.wav")
self.eatBigSound = pygame.mixer.Sound("sounds/eatBig_2.wav")
self.selectSound = pygame.mixer.Sound("sounds/select.wav")
self.nextScene = self
self.screen_shake = 0
self.menuScene = None
self.pauseMenu = Menu(self.textFont, ('P A U S E', config.WHITE),
('Continue', config.WHITE),
('Exit', config.WHITE))
self.gameOverMenu = Menu(self.textFont,
('G A M E O V E R', config.RED),
('Retry', config.WHITE),
('Exit', config.WHITE))
def restart(self):
self.snake.snake_length = 1
self.snake.snake_list = []
self.snake.x1 = round(config.DIS_WIDTH / 2)
self.snake.y1 = round(config.DIS_HEIGHT / 2)
self.controls.stop()
self.screen_shake = 0
self.foodSmall.spawn()
self.foodBig.spawn()
def update_score(self, display):
self.scoreText.text = "Score: {:}".format(
(self.snake.snake_length - 1) * 10)
self.scoreText.reRender()
self.scoreText.update(display)
def _handle_menu_controls(self):
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.nextScene = None
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_RETURN:
if self.cursorPos == 0:
if self.controls.pause:
self.controls.pause = False
elif self.game_over:
self.game_over = False
self.restart()
elif self.cursorPos == 1:
self.cursorPos = 0
self.controls.pause = False
self.game_over = False
self.nextScene = self.menuScene
if event.key == pygame.K_DOWN:
if not self.cursorPos >= 1:
self.selectSound.play()
self.cursorPos += 1
if event.key == pygame.K_UP:
if not self.cursorPos <= 0:
self.selectSound.play()
self.cursorPos -= 1
if event.key == pygame.K_ESCAPE:
self.controls.pause = False
def render(self):
self.nextScene = self
if self.controls.pause or self.game_over:
self._handle_menu_controls()
else:
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.nextScene = None
self.controls.handle_input()
self.snake.move(self.controls.x1_change, self.controls.y1_change,
self.scoreText.tH + self.textOffset[1])
self.snake.render()
if self.snake.eats_itself():
self.game_over = True
if self.foodSmall.is_eaten(self.snake.x1, self.snake.y1):
self.eatSound.play()
self.foodSmall.spawn()
if self.foodBig.isIdle:
self.foodBig.spawn()
self.snake.snake_length += 2
if self.foodBig.is_eaten(self.snake.x1, self.snake.y1):
self.eatBigSound.play()
self.foodBig.turn_idle()
self.snake.snake_length += 6
self.screen_shake = 13
def update(self, display):
offset = [0, 0]
if self.screen_shake:
offset[0] = randint(0, 14) - 7
offset[1] = randint(0, 14) - 7
self.screen_shake -= 1
# Background
display.fill(config.BLACK)
# Snake
self.snake.update(display, offset)
# Score
pygame.draw.rect(
display, config.WHITE,
[0, 0, config.DIS_WIDTH, self.scoreText.tH + self.textOffset[1]])
self.update_score(display)
# Food
self.foodSmall.update(display, offset)
self.foodBig.update(display, offset)
if self.controls.pause:
self.pauseMenu.update(display, self.cursorPos)
elif self.game_over:
self.gameOverMenu.update(display, self.cursorPos)
