imageencoder/methods/ascii_bin.py

import numpy as np
import scipy.misc as smp
from PIL import Image


bit_on  = [191, 191, 191] # RGB value if bit is 1
bit_off = [17, 16, 21] # RGB value if bit is 0

def translate_bit_color(bit:str):
    if( bit == "1" ):
        return bit_on
    else:
        return bit_off

def data_to_matrix(data:list) -> np.array:
    newdat = [ [translate_bit_color(bit) for bit in byte] for byte in data ]
    col_array = np.array(newdat, dtype=np.uint8)

    return col_array

def generate(inp:list) -> None:
    print("Converting data to matrix...")
    col_array = data_to_matrix(inp)

    print("Generating image...")
    img = Image.fromarray(col_array, "RGB")
    img.save("ascii_bin-image.jpg")
    print("Generated image \"ascii_bin-image.jpg\"")