From 217bd2bde6fdfd9e29f9294aa2968980c7a1c29b Mon Sep 17 00:00:00 2001
From: Karl Kroening <karlk@kralnet.us>
Date: Fri, 13 Jul 2018 04:28:21 +0200
Subject: [PATCH] Fix `-map` to not use brackets for passthroughs; fixes #102,
 #23

---
 ffmpeg/_run.py              | 14 +++++++++----
 ffmpeg/tests/test_ffmpeg.py | 41 +++++++++++++++++++++++++++++++++----
 2 files changed, 47 insertions(+), 8 deletions(-)

diff --git a/ffmpeg/_run.py b/ffmpeg/_run.py
index cbc0676..87e44ff 100644
--- a/ffmpeg/_run.py
+++ b/ffmpeg/_run.py
@@ -57,13 +57,19 @@ def _get_input_args(input_node):
     return args
 
 
-def _format_input_stream_name(stream_name_map, edge):
+def _format_input_stream_name(stream_name_map, edge, is_final_arg=False):
     prefix = stream_name_map[edge.upstream_node, edge.upstream_label]
     if not edge.upstream_selector:
         suffix = ''
     else:
         suffix = ':{}'.format(edge.upstream_selector)
-    return '[{}{}]'.format(prefix, suffix)
+    if is_final_arg and isinstance(edge.upstream_node, InputNode):
+        ## Special case: `-map` args should not have brackets for input
+        ## nodes.
+        fmt = '{}{}'
+    else:
+        fmt = '[{}{}]'
+    return fmt.format(prefix, suffix)
 
 
 def _format_output_stream_name(stream_name_map, edge):
@@ -113,8 +119,8 @@ def _get_output_args(node, stream_name_map):
 
     for edge in node.incoming_edges:
         # edge = node.incoming_edges[0]
-        stream_name = _format_input_stream_name(stream_name_map, edge)
-        if stream_name != '[0]' or len(node.incoming_edges) > 1:
+        stream_name = _format_input_stream_name(stream_name_map, edge, is_final_arg=True)
+        if stream_name != '0' or len(node.incoming_edges) > 1:
             args += ['-map', stream_name]
 
     kwargs = copy.copy(node.kwargs)
diff --git a/ffmpeg/tests/test_ffmpeg.py b/ffmpeg/tests/test_ffmpeg.py
index fd11cc7..04e363e 100644
--- a/ffmpeg/tests/test_ffmpeg.py
+++ b/ffmpeg/tests/test_ffmpeg.py
@@ -165,8 +165,8 @@ def test_combined_output():
     assert out.get_args() == [
         '-i', TEST_INPUT_FILE1,
         '-i', TEST_OVERLAY_FILE,
-        '-map', '[0]',
-        '-map', '[1]',
+        '-map', '0',
+        '-map', '1',
         TEST_OUTPUT_FILE1
     ]
 
@@ -186,6 +186,7 @@ def test_filter_with_selector():
     ]
 
 
+
 def test_get_item_with_bad_selectors():
     input = ffmpeg.input(TEST_INPUT_FILE1)
 
@@ -532,18 +533,50 @@ def test_multi_passthrough():
         '-i', 'in1.mp4',
         '-i', 'in2.mp4',
         'out1.mp4',
-        '-map', '[1]',  # FIXME: this should not be here (see #23)
+        '-map', '1',
         'out2.mp4'
     ]
     assert ffmpeg.get_args([out1, out2]) == [
         '-i', 'in2.mp4',
         '-i', 'in1.mp4',
         'out2.mp4',
-        '-map', '[1]',  # FIXME: this should not be here (see #23)
+        '-map', '1',
         'out1.mp4'
     ]
 
 
+def test_passthrough_selectors():
+    i1 = ffmpeg.input(TEST_INPUT_FILE1)
+    args = (
+        ffmpeg
+        .output(i1['1'], i1['2'], TEST_OUTPUT_FILE1)
+        .get_args()
+    )
+    assert args == [
+        '-i', TEST_INPUT_FILE1,
+        '-map', '0:1',
+        '-map', '0:2',
+        TEST_OUTPUT_FILE1,
+    ]
+
+
+def test_mixed_passthrough_selectors():
+    i1 = ffmpeg.input(TEST_INPUT_FILE1)
+    args = (
+        ffmpeg
+        .output(i1['1'].hflip(), i1['2'], TEST_OUTPUT_FILE1)
+        .get_args()
+    )
+    assert args == [
+        '-i', TEST_INPUT_FILE1,
+        '-filter_complex',
+        '[0:1]hflip[s0]',
+        '-map', '[s0]',
+        '-map', '0:2',
+        TEST_OUTPUT_FILE1,
+    ]
+
+
 def test_pipe():
     width = 32
     height = 32