Mercurial > hg > config
changeset 741:9681a0bd74d6
add utility for slicing video files
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Sun, 28 Jun 2015 11:56:13 -0700 |
parents | 8a9fe2f20bcb |
children | e12ca24fd7c6 |
files | python/ffslice.py |
diffstat | 1 files changed, 168 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python/ffslice.py Sun Jun 28 11:56:13 2015 -0700 @@ -0,0 +1,168 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +""" +slice a clip with `ffmpeg`: + +ffmpeg -ss <00:00:00> -t <sec> -i <input_file> -q 1 <output_file> # + +--ss = start time ; + +-t = length of video to create from the start time in seconds. +""" + +# imports +import argparse +import os +import subprocess +import sys + +# module globals +__all__ = ['main', 'Parser'] +here = os.path.dirname(os.path.realpath(__file__)) +string = (str, unicode) + + +def ensure_dir(directory): + """ensure a directory exists""" + if os.path.exists(directory): + assert os.path.isdir(directory) + return directory + os.makedirs(directory) + return directory + + +def convert_seconds_to_hhmmss(time_sec): + """converts `time_sec` to (hh,mm,ss)""" + + hh = int(time_sec / 3600) + mm = int((time_sec % 3600) / 60) + ss = int((time_sec % 3600) % 60) + return (hh,mm,ss) + + +def format_seconds_to_hhmmss(time_sec, separator=':'): + """converts `time_sec` to string 'hh:mm:ss'""" + return separator.join(['{:0>2d}'.format(i) + for i in convert_seconds_to_hhmmss(time_sec)]) + + +def duration(clip, ffprobe='ffprobe'): + """get the duration in seconds using `ffprobe` from ffmpeg""" + + command = [ffprobe, clip] + assert os.path.exists(clip), "Missing clip, duration not available" + + # probe the file + try: + output = subprocess.check_output(command, stderr=subprocess.STDOUT) + # (for some reason, ffprobe sends output to stderr) + except subprocess.CalledProcessError as e: + print (e.output) + raise + + duration = None + for line in output.splitlines(): + # Parse the output of `ffprobe` + # look for a line like: + # Duration: 00:00:59.73, start: 0.000000, bitrate: 7783 kb/s + + line = line.strip() + if line.startswith("Duration:"): + if duration: + raise AssertionError("Duplicate duration - already found: {}".format(duration)) + line = line.split(',')[0] + duration = line.split(':', 1)[1].strip() + + if duration: + hh, mm, ss = [float(i) for i in duration.split(':')] + duration = 3600*hh + 60*mm + ss + + return duration + + +class FFSliceParser(argparse.ArgumentParser): + """fflice CLI option parser""" + + default_slice_time = 300. + + def __init__(self): + argparse.ArgumentParser.__init__(self, description=__doc__) + self.add_argument('clips', metavar='clip', nargs='+', help="clips to slice") + self.add_argument('-d', '--directory', dest='directory', + default=os.getcwd(), + help="output directory [DEFAULT: %(default)s]") + self.add_argument('--durations', '--print-durations', dest='print_durations', + action='store_true', default=False, + help="print durations and exit") + self.add_argument('-n', dest='number', type=int, + help="number of slices") + self.add_argument('-t', '--time', dest='slice_time', + type=float, + help="time of each slice [DEFAULT: {}]".format(self.default_slice_time)) + self.add_argument('--dry-run', dest='dry_run', + action='store_true', default=False, + help="print out what will be done") + self.add_argument('--hhmmss', dest='hhmmss', + action='store_true', default=False, + help="display times in 'hh:mm:ss' format; thedefault is in seconds") + + self.add_argument('-v', '--verbose', dest='verbose', + action='store_true', default=False, + help="be verbose") + self.options = None + + def parse_args(self, *args, **kw): + options = argparse.ArgumentParser.parse_args(self, *args, **kw) + self.validate(options) + self.options = options + return options + + def validate(self, options): + """validate options""" + + missing = [clip for clip in options.clips + if not os.path.exists(clip)] + if missing: + self.error("Not found: {}".format(', '.join(missing))) + + if options.slice_time and options.number: + self.error("Cannot specify slice time and number of slices") + # TODO: allow specification of both + elif options.slice_time is None and options.number is None: + options.slice_time = self.default_slice_time + + ensure_dir(options.directory) + + def format_seconds(self, seconds): + """format seconds to string""" + if self.options.hhmmss: + return format_seconds_to_hhmmss(seconds) + return '{:.2}'.format(seconds) + + +def main(args=sys.argv[1:]): + """CLI""" + + # parse command line options + parser = FFSliceParser() + options = parser.parse_args(args) + + # compute durations + durations = {clip: duration(clip) for clip in options.clips} + if options.print_durations: + returncode = 0 + total = 0. + for clip in options.clips: + _duration = durations[clip] + if _duration is None: + print ("Duration not found: '{}'".format(clip)) + returncode = 1 + else: + print ('{} : {}'.format(clip, + parser.format_seconds(_duration))) + sys.exit(returncode) + +if __name__ == '__main__': + main() +