#!/usr/bin/python3
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
# GPL 2010 <j@mailb.org>

import os
import sys
from optparse import OptionParser
import subprocess
import tempfile

from PIL import Image


if __name__ == '__main__':
    width = None
    height = None
    size = None
    parser = OptionParser()
    parser.add_option('-x', '--size', dest='size', help='scale image to size')
    parser.add_option('-p', '--pos', dest='pos', help='frame position in seconds, float')
    parser.add_option('-i', '--input', dest='input', help='video input')
    parser.add_option('-o', '--output', dest='output', help='path to save frame to, jpg, png supported')
    (opts, args) = parser.parse_args()
    if None in (opts.input, opts.output, opts.pos):
        parser.print_help()
        sys.exit()

    f, fname = tempfile.mkstemp(suffix='.png')
    cmd = ['oxframe', '-i', opts.input, '-o', fname, '-p', opts.pos]
    p = subprocess.Popen(cmd)
    p.wait()
    frame = Image.open(fname)
    width = frame.size[0]
    height = frame.size[1]

    if opts.size:
       size = opts.size
    else:
        size = min(width, height) 

    if width > height:
        width = int(int(opts.size) * (float(width) / height))
        height = int(opts.size)
        width = width - width % 2
        ox = int(abs(height - width) / 2)
        oy = 0
    else:
        height = int(int(opts.size) * (float(height) / width))
        width = int(opts.size)
        height = height - height % 2
        ox = 0
        oy = int(abs(height - width) / 2)

    frame = frame.resize((width, height), Image.ANTIALIAS)

    frame = frame.crop((ox, oy, int(opts.size) + ox, oy + int(opts.size)))
    frame.save(opts.output)
    if os.path.exists(fname):
        os.unlink(fname)