encoder.py

from __future__ import print_function

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

def encode(input, output):
    a = open(input, "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

    print(len(bits), _ELEMENT)



    for i in tqdm(range(test.size[0]), desc="Encoding"):
        for j in range(test.size[1]):
            if _ELEMENT == len(bits):
                test.save(output, "JPEG")
                a.close()
                return
            bitmap[i, j] = int(bits[_ELEMENT])

            _ELEMENT += 1

    test.save(output, "JPEG")
    a.close()

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

    encode(argv[1], argv[2])

if __name__ == "__main__":
    main()