mirror of
https://github.com/kkroening/ffmpeg-python.git
synced 2025-04-05 04:22:51 +08:00
#1: split operators into modules
This commit is contained in:
Karl Kroening
parent
efbfd9f233
commit
696e52a989
@ -1,9 +1,4 @@
|
||||
.. ffmpeg-python documentation master file, created by
|
||||
sphinx-quickstart on Sat May 27 14:30:53 2017.
|
||||
You can adapt this file completely to your liking, but it should at least
|
||||
contain the root `toctree` directive.
|
||||
|
||||
Welcome to ffmpeg-python's documentation!
|
||||
ffmpeg-python: Python bindings for FFmpeg
|
||||
=========================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
@ -84,14 +84,12 @@
|
||||
<table style="width: 100%" class="indextable genindextable"><tr>
|
||||
<td style="width: 33%; vertical-align: top;"><ul>
|
||||
<li><a href="index.html#module-ffmpeg">ffmpeg (module)</a>
|
||||
</li>
|
||||
<li><a href="index.html#ffmpeg.file_input">file_input() (in module ffmpeg)</a>
|
||||
</li>
|
||||
</ul></td>
|
||||
<td style="width: 33%; vertical-align: top;"><ul>
|
||||
<li><a href="index.html#ffmpeg.file_output">file_output() (in module ffmpeg)</a>
|
||||
<li><a href="index.html#ffmpeg.file_input">file_input() (in module ffmpeg)</a>
|
||||
</li>
|
||||
<li><a href="index.html#ffmpeg.fluent">fluent() (in module ffmpeg)</a>
|
||||
<li><a href="index.html#ffmpeg.file_output">file_output() (in module ffmpeg)</a>
|
||||
</li>
|
||||
</ul></td>
|
||||
</tr></table>
|
||||
|
||||
@ -6,7 +6,7 @@
|
||||
<head>
|
||||
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
|
||||
|
||||
<title>Welcome to ffmpeg-python’s documentation! — ffmpeg-python documentation</title>
|
||||
<title>ffmpeg-python: Python bindings for FFmpeg — ffmpeg-python documentation</title>
|
||||
|
||||
<link rel="stylesheet" href="_static/alabaster.css" type="text/css" />
|
||||
<link rel="stylesheet" href="_static/pygments.css" type="text/css" />
|
||||
@ -41,125 +41,91 @@
|
||||
<div class="bodywrapper">
|
||||
<div class="body" role="main">
|
||||
|
||||
<div class="section" id="welcome-to-ffmpeg-python-s-documentation">
|
||||
<h1>Welcome to ffmpeg-python’s documentation!<a class="headerlink" href="#welcome-to-ffmpeg-python-s-documentation" title="Permalink to this headline">¶</a></h1>
|
||||
<div class="section" id="ffmpeg-python-python-bindings-for-ffmpeg">
|
||||
<h1>ffmpeg-python: Python bindings for FFmpeg<a class="headerlink" href="#ffmpeg-python-python-bindings-for-ffmpeg" title="Permalink to this headline">¶</a></h1>
|
||||
<div class="toctree-wrapper compound">
|
||||
</div>
|
||||
<span class="target" id="module-ffmpeg"></span><p>ffmpeg-python: Python bindings for FFmpeg</p>
|
||||
<dl class="function">
|
||||
<span class="target" id="module-ffmpeg"></span><dl class="function">
|
||||
<dt id="ffmpeg.colorchannelmixer">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">colorchannelmixer</code><span class="sig-paren">(</span><em>parent</em>, <em>*args</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.colorchannelmixer" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
<dd><p>Adjust video input frames by re-mixing color channels.</p>
|
||||
<p><a class="reference external" href="https://ffmpeg.org/ffmpeg-filters.html#toc-colorchannelmixer">FFmpeg colorchannelmixer filter</a></p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.concat">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">concat</code><span class="sig-paren">(</span><em>*parents</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.concat" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.drawbox">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">drawbox</code><span class="sig-paren">(</span><em>parent</em>, <em>x</em>, <em>y</em>, <em>width</em>, <em>height</em>, <em>color</em>, <em>thickness=None</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.drawbox" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.file_input">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">file_input</code><span class="sig-paren">(</span><em>filename</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.file_input" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>Input from a file.</p>
|
||||
<p>Corresponds to ffmpeg <cite>-i</cite> option.</p>
|
||||
<dl class="docutils">
|
||||
<dt>Args:</dt>
|
||||
<dd>filename: Input filename</dd>
|
||||
</dl>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.file_output">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">file_output</code><span class="sig-paren">(</span><em>parent</em>, <em>filename</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.file_output" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.fluent">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">fluent</code><span class="sig-paren">(</span><em>node_classes=set([<class ‘ffmpeg.nodes.Node’>])</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.fluent" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>Decorator to make function show up as an instance method on specified node classes.</p>
|
||||
<table class="docutils field-list" frame="void" rules="none">
|
||||
<col class="field-name" />
|
||||
<col class="field-body" />
|
||||
<tbody valign="top">
|
||||
<tr class="field-odd field"><th class="field-name">Parameters:</th><td class="field-body"><strong>node_classes</strong> – list of node classes</td>
|
||||
</tr>
|
||||
</tbody>
|
||||
</table>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.get_args">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">get_args</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.get_args" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.hflip">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">hflip</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.hflip" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.hue">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">hue</code><span class="sig-paren">(</span><em>parent</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.hue" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.merge_outputs">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">merge_outputs</code><span class="sig-paren">(</span><em>*parents</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.merge_outputs" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.overlay">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">overlay</code><span class="sig-paren">(</span><em>main_parent</em>, <em>overlay_parent</em>, <em>eof_action=’repeat’</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.overlay" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.overwrite_output">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">overwrite_output</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.overwrite_output" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.run">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">run</code><span class="sig-paren">(</span><em>parent</em>, <em>cmd=’ffmpeg’</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.run" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.setpts">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">setpts</code><span class="sig-paren">(</span><em>parent</em>, <em>expr</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.setpts" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.trim">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">trim</code><span class="sig-paren">(</span><em>parent</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.trim" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.vflip">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">vflip</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.vflip" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.zoompan">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">zoompan</code><span class="sig-paren">(</span><em>parent</em>, <em>**kwargs</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.zoompan" title="Permalink to this definition">¶</a></dt>
|
||||
<dd><p>FIXME</p>
|
||||
</dd></dl>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.file_input">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">file_input</code><span class="sig-paren">(</span><em>filename</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.file_input" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.file_output">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">file_output</code><span class="sig-paren">(</span><em>parent</em>, <em>filename</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.file_output" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.merge_outputs">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">merge_outputs</code><span class="sig-paren">(</span><em>*parents</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.merge_outputs" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.overwrite_output">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">overwrite_output</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.overwrite_output" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.get_args">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">get_args</code><span class="sig-paren">(</span><em>parent</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.get_args" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
<dl class="function">
|
||||
<dt id="ffmpeg.run">
|
||||
<code class="descclassname">ffmpeg.</code><code class="descname">run</code><span class="sig-paren">(</span><em>parent</em>, <em>cmd=’ffmpeg’</em><span class="sig-paren">)</span><a class="headerlink" href="#ffmpeg.run" title="Permalink to this definition">¶</a></dt>
|
||||
<dd></dd></dl>
|
||||
|
||||
</div>
|
||||
<div class="section" id="indices-and-tables">
|
||||
@ -179,7 +145,7 @@
|
||||
<div class="sphinxsidebarwrapper">
|
||||
<h3><a href="#">Table Of Contents</a></h3>
|
||||
<ul>
|
||||
<li><a class="reference internal" href="#">Welcome to ffmpeg-python’s documentation!</a></li>
|
||||
<li><a class="reference internal" href="#">ffmpeg-python: Python bindings for FFmpeg</a></li>
|
||||
<li><a class="reference internal" href="#indices-and-tables">Indices and tables</a></li>
|
||||
</ul>
|
||||
<div class="relations">
|
||||
|
||||
Binary file not shown.
@ -1 +1 @@
|
||||
Search.setIndex({docnames:["index"],envversion:52,filenames:["index.rst"],objects:{"":{ffmpeg:[0,0,0,"-"]},ffmpeg:{colorchannelmixer:[0,1,1,""],concat:[0,1,1,""],drawbox:[0,1,1,""],file_input:[0,1,1,""],file_output:[0,1,1,""],fluent:[0,1,1,""],get_args:[0,1,1,""],hflip:[0,1,1,""],hue:[0,1,1,""],merge_outputs:[0,1,1,""],overlay:[0,1,1,""],overwrite_output:[0,1,1,""],run:[0,1,1,""],setpts:[0,1,1,""],trim:[0,1,1,""],vflip:[0,1,1,""],zoompan:[0,1,1,""]}},objnames:{"0":["py","module","Python module"],"1":["py","function","Python function"]},objtypes:{"0":"py:module","1":"py:function"},terms:{"class":0,"function":0,arg:0,bind:0,cmd:0,color:0,colorchannelmix:0,concat:0,correspond:0,decor:0,drawbox:0,eof_act:0,expr:0,file:0,file_input:0,file_output:0,filenam:0,fixm:0,fluent:0,from:0,get_arg:0,height:0,hflip:0,hue:0,index:0,input:0,instanc:0,kwarg:0,list:0,main_par:0,make:0,merge_output:0,method:0,modul:0,node:0,node_class:0,none:0,option:0,overlai:0,overlay_par:0,overwrite_output:0,page:0,paramet:0,parent:0,repeat:0,run:0,search:0,set:0,setpt:0,show:0,specifi:0,thick:0,trim:0,vflip:0,width:0,zoompan:0},titles:["Welcome to ffmpeg-python\u2019s documentation!"],titleterms:{document:0,ffmpeg:0,indic:0,python:0,tabl:0,welcom:0}})
|
||||
Search.setIndex({docnames:["index"],envversion:52,filenames:["index.rst"],objects:{"":{ffmpeg:[0,0,0,"-"]},ffmpeg:{colorchannelmixer:[0,1,1,""],concat:[0,1,1,""],drawbox:[0,1,1,""],file_input:[0,1,1,""],file_output:[0,1,1,""],get_args:[0,1,1,""],hflip:[0,1,1,""],hue:[0,1,1,""],merge_outputs:[0,1,1,""],overlay:[0,1,1,""],overwrite_output:[0,1,1,""],run:[0,1,1,""],setpts:[0,1,1,""],trim:[0,1,1,""],vflip:[0,1,1,""],zoompan:[0,1,1,""]}},objnames:{"0":["py","module","Python module"],"1":["py","function","Python function"]},objtypes:{"0":"py:module","1":"py:function"},terms:{"class":[],"function":[],adjust:0,arg:0,base:[],bind:[],channel:0,cmd:0,color:0,colorchannelmix:0,concat:0,correspond:[],decor:[],drawbox:0,eof_act:0,expr:0,file:[],file_input:0,file_output:0,filenam:0,filter:0,filternod:[],fixm:[],fluent:[],frame:0,from:[],get_arg:0,globalnod:[],height:0,hflip:0,hue:0,index:0,input:0,inputnod:[],instanc:[],kwarg:0,list:[],main_par:0,make:[],merge_output:0,method:[],mix:0,modul:0,name:[],node:[],node_class:[],none:0,object:[],oper:[],option:[],outputnod:[],overlai:0,overlay_par:0,overwrite_output:0,page:0,paramet:[],parent:0,repeat:0,run:0,search:0,set:[],setpt:0,show:[],specifi:[],thick:0,trim:0,vflip:0,video:0,width:0,zoompan:0},titles:["ffmpeg-python: Python bindings for FFmpeg"],titleterms:{bind:0,document:[],ffmpeg:0,indic:0,python:0,tabl:0,welcom:[]}})
|
||||
239
ffmpeg/__init__.py
Executable file → Normal file
239
ffmpeg/__init__.py
Executable file → Normal file
@ -1,234 +1,5 @@
|
||||
#!./venv/bin/python
|
||||
|
||||
import hashlib
|
||||
import json
|
||||
import operator as _operator
|
||||
import subprocess
|
||||
|
||||
|
||||
class Node(object):
|
||||
def __init__(self, parents, name, *args, **kwargs):
|
||||
parent_hashes = [parent._hash for parent in parents]
|
||||
assert len(parent_hashes) == len(set(parent_hashes)), 'Same node cannot be included as parent multiple times'
|
||||
self._parents = parents
|
||||
self._name = name
|
||||
self._args = args
|
||||
self._kwargs = kwargs
|
||||
self._update_hash()
|
||||
|
||||
@classmethod
|
||||
def _add_operator(cls, func):
|
||||
setattr(cls, func.__name__, func)
|
||||
|
||||
def __repr__(self):
|
||||
formatted_props = ['{}'.format(arg) for arg in self._args]
|
||||
formatted_props += ['{}={!r}'.format(key, self._kwargs[key]) for key in sorted(self._kwargs)]
|
||||
return '{}({})'.format(self._name, ','.join(formatted_props))
|
||||
|
||||
def __eq__(self, other):
|
||||
return self._hash == other._hash
|
||||
|
||||
def _update_hash(self):
|
||||
props = {'args': self._args, 'kwargs': self._kwargs}
|
||||
my_hash = hashlib.md5(json.dumps(props, sort_keys=True)).hexdigest()
|
||||
parent_hashes = [parent._hash for parent in self._parents]
|
||||
hashes = parent_hashes + [my_hash]
|
||||
self._hash = hashlib.md5(','.join(hashes)).hexdigest()
|
||||
|
||||
|
||||
class InputNode(Node):
|
||||
def __init__(self, name, *args, **kwargs):
|
||||
super(InputNode, self).__init__(parents=[], name=name, *args, **kwargs)
|
||||
|
||||
|
||||
class FilterNode(Node):
|
||||
def _get_filter(self):
|
||||
params_text = self._name
|
||||
arg_params = ['{}'.format(arg) for arg in self._args]
|
||||
kwarg_params = ['{}={}'.format(k, self._kwargs[k]) for k in sorted(self._kwargs)]
|
||||
params = arg_params + kwarg_params
|
||||
if params:
|
||||
params_text += '={}'.format(':'.join(params))
|
||||
return params_text
|
||||
|
||||
|
||||
class OutputNode(Node):
|
||||
pass
|
||||
|
||||
|
||||
class GlobalNode(Node):
|
||||
def __init__(self, parent, name, *args, **kwargs):
|
||||
assert isinstance(parent, OutputNode), 'Global nodes can only be attached after output nodes'
|
||||
super(GlobalNode, self).__init__([parent], name, *args, **kwargs)
|
||||
|
||||
|
||||
def operator(node_classes={Node}):
|
||||
def decorator(func):
|
||||
[node_class._add_operator(func) for node_class in node_classes]
|
||||
return func
|
||||
return decorator
|
||||
|
||||
|
||||
def file_input(filename):
|
||||
return InputNode(file_input.__name__, filename=filename)
|
||||
|
||||
|
||||
@operator()
|
||||
def setpts(parent, expr):
|
||||
return FilterNode([parent], setpts.__name__, expr)
|
||||
|
||||
|
||||
@operator()
|
||||
def trim(parent, **kwargs):
|
||||
return FilterNode([parent], trim.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def overlay(main_parent, overlay_parent, eof_action='repeat', **kwargs):
|
||||
kwargs['eof_action'] = eof_action
|
||||
return FilterNode([main_parent, overlay_parent], overlay.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def hflip(parent):
|
||||
return FilterNode([parent], hflip.__name__)
|
||||
|
||||
|
||||
@operator()
|
||||
def vflip(parent):
|
||||
return FilterNode([parent], vflip.__name__)
|
||||
|
||||
|
||||
@operator()
|
||||
def drawbox(parent, x, y, width, height, color, thickness=None, **kwargs):
|
||||
if thickness:
|
||||
kwargs['t'] = thickness
|
||||
return FilterNode([parent], drawbox.__name__, x, y, width, height, color, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def concat(*parents, **kwargs):
|
||||
kwargs['n'] = len(parents)
|
||||
return FilterNode(parents, concat.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def zoompan(parent, **kwargs):
|
||||
return FilterNode([parent], zoompan.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def hue(parent, **kwargs):
|
||||
return FilterNode([parent], hue.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def colorchannelmixer(parent, *args, **kwargs):
|
||||
return FilterNode([parent], colorchannelmixer.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def overwrite_output(parent):
|
||||
return GlobalNode(parent, overwrite_output.__name__)
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode})
|
||||
def merge_outputs(*parents):
|
||||
return OutputNode(parents, merge_outputs.__name__)
|
||||
|
||||
|
||||
@operator(node_classes={InputNode, FilterNode})
|
||||
def file_output(parent, filename):
|
||||
return OutputNode([parent], file_output.__name__, filename=filename)
|
||||
|
||||
|
||||
def _get_stream_name(name):
|
||||
return '[{}]'.format(name)
|
||||
|
||||
|
||||
def _get_input_args(input_node):
|
||||
if input_node._name == file_input.__name__:
|
||||
args = ['-i', input_node._kwargs['filename']]
|
||||
else:
|
||||
assert False, 'Unsupported input node: {}'.format(input_node)
|
||||
return args
|
||||
|
||||
|
||||
def _topo_sort(start_node):
|
||||
marked_nodes = []
|
||||
sorted_nodes = []
|
||||
child_map = {}
|
||||
def visit(node, child):
|
||||
assert node not in marked_nodes, 'Graph is not a DAG'
|
||||
if child is not None:
|
||||
if node not in child_map:
|
||||
child_map[node] = []
|
||||
child_map[node].append(child)
|
||||
if node not in sorted_nodes:
|
||||
marked_nodes.append(node)
|
||||
[visit(parent, node) for parent in node._parents]
|
||||
marked_nodes.remove(node)
|
||||
sorted_nodes.append(node)
|
||||
unmarked_nodes = [start_node]
|
||||
while unmarked_nodes:
|
||||
visit(unmarked_nodes.pop(), None)
|
||||
return sorted_nodes, child_map
|
||||
|
||||
|
||||
def _get_filter_spec(i, node, stream_name_map):
|
||||
stream_name = _get_stream_name('v{}'.format(i))
|
||||
stream_name_map[node] = stream_name
|
||||
inputs = [stream_name_map[parent] for parent in node._parents]
|
||||
filter_spec = '{}{}{}'.format(''.join(inputs), node._get_filter(), stream_name)
|
||||
return filter_spec
|
||||
|
||||
|
||||
def _get_filter_arg(filter_nodes, stream_name_map):
|
||||
filter_specs = [_get_filter_spec(i, node, stream_name_map) for i, node in enumerate(filter_nodes)]
|
||||
return ';'.join(filter_specs)
|
||||
|
||||
|
||||
def _get_global_args(node):
|
||||
if node._name == overwrite_output.__name__:
|
||||
return ['-y']
|
||||
else:
|
||||
assert False, 'Unsupported global node: {}'.format(node)
|
||||
|
||||
|
||||
def _get_output_args(node, stream_name_map):
|
||||
args = []
|
||||
if node._name != merge_outputs.__name__:
|
||||
stream_name = stream_name_map[node._parents[0]]
|
||||
if stream_name != '[0]':
|
||||
args += ['-map', stream_name]
|
||||
if node._name == file_output.__name__:
|
||||
args += [node._kwargs['filename']]
|
||||
else:
|
||||
assert False, 'Unsupported output node: {}'.format(node)
|
||||
return args
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def get_args(parent):
|
||||
args = []
|
||||
# TODO: group nodes together, e.g. `-i somefile -r somerate`.
|
||||
sorted_nodes, child_map = _topo_sort(parent)
|
||||
input_nodes = [node for node in sorted_nodes if isinstance(node, InputNode)]
|
||||
output_nodes = [node for node in sorted_nodes if isinstance(node, OutputNode) and not
|
||||
isinstance(node, GlobalNode)]
|
||||
global_nodes = [node for node in sorted_nodes if isinstance(node, GlobalNode)]
|
||||
filter_nodes = [node for node in sorted_nodes if node not in (input_nodes + output_nodes + global_nodes)]
|
||||
stream_name_map = {node: _get_stream_name(i) for i, node in enumerate(input_nodes)}
|
||||
filter_arg = _get_filter_arg(filter_nodes, stream_name_map)
|
||||
args += reduce(_operator.add, [_get_input_args(node) for node in input_nodes])
|
||||
if filter_arg:
|
||||
args += ['-filter_complex', filter_arg]
|
||||
args += reduce(_operator.add, [_get_output_args(node, stream_name_map) for node in output_nodes])
|
||||
args += reduce(_operator.add, [_get_global_args(node) for node in global_nodes], [])
|
||||
return args
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def run(parent, cmd='ffmpeg'):
|
||||
args = [cmd] + parent.get_args()
|
||||
subprocess.check_call(args)
|
||||
from . import _filters, _ffmpeg, _run
|
||||
from ._filters import *
|
||||
from ._ffmpeg import *
|
||||
from ._run import *
|
||||
__all__ = _filters.__all__ + _ffmpeg.__all__ + _run.__all__
|
||||
|
||||
34
ffmpeg/_ffmpeg.py
Normal file
34
ffmpeg/_ffmpeg.py
Normal file
@ -0,0 +1,34 @@
|
||||
from .nodes import (
|
||||
FilterNode,
|
||||
GlobalNode,
|
||||
InputNode,
|
||||
operator,
|
||||
OutputNode,
|
||||
)
|
||||
|
||||
|
||||
def file_input(filename):
|
||||
return InputNode(file_input.__name__, filename=filename)
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def overwrite_output(parent):
|
||||
return GlobalNode(parent, overwrite_output.__name__)
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode})
|
||||
def merge_outputs(*parents):
|
||||
return OutputNode(parents, merge_outputs.__name__)
|
||||
|
||||
|
||||
@operator(node_classes={InputNode, FilterNode})
|
||||
def file_output(parent, filename):
|
||||
return OutputNode([parent], file_output.__name__, filename=filename)
|
||||
|
||||
|
||||
__all__ = [
|
||||
'file_input',
|
||||
'file_output',
|
||||
'merge_outputs',
|
||||
'overwrite_output',
|
||||
]
|
||||
79
ffmpeg/_filters.py
Normal file
79
ffmpeg/_filters.py
Normal file
@ -0,0 +1,79 @@
|
||||
from .nodes import (
|
||||
FilterNode,
|
||||
operator,
|
||||
)
|
||||
|
||||
|
||||
@operator()
|
||||
def setpts(parent, expr):
|
||||
return FilterNode([parent], setpts.__name__, expr)
|
||||
|
||||
|
||||
@operator()
|
||||
def trim(parent, **kwargs):
|
||||
return FilterNode([parent], trim.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def overlay(main_parent, overlay_parent, eof_action='repeat', **kwargs):
|
||||
kwargs['eof_action'] = eof_action
|
||||
return FilterNode([main_parent, overlay_parent], overlay.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def hflip(parent):
|
||||
return FilterNode([parent], hflip.__name__)
|
||||
|
||||
|
||||
@operator()
|
||||
def vflip(parent):
|
||||
return FilterNode([parent], vflip.__name__)
|
||||
|
||||
|
||||
@operator()
|
||||
def drawbox(parent, x, y, width, height, color, thickness=None, **kwargs):
|
||||
if thickness:
|
||||
kwargs['t'] = thickness
|
||||
return FilterNode([parent], drawbox.__name__, x, y, width, height, color, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def concat(*parents, **kwargs):
|
||||
kwargs['n'] = len(parents)
|
||||
return FilterNode(parents, concat.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def zoompan(parent, **kwargs):
|
||||
return FilterNode([parent], zoompan.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def hue(parent, **kwargs):
|
||||
return FilterNode([parent], hue.__name__, **kwargs)
|
||||
|
||||
|
||||
@operator()
|
||||
def colorchannelmixer(parent, *args, **kwargs):
|
||||
"""Adjust video input frames by re-mixing color channels.
|
||||
|
||||
`FFmpeg colorchannelmixer filter`_
|
||||
|
||||
.. _FFmpeg colorchannelmixer filter:
|
||||
https://ffmpeg.org/ffmpeg-filters.html#toc-colorchannelmixer
|
||||
"""
|
||||
return FilterNode([parent], colorchannelmixer.__name__, **kwargs)
|
||||
|
||||
|
||||
__all__ = [
|
||||
'colorchannelmixer',
|
||||
'concat',
|
||||
'drawbox',
|
||||
'hflip',
|
||||
'hue',
|
||||
'overlay',
|
||||
'setpts',
|
||||
'trim',
|
||||
'vflip',
|
||||
'zoompan',
|
||||
]
|
||||
112
ffmpeg/_run.py
Normal file
112
ffmpeg/_run.py
Normal file
@ -0,0 +1,112 @@
|
||||
import operator as _operator
|
||||
import subprocess as _subprocess
|
||||
|
||||
from ._ffmpeg import (
|
||||
file_input,
|
||||
file_output,
|
||||
merge_outputs,
|
||||
overwrite_output,
|
||||
)
|
||||
from .nodes import (
|
||||
GlobalNode,
|
||||
InputNode,
|
||||
operator,
|
||||
OutputNode,
|
||||
)
|
||||
|
||||
def _get_stream_name(name):
|
||||
return '[{}]'.format(name)
|
||||
|
||||
|
||||
def _get_input_args(input_node):
|
||||
if input_node._name == file_input.__name__:
|
||||
args = ['-i', input_node._kwargs['filename']]
|
||||
else:
|
||||
assert False, 'Unsupported input node: {}'.format(input_node)
|
||||
return args
|
||||
|
||||
|
||||
def _topo_sort(start_node):
|
||||
marked_nodes = []
|
||||
sorted_nodes = []
|
||||
child_map = {}
|
||||
def visit(node, child):
|
||||
assert node not in marked_nodes, 'Graph is not a DAG'
|
||||
if child is not None:
|
||||
if node not in child_map:
|
||||
child_map[node] = []
|
||||
child_map[node].append(child)
|
||||
if node not in sorted_nodes:
|
||||
marked_nodes.append(node)
|
||||
[visit(parent, node) for parent in node._parents]
|
||||
marked_nodes.remove(node)
|
||||
sorted_nodes.append(node)
|
||||
unmarked_nodes = [start_node]
|
||||
while unmarked_nodes:
|
||||
visit(unmarked_nodes.pop(), None)
|
||||
return sorted_nodes, child_map
|
||||
|
||||
|
||||
def _get_filter_spec(i, node, stream_name_map):
|
||||
stream_name = _get_stream_name('v{}'.format(i))
|
||||
stream_name_map[node] = stream_name
|
||||
inputs = [stream_name_map[parent] for parent in node._parents]
|
||||
filter_spec = '{}{}{}'.format(''.join(inputs), node._get_filter(), stream_name)
|
||||
return filter_spec
|
||||
|
||||
|
||||
def _get_filter_arg(filter_nodes, stream_name_map):
|
||||
filter_specs = [_get_filter_spec(i, node, stream_name_map) for i, node in enumerate(filter_nodes)]
|
||||
return ';'.join(filter_specs)
|
||||
|
||||
|
||||
def _get_global_args(node):
|
||||
if node._name == overwrite_output.__name__:
|
||||
return ['-y']
|
||||
else:
|
||||
assert False, 'Unsupported global node: {}'.format(node)
|
||||
|
||||
|
||||
def _get_output_args(node, stream_name_map):
|
||||
args = []
|
||||
if node._name != merge_outputs.__name__:
|
||||
stream_name = stream_name_map[node._parents[0]]
|
||||
if stream_name != '[0]':
|
||||
args += ['-map', stream_name]
|
||||
if node._name == file_output.__name__:
|
||||
args += [node._kwargs['filename']]
|
||||
else:
|
||||
assert False, 'Unsupported output node: {}'.format(node)
|
||||
return args
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def get_args(parent):
|
||||
args = []
|
||||
# TODO: group nodes together, e.g. `-i somefile -r somerate`.
|
||||
sorted_nodes, child_map = _topo_sort(parent)
|
||||
input_nodes = [node for node in sorted_nodes if isinstance(node, InputNode)]
|
||||
output_nodes = [node for node in sorted_nodes if isinstance(node, OutputNode) and not
|
||||
isinstance(node, GlobalNode)]
|
||||
global_nodes = [node for node in sorted_nodes if isinstance(node, GlobalNode)]
|
||||
filter_nodes = [node for node in sorted_nodes if node not in (input_nodes + output_nodes + global_nodes)]
|
||||
stream_name_map = {node: _get_stream_name(i) for i, node in enumerate(input_nodes)}
|
||||
filter_arg = _get_filter_arg(filter_nodes, stream_name_map)
|
||||
args += reduce(_operator.add, [_get_input_args(node) for node in input_nodes])
|
||||
if filter_arg:
|
||||
args += ['-filter_complex', filter_arg]
|
||||
args += reduce(_operator.add, [_get_output_args(node, stream_name_map) for node in output_nodes])
|
||||
args += reduce(_operator.add, [_get_global_args(node) for node in global_nodes], [])
|
||||
return args
|
||||
|
||||
|
||||
@operator(node_classes={OutputNode, GlobalNode})
|
||||
def run(parent, cmd='ffmpeg'):
|
||||
args = [cmd] + parent.get_args()
|
||||
_subprocess.check_call(args)
|
||||
|
||||
|
||||
__all__ = [
|
||||
'get_args',
|
||||
'run',
|
||||
]
|
||||
66
ffmpeg/nodes.py
Normal file
66
ffmpeg/nodes.py
Normal file
@ -0,0 +1,66 @@
|
||||
import hashlib
|
||||
import json
|
||||
|
||||
|
||||
class Node(object):
|
||||
"""Node base"""
|
||||
def __init__(self, parents, name, *args, **kwargs):
|
||||
parent_hashes = [parent._hash for parent in parents]
|
||||
assert len(parent_hashes) == len(set(parent_hashes)), 'Same node cannot be included as parent multiple times'
|
||||
self._parents = parents
|
||||
self._name = name
|
||||
self._args = args
|
||||
self._kwargs = kwargs
|
||||
self._update_hash()
|
||||
|
||||
def __repr__(self):
|
||||
formatted_props = ['{}'.format(arg) for arg in self._args]
|
||||
formatted_props += ['{}={!r}'.format(key, self._kwargs[key]) for key in sorted(self._kwargs)]
|
||||
return '{}({})'.format(self._name, ','.join(formatted_props))
|
||||
|
||||
def __eq__(self, other):
|
||||
return self._hash == other._hash
|
||||
|
||||
def _update_hash(self):
|
||||
props = {'args': self._args, 'kwargs': self._kwargs}
|
||||
my_hash = hashlib.md5(json.dumps(props, sort_keys=True)).hexdigest()
|
||||
parent_hashes = [parent._hash for parent in self._parents]
|
||||
hashes = parent_hashes + [my_hash]
|
||||
self._hash = hashlib.md5(','.join(hashes)).hexdigest()
|
||||
|
||||
|
||||
class InputNode(Node):
|
||||
"""InputNode type"""
|
||||
def __init__(self, name, *args, **kwargs):
|
||||
super(InputNode, self).__init__(parents=[], name=name, *args, **kwargs)
|
||||
|
||||
|
||||
class FilterNode(Node):
|
||||
"""FilterNode"""
|
||||
def _get_filter(self):
|
||||
params_text = self._name
|
||||
arg_params = ['{}'.format(arg) for arg in self._args]
|
||||
kwarg_params = ['{}={}'.format(k, self._kwargs[k]) for k in sorted(self._kwargs)]
|
||||
params = arg_params + kwarg_params
|
||||
if params:
|
||||
params_text += '={}'.format(':'.join(params))
|
||||
return params_text
|
||||
|
||||
|
||||
class OutputNode(Node):
|
||||
"""OutputNode"""
|
||||
pass
|
||||
|
||||
|
||||
class GlobalNode(Node):
|
||||
def __init__(self, parent, name, *args, **kwargs):
|
||||
assert isinstance(parent, OutputNode), 'Global nodes can only be attached after output nodes'
|
||||
super(GlobalNode, self).__init__([parent], name, *args, **kwargs)
|
||||
|
||||
|
||||
def operator(node_classes={Node}, name=None):
|
||||
def decorator(func):
|
||||
func_name = name or func.__name__
|
||||
[setattr(node_class, func_name, func) for node_class in node_classes]
|
||||
return func
|
||||
return decorator
|
||||
@ -1,4 +1,3 @@
|
||||
|
||||
import ffmpeg
|
||||
import os
|
||||
import subprocess
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user