import pygame
from DataModels.Cell import Cell
from VowpalWabbit.vowpal_utils import MAP_CONTENT

class Dump( Cell ):
    def __init__( self, x, y, max_rubbish, dump_type, yellow = 0, green = 0, blue = 0 ):
        Cell.__init__( self, x, y, max_rubbish, yellow, green, blue, dump_type )
        self.dump_type = dump_type
        self.unvisited = True

    def return_trash(self, collector):
        dump_type = self.dump_type.lower()[5:]
        self.container.yellow, self.container.green, self.container.blue = collector.container.empty(dump_type,
            [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] = self.dump_type[5]
        else:
            MAP_CONTENT[self.y][self.x] = "V"