diff --git a/decode.py b/decode.py index 2774d58..eb5b156 100644 --- a/decode.py +++ b/decode.py @@ -26,6 +26,9 @@ from PIL import Image import sys +if not len(sys.argv) in [2, 3]: + print(f"Usage: {sys.argv[0]} [output.png]") + # RGB, RGBA frm_sizes = [3, 4] frm_names = ["RGB", "RGBA"] @@ -41,4 +44,7 @@ height = (len(data)//frm_size)//width print(width, height) img = Image.frombuffer(frm_names[formt], (width, height), data, "raw", frm_names[formt], 0, 1) -img.save('.'.join(sys.argv[1].split(".")[:1])+".png") +if len(sys.argv) == 2: + img.save('.'.join(sys.argv[1].split(".")[:1])+".png") +elif len(sys.argv) == 3: + img.save(sys.argv[2])