diff --git a/modules/mapa.py b/modules/mapa.py index a67f213..ef384b0 100644 --- a/modules/mapa.py +++ b/modules/mapa.py @@ -1,5 +1,6 @@ # funções associadas a operações com o Mapa +from modules import agente from modules.agente import Agente @@ -28,3 +29,11 @@ class Mapa: for n in range(0, self.dimensao[1]): print(f'{self.mundo[m][n].pontuacao:2} ', end='') print('') + + def posicao(self, posicao: tuple[int, int]) -> Agente | None: + # verificar limites + if posicao[0] > self.dimensao[0] or posicao[0] < 0: + return None + if posicao[1] > self.dimensao[1] or posicao[1] < 0: + return None + return self.mundo[posicao[0]][posicao[1]]