<reponame>charles-hefele/gym-digger<filename>gym_digger/envs/maps.py
import numpy as np

MAPS = {
    '2x2_a': np.array([
        [1, 1],
        [1, 1]
    ]),
    '2x2_b': np.array([
        [2, 1],
        [1, 2]
    ]),
    '2x2_c': np.array([
        [3, 1],
        [0, 2]
    ]),
    '2x2_d': np.array([
        [3, 4],
        [4, 2]
    ]),
    '2x2_e': np.array([
        [3, 5],
        [4, 2]
    ]),
    '2x2_f': np.array([
        [3, 6],
        [4, 2]
    ]),
    '2x2_g': np.array([
        [3, 7],
        [4, 2]
    ]),
    '2x2_h': np.array([
        [3, 8],
        [4, 2]
    ]),
    '2x2_i': np.array([
        [3, 9],
        [4, 2]
    ]),
    '2x2_j': np.array([
        [3, 10],
        [4, 2]
    ]),
    '2x2_k': np.array([
        [3, 25],
        [4, 2]
    ]),
    '2x2_l': np.array([
        [3, 100],
        [4, 2]
    ]),
    '2x2_m': np.array([
        [3, 50],
        [4, 2]
    ]),
    '3x3_a': np.array([
        [1, 1, 1],
        [1, 1, 1],
        [1, 1, 1]
    ]),
    '3x3_b': np.array([
        [1, 2, 1],
        [2, 1, 2],
        [1, 2, 1]
    ]),
    '3x3_c': np.array([
        [2, 2, 2],
        [2, 2, 2],
        [2, 2, 2]
    ]),
    '3x3_d': np.array([
        [3, 3, 3],
        [3, 3, 3],
        [3, 3, 3]
    ]),
    '3x3_e': np.array([
        [4, 4, 4],
        [4, 4, 4],
        [4, 4, 4]
    ]),
    '3x3_f': np.array([
        [5, 5, 5],
        [5, 5, 5],
        [5, 5, 5]
    ]),
    '3x3_g': np.array([
        [6, 6, 6],
        [6, 6, 6],
        [6, 6, 6]
    ]),
    '3x3_h': np.array([
        [7, 7, 7],
        [7, 7, 7],
        [7, 7, 7]
    ]),
    '3x3_i': np.array([
        [8, 8, 8],
        [8, 8, 8],
        [8, 8, 8]
    ]),
    '3x3_j': np.array([
        [9, 9, 9],
        [9, 9, 9],
        [9, 9, 9]
    ]),
    '3x3_k': np.array([
        [10, 10, 10],
        [10, 10, 10],
        [10, 10, 10]
    ]),
    '4x4_a': np.array([
        [2, 2, 1, 3],
        [1, 0, 0, 1],
        [2, 0, 1, 1],
        [1, 3, 1, 0]
    ]),
    '4x4_b': np.array([
        [0, 0, 2, 3],
        [0, 0, 0, 2],
        [0, 2, 0, 0],
        [2, 3, 1, 0]
    ]),
    '4x4_c': np.array([
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 0]
    ]),
    '4x4_d': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 0]
    ]),
    '4x4_e': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_f': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 1, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_g': np.array([
        [0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_h': np.array([
        [0, 2, 0, 4],
        [1, 3, 0, 2],
        [0, 1, 1, 0],
        [0, 2, 3, 0]
    ]),
    '6x6_a': np.array([
        [0, 0, 4, 4, 3, 0],
        [0, 1, 2, 5, 0, 0],
        [0, 0, 0, 0, 3, 1],
        [0, 3, 3, 0, 2, 1],
        [3, 4, 3, 0, 2, 0],
        [0, 2, 0, 0, 0, 0]
    ]),
    '6x6_b': np.array([
        [1, 1, 0, 1, 1, 0],
        [2, 0, 0, 2, 0, 0],
        [0, 3, 4, 0, 3, 3],
        [0, 4, 2, 0, 3, 4],
        [0, 3, 0, 2, 3, 5],
        [0, 0, 0, 1, 2, 0]
    ]),
    '7x7_a': np.array([
        [0, 0, 2, 1, 0, 4, 0],
        [0, 0, 1, 0, 4, 4, 6],
        [0, 0, 0, 0, 0, 5, 0],
        [0, 1, 1, 0, 0, 0, 0],
        [2, 2, 2, 0, 3, 0, 0],
        [2, 1, 0, 3, 4, 3, 0],
        [0, 0, 0, 2, 2, 0, 0]
    ]),
    '10x10_a': np.array([
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 3, 2, 0, 0, 0, 4, 0, 0],
        [0, 0, 1, 2, 0, 0, 4, 5, 4, 0],
        [0, 0, 0, 0, 0, 0, 0, 4, 0, 0],
        [0, 1, 1, 0, 0, 0, 0, 0, 0, 0],
        [0, 2, 2, 2, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 3, 2, 0],
        [0, 1, 2, 2, 0, 0, 3, 3, 2, 0],
        [0, 0, 3, 3, 0, 0, 0, 2, 2, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
    ]),
    '10x10_b': np.array([
        [0, 0, 3, 0, 0, 0, 0, 0, 3, 0],
        [0, 3, 5, 4, 0, 0, 0, 4, 3, 3],
        [0, 0, 4, 2, 0, 0, 2, 3, 4, 0],
        [0, 0, 0, 0, 0, 0, 0, 3, 0, 0],
        [0, 2, 1, 1, 0, 0, 0, 0, 0, 0],
        [0, 0, 2, 0, 0, 0, 0, 6, 0, 0],
        [0, 0, 0, 0, 0, 0, 6, 7, 5, 0],
        [0, 0, 0, 0, 0, 0, 6, 6, 5, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
    ])
}
import numpy as np

MAPS = {
    '2x2_a': np.array([
        [1, 1],
        [1, 1]
    ]),
    '2x2_b': np.array([
        [2, 1],
        [1, 2]
    ]),
    '2x2_c': np.array([
        [3, 1],
        [0, 2]
    ]),
    '2x2_d': np.array([
        [3, 4],
        [4, 2]
    ]),
    '2x2_e': np.array([
        [3, 5],
        [4, 2]
    ]),
    '2x2_f': np.array([
        [3, 6],
        [4, 2]
    ]),
    '2x2_g': np.array([
        [3, 7],
        [4, 2]
    ]),
    '2x2_h': np.array([
        [3, 8],
        [4, 2]
    ]),
    '2x2_i': np.array([
        [3, 9],
        [4, 2]
    ]),
    '2x2_j': np.array([
        [3, 10],
        [4, 2]
    ]),
    '2x2_k': np.array([
        [3, 25],
        [4, 2]
    ]),
    '2x2_l': np.array([
        [3, 100],
        [4, 2]
    ]),
    '2x2_m': np.array([
        [3, 50],
        [4, 2]
    ]),
    '3x3_a': np.array([
        [1, 1, 1],
        [1, 1, 1],
        [1, 1, 1]
    ]),
    '3x3_b': np.array([
        [1, 2, 1],
        [2, 1, 2],
        [1, 2, 1]
    ]),
    '3x3_c': np.array([
        [2, 2, 2],
        [2, 2, 2],
        [2, 2, 2]
    ]),
    '3x3_d': np.array([
        [3, 3, 3],
        [3, 3, 3],
        [3, 3, 3]
    ]),
    '3x3_e': np.array([
        [4, 4, 4],
        [4, 4, 4],
        [4, 4, 4]
    ]),
    '3x3_f': np.array([
        [5, 5, 5],
        [5, 5, 5],
        [5, 5, 5]
    ]),
    '3x3_g': np.array([
        [6, 6, 6],
        [6, 6, 6],
        [6, 6, 6]
    ]),
    '3x3_h': np.array([
        [7, 7, 7],
        [7, 7, 7],
        [7, 7, 7]
    ]),
    '3x3_i': np.array([
        [8, 8, 8],
        [8, 8, 8],
        [8, 8, 8]
    ]),
    '3x3_j': np.array([
        [9, 9, 9],
        [9, 9, 9],
        [9, 9, 9]
    ]),
    '3x3_k': np.array([
        [10, 10, 10],
        [10, 10, 10],
        [10, 10, 10]
    ]),
    '4x4_a': np.array([
        [2, 2, 1, 3],
        [1, 0, 0, 1],
        [2, 0, 1, 1],
        [1, 3, 1, 0]
    ]),
    '4x4_b': np.array([
        [0, 0, 2, 3],
        [0, 0, 0, 2],
        [0, 2, 0, 0],
        [2, 3, 1, 0]
    ]),
    '4x4_c': np.array([
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 0]
    ]),
    '4x4_d': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 0]
    ]),
    '4x4_e': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 0, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_f': np.array([
        [1, 1, 1, 0],
        [1, 1, 0, 0],
        [1, 0, 1, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_g': np.array([
        [0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 1]
    ]),
    '4x4_h': np.array([
        [0, 2, 0, 4],
        [1, 3, 0, 2],
        [0, 1, 1, 0],
        [0, 2, 3, 0]
    ]),
    '6x6_a': np.array([
        [0, 0, 4, 4, 3, 0],
        [0, 1, 2, 5, 0, 0],
        [0, 0, 0, 0, 3, 1],
        [0, 3, 3, 0, 2, 1],
        [3, 4, 3, 0, 2, 0],
        [0, 2, 0, 0, 0, 0]
    ]),
    '6x6_b': np.array([
        [1, 1, 0, 1, 1, 0],
        [2, 0, 0, 2, 0, 0],
        [0, 3, 4, 0, 3, 3],
        [0, 4, 2, 0, 3, 4],
        [0, 3, 0, 2, 3, 5],
        [0, 0, 0, 1, 2, 0]
    ]),
    '7x7_a': np.array([
        [0, 0, 2, 1, 0, 4, 0],
        [0, 0, 1, 0, 4, 4, 6],
        [0, 0, 0, 0, 0, 5, 0],
        [0, 1, 1, 0, 0, 0, 0],
        [2, 2, 2, 0, 3, 0, 0],
        [2, 1, 0, 3, 4, 3, 0],
        [0, 0, 0, 2, 2, 0, 0]
    ]),
    '10x10_a': np.array([
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 3, 2, 0, 0, 0, 4, 0, 0],
        [0, 0, 1, 2, 0, 0, 4, 5, 4, 0],
        [0, 0, 0, 0, 0, 0, 0, 4, 0, 0],
        [0, 1, 1, 0, 0, 0, 0, 0, 0, 0],
        [0, 2, 2, 2, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 3, 2, 0],
        [0, 1, 2, 2, 0, 0, 3, 3, 2, 0],
        [0, 0, 3, 3, 0, 0, 0, 2, 2, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
    ]),
    '10x10_b': np.array([
        [0, 0, 3, 0, 0, 0, 0, 0, 3, 0],
        [0, 3, 5, 4, 0, 0, 0, 4, 3, 3],
        [0, 0, 4, 2, 0, 0, 2, 3, 4, 0],
        [0, 0, 0, 0, 0, 0, 0, 3, 0, 0],
        [0, 2, 1, 1, 0, 0, 0, 0, 0, 0],
        [0, 0, 2, 0, 0, 0, 0, 6, 0, 0],
        [0, 0, 0, 0, 0, 0, 6, 7, 5, 0],
        [0, 0, 0, 0, 0, 0, 6, 6, 5, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
    ])
}