From 8917a71f47059c5bb4540f03ae858a11f4a39734 Mon Sep 17 00:00:00 2001
From: j <0x006A@0x2620.org>
Date: Wed, 2 Oct 2013 10:56:30 +0000
Subject: [PATCH] use avconv to extract frames from mp4 videos

---
 pandora/archive/extract.py | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/pandora/archive/extract.py b/pandora/archive/extract.py
index dede6c82..836ab271 100644
--- a/pandora/archive/extract.py
+++ b/pandora/archive/extract.py
@@ -278,21 +278,33 @@ def run_command(cmd, timeout=10):
     return p.returncode
 
 
-def frame(videoFile, frame, position, height=128, redo=False):
+def frame(video, frame, position, height=128, redo=False):
     '''
         params:
-            videoFile input
+            video     input
             frame     output
             position as float in seconds
             height of frame
             redo boolean to extract file even if it exists
     '''
-    if exists(videoFile):
-        frameFolder = os.path.dirname(frame)
+    if exists(video):
+        folder = os.path.dirname(frame)
         if redo or not exists(frame):
-            ox.makedirs(frameFolder)
-            cmd = ['oxframe', '-i', videoFile, '-o', frame,
-                '-p', str(position), '-y', str(height)]
+            ox.makedirs(folder)
+            if video.endswith('.mp4'):
+                cmd = [
+                    AVCONV, '-y',
+                    '-ss', str(position),
+                    '-i', video,
+                    '-an', '-vframes', '1',
+                    '-vf', 'scale=-1:%s' % height
+                ]
+                if not frame.endswith('.png'):
+                    cmd += ['-f', 'mjpeg']
+                cmd += [frame]
+            else:
+                cmd = ['oxframe', '-i', video, '-o', frame,
+                    '-p', str(position), '-y', str(height)]
             run_command(cmd)