Display simulation time data

2 files changed, 23 insertions, 1 deletions

diff --git a/kernel/fstdata.cc b/kernel/fstdata.cc
index 5d6d85ed8..a7a2c80f7 100644
--- a/kernel/fstdata.cc
+++ b/kernel/fstdata.cc
@@ -21,10 +21,30 @@
 
 USING_YOSYS_NAMESPACE
 
+
 FstData::FstData(std::string filename) : ctx(nullptr)
 {
+	const std::vector<std::string> g_units = { "s", "ms", "us", "ns", "ps", "fs", "as", "zs" };
 	ctx = (fstReaderContext *)fstReaderOpen(filename.c_str());
-	timescale = pow(10.0, (int)fstReaderGetTimescale(ctx));
+	int scale = (int)fstReaderGetTimescale(ctx);	
+	timescale = pow(10.0, scale);
+	timescale_str = "";
+	int unit = 0;
+	int zeros = 0;
+	if (scale > 0)  {
+		zeros = scale;
+	} else {
+		if ((scale % 3) == 0) {
+			zeros = (-scale % 3);
+			unit = (-scale / 3);
+		} else {
+			zeros = 3 - (-scale % 3);
+			unit = (-scale / 3) + 1;
+		}
+	}
+	for (int i=0;i<zeros; i++) timescale_str += "0";
+	if (zeros>0)timescale_str += " ";
+	timescale_str += g_units[unit];
 	extractVarNames();
 }
 
diff --git a/kernel/fstdata.h b/kernel/fstdata.h
index 8095981a2..c069ff5e5 100644
--- a/kernel/fstdata.h
+++ b/kernel/fstdata.h
@@ -55,6 +55,7 @@ class FstData
 	std::string valueAt(fstHandle signal, uint64_t time);
 	fstHandle getHandle(std::string name);
 	double getTimescale() { return timescale; }
+	const char *getTimescaleString() { return timescale_str.c_str(); }
 private:
 	void extractVarNames();
 
@@ -69,6 +70,7 @@ private:
 	std::vector<uint64_t> sample_times;
 	size_t sample_times_ndx;
 	double timescale;
+	std::string timescale_str;
 	uint64_t start_time;
 	uint64_t end_time;
 	std::vector<uint64_t> edges;