avocados/avocado.py

154 lines
4.7 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*- #
import pygame, random, os
from support import operations
class Avocado:
def __init__(self, screen, boundaries, properties, target, level, filename=os.path.join('img', 'AvoCado_0_PINK.png')):
# Set up our instance variables
self.screen = screen
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')}
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
# Initialize the image
self.i = pygame.image.load(filename).convert_alpha()
self.i = pygame.image.load(self.filename)
self.image = pygame.transform.scale(self.i, (self.w, self.y))
self.rect = self.image.get_rect()
# Set the avocado's initial position and velocity
self.init_pos()
self.vx = 2
self.vy = 4 * (level * 0.5)
# Avocado state
self.is_still_falling = True
self.has_been_pinned = False
# Avocado sounds
self.click = self.loadClick()
self.clickFail = self.loadFailClick()
def setTargetColor(self, targetColor):
self.target = targetColor
def blitme(self):
self.screen.blit(self.image, self.rect)
def init_pos(self):
self.rect.x = random.randint(0, self.screen_width)
self.rect.y = random.randint(20, 70)
def isHit(self, click):
"""
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
"""
mousex, mousey = click
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
self.is_still_falling = False
self.click.play()
return True, self.rect.center
else:
self.clickFail.play()
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()
def isFalling(self):
return self.is_still_falling
def isPinned(self):
return self.has_been_pinned
def checkBoundaries(self):
# Checking screen boundaries
if self.rect.right > self.screen_width or self.rect.left < 0:
self.checkObstacle = True
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
# 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)):
if self.rect.bottom > top:
self.vx = -self.vx
def move(self):
if not self.has_been_pinned:
self.checkBoundaries()
if self.hasLanded():
self.destroy()
# Ah, weird…
# self.image = pygame.transform.rotate(self.image, 1)
self.rect.x += self.vx
self.rect.y += self.vy
return True
def hasLanded(self):
if self.rect.top > self.screen_height:
self.is_still_falling = False
print('DEBUG :: splatch!')
return True
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)