#!/usr/bin/python2

# Sozi - A presentation tool using the SVG standard
#
# Copyright (C) 2010-2013 Guillaume Savaton
#
# This program is dual licensed under the terms of the MIT license
# or the GNU General Public License (GPL) version 3.
# A copy of both licenses is provided in the doc/ folder of the
# official release of Sozi.
# 
# See http://sozi.baierouge.fr/wiki/en:license for details.

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


if __name__ == '__main__':
    option_parser = OptionParser()

    option_parser.description = "Export a Sozi presentation to a video"
    option_parser.usage = "sozi2video.py [options] url.svg"
    
    option_parser.add_option("-W", "--width", type="int", dest="width_px", default=1024,
        help="Page width, in pixels (default is 1024)")
    option_parser.add_option("-H", "--height", type="int", dest="height_px", default=768,
        help="Page height, in pixels (default is 768)")

    option_parser.add_option("-b", "--bit-rate", type="string", dest="bit_rate", default="2M",
        help="The target video bit rate (default is 2M)")
        
    option_parser.add_option("-o", "--output", type="string", dest="output",
        help="The target video file name")

    options, args = option_parser.parse_args()

    if len(args) == 0:
        option_parser.print_usage(sys.stderr)
        sys.exit()
    
    # Set input and output file name
    input_file_name = args[0]
    
    if options.output is not None:
        output_file_name = options.output
    else:
        output_file_name = os.path.basename(input_file_name)
        output_file_name = os.path.splitext(output_file_name)[0] + ".ogv"        
    
    # Create a temporary directory for intermediate files
    # changement de nom et de repertoire pour les fichiers intermediaires par Lionel Allorge le 18/07/2014
    # tmp_dir = tempfile.mkdtemp()
    tmp_dir = os.path.splitext(output_file_name)[0]
    
    # Export Sozi frames to individual PNG files
    js = os.path.join(os.path.dirname(__file__), "sozi2video.js") 
    subprocess.call(["phantomjs", js, input_file_name, tmp_dir, str(options.width_px), str(options.height_px)])
    
    # Merge all frames to a single video file
    if len(os.listdir(tmp_dir)):
        sys.stdout.write("Writing video to: {0}\n".format(output_file_name))
        frame_pngs = os.path.join(tmp_dir, "frame-%d.png")
        # changement de ffmpeg vers avconv par Lionel Allorge le 18/07/2014
        # subprocess.call(["ffmpeg", "-r", "50", "-f", "image2",
        subprocess.call(["avconv", "-r", "50", "-f", "image2",

            "-i", frame_pngs, "-b", options.bit_rate, output_file_name])

    # Remove the temporary directory and its content
    # suppression de l'effacement des images intermediaires par Lionel Allorge le 18/07/2014
    # shutil.rmtree(tmp_dir, ignore_errors=True)