from vaixells import TaulerVaixells

class TaulerMaxTirades(TaulerVaixells):

    def __init__(self, dim=5, max_tirades=10):
        super().__init__(dim)
        self.max_tirades = max_tirades

    def dispara(self, posicio):
        if self.num_tirades < self.max_tirades:
            super().dispara(posicio)
    
    def caselles(self):
        nv, nt, na = 0, 0, 0
        for x in range(self.dimensio()):
            for y in range(self.dimensio()):
                k = self[x, y]
                if k == '-':
                    na = na + 1
                elif k == 'V':
                    nv = nv + 1
                else:
                    nt = nt + 1
        return nv, nt, na