import memory

class MemoryPlus(memory.Memory):

    def __init__(self, nfils, ncols):
        super().__init__(nfils, ncols)
        self.npunts1 = 0
        self.npunts2 = 0

    def capgira(self, p, q, jug):
        n = 0
        if self[p] == self[q]:
            self[p] = '*'
            self[q] = '*'
            n = 2
        elif self[p] == '@':
            self[p] = '*'
            n = 1
        elif self[q] == '@':
            self[q] = '*'
            n = 1
        if jug == 1:
            self.npunts1 = self.npunts1 + n
        elif jug == 2:
            self.npunts2 = self.npunts2 + n