You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.9 KiB
64 lines
1.9 KiB
10 years ago
|
#!/usr/bin/python3
|
||
|
|
# (c) 2013 MORI Taeyeon
|
||
|
|
# Create Video Thumbnails using FFmpeg
|
||
|
|
|
||
|
|
from __future__ import print_function, division, absolute_import, unicode_literals
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
import random
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
ffmpeg_cmd = ["ffmpeg"]
|
||
|
|
ffprobe_cmd = ["ffmpeg", "-i"]
|
||
|
|
regexp = re.compile(r"Duration: (\d+):(\d\d):(\d\d).\d\d,")
|
||
|
|
|
||
|
|
def getdur(file):
|
||
|
|
cmd = ffprobe_cmd + [file]
|
||
|
|
pipe = subprocess.Popen(cmd, stderr=subprocess.PIPE)
|
||
|
|
pipe.wait()
|
||
|
|
try:
|
||
|
|
for line in pipe.stderr:
|
||
|
|
sys.stderr.buffer.write(line)
|
||
|
|
line = str(line, "utf8").rstrip()
|
||
|
|
if "Duration:" in line:
|
||
|
|
dur = regexp.search(line)
|
||
|
|
return int(dur.group(1))*3600+int(dur.group(2))*60+int(dur.group(3))
|
||
|
|
else:
|
||
|
|
raise ValueError("Duration not in FFMpeg output!")
|
||
|
|
finally:
|
||
|
|
sys.stderr.buffer.flush()
|
||
|
|
sys.stderr.flush()
|
||
|
|
|
||
|
|
def randpos(file):
|
||
|
|
dur = getdur(file)
|
||
|
|
pos = random.randint(1, dur)
|
||
|
|
print("Dur: %i, Pos: %i" % (dur, pos))
|
||
|
|
return pos
|
||
|
|
|
||
|
|
def mkthumb(file, pos, target, res=None):
|
||
|
|
cmd = ffmpeg_cmd + ["-ss", pos, "-i", file, "-an", "-vframes", 1, "-f", "image2"]
|
||
|
|
if res is not None:
|
||
|
|
cmd.extend(["-s", res])
|
||
|
|
cmd.append(target)
|
||
|
|
return subprocess.check_call(map(str,cmd))
|
||
|
|
|
||
|
|
def main(argv):
|
||
|
|
parser = argparse.ArgumentParser(prog=argv[0])
|
||
|
|
parser.add_argument("input", help="input file name")
|
||
|
|
parser.add_argument("target", help="thumbnail file name")
|
||
|
|
parser.add_argument("-p", help="Position in Video (random if omitted)", type=int)
|
||
|
|
parser.add_argument("-s", help="scale picture (WxH)")
|
||
|
|
args = parser.parse_args(argv[1:])
|
||
|
|
os.stat(args.input)
|
||
|
|
if args.p:
|
||
|
|
mkthumb(args.input, args.p, args.target, args.s)
|
||
|
|
else:
|
||
|
|
mkthumb(args.input, randpos(args.input), args.target, args.s)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import sys
|
||
|
|
sys.exit(main(sys.argv))
|
||
|
|
|