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], "") - 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], " [output_file_name]")