import argparse
from os import path
from . import __version__
from .adathres import adathres_bts
from .dither import grid_bts, noise_bts
from .quant import pngquant
from .trunc import trunc_bts
from .util import *

modes = ['grid', 'noise', 'trunc', 'quant', 'thres']

def main():
    parser = argparse.ArgumentParser(description='ImgYaso')

    parser.add_argument('fname', help='file name to be processed')
    parser.add_argument('-c', '--colors', type=int, default=8, help='num of colors')
    parser.add_argument('-m', '--mode', default=modes[0], choices=modes, help='processing mode')
    parser.add_argument('-o', '--ofname', help='output file name')
    args = parser.parse_args()

    if not path.exists(args.fname):
        print('file not found')
        return
        
    with open(args.fname, 'rb') as f:
        img = f.read()
    img = conv2png(img)
    
    if args.mode == 'grid':
        img = grid_bts(img)
    elif args.mode == 'noise':
        img = noise_bts(img)
    elif args.mode == 'trunc':
        img = trunc_bts(img, args.colors)
    elif args.mode == 'quant':
        img = pngquant(img, args.colors)
    elif args.mode == 'thres':
        img = adathres_bts(img)
    
    fname = args.ofname or args.fname
    with open(fname, 'wb') as f:
        f.write(img)
        
    print('done...')
    
if __name__ == '__main__': main()