diff options
Diffstat (limited to 'src/mem')
| -rw-r--r-- | src/mem/CommMonitor.py | 7 | ||||
| -rw-r--r-- | src/mem/comm_monitor.cc | 27 |
2 files changed, 29 insertions, 5 deletions
diff --git a/src/mem/CommMonitor.py b/src/mem/CommMonitor.py index 4e8cfe270..ba871357d 100644 --- a/src/mem/CommMonitor.py +++ b/src/mem/CommMonitor.py @@ -53,6 +53,13 @@ class CommMonitor(MemObject): master = MasterPort("Master port") slave = SlavePort("Slave port") + # Boolean to enable or disable the trace. Writes to an a file named based on + # SimObject hierarchy. + trace_enable = Param.Bool(False, "Enable trace capture") + + # Boolean to compress the trace or not. + trace_compress = Param.Bool(True, "Enable trace compression") + # packet trace output file, disabled by default trace_file = Param.String("", "Packet trace output file") diff --git a/src/mem/comm_monitor.cc b/src/mem/comm_monitor.cc index 70b8a6451..3bfaad289 100644 --- a/src/mem/comm_monitor.cc +++ b/src/mem/comm_monitor.cc @@ -58,11 +58,28 @@ CommMonitor::CommMonitor(Params* params) traceStream(NULL), system(params->system) { - // If we are using a trace file, then open the file, - if (params->trace_file != "") { - // If the trace file is not specified as an absolute path, - // append the current simulation output directory - std::string filename = simout.resolve(params->trace_file); + // If we are using a trace file, then open the file + if (params->trace_enable) { + std::string filename; + if (params->trace_file != "") { + // If the trace file is not specified as an absolute path, + // append the current simulation output directory + filename = simout.resolve(params->trace_file); + + std::string suffix = ".gz"; + // If trace_compress has been set, check the suffix. Append + // accordingly. + if (params->trace_compress && + filename.compare(filename.size() - suffix.size(), suffix.size(), + suffix) != 0) + filename = filename + suffix; + } else { + // Generate a filename from the name of the SimObject. Append .trc + // and .gz if we want compression enabled. + filename = simout.resolve(name() + ".trc" + + (params->trace_compress ? ".gz" : "")); + } + traceStream = new ProtoOutputStream(filename); // Create a protobuf message for the header and write it to |