1 files changed, 7 insertions, 2 deletions
diff --git a/TOOLS/matroska.py b/TOOLS/matroska.py
index d75c45a715..e715c229df 100755
--- a/TOOLS/matroska.py
+++ b/TOOLS/matroska.py
@@ -459,10 +459,15 @@ if __name__ == "__main__":
     def parse_toplevel(s):
         parse_one(s, 0, None, 1 << 63)
 
+    if len(sys.argv) < 3:
+        outfile = sys.stdout
+    else:
+        outfile = open(sys.argv[2], "w")
+
     if sys.argv[1] == '--generate-header':
-        generate_C_header(sys.stdout)
+        generate_C_header(outfile)
     elif sys.argv[1] == '--generate-definitions':
-        generate_C_definitions(sys.stdout)
+        generate_C_definitions(outfile)
     else:
         s = open(sys.argv[1], "rb")
         while 1: