PyAV/scripts/generate_video.py at shared-options · Python3pkg/PyAV · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
#!/usr/bin/env python

from __future__ import print_function

import subprocess
import optparse
import os
import sys

def which(program):
    import os
    def is_exe(fpath):
        return os.path.isfile(fpath) and os.access(fpath, os.X_OK)

    fpath, fname = os.path.split(program)
    if fpath:
        if is_exe(program):
            return program
    else:
        for path in os.environ["PATH"].split(os.pathsep):
            path = path.strip('"')
            exe_file = os.path.join(path, program)
            if is_exe(exe_file):
                return exe_file

    return None

ffmpeg_cmd = None
avconv_cmd = None

if os.name == 'nt':
   ffmpeg_bin, avconv_bin = 'ffmpeg.exe', 'avconv.exe'
else:
   ffmpeg_bin, avconv_bin = 'ffmpeg', 'avconv'


if which(ffmpeg_bin):
    ffmpeg_cmd = [which(ffmpeg_bin), '-y', '-f', 'lavfi', '-i']
if which(avconv_bin):
    avconv_cmd = [which(avconv_bin),'-y', '-filter_complex']

if not ffmpeg_cmd and not avconv_cmd:
    print('Unable to find ffmpeg or avconve command')
    sys.exit(-1)

def testsrc(
        size, frame_rate, time, out_path, vcodec=None, bitrate=None, pix_fmt=None, audio=False, language=None,
        use_avconv=False):
    if use_avconv or os.environ.get('LIBRARY', None) == 'libav':
        exec_cmd = avconv_cmd
    elif ffmpeg_cmd:
        exec_cmd = ffmpeg_cmd
    else:
        exec_cmd = avconv_cmd

    exec_cmd.extend(['testsrc=size=%s:rate=%s' % (size, str(frame_rate))])

    if vcodec:
        exec_cmd.extend(['-vcodec',vcodec])

    if pix_fmt:
        exec_cmd.extend(['-pix_fmt', pix_fmt])

    if audio:
        if avconv_cmd:
            exec_cmd.extend(['-filter_complex', 'aevalsrc=0', '-map', '0', '-map', '1'])
        else:
            exec_cmd.extend(['-f', 'lavfi', '-i', 'aevalsrc=0', '-map', '0', '-map', '1'])
        if language:
            exec_cmd.extend(['-metadata:s:a:0', 'language={0}'.format(language)])

    if bitrate:
        exec_cmd.extend(['-b', str(bitrate)])

    exec_cmd.extend(['-t', str(time)])
    exec_cmd.extend([out_path])
    print(subprocess.list2cmdline(exec_cmd))
    subprocess.check_call(exec_cmd)

def main():
    parser = optparse.OptionParser()
    parser.add_option('-r', '--rate', help="Frame Rate", default=24, type="string")
    parser.add_option('-t', '--time', help="Duration in seconds.", default=5, type="int")
    parser.add_option('-s', '--size', help="size of testsrc", default='640x480')
    parser.add_option('-v', '--vcodec', help="video codec", default='mpeg4')
    parser.add_option('-b', '--bitrate', help="video bitrate")
    parser.add_option('-p', '--pix_fmt', help='pixel format')
    parser.add_option('-a', '--with-audio', help='add audio stream', action="store_true", default=False)
    parser.add_option('-l', '--language', help='audio language')
    parser.add_option('--use_avconv', help="force using avconv", action="store_true", default=False)

    (options, args) = parser.parse_args()

    if not args:
        parser.error("not enough args")

    testsrc(options.size,
            options.rate,
            options.time,
            args[0],
            options.vcodec,
            options.bitrate,
            options.pix_fmt,
            options.with_audio,
            options.language,
            use_avconv=options.use_avconv)


if __name__ == "__main__":
    main()