Scribd
Upload
13 views

Class Agent

class

Uploaded by

2021csm.r350
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
13 views

Class Agent

class

Uploaded by

2021csm.r350
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
You are on page 1/ 4

class Agent:

def __init__(self):

self._wumpusWorld = [

['','','',''], # Rooms [1,1] to [4,1]

['','W','P',''], # Rooms [1,2] to [4,2]

['','','',''], # Rooms [1,3] to [4,3]

['','','',''], # Rooms [1,4] to [4,4]

] # This is the wumpus world shown in the assignment question.

# A different instance of the wumpus world will be used for evaluation.

self._curLoc = [1,1]

self._isAlive = True

self._hasExited = False

def _FindIndicesForLocation(self,loc):

x,y = loc

i,j = y-1, x-1

return i,j

def _CheckForPitWumpus(self):

ww = self._wumpusWorld

i,j = self._FindIndicesForLocation(self._curLoc)

if 'P' in ww[i][j] or 'W' in ww[i][j]:

print(ww[i][j])

self._isAlive = False

print('Agent is DEAD.')

return self._isAlive

def TakeAction(self,action): # The function takes an action and returns whether the Agent is alive

# after taking the action.

validActions = ['Up','Down','Left','Right']

assert action in validActions, 'Invalid Action.'


if self._isAlive == False:

print('Action cannot be performed. Agent is DEAD. Location:{0}'.format(self._curLoc))

return False

if self._hasExited == True:

print('Action cannot be performed. Agent has exited the Wumpus world.'.format(self._curLoc))

return False

index = validActions.index(action)

validMoves = [[0,1],[0,-1],[-1,0],[1,0]]

move = validMoves[index]

newLoc = []

for v, inc in zip(self._curLoc,move):

z = v + inc #increment location index

z = 4 if z>4 else 1 if z<1 else z #Ensure that index is between 1 and 4

newLoc.append(z)

self._curLoc = newLoc

print('Action Taken: {0}, Current Location {1}'.format(action,self._curLoc))

if self._curLoc[0]==4 and self._curLoc[1]==4:

self._hasExited=True

return self._CheckForPitWumpus()

def _FindAdjacentRooms(self):

cLoc = self._curLoc

validMoves = [[0,1],[0,-1],[-1,0],[1,0]]

adjRooms = []

for vM in validMoves:

room = []

valid = True

for v, inc in zip(cLoc,vM):

z = v + inc

if z<1 or z>4:
valid = False

break

else:

room.append(z)

if valid==True:

adjRooms.append(room)

return adjRooms

def PerceiveCurrentLocation(self): #This function perceives the current location.

breeze, stench = False, False

ww = self._wumpusWorld

if self._isAlive == False:

print('Agent cannot perceive. Agent is DEAD. Location:{0}'.format(self._curLoc))

return [None,None]

if self._hasExited == True:

print('Agent cannot perceive. Agent has exited the Wumpus World.'.format(self._curLoc))

return [None,None]

adjRooms = self._FindAdjacentRooms()

for room in adjRooms:

i,j = self._FindIndicesForLocation(room)

if 'P' in ww[i][j]:

breeze = True

if 'W' in ww[i][j]:

stench = True

return [breeze,stench]

def FindCurrentLocation(self):

return self._curLoc
def main():

ag = Agent()

print('curLoc',ag.FindCurrentLocation())

print('Percept [breeze, stench] :',ag.PerceiveCurrentLocation())

ag.TakeAction('Right')

print('Percept',ag.PerceiveCurrentLocation())

ag.TakeAction('Right')

print('Percept',ag.PerceiveCurrentLocation())

ag.TakeAction('Right')

print('Percept',ag.PerceiveCurrentLocation())

ag.TakeAction('Up')

print('Percept',ag.PerceiveCurrentLocation())

ag.TakeAction('Up')

print('Percept',ag.PerceiveCurrentLocation())

ag.TakeAction('Up')

print('Percept',ag.PerceiveCurrentLocation())

if __name__=='__main__':

main()

You might also like