SZI2019SmieciarzWmi/DataModels/House.py

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"