avocados/avocado.py

154 lines
4.7 KiB
Python
Raw Normal View History

2014-08-30 17:59:20 +02:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*- #
2014-08-30 17:59:20 +02:00
import pygame, random, os
2014-08-30 20:53:47 +02:00
from support import operations
2014-08-30 17:59:20 +02:00
class Avocado:
def __init__(self, screen, boundaries, properties, target, level, filename=os.path.join('img', 'AvoCado_0_PINK.png')):
2014-08-30 21:20:42 +02:00
# Set up our instance variables
2014-08-30 20:53:47 +02:00
self.screen = screen
2014-08-30 20:47:21 +02:00
self.screen_width, self.screen_height = screen.get_size()
self.avocados = {(255,0,0): os.path.join('img', 'AvoCado_0_RED.png'), \
(0,255,0): os.path.join('img', 'AvoCado_0_GREEN.png'), \
(0,0,255): os.path.join('img', 'AvoCado_0_BLUE.png'), \
(255,255,0): os.path.join('img', 'AvoCado_0_YELLOW.png'), \
(255,192,203): os.path.join('img', 'AvoCado_0_PINK.png')}
2014-08-31 11:52:40 +02:00
self.color = properties['color']
self.w, self.y = properties['size']
self.filename = self.avocados[self.color]
self.target = target
self.boundaries = boundaries
self.checkObstacle = True
2014-08-30 20:53:47 +02:00
2014-08-30 21:20:42 +02:00
# Initialize the image
2014-08-30 18:20:49 +02:00
self.i = pygame.image.load(filename).convert_alpha()
self.i = pygame.image.load(self.filename)
2014-08-30 20:55:40 +02:00
self.image = pygame.transform.scale(self.i, (self.w, self.y))
2014-08-30 21:20:42 +02:00
self.rect = self.image.get_rect()
2014-08-30 18:20:49 +02:00
2014-08-30 21:20:42 +02:00
# Set the avocado's initial position and velocity
2014-08-30 19:01:48 +02:00
self.init_pos()
2014-08-30 23:34:52 +02:00
self.vx = 2
2014-08-31 02:36:21 +02:00
self.vy = 4 * (level * 0.5)
2014-08-30 23:34:52 +02:00
2014-08-31 01:07:12 +02:00
# Avocado state
self.is_still_falling = True
self.has_been_pinned = False
2014-08-30 17:59:20 +02:00
2014-08-31 01:07:12 +02:00
# Avocado sounds
self.click = self.loadClick()
self.clickFail = self.loadFailClick()
2014-08-31 01:07:12 +02:00
2014-08-30 22:48:49 +02:00
2014-08-30 23:03:51 +02:00
def setTargetColor(self, targetColor):
self.target = targetColor
2014-08-30 21:20:42 +02:00
def blitme(self):
self.screen.blit(self.image, self.rect)
2014-08-30 19:01:48 +02:00
def init_pos(self):
2014-08-30 21:20:42 +02:00
self.rect.x = random.randint(0, self.screen_width)
self.rect.y = random.randint(20, 70)
2014-08-30 18:43:19 +02:00
def isHit(self, click):
2014-08-30 17:59:20 +02:00
"""
Checks whether this object collides with the given position
of a mouse-click. Return true is the correct color was hit and
false if it was the wrong one.
Just returns void if no avocado was hit
2014-08-30 17:59:20 +02:00
"""
2014-08-30 21:11:10 +02:00
mousex, mousey = click
2014-08-30 21:24:58 +02:00
if self.rect.left < mousex and self.rect.right > mousex and \
self.rect.top < mousey and self.rect.bottom > mousey:
if self.color == self.target:
self.has_been_pinned = True
2014-08-31 02:24:02 +02:00
self.is_still_falling = False
self.click.play()
return True, self.rect.center
else:
self.clickFail.play()
2014-08-31 10:45:06 +02:00
return False, (0, 0)
return None, (0, 0)
# BUG - isHit is called 4 times upon click which makes it return the
# first time but fail the consecutive times
#else:
# self.clickFail.play()
2014-08-30 21:20:42 +02:00
2014-08-31 02:24:02 +02:00
def isFalling(self):
return self.is_still_falling
2014-08-31 01:07:12 +02:00
def isPinned(self):
return self.has_been_pinned
2014-08-30 21:20:42 +02:00
2014-08-30 17:59:20 +02:00
2014-08-31 10:20:12 +02:00
def checkBoundaries(self):
# Checking screen boundaries
2014-08-31 10:20:12 +02:00
if self.rect.right > self.screen_width or self.rect.left < 0:
self.checkObstacle = True
2014-08-31 10:20:12 +02:00
self.vx = -self.vx
# Checking for obstacle collisions
for obstacle in self.boundaries:
left, top, width, height = obstacle
right = left + width
bottom = top + height
if self.checkObstacle \
and (self.rect.right < right and self.rect.left > left):
self.checkObstacle = False
2014-08-31 12:08:00 +02:00
# print(self.rect.bottom, top)
# if self.rect.bottom > top:
# self.vy = -self.vy
if self.checkObstacle \
and ((self.rect.right > right and self.rect.left < right) \
or (self.rect.left < left and self.rect.right > left)):
2014-08-31 12:08:00 +02:00
if self.rect.bottom > top:
self.vx = -self.vx
2014-08-31 10:20:12 +02:00
2014-08-30 18:20:49 +02:00
def move(self):
2014-08-30 23:34:15 +02:00
if not self.has_been_pinned:
2014-08-31 10:20:12 +02:00
self.checkBoundaries()
2014-08-30 21:20:42 +02:00
2014-08-30 23:34:15 +02:00
if self.hasLanded():
self.destroy()
2014-08-30 21:20:42 +02:00
2014-08-31 13:07:45 +02:00
# Ah, weird…
# self.image = pygame.transform.rotate(self.image, 1)
2014-08-30 23:34:15 +02:00
self.rect.x += self.vx
self.rect.y += self.vy
2014-08-30 18:20:49 +02:00
return True
2014-08-30 21:20:42 +02:00
def hasLanded(self):
2014-08-31 10:29:04 +02:00
if self.rect.top > self.screen_height:
2014-08-31 01:07:12 +02:00
self.is_still_falling = False
2014-08-31 10:29:04 +02:00
print('DEBUG :: splatch!')
2014-08-30 21:20:42 +02:00
return True
2014-08-31 01:07:12 +02:00
def loadClick(self, sound=True):
if not sound:
return
return pygame.mixer.Sound("audio/click.wav")
def loadFailClick(self, sound=True):
if not sound:
return
return pygame.mixer.Sound("audio/poop.wav")
def destroy(self):
del(self)