Handle stdout with mkv output

Signed-off-by: Maxime Gervais <[email protected]>
MediaArea · Dec 4, 2023 · 7345af4 · 7345af4
1 parent 2c39ef3
commit 7345af4
Showing 1 changed file with 19 additions and 6 deletions.
diff --git a/Source/Common/DecklinkWrapper.cpp b/Source/Common/DecklinkWrapper.cpp
@@ -9,6 +9,7 @@
 #include "Common/Merge.h"
 
 #include <DeckLinkAPIDispatch.cpp>
+#include <iostream>
 #include <ctime>
 
 using namespace std;
@@ -594,8 +595,13 @@ void DecklinkWrapper::CreateCaptureSession(FileWrapper* Wrapper_)
         uint32_t Den = DeckLinkVideoMode == bmdModeNTSC ? 1001 : 1;
 
         output Output;
-        Output.Output = new ofstream(OutputFile, ios_base::binary | ios_base::trunc);
-        Output.Writer = new matroska_writer(Output.Output, 720, Lines, Num, Den, DeckLinkTimecodeFormat != (uint32_t)-1);
+        if (OutputFile == "-")
+          Output.Writer = new matroska_writer(&cout, 720, Lines, Num, Den, DeckLinkTimecodeFormat != (uint32_t)-1);
+        else
+        {
+            Output.Output = new ofstream(OutputFile, ios_base::binary | ios_base::trunc);
+            Output.Writer = new matroska_writer(Output.Output, 720, Lines, Num, Den, DeckLinkTimecodeFormat != (uint32_t)-1);
+        }
         Outputs.push_back(Output);
     }
 
@@ -667,10 +673,17 @@ void DecklinkWrapper::StopCaptureSession()
 
     for (output Output: Outputs)
     {
-        Output.Writer->close(Output.Output);
-        Output.Output->close();
-        delete Output.Writer;
-        delete Output.Output;
+        if (Output.Writer)
+        {
+             Output.Writer->close(Output.Output);
+             delete Output.Writer;
+        }
+
+        if (Output.Output)
+        {
+            Output.Output->close();
+            delete Output.Output;
+        }
     }
     Outputs.clear();