import click


@click.command()
@click.option('-i', '--orig', 'opt_dir_in_orig', required=True,
  help='Input directory')
@click.option('-n', '--new', 'opt_dir_in_new', required=True,
  help='Input directory files to compare to')
@click.pass_context
def cli(ctx, opt_dir_in_orig, opt_dir_in_new):
  """Compare quality of super resolution images"""

  import os

  import sys
  from os.path import join
  from pathlib import Path
  from glob import glob

  from random import randint
  from PIL import Image, ImageOps, ImageFilter
  from pathlib import Path
  import cv2 as cv

  from app.settings import types
  from app.utils import click_utils
  from app.settings import app_cfg as cfg
  from app.utils import file_utils, im_utils, logger_utils
  
  log = logger_utils.Logger.getLogger()

  fp_ims = glob(join(opt_dir_in_new, '*.jpg'))
  fp_ims += glob(join(opt_dir_in_new, '*.png'))

  log.info('{}'.format(len(fp_ims)))

  while True:
    rn = randint(0, len(fp_ims) - 1)
    fp_im_new = fp_ims[rn]
    fp_im_orig = fp_im_new.replace(opt_dir_in_new, opt_dir_in_orig)
    log.info('new: {}'.format(fp_im_new))
    log.info('orig: {}'.format(fp_im_orig))

    im_new = cv.imread(fp_im_new)
    im_orig = cv.imread(fp_im_orig)

    # show
    cv.imshow('new', im_new)
    cv.imshow('orig', im_orig)

    # handle key io
    k = cv.waitKey(0) & 0xFF
    if k == 27 or k == ord('q'):  # ESC
      # exits the app
      cv.destroyAllWindows()
      sys.exit('Exiting because Q or ESC was pressed')
    elif k == ord(' ') or k == 81 or k == 83:
      continue