139 lines
4.1 KiB
Python
139 lines
4.1 KiB
Python
import pygame
|
|
import PySimpleGUI as sg
|
|
import os
|
|
|
|
"""
|
|
Demo - Simple Snake Game using PyGame and PySimpleGUI
|
|
This demo may not be fully functional in terms of getting the coordinate
|
|
systems right or other problems due to a lack of understanding of PyGame
|
|
The purpose of the demo is to show one way of adding a PyGame window into your PySimpleGUI window
|
|
Note, you must click on the game area in order for PyGame to get keyboard strokes, etc.
|
|
Tried using set_focus to switch to the PyGame canvas but still needed to click on game area
|
|
"""
|
|
|
|
# --- Globals ---
|
|
# Colors
|
|
BLACK = (0, 0, 0)
|
|
WHITE = (255, 255, 255)
|
|
|
|
# Set the width and height of each snake segment
|
|
segment_width = 15
|
|
segment_height = 15
|
|
# Margin between each segment
|
|
segment_margin = 3
|
|
|
|
# Set initial speed
|
|
x_change = segment_width + segment_margin
|
|
y_change = 0
|
|
|
|
|
|
class Segment(pygame.sprite.Sprite):
|
|
""" Class to represent one segment of the snake. """
|
|
# -- Methods
|
|
# Constructor function
|
|
def __init__(self, x, y):
|
|
# Call the parent's constructor
|
|
super().__init__()
|
|
|
|
# Set height, width
|
|
self.image = pygame.Surface([segment_width, segment_height])
|
|
self.image.fill(WHITE)
|
|
|
|
# Make our top-left corner the passed-in location.
|
|
self.rect = self.image.get_rect()
|
|
self.rect.x = x
|
|
self.rect.y = y
|
|
|
|
# --------------------------- GUI Setup & Create Window -------------------------------
|
|
|
|
layout = [[sg.T('Snake Game - PySimpleGUI + PyGame')],
|
|
[sg.Graph((800,600), (0,0), (800,600), background_color='lightblue', key='_GRAPH_')],
|
|
[sg.Exit()]]
|
|
|
|
window = sg.Window('Snake Game using PySimpleGUI and PyGame', layout).Finalize()
|
|
|
|
# ------------------------ Do the magic that integrates PyGame and Graph Element ------------------
|
|
graph = window.Element('_GRAPH_') # type: sg.Graph
|
|
embed = graph.TKCanvas
|
|
os.environ['SDL_WINDOWID'] = str(embed.winfo_id())
|
|
os.environ['SDL_VIDEODRIVER'] = 'windib'
|
|
|
|
# ----------------------------- PyGame Code -----------------------------
|
|
# Call this function so the Pygame library can initialize itself
|
|
# pygame.init()
|
|
screen = pygame.display.set_mode((800,600))
|
|
screen.fill(pygame.Color(255,255,255))
|
|
|
|
pygame.display.init()
|
|
pygame.display.update()
|
|
|
|
# Set the title of the window
|
|
pygame.display.set_caption('Snake Example')
|
|
|
|
allspriteslist = pygame.sprite.Group()
|
|
|
|
# Create an initial snake
|
|
snake_segments = []
|
|
for i in range(15):
|
|
x = 250 - (segment_width + segment_margin) * i
|
|
y = 30
|
|
segment = Segment(x, y)
|
|
snake_segments.append(segment)
|
|
allspriteslist.add(segment)
|
|
|
|
clock = pygame.time.Clock()
|
|
|
|
while True:
|
|
event, values = window.Read(timeout=10)
|
|
if event in (None, 'Exit'):
|
|
break
|
|
pygame.display.update()
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
break
|
|
# Set the speed based on the key pressed
|
|
# We want the speed to be enough that we move a full
|
|
# segment, plus the margin.
|
|
if event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_LEFT:
|
|
x_change = (segment_width + segment_margin) * -1
|
|
y_change = 0
|
|
if event.key == pygame.K_RIGHT:
|
|
x_change = (segment_width + segment_margin)
|
|
y_change = 0
|
|
if event.key == pygame.K_UP:
|
|
x_change = 0
|
|
y_change = (segment_height + segment_margin) * -1
|
|
if event.key == pygame.K_DOWN:
|
|
x_change = 0
|
|
y_change = (segment_height + segment_margin)
|
|
|
|
# Get rid of last segment of the snake
|
|
# .pop() command removes last item in list
|
|
old_segment = snake_segments.pop()
|
|
allspriteslist.remove(old_segment)
|
|
|
|
# Figure out where new segment will be
|
|
x = snake_segments[0].rect.x + x_change
|
|
y = snake_segments[0].rect.y + y_change
|
|
segment = Segment(x, y)
|
|
|
|
# Insert new segment into the list
|
|
snake_segments.insert(0, segment)
|
|
allspriteslist.add(segment)
|
|
|
|
# -- Draw everything
|
|
# Clear screen
|
|
screen.fill(BLACK)
|
|
|
|
allspriteslist.draw(screen)
|
|
|
|
# Flip screen
|
|
pygame.display.flip()
|
|
|
|
# Pause
|
|
clock.tick(5)
|
|
|
|
window.Close()
|