from __future__ import print_function

import bitstring
from PIL import Image
from tqdm import tqdm
from sys import argv

if len(argv) != 3:
    print("Usage: " + argv[0] + " <input> <output>")
    exit()

a = open(argv[1], "rb")
b = bitstring.Bits(a)
print("Input file size: " + str(len(b)/8) + " bytes")
test = Image.new("1", (1920, 1080))
bitmap = test.load()
bits = list(b.bin)

_ELEMENT = 0

for i in tqdm(range(test.size[0])):
    for j in tqdm(range(test.size[1])):
        bitmap[i, j] = int(bits[_ELEMENT])
        if _ELEMENT == len(bits):
            break
        _ELEMENT += 1

test.save(argv[2], "JPEG")
a.close()