From 8123f5d698b50aedcb8ad3a8b4a7cbb0a77279f4 Mon Sep 17 00:00:00 2001
From: Lukas Steiner <lsteiner@rhrk.uni-kl.de>
Date: Mon, 16 Aug 2021 11:03:26 +0200
Subject: [PATCH] Add output file name to CLI.

---
 DRAMSys/traceAnalyzer/scripts/vcdExport.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/DRAMSys/traceAnalyzer/scripts/vcdExport.py b/DRAMSys/traceAnalyzer/scripts/vcdExport.py
index d7484aa8..70d03d32 100755
--- a/DRAMSys/traceAnalyzer/scripts/vcdExport.py
+++ b/DRAMSys/traceAnalyzer/scripts/vcdExport.py
@@ -275,8 +275,12 @@ def dumpVcd(pathToTrace):
         return f.read()
 
 if __name__ == "__main__":
-    if len(sys.argv) < 2:
-        print("Usage:", sys.argv[0], "<tracefile>")
-    else:
+    if len(sys.argv) == 2:
         dump = dumpVcd(sys.argv[1])
         print(dump)
+    elif len(sys.argv) == 3:
+        dump = dumpVcd(sys.argv[1])
+        with open(sys.argv[2], 'x') as outputFile:
+            outputFile.write(dump)
+    else:
+        print("Usage: ", sys.argv[0], "<trace_file> [output_file_name]")