import pygame from machine import StateMachine, State class Example(State): def __init__(self): # This will save the this instance to state machine State.__init__(self) # All draw code goes here def draw(self, surface): surface.fill((0, 40, 0)) # Process all event here def event(self, event): if event.type == pygame.QUIT: # call state machine self.state.machine.quit() if __name__ == '__main__': pygame.init() StateMachine.screen_center() # Create the state machine StateMachine('Example', 800, 600) # Create Example and set it to first state StateMachine.main_loop(Example()) pygame.quit()
def main(): pygame.init() StateMachine.screen_center() StateMachine("Test", 800, 600) Test2() StateMachine.main_loop(Test())