openAI的仿真环境Gym Retro的Python API接口（续1）—— 游戏过程记录及回放

如题，本文主要介绍仿真环境Gym Retro的Python API接口 。

官网地址：

​​https://retro.readthedocs.io/en/latest/python.html​​

本文环境配置：​​如何使用Python环境下的2D经典游戏仿真器（openai推出的）retro库运行游戏"刺猬索尼克" (SonicTheHedgehog-Genesis)​​

==============================================

游戏状态的记录，示例代码：

import retro

env = retro.make(game='Airstriker-Genesis', record='.')
env.reset()
while True:
_obs, _rew, done, _info = env.step(env.action_space.sample())
if done:
break

运行后将游戏的过程保存为文件：Airstriker-Genesis-Level1-000000.bk2

游戏过程的记录只需要在环境生成函数  retro.make  中指定 record 参数即可。

注意：

游戏过程文件只保存了游戏开始时的状态及过程中的所有按键，所以要对过程记录文件进行游戏回放的话我还是需要有相应的游戏ROM文件的。

游戏记录的回放，示例代码：

import retro

movie = retro.Movie('Airstriker-Genesis-Level1-000000.bk2')
movie.step()

env = retro.make(
game=movie.get_game(),
state=None,
# bk2s can contain any button presses, so allow everything
use_restricted_actions=retro.Actions.ALL,
players=movie.players,
obs_type=retro.Observations.IMAGE
)
env.initial_state = movie.get_state()
env.reset()

while movie.step():
keys = []
for p in range(movie.players):
for i in range(env.num_buttons):
keys.append(movie.get_key(i, p))

env.render()
observation, reward, done, info = env.step(keys)
print(keys)
print(observation)
print(reward)
print(done)
print(info)

由前文可知，

use_restricted_actions=retro.Actions.ALL    代表动作为 MultiBinary 类型，因此执行的动作形式大致为：

[True, False, False, False, False, False, True, False, True, False, False, False]

这个类型的。

其中，代码：

for p in range(movie.players):
for i in range(env.num_buttons):
keys.append(movie.get_key(i, p))

就是从记录文件中取出当前步骤中不同player的动作（MultiBinary 类型），该动作是分别判断MultiBinary 类型不同位置对应的动作（env.num_buttons）是否被选择，具体代码为：

movie.get_key(i, p)