viztracer 1.1.0__cp314-cp314-win_amd64.whl

viztracer/functree.py

@@ -0,0 +1,135 @@
+# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+# For details: https://github.com/gaogaotiantian/viztracer/blob/master/NOTICE.txt
+
+import bisect
+import copy
+import re
+from typing import Any, Generator
+
+
+class FuncTreeNode:
+    name_regex = r"(.*) \((.*?):([0-9]+)\)"
+
+    def __init__(self, event: dict[str, Any] | None = None) -> None:
+        self.filename: str | None = None
+        self.lineno: int | None = None
+        self.is_python: bool | None = False
+        self.funcname: str | None = None
+        self.parent: FuncTreeNode | None = None
+        self.children: list[FuncTreeNode] = []
+        self.start: float = - (2 ** 64)
+        self.end: float = 2 ** 64
+        self.event: dict[str, Any] = {}
+        if event is None:
+            self.event = {"name": "__ROOT__"}
+            self.fullname = "__ROOT__"
+        else:
+            self.event = copy.copy(event)
+            self.start = self.event["ts"]
+            self.end = self.event["ts"] + self.event["dur"]
+            self.fullname = self.event["name"]
+            m = re.match(self.name_regex, self.fullname)
+            if m:
+                self.is_python = True
+                self.funcname = m.group(1)
+                self.filename = m.group(2)
+                self.lineno = int(m.group(3))
+
+    def is_ancestor(self, other: "FuncTreeNode") -> bool:
+        return self.start < other.start and self.end > other.end
+
+    def is_same(self, other: "FuncTreeNode") -> bool:
+        return (self.fullname == other.fullname
+                and len(self.children) == len(other.children)
+                and all(t[0].is_same(t[1]) for t in zip(self.children, other.children)))
+
+    def adopt(self, other: "FuncTreeNode") -> None:
+        new_children = []
+        if self.is_ancestor(other):
+            # Build a list is slow
+            # In almost all cases, end_idx should be the last, because that's
+            # how we record entries.
+            # In many cases, if two entries are siblings, start_idx is the
+            # last too.
+            # Try to skip building the list by checking these common situations
+            # first.
+            if not self.children:
+                # if it's empty, then both indexes are 0
+                start_idx = end_idx = 0
+            else:
+                if other.start > self.children[-1].start:
+                    start_idx = len(self.children)
+                elif other.start < self.children[0].start:
+                    start_idx = 0
+                else:
+                    start_array = [n.start for n in self.children]
+                    start_idx = bisect.bisect(start_array, other.start)
+                if other.end > self.children[-1].end:
+                    end_idx = len(self.children)
+                else:
+                    end_array = [n.end for n in self.children]
+                    end_idx = bisect.bisect(end_array, other.end)
+            if (start_idx == end_idx + 1):
+                self.children[end_idx].adopt(other)
+            elif (start_idx == end_idx):
+                other.parent = self
+                self.children.insert(start_idx, other)
+            elif (start_idx < end_idx):
+                def change_parent(node):
+                    node.parent = other
+                new_children = self.children[start_idx:end_idx]
+                # force map to run
+                list(map(change_parent, new_children))
+                other.children = new_children
+                other.parent = self
+                self.children = self.children[:start_idx] + [other] + self.children[end_idx:]
+            else:  # pragma: no cover
+                raise Exception("This should not be possible")
+        elif self.parent is not None:
+            self.parent.adopt(other)
+        else:  # pragma: no cover
+            raise Exception("This should not be possible")
+
+
+class FuncTree:  # pragma: no cover
+    def __init__(self, pid: int = 0, tid: int = 0) -> None:
+        self.root: FuncTreeNode = FuncTreeNode()
+        self.curr: FuncTreeNode = self.root
+        self.pid: int = pid
+        self.tid: int = tid
+
+    def is_same(self, other: "FuncTree") -> bool:
+        return self.root.is_same(other.root)
+
+    def add_event(self, event: dict[str, Any]) -> None:
+        node = FuncTreeNode(event)
+
+        self.curr.adopt(node)
+        self.curr = node
+
+    def first_ts(self) -> float:
+        return self.root.children[0].event["ts"]
+
+    def first_node(self) -> FuncTreeNode:
+        return self.root.children[0]
+
+    def node_by_timestamp(self, ts: float) -> FuncTreeNode:
+        starts = [node.start for node in self.root.children]
+        idx = bisect.bisect(starts, ts)
+        if idx == 0:
+            return self.root.children[0]
+        else:
+            return self.root.children[idx - 1]
+
+    def normalize(self, first_ts: float) -> None:
+        for node in self.inorder_traverse():
+            node.start -= first_ts
+            node.end -= first_ts
+
+    def inorder_traverse(self) -> Generator[FuncTreeNode, None, None]:
+        lst = [self.root]
+        while lst:
+            ret = lst.pop()
+            lst.extend(ret.children[::-1])
+            yield ret
+        return

viztracer/html/flamegraph.html

