12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
import { getRowPadding } from "./row_padding.ts";
/** Return value for {@linkcode createCapture}. */export interface CreateCapture {  /**   * Texture to be used as view to render to.   */  texture: GPUTexture;
  /**   * Represents the output buffer of the rendered texture.   * Can then be used to access and retrieve raw image data.   */  outputBuffer: GPUBuffer;}
/** * Creates a texture and buffer to use as a capture. * * @example * ```ts * import { createCapture } from "https://deno.land/std@$STD_VERSION/webgpu/create_capture.ts"; * import { getRowPadding } from "https://deno.land/std@$STD_VERSION/webgpu/row_padding.ts"; * * const adapter = await navigator.gpu.requestAdapter(); * const device = await adapter?.requestDevice()!; * * const dimensions = { *   width: 200, *   height: 200, * }; * * const { texture, outputBuffer } = createCapture(device, dimensions.width, dimensions.height); * * const encoder = device.createCommandEncoder(); * encoder.beginRenderPass({ *   colorAttachments: [ *     { *       view: texture.createView(), *       storeOp: "store", *       loadOp: "clear", *       clearValue: [1, 0, 0, 1], *     }, *   ], * }).end(); * * const { padded } = getRowPadding(dimensions.width); * * encoder.copyTextureToBuffer( *   { *     texture, *   }, *   { *     buffer: outputBuffer, *     bytesPerRow: padded, *   }, *   dimensions, * ); * * device.queue.submit([encoder.finish()]); * * // outputBuffer contains the raw image data, can then be used * // to save as png or other formats. * ``` */export function createCapture(  device: GPUDevice,  width: number,  height: number,): CreateCapture {  const { padded } = getRowPadding(width);  const outputBuffer = device.createBuffer({    label: "Capture",    size: padded * height,    usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST,  });  const texture = device.createTexture({    label: "Capture",    size: {      width,      height,    },    format: "rgba8unorm-srgb",    usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.COPY_SRC,  });
  return { texture, outputBuffer };}