commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

viz auto recenter on out of view graph [pr] (#6986)

This commit is contained in:
qazal 2024-10-11 02:40:06 +03:00 committed by GitHub
parent 159ee04489
commit 54dcea235d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tinygrad/viz/index.html
View File

@ -187,6 +187,19 @@
...hljs.getLanguage('cpp'),
contains: [{ begin: '\\b(?:float|half)[0-9]+\\b', className: 'type' }, ...hljs.getLanguage('cpp').contains]
}));
function recenterRects(svg, zoom) {
const svgBounds = svg.node().getBoundingClientRect();
for (const rect of svg.node().querySelectorAll("rect")) {
const rectBounds = rect.getBoundingClientRect();
const outOfBounds = rectBounds.top < svgBounds.top || rectBounds.left < svgBounds.left ||
rectBounds.bottom > svgBounds.bottom || rectBounds.right > svgBounds.right;
// if there's at least one rect in view we don't do anything
if (!outOfBounds) return;
}
svg.call(zoom.transform, d3.zoomIdentity)
}
function renderGraph(graph, additions) {
const g = new dagreD3.graphlib.Graph({ compound: true }).setGraph({ rankdir: "LR" }).setDefaultEdgeLabel(function() { return {}; });
g.setNode("addition", {label: "", clusterLabelPos: "top", style: additions.length !== 0 ? "fill: rgba(26, 27, 38, 0.5); rx: 8; ry: 8;" : "display: none;"});
@ -207,6 +220,7 @@
const transform = d3.event.transform;
inner.attr("transform", transform);
});
recenterRects(svg, zoom);
svg.call(zoom);
const render = new dagreD3.render();
render(inner, g);