cleanup

profiler_vis/report2tracing.py

@@ -2,6 +2,26 @@ import json
 import csv
 import sys
 
+class FancyFloat(float):
+    def __repr__(self):
+        return format(Decimal(self), "f")
+
+class JsonRpcEncoder(json.JSONEncoder):
+    def decimalize(self, val):
+        if isinstance(val, dict):
+            return {k:self.decimalize(v) for k,v in val.items()}
+
+        if isinstance(val, (list, tuple)):
+            return type(val)(self.decimalize(v) for v in val)
+
+        if isinstance(val, float):
+            return FancyFloat(val)
+
+        return val
+
+    def encode(self, val):
+        return super().encode(self.decimalize(val))
+
 if len(sys.argv) == 1:
     print("No file was provided")
 else:
@@ -9,24 +29,35 @@ else:
     call_stack = []
     with open(sys.argv[1], "r") as in_file:
         csv_reader = csv.reader(in_file, delimiter=',')
+        pf = 1
+        first_line = True
         for line in csv_reader:
+            if first_line:
+                pf = float(line[0]) / 1000
+                first_line = False
+                continue
             if line[0] == "->":
                 call_stack.append(line)
             elif line[0] == "<-":
                 call = call_stack.pop()
-                trace_events.append({
+                dict = {
                     "pid": 1,
                     "tid": 1,
-                    "ts" : int(call[1]),
-                    "dur": int(line[1])-int(call[1]),
+                    "ts" : float(call[1]),
+                    "dur": (float(line[1])-float(call[1])),
                     "ph" : "X",
                     "name": call[2],
                     "args": {
-                        "file": f"({call[3]}:{call[4]})"
+                        "file":  f"({call[3]}:{call[4]})",
                     }
-                })
+                }
+                if call[5]:
+                    dict["args"]["info1"] = call[5]
+                if call[6]:
+                    dict["args"]["info2"] = call[6]
+                trace_events.append(dict)
             else:
                 print("invalid syntax")
                 break
     with open("out.json", "w") as out_file:
-        out_file.write(json.dumps({"traceEvents": trace_events}, indent=4))
+        out_file.write(json.dumps({"traceEvents": trace_events}))