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

viz: prune out of screen rects (#11217)

Browse source
This commit is contained in:
qazal 2025-07-13 21:49:59 +03:00 • committed by GitHub
commit 40847ca29c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tinygrad/viz/js/index.js
View file

@ -197,12 +197,15 @@ async function renderProfiler() {
// rescale to match current zoom
const xscale = d3.scaleLinear().domain([0, et-st]).range([0, canvas.clientWidth]);
xscale.domain(xscale.range().map(zoomLevel.invertX, zoomLevel).map(xscale.invert, xscale));
const zoomDomain = transform != null ? xscale.domain() : null;
let yscale = null;
if (data.axes.y != null) {
yscale = d3.scaleLinear().domain(data.axes.y.domain).range(data.axes.y.range);
}
// draw shapes
for (const e of data.shapes) {
const [start, end] = Array.isArray(e.x) ? [e.x[0], e.x[e.x.length-1]] : [e.x, e.x+e.dur];
if (zoomDomain != null && (start>zoomDomain[1]|| end<zoomDomain[0])) continue;
if (Array.isArray(e.x)) {
const x = e.x.map(xscale);
ctx.beginPath();
@ -217,8 +220,8 @@ async function renderProfiler() {
continue;
}
// zoom only changes x and width
const x = xscale(e.x);
const width = xscale(e.x+e.dur)-x;
const x = xscale(start);
const width = xscale(end)-x;
ctx.fillStyle = e.fillColor;
ctx.fillRect(x, e.y, width, e.height);
rectLst.push({ y0:e.y, y1:e.y+e.height, x0:x, x1:x+width, ref:e.ref, tooltipText:formatTime(e.dur) });