Function routeFakeDelayedPsnrSession

e2e/webgpu-psnr-app.spec.ts:410–456

(page: Page, delaysMs: number[])

}

async function routeFakeDelayedPsnrSession(page: Page, delaysMs: number[]): Promise<void> {
  const delays = delaysMs.map((delay) => Math.max(0, Math.trunc(delay)));
  const installState = (nextDelays: number[]) => {
    Object.defineProperty(window, '__COLMAP_WEBVIEW_FAKE_PSNR__', {
      configurable: true,
      value: {
        delays: nextDelays,
        sessionCount: 0,
        computeCount: 0,
        disposedCount: 0,
        completed: [] as number[],
      },
    });
  };

  await page.addInitScript(installState, delays);
  await page.evaluate(installState, delays);

  await page.route('**/src/splat/webgpu/psnrSplatSession.ts*', async (route) => {
    await route.fulfill({
      contentType: 'application/javascript',
      body: `
        const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));

        export async function createWebGpuSplatPsnrSession() {
          const state = window.__COLMAP_WEBVIEW_FAKE_PSNR__;
          ++state.sessionCount;
          return {
            async computeImageMetric(options) {
              const computeId = ++state.computeCount;
              await delay(state.delays[computeId - 1] ?? 0);
              state.completed.push(computeId);
              return {
                psnr: 20 + computeId,
                mse: computeId,
                validPixelCount: options.width * options.height,
              };
            },
            dispose() {
              state.disposedCount += 1;
            },
          };
        }
      `,
    });
  });
}

async function resetFakeDelayedPsnrSession(page: Page, delaysMs: number[]): Promise<void> {
  const delays = delaysMs.map((delay) => Math.max(0, Math.trunc(delay)));