From 3cb952352ff0c937c1da3bf1d80142e0d8ee1e46 Mon Sep 17 00:00:00 2001
From: Davide Depau <davide@depau.eu>
Date: Thu, 25 May 2017 17:17:55 +0200
Subject: [PATCH] add more filters, options

* add filters: vflip, zoompan, hue, colorchannelmixer
* add options for trim, overlay
* filter code made more generic and extendable
---
 ffmpeg/__init__.py | 106 ++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 95 insertions(+), 11 deletions(-)

diff --git a/ffmpeg/__init__.py b/ffmpeg/__init__.py
index 9e8262f..4e40606 100755
--- a/ffmpeg/__init__.py
+++ b/ffmpeg/__init__.py
@@ -74,29 +74,60 @@ class _FilterNode(_Node):
     def _get_filter(self):
         raise NotImplementedError()
 
+    def _get_params_from_dict(self, d):
+        params = ""
+        for k in self.kwargs:
+            params += k + "={}:".format(self.kwargs[k])
+        if len(params) > 0:
+            params = params[:-1]
+        return params
+
+    def _get_params_from_list(self, l):
+        return ":".join(["{}".format(i) for i in l])
+
+    def _get_filter_from_dict(self, d):
+        p = self._get_params_from_dict(d)
+        if len(p) > 0:
+            return self.NAME + "=" + p
+        return self.NAME
+
+
 
 class _TrimNode(_FilterNode):
     NAME = 'trim'
 
-    def __init__(self, parent, start_frame, end_frame, setpts='PTS-STARTPTS'):
+    def __init__(self, parent, **kwargs):
         super(_TrimNode, self).__init__([parent])
-        self.start_frame = start_frame
-        self.end_frame = end_frame
-        self.setpts = setpts
+        # if "setpts" not in kwargs:
+        #     kwargs["setpts"] = 'PTS-STARTPTS'
+        self.kwargs = kwargs
 
     def _get_filter(self):
-        return 'trim=start_frame={}:end_frame={},setpts={}'.format(self.start_frame, self.end_frame, self.setpts)
+        params = ""
+        for k in self.kwargs:
+            if k == "setpts":
+                continue
+            params += k
+            params += "={}:".format(self.kwargs[k])
+        if len(params) > 0:
+            params = params[:-1]
+
+        if "setpts" in self.kwargs:
+            params += "setpts={}".format(self.kwargs["setpts"])
+
+        return self.NAME + '=' + params
 
 
 class _OverlayNode(_FilterNode):
     NAME = 'overlay'
 
-    def __init__(self, main_parent, overlay_parent, eof_action='repeat'):
+    def __init__(self, main_parent, overlay_parent, **kwargs):
         super(_OverlayNode, self).__init__([main_parent, overlay_parent])
         self.eof_action = eof_action
+        self.kwargs = kwargs
 
     def _get_filter(self):
-        return 'overlay=eof_action={}'.format(self.eof_action)
+        return self._get_filter_from_dict(self.kwargs)
 
 
 class _HFlipNode(_FilterNode):
@@ -106,23 +137,38 @@ class _HFlipNode(_FilterNode):
         super(_HFlipNode, self).__init__([parent])
 
     def _get_filter(self):
-        return 'hflip'
+        return self.NAME
+
+
+class _VFlipNode(_FilterNode):
+    NAME = 'vflip'
+
+    def __init__(self, parent):
+        super(_VFlipNode, self).__init__([parent])
+
+    def _get_filter(self):
+        return self.NAME
+
 
 
 class _DrawBoxNode(_FilterNode):
     NAME = 'drawbox'
 
-    def __init__(self, parent, x, y, width, height, color, thickness=1):
+    def __init__(self, parent, x, y, width, height, color, **kwargs):
         super(_DrawBoxNode, self).__init__([parent])
         self.x = x
         self.y = y
         self.width = width
         self.height = height
         self.color = color
-        self.thickness = thickness
+        self.kwargs = kwargs
 
     def _get_filter(self):
-        return 'drawbox={}:{}:{}:{}:{}:t={}'.format(self.x, self.y, self.width, self.height, self.color, self.thickness)
+        f = 'drawbox={}:{}:{}:{}:{}'.format(self.x, self.y, self.width, self.height, self.color)
+        p = self._get_params_from_dict(self.kwargs)
+        if len(p) > 0:
+            return f + ":" + p
+        return f
 
 
 class _ConcatNode(_Node):
@@ -135,7 +181,45 @@ class _ConcatNode(_Node):
     def _get_filter(self):
         return 'concat=n={}'.format(len(self.parents))
 
+class _ZoomPanNode(_FilterNode):
+    NAME = 'zoompan'
 
+    def __init__(self, parent, **kwargs):
+        super(_ZoomPanNode, self).__init__([parent])
+        self.kwargs = kwargs
+
+    def _get_filter(self):
+        return self._get_filter_from_dict(self.kwargs)
+
+class _HueNode(_FilterNode):
+    NAME = 'hue'
+
+    def __init__(self, parent, **kwargs):
+        super(_HueNode, self).__init__([parent])
+        self.kwargs = kwargs
+
+    def _get_filter(self):
+        return self._get_filter_from_dict(self.kwargs)
+
+class _ColorChannelMixerNode(_FilterNode):
+    NAME = 'colorchannelmixer'
+
+    def __init__(self, parent, *args, **kwargs):
+        super(_ColorChannelMixerNode, self).__init__([parent])
+        self.args = args
+        self.kwargs = kwargs
+
+    def _get_filter(self):
+        f = self.NAME + "="
+        if self.args:
+            f += self._get_params_from_list(self.args)
+            if self.kwargs:
+                f += ":"
+        if self.kwargs:
+            f += self._get_params_from_dict(self.kwargs)
+        return f
+
+    
 class _OutputNode(_Node):
     @classmethod
     def _get_stream_name(cls, name):