steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-18 18:52:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Code Comments, Getting Dirty Env, Naming

Browse Source
This commit is contained in:
Steffen Illium
2021-05-11 10:31:35 +02:00
parent f98f689f5e
commit 7d926c403d
5 changed files with 98 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
environments/helpers.py
View File

@ -29,26 +29,28 @@ def check_agent_move(state, dim, action):
x, y = agent_pos[0]
x_new, y_new = x, y
# Actions
if action == 0: # North
if action == 0: # North
x_new -= 1
elif action == 1: # East
elif action == 1: # East
y_new += 1
elif action == 2: # South
elif action == 2: # South
x_new += 1
elif action == 3: # West
elif action == 3: # West
y_new -= 1
elif action == 4: # NE
elif action == 4: # NE
x_new -= 1
y_new += 1
elif action == 5: # SE
elif action == 5: # SE
x_new += 1
y_new += 1
elif action == 6: # SW
elif action == 6: # SW
x_new += 1
y_new -= 1
elif action == 7: # NW
elif action == 7: # NW
x_new -= 1
y_new -= 1
else:
pass
# Check validity
valid = not (
x_new < 0 or y_new < 0
@ -58,10 +60,7 @@ def check_agent_move(state, dim, action):
return (x, y), (x_new, y_new), valid
if __name__ == '__main__':
x = parse_level(Path(__file__).parent / 'factory' / 'levels' / 'simple.txt')
y = one_hot_level(x)
print(np.argwhere(y == 0))
parsed_level = parse_level(Path(__file__).parent / 'factory' / 'levels' / 'simple.txt')
y = one_hot_level(parsed_level)
print(np.argwhere(y == 0))