20 lines
714 B
Python
20 lines
714 B
Python
from DataModels.Cell import Cell
|
|
from VowpalWabbit.vowpal_utils import MAP_CONTENT
|
|
|
|
class House(Cell):
|
|
def __init__(self, x, y, max_rubbish, yellow=0, green=0, blue=0):
|
|
Cell.__init__(self, x, y, max_rubbish, yellow, green, blue)
|
|
self.unvisited = True
|
|
|
|
def return_trash(self, collector):
|
|
self.container.yellow, self.container.green, self.container.blue = collector.container.add(
|
|
[self.container.yellow, self.container.green, self.container.blue])
|
|
self.update_image()
|
|
|
|
def Visit(self):
|
|
self.unvisited = not self.unvisited
|
|
if self.unvisited:
|
|
MAP_CONTENT[self.y][self.x] = "H"
|
|
else:
|
|
MAP_CONTENT[self.y][self.x] = "V"
|