From d218a01e4dcf5d6c2376c2c6ffac5ef6455bb946 Mon Sep 17 00:00:00 2001
From: "E. Almqvist" <elalmqvist@gmail.com>
Date: Mon, 25 Jan 2021 18:45:59 +0100
Subject: [PATCH] Image generation

---
 ascii_bin-image.jpg | Bin 0 -> 791 bytes
 ascii_bin.py        |   8 +++++---
 generator.py        |  10 +++++++---
 3 files changed, 12 insertions(+), 6 deletions(-)
 create mode 100644 ascii_bin-image.jpg

diff --git a/ascii_bin-image.jpg b/ascii_bin-image.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..996b84974f30ceaa37463fcda06efff186f8de0d
GIT binary patch
literal 791
zcmex=<NpH&0WUXCHwH#V1_nkTWcYuZ!I^=Bjg6g+m4ls~os*M;i${c)hnt&6Qb?Fz
zL{>^(PF6}rMnOeST|r4lSw=>~TvNxu(8R<<Uft5x!pKI?*u)5A2qPyaCl5D|1TU|I
zk)n*E5y{~H0R}-11|9|hW=16jCP7AKLB{__803NOWMu>c1}I=;VrF4wW9Q)H;sz?%
zD!{<R$jr>d!pzFb!U9xX3zTPI5o8roG<0MW4oqZMDikqloVbuf*=gfJ(V&YTRE(2~
znmD<{#3dx9RMpfqG__1j&CD$<t(;w4-P}Dqy@EqR!@?sXqmoln)6z3Cvx-Yf%gQS%
ztD0L{+uA!iyCzMZGIiSY88c@sTD)ZGvgIpQuG+MD%hqk%ckJAC=<t!F$Bv&kdFs;T
zD_5^wzj5={!$*&wJbm{3#miS8KYjl4_1pI!KYxMz#mK-6@fMJTcnr;7f<S*Uv9K_+
zu!H=?$W#u*%z`YeiiT`Lj)Clng~CckjT|CQ6Blkg$f;}`^g%SK=pvVxipfLOk07ss
zeMX$en#l4Q++zrT-D2QjW&}navmk>#!@AQ;?`G^Osd~FjZSvt?;ZypnpRjEfJ}j;C
zi6ym9Y|4@a>s}VD-+4ixIqzKc>w4CTEbnNIFRRabyZhR%`}22Q+SIUn7ZyKgieS`f
zvhi5WQ{5@J%HDA4tKj;~7hS8RVjT+?+=#x{;Pb(`G{t<CkOzZ_<AGyFm)LkVS#6u&
yx_I+Pxuxj~v!rsnqyJ9!*e0}w`OKY?mw7B}=d78;)+v(Dv(Q}cj`cf+|2F|w5**+F

literal 0
HcmV?d00001

diff --git a/ascii_bin.py b/ascii_bin.py
index 0bcebfd..5f79023 100644
--- a/ascii_bin.py
+++ b/ascii_bin.py
@@ -1,5 +1,7 @@
 import numpy as np
 import scipy.misc as smp
+from PIL import Image
+
 
 bit_on  = [255, 255, 255] # RGB value if bit is 1
 bit_off = [0, 0, 0] # RGB value if bit is 0
@@ -16,8 +18,8 @@ def data_to_matrix(data:list) -> np.array:
 
     return col_array
 
-def main(inp:list) -> None:
+def generate(inp:list) -> None:
     col_array = data_to_matrix(inp)
 
-    img = smp.toimage(col_array)
-    img.show()
+    img = Image.fromarray(col_array, "RGB")
+    img.save("ascii_bin-image.jpg")
diff --git a/generator.py b/generator.py
index 8cec4d7..653dfde 100755
--- a/generator.py
+++ b/generator.py
@@ -1,10 +1,10 @@
 #!/usr/bin/env python
 
 import sys
-from ascii_bin import main as ascii_bin
+import ascii_bin
 
 method_pointers = {
-    "ascii_bin": ascii_bin
+    "ascii_bin": ascii_bin.generate
 }
 
 
@@ -24,4 +24,8 @@ def generate_image(file:str, method:str = "ascii_bin", bytesize:int = 8) -> None
 
 
 if __name__ == "__main__":
-    generate_image(sys.argv[3], sys.argv[1], int(sys.argv[2]))
+    try:
+        generate_image(sys.argv[3], sys.argv[1], int(sys.argv[2]))
+    except Exception as err:
+        print("\033[91m./generator.py (method:str) (bytesize:int) (dimensions:str (WxH)) (input_file:str)\033[0m")
+        raise err