import sys
import os
from skimage.io import imsave
import numpy as np
from glob import glob

out = os.environ.get('OUTDIR', os.getcwd())

files = [f for arg in sys.argv[1:] for f in glob(arg)]

colors = [(0,0,0), 
          (255, 0, 0), 
          (0, 255, 0), 
          (255,255,0),  
          (0, 0, 255),  
          (255, 0, 255),  
          (0, 255, 255),  
          (255, 255, 255)]  
colors = np.array(colors, dtype=np.uint8)


for f in files:
    outname = os.path.splitext(f)[0] + '.png'
    print(f, '-->', outname)
    data = np.loadtxt(f, dtype=int)
    im = colors[data+1]
    imsave(outname, im)