game_state.py

from abc import ABCMeta, abstractmethod
import os
import pygame
from pygame.locals import *
import sys
from projectile import Sushi, Sardine, Dory
from whale import Whale
from Enemies.octopus import Octopus
from Enemies.jelly import Jelly
from Enemies.hermit import Hermit
from Enemies.crab import Crab
from Enemies.sunfish import Sunfish

#STATECHANGE is the first userevent of 9
STATECHANGE = USEREVENT+0

class GameState:

	def __init__(self):
		pygame.init()
		self.size = self.width, self.height = 750, 500
		self.screen = pygame.display.set_mode(self.size)
		self.doors = []
		self.cleared = False

	def handle_event(self, event):
		if event.type == QUIT:
			sys.exit()



class StartState(GameState): 
	
	def __init__(self):
		super().__init__()
		self.whale = Whale()			
		startscreen_path = os.path.join(os.path.realpath(''), 'Images', 'startscreen.png')
		self.start_screen = pygame.image.load(startscreen_path).convert()
			
		button_path = os.path.join(os.path.realpath(''), 'Images', 'playbutton.png')
		self.play_button = pygame.image.load(button_path).convert()
				
		pressed_path = os.path.join(os.path.realpath(''), 'Images', 'pressedbutton.png')
		self.pressed_button = pygame.image.load(pressed_path).convert()
	
		self.button_rect = self.play_button.get_rect(center=(self.width/2, 3*self.height/4))
		
		self.button = self.play_button
	
	def update(self):
		self.screen.blit(self.start_screen, (0,0))
		mouse_pos = pygame.mouse.get_pos()
		
		if self.button_rect.collidepoint(mouse_pos):
			self.button = self.pressed_button
		else:
			self.button = self.play_button
		
		self.screen.blit(self.button, self.button_rect.topleft)
		pygame.display.update()

	def handle_event(self, event):
		super().handle_event(event)
		if event.type == MOUSEBUTTONDOWN:
			if self.button_rect.collidepoint(event.pos):
				startgame_event = pygame.event.Event(STATECHANGE, event_id="won", new_state="Room_1")
				pygame.event.post(startgame_event)
