04_convert_txt_to_png.py

# -*- coding: utf-8 -*-
"""04_convert_txt_to_png.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1xY_OCBQqT-9u8zmhnDEJdf0s7stHwv3H
"""



import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

tiles_dir = './tiles'

chars2pngs = {    
    "-": Image.open(f'./tiles/smb-background.png'),
    "X": Image.open(f'./tiles/smb-unpassable.png'),
    "S": Image.open(f'./tiles/smb-breakable.png'),
    "?": Image.open(f'./tiles/smb-question.png'),
    "Q": Image.open(f'./tiles/smb-question.png'),
    "o": Image.open(f'./tiles/smb-coin.png'),
    "E": Image.open(f'./tiles/smb-enemy.png'),
    "<": Image.open(f'./tiles/smb-tube-top-left.png'),
    ">": Image.open(f'./tiles/smb-tube-top-right.png'),
    "[": Image.open(f'./tiles/smb-tube-lower-left.png'),
    "]": Image.open(f'./tiles/smb-tube-lower-right.png'),
    "x": Image.open(f'./tiles/smb-path.png'),  # self-created
    "Y": Image.open(f'./tiles/Y.png'),  # self-created
    "N": Image.open(f'./tiles/N.png'),  # self-created
    "B": Image.open(f'./tiles/cannon_top.png'),
    "b": Image.open(f'./tiles/cannon_bottom.png'), 
}

def char_array_to_image(array):
    """
    Convert a 16-by-16 array of integers into a PIL.Image object
    param: array: a 16-by-16 array of integers
    """
    image = Image.new('RGB',(array.shape[1] * 16, array.shape[0] * 16))
    for row, seg in enumerate(array):
        for col, char in enumerate(seg):
            image.paste(chars2pngs[char], (col * 16, row * 16))
    return image

for i in range(10):
    with open(f'./generated_levels_txt/{i+1}.txt',  'r') as txt_f:
        infile = np.array([list(line.rstrip()) for line in txt_f.readlines()])
    plt.figure(figsize=(20, 4))
    plt.imshow(char_array_to_image(infile))
    plt.axis('off')
    plt.savefig(f'./generated_levels_png/{i}.png', dpi=200, bbox_inches='tight')