State.py

import pygame
import Colors
import Images
from Entities import *
import PyPong

ACTIVE = "game"

class GameState(object):

	def init(self, screen):
		self.screen = screen
		self.__paddle1 = Paddle(self.screen, (50,50), True)
		self.__paddle2 = Paddle(self.screen, (PyPong.WIDTH - 50, 50), False)
		self.__table = Table(self.screen, (200, 350))
		self.__net = Net(self.screen, (499, 300))
		self.__ball = Ball(self.screen, (200, 100))
		self.__entities = [self.__paddle1, self.__paddle2, self.__table, self.__net, self.__ball]
		
	def tick(self):
		self.__ball.tick()
		self.__table.tick()
		self.__net.tick()
		self.__paddle1.tick()
		self.__paddle2.tick()
		check_impacts(self.__entities)
		
	def render(self):
		self.screen.fill(Colors.WHITE)
		self.__table.render()
		self.__net.render()
		self.__ball.render()
		self.__paddle1.render()
		self.__paddle2.render()
		

class MenuState(object):

	def init(self, screen):
		self.screen = screen
		
	def tick(self):
		pass
		
	def render(self):
		pass
		
		

game_state = GameState()
menu_state = MenuState()

def active_state():
	if ACTIVE == "menu":
		return menu_state
	else:
		return game_state