class Room(GameState):	
	
	def __init__(self, whale=None, enemies=None):
		super().__init__()
		self.background = 210, 210, 210 
		
		heart_path = os.path.join(os.path.realpath(''), 'Images', 'heart.gif')
		self.heart = pygame.image.load(heart_path).convert()
	
		halfheart_path = os.path.join(os.path.realpath(''), 'Images', 'halfheart.gif')
		self.halfheart = pygame.image.load(halfheart_path).convert()
		self.healthbar = pygame.sprite.Group()
		
		if whale == None:
			self.whale = Whale()
		else:
			self.whale = whale
		
		if enemies == None:
			enemies = [Octopus()]
		
		self.set_enemies(enemies)
		self.whale_is_dead = False
		
		self.player_projectiles = pygame.sprite.Group()
		self.enemy_projectiles = pygame.sprite.Group()
		
		self.sprites = pygame.sprite.Group(self.whale, self.enemy_sprites, 
						   self.player_projectiles, self.enemy_projectiles, self.healthbar)
		self.update_healthbar()
		self.next_state = "LevelOne"
	
	def set_enemies(self, enemies):
		self.enemy_sprites = pygame.sprite.Group(enemies)
	
	def calc_dmg(self):
		player_projectiles = self.player_projectiles
		enemies = self.enemy_sprites
			
		for enemy in enemies:
			for projectile in player_projectiles:
				if projectile.hitbox.colliderect(enemy.hitbox):
					enemy.health -= projectile.damage
					if enemy.health <= 0:
						enemy.kill()
					projectile.kill()
		
		enemy_hitboxes = [enemy.hitbox for enemy in self.enemy_sprites]
		whale_collide = self.whale.hitbox.collidelist(enemy_hitboxes)
		
		if whale_collide != -1:
			if not self.whale.invincible:
				self.whale.invincible = 50
				self.whale.health -= 1
			if self.whale.health <= 0:
				self.whale.kill()
				self.whale_is_dead = True

	def update_healthbar(self):
		pos = [0, 0]
		for i in range(int(self.whale.health/2)):
			self.screen.blit(self.heart.copy(), pos)
			pos[0] += 50
		if self.whale.health%2 == 1:
			self.screen.blit(self.halfheart, pos)

	def handle_event(self, event):
			super().handle_event(event)

			if event.type == KEYDOWN:
				if event.key == K_UP or event.key == K_w:
					self.whale.state = 'moveup'
				if event.key == K_DOWN or event.key == K_s:
					self.whale.state = 'movedown'
				if event.key == K_LEFT or event.key == K_a:
					self.whale.state = 'moveleft'
				if event.key == K_RIGHT or event.key == K_d:
					self.whale.state = 'moveright'

			elif event.type == KEYUP:
				if (event.key == K_UP or event.key == K_w) and self.whale.state == 'moveup':
					self.whale.state = 'still'
				if (event.key == K_DOWN or event.key == K_s) and self.whale.state == 'movedown':
					self.whale.state = 'still'
				if (event.key == K_LEFT or event.key == K_a) and self.whale.state == 'moveleft':
					self.whale.state = 'still'
				if (event.key == K_RIGHT or event.key == K_d) and self.whale.state == 'moveright':
					self.whale.state = 'still'

			elif event.type == MOUSEBUTTONDOWN:
				bullet_sprite = self.whale.shoot()
				self.player_projectiles.add(bullet_sprite)
				self.sprites.add(bullet_sprite)
				
	def is_cleared(self):
		if self.enemy_sprites.sprites() == []:
			self.cleared = True

	def update(self):
		
		self.screen.fill(self.background)
		self.sprites.update()
		self.sprites.draw(self.screen)
		self.calc_dmg()
		self.update_healthbar()
		self.is_cleared()
		
		#Debugging code for rects
		"""pygame.draw.rect(self.screen, (125, 65, 190), self.whale.hitbox, 1)
		
		for projectile in self.player_projectiles:
			pygame.draw.rect(self.screen, (125, 65, 190), projectile.hitbox, 1)	
		
		for sprite in self.enemy_sprites.sprites():
			pygame.draw.rect(self.screen, (125, 65, 190), sprite.hitbox, 1)	
		"""
		
		if self.whale_is_dead: 
			playerlose_event = pygame.event.Event(STATECHANGE, event_id="lose", new_state="Lose")
			pygame.event.post(playerlose_event)
	
		elif self.cleared: 
			#print("Cleared")
			playerwon_event = pygame.event.Event(STATECHANGE, event_id="won", new_state=self.next_state)
			pygame.event.post(playerwon_event)
		
		
		pygame.display.update()



class Win(GameState): 
	
	def __init__(self):
		super().__init__()
		winscreen_path = os.path.join(os.path.realpath(''), 'Images', 'winscreen.gif')
		self.win_screen = pygame.image.load(winscreen_path).convert()
		self.invincible = 0

	def update(self):
		self.screen.blit(self.win_screen, (0,0))
		if self.invincible <= 100:
			self.invincible += 1
		pygame.display.update()


	def handle_event(self, event):
		super().handle_event(event)
		if event.type == MOUSEBUTTONDOWN and self.invincible > 100:
			win_event = pygame.event.Event(STATECHANGE, event_id="start_over", new_state="StartState")
			pygame.event.post(win_event)

class Lose(GameState): 
	
	def __init__(self):
		super().__init__()
		losescreen_path = os.path.join(os.path.realpath(''), 'Images', 'losescreen.gif')
		self.lose_screen = pygame.image.load(losescreen_path).convert()
		self.invincible = 0	
	
	def update(self):
		self.screen.blit(self.lose_screen, (0,0))
		if self.invincible <= 100:
			self.invincible += 1
		pygame.display.update()

	def handle_event(self, event):
		super().handle_event(event)
		if event.type == MOUSEBUTTONDOWN and self.invincible > 100:
			lose_event = pygame.event.Event(STATECHANGE, event_id="start_over", new_state="StartState")
			pygame.event.post(lose_event)