sentinel/static/app/utils/profiling/renderers/selectedFrameRenderer.tsx

import type {mat3} from 'gl-matrix';

import type {Rect} from 'sentry/utils/profiling/speedscope';

import {getContext} from '../gl/utils';

class SelectedFrameRenderer {
  canvas: HTMLCanvasElement;
  context: CanvasRenderingContext2D;

  constructor(canvas: HTMLCanvasElement) {
    this.canvas = canvas;
    this.context = getContext(canvas, '2d');
  }

  // We allow for passing of different contexts, this allows us to use a
  // single instance of the renderer to draw overlays on multiple canvases
  draw(
    frames: Rect[],
    style: {BORDER_COLOR: string; BORDER_WIDTH: number},
    configViewToPhysicalSpace: mat3,
    context: CanvasRenderingContext2D = this.context
  ): void {
    context.strokeStyle = style.BORDER_COLOR;
    context.lineWidth = style.BORDER_WIDTH;

    for (let i = 0; i < frames.length; i++) {
      const frameInPhysicalSpace = frames[i].transformRect(configViewToPhysicalSpace);

      context.beginPath();

      // We draw the border in the center of the flamegraph, so we need to decrease
      // the width by border width and negatively offset it by half the border width
      context.strokeRect(
        frameInPhysicalSpace.x + style.BORDER_WIDTH,
        frameInPhysicalSpace.y + style.BORDER_WIDTH,
        frameInPhysicalSpace.width - style.BORDER_WIDTH * 2,
        frameInPhysicalSpace.height - style.BORDER_WIDTH * 2
      );
    }
  }
}

export {SelectedFrameRenderer};