Add pyconcat app

Clement Champetier · Clement Champetier · commit 5915bb8f5610 · 2015-11-05T18:09:14.000+01:00
v1 of the app.
Next TODO:
* manage when first add is audio stream.
* manage input file with several streams.
diff --git a/app/pyConcat/pyconcat.py b/app/pyConcat/pyconcat.py
@@ -0,0 +1,63 @@
+#!/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='pyconcat',
+        description='''Concatenate first stream of each given file to create an output file.''',
+    )
+
+    # requirements
+    parser.add_argument('inputs', nargs='+', action='store', help='list of files to concatenate')
+    # options
+    parser.add_argument("-o", "--outputFile", dest="outputFileName", type=str, default="output.mov", help="Set the output filename (output.mov by default).")
+    # Parse command-line
+    args = parser.parse_args()
+
+except ImportError:
+    print("pyconcat currently expects python2.7+")
+    exit(1)
+
+# setup avtranscoder
+logger = av.Logger().setLogLevel(av.AV_LOG_QUIET)
+av.preloadCodecsAndFormats()
+
+# output
+outputFile = av.OutputFile( args.outputFileName );
+
+# get all input files
+inputFiles = []
+streamTypeToConcat = 0
+for input in args.inputs:
+    inputFile = av.InputFile(input)
+    streamTypeToConcat = inputFile.getStream(0).getProperties().getStreamType()
+    inputFiles.append(inputFile)
+
+if streamTypeToConcat == av.AVMEDIA_TYPE_VIDEO:
+    outputFile.addVideoStream( inputFiles[-1].getStream(0).getVideoCodec() )
+elif streamTypeToConcat == av.AVMEDIA_TYPE_AUDIO:
+    outputFile.addVideoStream( inputFiles[-1].getStream(0).getAudioCodec() )
+
+### process
+outputFile.beginWrap()
+
+data = av.Frame()
+# for each input
+for inputFile in inputFiles:
+    packetRead = True
+    # read all packets of first stream
+    while packetRead:
+        # read
+        packetRead = inputFile.readNextPacket( data, 0 )
+        # wrap
+        outputFile.wrap( data, 0 )
+
+outputFile.endWrap()
