#!/usr/bin/env python

from pyAvTranscoder import avtranscoder as av

# Get command line arguments

args = []

try:

# python2.7+

import argparse

# Create command-line interface

parser = argparse.ArgumentParser(

prog='pyrewrap',

description='''Rewrap a video file and can move the index (moov atom) to the beginning of the file.''',

)

# requirements

parser.add_argument('inputFileName', type=str, help='It could be any video file. Support file without extension.')

# options

parser.add_argument("-o", "--outputFile", dest="outputFileName", type=str, default="output.mov", help="Set the output filename (output.mov by default).")

parser.add_argument("-c", "--format", dest="format", type=str, default="mov", help="Specify the output format.")

parser.add_argument("-f", "--faststart", dest="faststart", action="store_true", default=False, help="Specify if the faststart option must be apply during rewrapping process (warning: 'mov' specific option).")

# Parse command-line

args = parser.parse_args()

except ImportError:

import optparse

# Create command-line interface

parser = optparse.OptionParser(

usage='usage: %prog -o <outputfile> -c <format> [-f] -i <inputfile>',

prog='pyrewrap',

description='''Rewrap a video file and can move the index (moov atom) to the beginning of the file.''',

)

# requirements

parser.add_option("-i", "--inputFile", dest='inputFileName', type="string", help='It could be any video file. Support file without extension.')

# options

parser.add_option("-o", "--outputFile", dest="outputFileName", type="string", default="output.mov", help="Set the output filename (output.mov by default).")

parser.add_option("-c", "--format", dest="format", type="string", default="mov", help="Specify the output format.")

parser.add_option("-f", "--faststart", dest="faststart", action="store_true", default=False, help="Specify if the faststart option must be apply during rewrapping process (warning: 'mov' specific option).")

# Parse command-line

args, other = parser.parse_args()

if args.inputFileName is None:

parser.print_help()

exit(1)

# setup avtranscoder

av.Logger().setLogLevel(av.AV_LOG_QUIET)

av.preloadCodecsAndFormats()

# create input file

inputFile = av.InputFile(args.inputFileName)

if len(inputFile.getProperties().getVideoProperties()) == 0:

print("No video stream found in file ", args.inputFileName)

exit(1)

# create output file (need to set format profile of encoding to force output format to mp4)

formatProfile = av.ProfileMap()

formatProfile[ av.avProfileIdentificator ] = "mp4WrapFormatPreset"

formatProfile[ av.avProfileIdentificatorHuman ] = "MP4 rewraping format preset"

formatProfile[ av.avProfileType ] = av.avProfileTypeFormat

formatProfile[ av.avProfileFormat ] = args.format

if args.faststart:

formatProfile[ "movflags" ] = "+faststart"

outputFile = av.OutputFile( args.outputFileName )

outputFile.setupWrapping( formatProfile )

# create transcoder

transcoder = av.Transcoder( outputFile )

for streamIndex in range(0, inputFile.getProperties().getNbStreams()):

transcoder.addStream(av.InputStreamDesc(args.inputFileName, streamIndex))

# launch process

progress = av.ConsoleProgress()

transcoder.process(progress)