2019-03-20 11:20:10 +01:00
|
|
|
import pygame
|
2019-03-25 16:00:01 +01:00
|
|
|
from sprites.cell import Cell, CELL_SIZE
|
|
|
|
from config import PLAY_HEIGHT, PLAY_WIDTH
|
2019-03-21 01:25:19 +01:00
|
|
|
|
2019-03-20 11:20:10 +01:00
|
|
|
class Garbage_collector(Cell):
|
2019-03-21 01:25:19 +01:00
|
|
|
def __init__(self, x, y):
|
2019-03-25 16:03:25 +01:00
|
|
|
GC_CAPACITY = 10
|
|
|
|
|
2019-03-21 01:25:19 +01:00
|
|
|
Cell.__init__(self, x, y)
|
|
|
|
self.image = pygame.image.load("images/garbage_collector.png")
|
|
|
|
self.move_options = {
|
2019-03-25 16:00:01 +01:00
|
|
|
"up": lambda forbidden: ('y', self.y - 1) if (self.x, self.y - 1) not in forbidden and self.y - 1 >= 0 else ('y', self.y),
|
|
|
|
"down": lambda forbidden: ('y', self.y + 1) if (self.x, self.y + 1) not in forbidden and self.y + 1 < PLAY_HEIGHT // CELL_SIZE else ('y', self.y),
|
|
|
|
"left": lambda forbidden: ('x', self.x - 1) if (self.x - 1, self.y) not in forbidden and self.x - 1 >= 0 else ('x', self.x),
|
|
|
|
"right": lambda forbidden: ('x', self.x + 1) if (self.x + 1, self.y) not in forbidden and self.x + 1 < PLAY_WIDTH // CELL_SIZE else ('x', self.x)
|
2019-03-21 01:25:19 +01:00
|
|
|
}
|
2019-03-25 16:03:25 +01:00
|
|
|
self.trash_space_taken = {
|
|
|
|
"plastic": 0,
|
|
|
|
"glass": 0,
|
|
|
|
"metal": 0
|
|
|
|
}
|
|
|
|
self.trash_collected = 0
|
2019-03-20 11:20:10 +01:00
|
|
|
|
2019-03-21 01:25:19 +01:00
|
|
|
def move(self, direction, forbidden):
|
|
|
|
(destination, value) = self.move_options[direction](forbidden)
|
|
|
|
if(destination is 'x'):
|
|
|
|
self.x = value
|
|
|
|
elif(destination is 'y'):
|
|
|
|
self.y = value
|
|
|
|
self.update()
|
2019-03-25 16:03:25 +01:00
|
|
|
|
|
|
|
def get_collect_data(self):
|
|
|
|
return self.trash_collected
|
|
|
|
|
|
|
|
def get_space_data(self):
|
|
|
|
return self.trash_space_taken
|