// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license. // @ts-check /// /// /// /// import { primordials } from "ext:core/mod.js"; import { op_webgpu_surface_configure, op_webgpu_surface_create, op_webgpu_surface_get_current_texture, op_webgpu_surface_present, } from "ext:core/ops"; const { ObjectPrototypeIsPrototypeOf, Symbol, SymbolFor, TypeError, } = primordials; import * as webidl from "ext:deno_webidl/00_webidl.js"; import { createFilteredInspectProxy } from "ext:deno_console/01_console.js"; import { loadWebGPU, webgpu } from "ext:deno_webgpu/00_init.js"; const _surfaceRid = Symbol("[[surfaceRid]]"); const _configuration = Symbol("[[configuration]]"); const _canvas = Symbol("[[canvas]]"); const _currentTexture = Symbol("[[currentTexture]]"); const _present = Symbol("[[present]]"); class GPUCanvasContext { /** @type {number} */ [_surfaceRid]; [_configuration]; [_canvas]; /** @type {GPUTexture | undefined} */ [_currentTexture]; get canvas() { webidl.assertBranded(this, GPUCanvasContextPrototype); return this[_canvas]; } constructor() { webidl.illegalConstructor(); } configure(configuration) { webidl.assertBranded(this, GPUCanvasContextPrototype); const prefix = "Failed to execute 'configure' on 'GPUCanvasContext'"; webidl.requiredArguments(arguments.length, 1, { prefix }); configuration = webidl.converters.GPUCanvasConfiguration(configuration, { prefix, context: "Argument 1", }); const { _device, assertDevice } = webgpu; this[_device] = configuration.device[_device]; this[_configuration] = configuration; const device = assertDevice(this, { prefix, context: "configuration.device", }); const { err } = op_webgpu_surface_configure({ surfaceRid: this[_surfaceRid], deviceRid: device.rid, format: configuration.format, viewFormats: configuration.viewFormats, usage: configuration.usage, width: configuration.width, height: configuration.height, alphaMode: configuration.alphaMode, }); device.pushError(err); } unconfigure() { const { _device } = webgpu; webidl.assertBranded(this, GPUCanvasContextPrototype); this[_configuration] = null; this[_device] = null; } getCurrentTexture() { webidl.assertBranded(this, GPUCanvasContextPrototype); const prefix = "Failed to execute 'getCurrentTexture' on 'GPUCanvasContext'"; if (this[_configuration] === null) { throw new DOMException("context is not configured.", "InvalidStateError"); } const { createGPUTexture, assertDevice } = webgpu; const device = assertDevice(this, { prefix, context: "this" }); if (this[_currentTexture]) { return this[_currentTexture]; } const { rid } = op_webgpu_surface_get_current_texture( device.rid, this[_surfaceRid], ); const texture = createGPUTexture( { size: { width: this[_configuration].width, height: this[_configuration].height, depthOrArrayLayers: 1, }, mipLevelCount: 1, sampleCount: 1, dimension: "2d", format: this[_configuration].format, usage: this[_configuration].usage, }, device, rid, ); device.trackResource(texture); this[_currentTexture] = texture; return texture; } // Required to present the texture; browser don't need this. [_present]() { const { assertDevice } = webgpu; webidl.assertBranded(this, GPUCanvasContextPrototype); const prefix = "Failed to execute 'present' on 'GPUCanvasContext'"; const device = assertDevice(this[_currentTexture], { prefix, context: "this", }); op_webgpu_surface_present(device.rid, this[_surfaceRid]); this[_currentTexture].destroy(); this[_currentTexture] = undefined; } [SymbolFor("Deno.privateCustomInspect")](inspect, inspectOptions) { return inspect( createFilteredInspectProxy({ object: this, evaluate: ObjectPrototypeIsPrototypeOf(GPUCanvasContextPrototype, this), keys: [ "canvas", ], }), inspectOptions, ); } } const GPUCanvasContextPrototype = GPUCanvasContext.prototype; function createCanvasContext(options) { // lazy load webgpu if needed loadWebGPU(); const canvasContext = webidl.createBranded(GPUCanvasContext); canvasContext[_surfaceRid] = options.surfaceRid; canvasContext[_canvas] = options.canvas; return canvasContext; } // External webgpu surfaces // TODO(@littledivy): This will extend `OffscreenCanvas` when we add it. class UnsafeWindowSurface { #ctx; #surfaceRid; constructor(system, win, display) { this.#surfaceRid = op_webgpu_surface_create(system, win, display); } getContext(context) { if (context !== "webgpu") { throw new TypeError("Only 'webgpu' context is supported."); } this.#ctx = createCanvasContext({ surfaceRid: this.#surfaceRid }); return this.#ctx; } present() { this.#ctx[_present](); } } export { GPUCanvasContext, UnsafeWindowSurface };