standalone.py

from __future__ import division, print_function, unicode_literals

# This code is so you can run the samples without installing the package
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../..'))
#

import time
import random

import mazeexp as mx

def main(argv):

    mode_id = 0
    if '-m' in argv or '--mode' in argv:
        indexes = [i for i,x in enumerate(argv) if x == '-m' or x == '--mode']
        mode_id = argv[indexes[0]+1]
        print('Changed mode to ' + mode_id)

    engine = mx.MazeExplorer(mode_id)

    if '-r' in argv or '--random' in argv:
        print('Random test agent...')

        engine.reset()

        while not engine.director.window.has_exit:
            action = random.randint(0, engine.actions_num-1)
            observation, reward, terminal, info = engine.act(action)
            #print(observation)
            print(action, [min(observation),max(observation)], reward, terminal)
    elif '-t' in argv or '--test' in argv:
        print('Dupicate deaccel bug agent...')

        engine.reset()

        cnt = 0
        while not engine.director.window.has_exit:
            action = 2
            if cnt % 4 == 0:
                switch = random.randint(0, 1)
                if switch == 0:
                    action = 0
                else:
                    action = 4

            # Wait random time
            sleep = 5*random.random()
            print('sleep', sleep)
            time.sleep(sleep)
            observation, reward, terminal, info = engine.act(action)
            #print(observation)
            #print(action, [min(observation),max(observation)], reward, terminal)
            cnt += 1
    elif '-s' in argv or '--step' in argv:
        print('Step by step...')
        engine.reset()
        cnt = 0
        while not engine.director.window.has_exit:
            sleep = 5*random.random()
            print('sleep', sleep)
            time.sleep(sleep)
            engine.step()

            #print(cnt, engine.world_layer.get_state())
            cnt += 1
    else:
        engine.run()

if __name__ == "__main__":
   main(sys.argv[1:])