import pygame

class Jatekos(pygame.sprite.Sprite):
	def __init__(self, *groups):
		super(Jatekos, self).__init__(*groups)		
		self.image = pygame.image.load('jatekos.png')
		self.rect = pygame.rect.Rect((320, 240), self.image.get_size())
		self.resting = False
		self.dy = 0
	
	def update(self, dt, game):
		last = self.rect.copy()
		
		key = pygame.key.get_pressed()
		if key[pygame.K_LEFT]:
			self.rect.x -= 300 * dt
		if key[pygame.K_RIGHT]:
			self.rect.x += 300 * dt
		# ~ if key[pygame.K_UP]:
			# ~ self.rect.y -= 300 * dt
		# ~ if key[pygame.K_DOWN]:
			# ~ self.rect.y += 300 * dt

		if self.resting and key[pygame.K_SPACE]:
			self.dy = -500
		self.dy = min(400, self.dy + 40)
		
		self.rect.y += self.dy * dt

		new = self.rect
		self.resting = False
		for cell in pygame.sprite.spritecollide(self, game.walls, False):
			# ~ self.rect = last
			cell = cell.rect
			if last.right <= cell.left and new.right > cell.left:
				new.right = cell.left
			if last.left >= cell.right and new.left < cell.right:
				new.left = cell.right
			if last.bottom <= cell.top and new.bottom > cell.top:
				self.resting = True
				new.bottom = cell.top
				self.dy = 0
			if last.top >= cell.bottom and new.top < cell.bottom:
				new.top = cell.bottom
				self.dy = 0

class Jatek(object):
	def main(self, kepernyo):
		clock = pygame.time.Clock()
		
		background = pygame.image.load('hatter.png')
		sprites = pygame.sprite.Group()
		self.player = Jatekos(sprites)
		
		
		self.walls = pygame.sprite.Group()
		szikla = pygame.image.load('szikla.png')
		for x in range(0, 640, 32):
			for y in range(0, 480, 32):
				if x in (0, 640-32) or y in (0, 480-32):
					wall = pygame.sprite.Sprite(self.walls)
					wall.image = szikla
					wall.rect = pygame.rect.Rect((x, y), wall.image.get_size())
		sprites.add(self.walls)
		
		while True:
			dt = clock.tick(30)
			
			for event in pygame.event.get():
				if event.type == pygame.QUIT:
					return
				if event.type == pygame.KEYDOWN and \
						event.key == pygame.K_ESCAPE:
					return
			sprites.update(dt / 1000., self)
			kepernyo.blit(background, (0, 0))
			sprites.draw(kepernyo)
			pygame.display.flip()

if __name__ == '__main__':
	pygame.init()
	kepernyo = pygame.display.set_mode((640, 480))
	Jatek().main(kepernyo)