@@ -0,0 +1,34 @@
+<head>
+  <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/d3-flame-graph@3.1.1/dist/d3-flamegraph.css">
+  <link rel="stylesheet" href="https://stackpath.bootstrapcdn.com/bootstrap/4.5.2/css/bootstrap.min.css" integrity="sha384-JcKb8q3iqJ61gNV9KGb8thSsNjpSL0n8PARn9HuZOnIxN0hoP+VmmDGMN5t9UJ0Z" crossorigin="anonymous">
+</head>
+<body>
+  <div id="chart" class="container text-center"></div>
+  <script src="https://code.jquery.com/jquery-3.5.1.slim.min.js" integrity="sha384-DfXdz2htPH0lsSSs5nCTpuj/zy4C+OGpamoFVy38MVBnE+IbbVYUew+OrCXaRkfj" crossorigin="anonymous"></script>
+  <script src="https://stackpath.bootstrapcdn.com/bootstrap/4.5.2/js/bootstrap.min.js" integrity="sha384-B4gt1jrGC7Jh4AgTPSdUtOBvfO8shuf57BaghqFfPlYxofvL8/KUEfYiJOMMV+rV" crossorigin="anonymous"></script>
+  <script type="text/javascript" src="https://d3js.org/d3.v4.min.js"></script>
+  <script type="text/javascript" src="https://cdn.jsdelivr.net/npm/d3-flame-graph@3.1.1/dist/d3-flamegraph.min.js"></script>
+  <script type="text/javascript">
+  var data = $data;
+
+  var charts = [];
+  for (var thread in data) {
+    var chart = flamegraph()
+      .width(1000)
+      .title(thread)
+      .minFrameSize(5)
+      .resetHeightOnZoom(true)
+      .setColorMapper((d, originalColor) =>
+        d.highlight ? "#6aff8f" : originalColor)
+      .label(function(d) {
+        return "name: " + d.data.name + " time: " + d.data.value.toString() + " ns"
+      });
+    var div = document.createElement("div");
+    div.setAttribute("id", thread);
+    document.getElementById("chart").appendChild(div);
+    d3.select("#" + thread)
+      .datum(data[thread])
+      .call(chart);
+  }
+  </script>
+</body>

viztracer/html/trace_viewer_embedder.html

