from gym.envs.registration import register
import sys

register(
    id='AntMaze-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "AntMaze", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='PointMaze-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointMaze", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='PointMaze1-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointMaze1", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='PointMaze1Pos-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointMaze1", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": False, 'test': 'Test'}
)

register(
    id='PointMaze1Rand-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointMaze1", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': True,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='PointMazeRand-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointMaze", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': True,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)


register(
    id='AntMaze1-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "AntMaze1", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='AntMaze1Rand-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "AntMaze1", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': True,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)


register(
    id='PointPush-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointPush", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='PointPushRand-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "PointPush", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': True,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='AntPush-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "AntPush", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': False,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)

register(
    id='AntPushRand-v1',
    entry_point='envs.mujoco_maze.create_maze_env:create_maze_env',
    kwargs={'env_name': "AntPush", 'goal_args': [[-5, -5], [5, 5]], 'maze_size_scaling': 4, 'random_start': True,
            "fix_goal": True, "top_down_view": True, 'test': 'Test'}
)
