diff --git a/RRF-Gen.py b/RRF-Gen.py index 6cad09a..cbfe018 100644 --- a/RRF-Gen.py +++ b/RRF-Gen.py @@ -2,12 +2,16 @@ import math import sys from PIL import Image +import os -image = Image.open(sys.argv[1]).convert('L') +inputFilename = sys.argv[1] +outputFilename = os.path.splitext(inputFilename)[0] + ".img" + +image = Image.open(inputFilename).convert('L') bitsPerRow = math.ceil(image.width/8) * 8 paddingBits = bitsPerRow - image.width -outputFile = open(sys.argv[2], "wb") +outputFile = open(outputFilename, "wb") outputFile.write(bytes((image.width,))) outputFile.write(bytes((image.height,))) @@ -30,4 +34,4 @@ for row in range(image.height): outputByte += "0" outputFile.write(bytes((int(outputByte,2),))) -outputFile.close() \ No newline at end of file +outputFile.close()