#!/usr/bin/env python

import blur
import numpy
import argparse

import imageio

def main(infile, outfile, iterations):
    b = blur.blur()
    img = imageio.v2.imread(infile, pilmode="RGB")
    img = imageio.core.asarray(img)
    img = img / 2
    img = img.astype(numpy.uint8)
    
    blurred = b.main(iterations, img)
    blurred = blurred * 2
    
    # The .get() is to retrieve a Numpy array from the PyOpenCL array
    # being returned.
    imageio.imwrite(outfile, blurred.get().astype(numpy.uint8))

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Simple Gaussian blur of a PNG file.')
    parser.add_argument('--output-file', metavar='OUTFILE', type=str, required=True,
                        help='Where the result is written')
    parser.add_argument('--iterations', metavar='INT', type=int, default=1,
                        help='Number of iterations to run')
    parser.add_argument('filename', metavar='INFILE', type=str,
                        help='The PNG file to blur')
    args = parser.parse_args()
    main(args.filename, args.output_file, args.iterations)