mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

little smaller viz/worker.js [pr] (#10402)

Browse source
This commit is contained in:
qazal 2025-05-18 23:44:46 +03:00 • committed by GitHub
commit e55ee28b29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tinygrad/viz/js/worker.js
View file

@ -1,19 +1,9 @@
const NODE_PADDING = 10;
const LINE_HEIGHT = 14;
const canvas = new OffscreenCanvas(0, 0);
const ctx = canvas.getContext('2d');
const ctx = canvas.getContext("2d");
ctx.font = `${LINE_HEIGHT}px sans-serif`;
function getTextDims(text) {
let [maxWidth, height] = [0, 0];
for (line of text.split("\n")) {
const { width } = ctx.measureText(line);
if (width > maxWidth) maxWidth = width;
height += LINE_HEIGHT;
}
return [maxWidth, height];
}
onmessage = (e) => {
const { graph, additions } = e.data;
const g = new dagre.graphlib.Graph({ compound: true });
@ -21,12 +11,14 @@ onmessage = (e) => {
if (additions.length !== 0) g.setNode("addition", {label:"", style:"fill: rgba(26, 27, 38, 0.5); stroke: none;", padding:0});
for (const [k, {label, src, color}] of Object.entries(graph)) {
// adjust node dims by label size + add padding
const [labelWidth, labelHeight] = getTextDims(label);
g.setNode(k, {label, color, width:labelWidth+NODE_PADDING*2, height:labelHeight+NODE_PADDING*2, padding:NODE_PADDING});
const edgeCounts = {}
for (const s of src) {
edgeCounts[s] = (edgeCounts[s] || 0)+1;
let [width, height] = [0, 0];
for (line of label.split("\n")) {
width = Math.max(width, ctx.measureText(line).width);
height += LINE_HEIGHT;
}
g.setNode(k, {label, color, width:width+NODE_PADDING*2, height:height+NODE_PADDING*2, padding:NODE_PADDING});
const edgeCounts = {}
for (const s of src) edgeCounts[s] = (edgeCounts[s] || 0)+1;
for (const s of src) g.setEdge(s, k, { label: edgeCounts[s] > 1 ? edgeCounts[s] : null });
if (additions.includes(parseInt(k))) g.setParent(k, "addition");
}