import pygame, sys, state
from pygame import *
pygame.init()
displaySurface = pygame.display.set_mode((800, 600))
states = {
"MainMenu" : state.MenuState(),
"Action" : state.ActionState() }
curState = state.MenuState()
while True:
for e in pygame.event.get():
if e.type == QUIT:
sys.exit()
else:
curState.event(e)
displaySurface.fill((0,0,0))
curState.paint()
pygame.display.flip()
from pygame import *
class State:
def event(self, event):
pass
def paint(self):
pass
class MenuState(State):
def event(self, event):
if event.type == KEYDOWN and event.key == K_RETURN:
main.curState = main.states["Action"]
def paint(self):
# draw stuff onto the screen (GUI, demo background, etc.)
main.displaySurface.blit(...)
class ActionState(State):
def event(self, event):
if event.type == KEYDOWN and event.key == K_RETURN:
main.curState = main.states["MainMenu"]
def paint(self):
main.displaySurface.blit(...)