Spaces:
Runtime error
Runtime error
File size: 5,461 Bytes
1aa4792 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
# [Translator classes and functions for Atari Boxing environment]
class BasicLevelTranslator:
def __init__(self,):
pass
def translate(self, state):
player_x, player_y, enemy_x, enemy_y, enemy_score, clock, player_score = state
return f"The player is at position ({player_x, player_y}, your opponent is at position ({enemy_x, enemy_y}) ), " \
f"your oppoent's score is {enemy_score}, your score is {player_score}. Move left and right will change the player_x while move up and down will change the player_y"
class GameDescriber:
def __init__(self, args):
self.is_only_local_obs = args.is_only_local_obs == 1
self.max_episode_len = args.max_episode_len
self.action_desc_dict = {
}
self.reward_desc_dict = {
}
def describe_goal(self):
return "The goal is to knock out your opponent."
def translate_terminate_state(self, state, episode_len, max_episode_len):
return ""
def translate_potential_next_state(self, state, action):
return ""
def describe_game(self):
return "In the Boxing game, you fight an opponent in a boxing ring. You score points for hitting the opponent. " \
"If you score 100 points, your opponent is knocked out.Scoring Points: When you get near enough to your opponent to throw a punch, " \
"press the red button. Each punch moves your opponent slightly back and away from the punch." \
" If you move him to the ropes, he can't easily duck the next punch, " \
"and you can set up a real scoring barrage. But don't get caughton the ropes yourself! " \
"Watch your distance. If you move in too close, the computer gets tougher;" \
" but if you're too far away, you can't land scoring punches. "
def describe_action(self):
return "Your Next Move: \n Please choose an action. Type '1' for NOOP (no operation), '2' to hit your opponent, " \
"'3' to move up, '4' to move right, '5' to move left, '6' to move down, '7' to move up-right, " \
"'8' to move up-left, '9' to move down-right, '10' to move down-left, '11' to hit your opponent and move up, " \
"'12' to hit your opponent and move right, '13' to hit your opponent and move left, '14' to hit your opponent and move down, " \
"'15' to hit your opponent and move up-right, '16' to hit your opponent and move up-left, '17' to hit your opponent and move down-right, " \
"or '18' to hit your opponent and move down-left. Ensure you only provide the action number " \
"from the valid action list, i.e., [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]."
class BasicStateSequenceTranslator(BasicLevelTranslator):
def translate(self, infos, is_current=False):
descriptions = []
if is_current:
state_desc = BasicLevelTranslator().translate(infos[-1]['state'])
return state_desc
for i, info in enumerate(infos):
assert 'state' in info, "info should contain state information"
state_desc = BasicLevelTranslator().translate(info['state'])
if info['action'] == 1:
action_desc = f"Take Action: 'Do nothing'"
elif info['action'] == 2:
action_desc = f"Take Action: 'Hit your opponent'"
elif info['action'] == 3:
action_desc = f"Take Action: 'Move up'"
elif info['action'] == 4:
action_desc = f"Take Action: 'Move right'"
elif info['action'] == 5:
action_desc = f"Take Action: 'Move left'"
elif info['action'] == 6:
action_desc = f"Take Action: 'Move down'"
elif info['action'] == 7:
action_desc = f"Take Action: 'Move up-right'"
elif info['action'] == 8:
action_desc = f"Take Action: 'Move up-lefr'"
elif info['action'] == 9:
action_desc = f"Take Action: 'Move down-right'"
elif info['action'] == 10:
action_desc = f"Take Action: 'Move down-left'"
elif info['action'] == 11:
action_desc = f"Take Action: 'Hit your opponent and move up'"
elif info['action'] == 12:
action_desc = f"Take Action: 'Hit your opponent and move right'"
elif info['action'] == 13:
action_desc = f"Take Action: 'Hit your opponent and move left'"
elif info['action'] == 14:
action_desc = f"Take Action: 'Hit your opponent and move down'"
elif info['action'] == 15:
action_desc = f"Take Action: 'Hit your opponent and move up-right'"
elif info['action'] == 16:
action_desc = f"Take Action: 'Hit your opponent and move up-left'"
elif info['action'] == 17:
action_desc = f"Take Action: 'Hit your opponent and move down-right'"
else:
action_desc = f"Take Action: 'Hit your opponent and move down-left'"
reward_desc = f"Result: Reward of {info['reward']}, "
next_state_desc = BasicLevelTranslator().translate(info['next_state'])
descriptions.append(f"{state_desc}.\n {action_desc} \n {reward_desc} \n Transit to {next_state_desc}")
return descriptions
|