Function createHarness

src/splat/webgpu/psnrSplatSession.test.ts:166–221

(options: {
  device?: GPUDevice;
  renderSession?: SplatRenderSession;
  createRenderSession?: WebGpuSplatPsnrSessionDeps['createRenderSession'];
  computePsnrFromTextures?: WebGpuSplatPsnrSessionDeps['computePsnrFromTextures'];
  computePsnrTextureReductionFromTextures?: WebGpuSplatPsnrSessionDeps['computePsnrTextureReductionFromTextures'];
  createMetricFrame?: WebGpuSplatPsnrSessionDeps['createMetricFrame'];
} = {})

}

function createHarness(options: {
  device?: GPUDevice;
  renderSession?: SplatRenderSession;
  createRenderSession?: WebGpuSplatPsnrSessionDeps['createRenderSession'];
  computePsnrFromTextures?: WebGpuSplatPsnrSessionDeps['computePsnrFromTextures'];
  computePsnrTextureReductionFromTextures?: WebGpuSplatPsnrSessionDeps['computePsnrTextureReductionFromTextures'];
  createMetricFrame?: WebGpuSplatPsnrSessionDeps['createMetricFrame'];
} = {}) {
  const splatFile = buildFile('scene.spz', 'spz');
  const loadedCloud = makeLoadedCloud(splatFile);
  const sceneRelease = vi.fn();
  const sceneRef = makeSceneRef(sceneRelease);
  const resourceManager = {
    acquire: vi.fn(() => sceneRef),
    dispose: vi.fn(),
  };
  const renderSession = options.renderSession ?? makeRenderSession();
  const bitmap = makeBitmap();
  const groundTruthTexture = {
    texture: makeTexture('ground-truth'),
    width: 4,
    height: 3,
    dispose: vi.fn(),
  };
  const metric = {
    sumSquaredError: 492,
    psnr: 32,
    mse: 41,
    validPixelCount: 12,
  };
  const deps = {
    loadGaussianCloudFromFile: vi.fn(async () => loadedCloud),
    createSceneResourceManager: vi.fn(() => resourceManager),
    createRenderSession: options.createRenderSession ?? vi.fn(() => renderSession),
    createBitmap: vi.fn(async () => bitmap),
    createGroundTruthTexture: vi.fn(() => groundTruthTexture),
    computePsnrFromTextures: options.computePsnrFromTextures ?? vi.fn(async () => metric),
    computePsnrTextureReductionFromTextures: options.computePsnrTextureReductionFromTextures
      ?? vi.fn(async () => ({ sumSquaredError: 0, validPixelCount: 1 })),
    createMetricFrame: options.createMetricFrame ?? vi.fn(() => makeFrame()),
  } satisfies WebGpuSplatPsnrSessionDeps;

  return {
    bitmap,
    deps,
    groundTruthTexture,
    loadedCloud,
    metric,
    renderSession,
    resourceManager,
    sceneRef,
    sceneRelease,
    splatFile,
    device: options.device ?? makeDevice().device,
  };
}

describe('WebGPU splat PSNR session', () => {