@@ -0,0 +1,203 @@
+<!DOCTYPE html>
+<!--
+Copyright (c) 2014 The Chromium Authors. All rights reserved.
+Use of this source code is governed by a BSD-style license that can be
+found in the LICENSE file.
+Original LICENSE file: 
+https://github.com/catapult-project/catapult/blob/master/tracing/LICENSE
+-->
+<head>
+<!-- WebComponent V0 polyfill. You may want to host this in a more suitable
+     place. See https://crbug.com/1036492 -->
+<script src="https://cdnjs.cloudflare.com/ajax/libs/webcomponentsjs/0.7.24/webcomponents.min.js"></script>
+
+<script src="https://cdnjs.cloudflare.com/ajax/libs/prism/1.22.0/components/prism-core.min.js" integrity="sha512-hqRrGU7ys5tkcqxx5FIZTBb7PkO2o3mU6U5+qB9b55kgMlBUT4J2wPwQfMCxeJW1fC8pBxuatxoH//z0FInhrA==" crossorigin="anonymous"></script>
+<script src="https://cdnjs.cloudflare.com/ajax/libs/prism/1.22.0/components/prism-python.min.js" integrity="sha512-EXseTM1JV8e3AonU5PfOUmzbPkUTRm4c4hVxv206gVeWZE/FwzSQBNWrUBaHsN3Idq9k76BEjUlXpluX1UFx6Q==" crossorigin="anonymous"></script>
+<script>
+'use strict';
+
+function onTraceViewerImportFail() {
+  document.addEventListener('DOMContentLoaded', function() {
+    document.body.textContent =
+        'tracing/bin/trace_viewer_full.html is missing. ' +
+        'Run vulcanize_trace_viewer from $$TRACE_VIEWER and reload.';
+  });
+}
+</script>
+<!-- <link rel="import" href="trace_viewer_full.html"
+      onerror="onTraceViewerImportFail(event)"> -->
+
+$trace_viewer_full
+
+
+<style>
+  html, body {
+    box-sizing: border-box;
+    overflow: hidden;
+    margin: 0px;
+    padding: 0;
+    width: 100%;
+    height: 100%;
+  }
+  #trace-viewer {
+    width: 100%;
+    height: 100%;
+  }
+  #trace-viewer:focus {
+    outline: none;
+  }
+</style>
+<script>
+'use strict';
+
+var json_data;
+(function() {
+  let viewer;
+  let url;
+  let model;
+
+  // You can set this to true if you want to hide the WebComponentsV0 polyfill
+  // warning.
+  window.__hideTraceViewerPolyfillWarning = true;
+  json_data = $json_data
+
+  function load() {
+    const req = new XMLHttpRequest();
+    const isBinary = /[.]gz$$/.test(url) || /[.]zip$$/.test(url);
+    req.overrideMimeType('text/plain; charset=x-user-defined');
+    req.open('GET', url, true);
+    if (isBinary) req.responseType = 'arraybuffer';
+
+    req.onreadystatechange = function(event) {
+      if (req.readyState !== 4) return;
+
+      window.setTimeout(function() {
+        if (req.status === 200) {
+          onResult(isBinary ? req.response : req.responseText);
+        } else {
+          onResultFail(req.status);
+        }
+      }, 0);
+    };
+    req.send(null);
+  }
+
+  function onResultFail(err) {
+    const overlay = new tr.ui.b.Overlay();
+    overlay.textContent = err + ': ' + url + ' could not be loaded';
+    overlay.title = 'Failed to fetch data';
+    overlay.visible = true;
+  }
+
+  function onResult(result) {
+    model = new tr.Model();
+    const i = new tr.importer.Import(model);
+    const p = i.importTracesWithProgressDialog([result]);
+    p.then(onModelLoaded, onImportFail);
+  }
+
+  function onModelLoaded() {
+    viewer.model = model;
+    viewer.viewTitle = url;
+  }
+
+  function onImportFail(err) {
+    const overlay = new tr.ui.b.Overlay();
+    overlay.textContent = tr.b.normalizeException(err).message;
+    overlay.title = 'Import error';
+    overlay.visible = true;
+  }
+
+  document.addEventListener('WebComponentsReady', function() {
+    const container = document.createElement('track-view-container');
+    container.id = 'track_view_container';
+
+    viewer = document.createElement('tr-ui-timeline-view');
+    viewer.track_view_container = container;
+    Polymer.dom(viewer).appendChild(container);
+
+    viewer.id = 'trace-viewer';
+    viewer.globalMode = true;
+    Polymer.dom(document.body).appendChild(viewer);
+
+    //url = '../test_data/big_trace.json';
+    //load();
+    model = new tr.Model();
+    const i = new tr.importer.Import(model);
+    const p = i.importTracesWithProgressDialog([json_data]);
+    p.then(onModelLoaded, onImportFail);
+  });
+
+}());
+
+function updateCodeSection() {
+  try {
+    var viewer = document.getElementById("trace-viewer");
+    if (viewer.shadowRoot) {
+      var tabRoot = viewer.shadowRoot
+                          .getElementById("analysis")
+                          .getElementsByTagName("tr-ui-b-tab-view")[0]
+                          .shadowRoot
+      var threadRoot = tabRoot.getElementById("subView")
+                              .getElementsByTagName("tr-ui-a-single-thread-slice-sub-view")[0]
+                              .shadowRoot;
+      var table = threadRoot.getElementById("content")
+                            .shadowRoot
+                            .getElementById("table")
+      var codeViewNode = threadRoot.getElementById("codeView");
+      var codeNode = threadRoot.getElementById("code-viewer");
+
+      var funcName = table.tableRows_[0].value.textContent;
+    } else {
+      var codeViewNode = document.getElementById("codeView");
+      var codeNode = document.getElementById("code-viewer");
+      var funcName = document.getElementById("table").tableRows_[0].value.textContent.split("\n")[0];
+    }
+
+    resizeCodeSection();
+
+    codeNode.innerHTML = "Unable to locate the source code";
+    var file_data = json_data["file_info"]["functions"][funcName];
+    if (file_data) {
+      var file_name = file_data[0];
+      var lineno = file_data[1];
+      var raw_code = json_data["file_info"]["files"][file_name][0];
+      var total_lineno = json_data["file_info"]["files"][file_name][1];
+      codeNode.innerHTML = raw_code;
+      Prism.highlightAll()
+      setTimeout(() => {
+        var portion = 5 + codeNode.offsetHeight * (lineno-1) / total_lineno;
+        codeViewNode.scrollTo(0, portion);
+      }, 100);
+    }
+  } catch (error) {
+  }
+}
+
+function resizeCodeSection() {
+  try {
+    var viewer = document.getElementById("trace-viewer");
+    if (viewer.shadowRoot) {
+      var tabRoot = viewer.shadowRoot
+                          .getElementById("analysis")
+                          .getElementsByTagName("tr-ui-b-tab-view")[0]
+                          .shadowRoot
+      var threadRoot = tabRoot.getElementById("subView")
+                              .getElementsByTagName("tr-ui-a-single-thread-slice-sub-view")[0]
+                              .shadowRoot;
+  
+      var height = viewer.shadowRoot.getElementById("analysis").offsetHeight - tabRoot.getElementById("tabs").offsetHeight - 7;
+      threadRoot.getElementById("codeView").style.height = height.toString() + "px";
+    } else {
+      var height = document.getElementById("analysis").offsetHeight - document.getElementById("tabs").offsetHeight - 7;
+      document.getElementById("codeView").style.height = height.toString() + "px";
+    }
+  } catch (error) {
+  }
+}
+
+</script>
+</head>
+<body>
+</body>
+</html>