wgpu-kit 0.9.11 → 1.0.0

package/dist/image.js

@@ -1,318 +0,0 @@
-// src/core/errors.ts
-var WgpuKitError = class extends Error {
-  constructor(message) {
-    super(message);
-    this.name = new.target.name;
-  }
-};
-var WebGPUUnavailableError = class extends WgpuKitError {
-  constructor(reason) {
-    super(
-      `\u5F53\u524D\u73AF\u5883\u4E0D\u53EF\u7528 WebGPU: ${reason}
-  \u6392\u67E5:\u2460 \u6D4F\u89C8\u5668\u9700 Chrome/Edge 113+ \u6216 Safari 18+;\u2461 \u65E0\u5934\u73AF\u5883\u9700\u5F00\u542F WebGPU;\u2462 \u68C0\u67E5 GPU \u9A71\u52A8\u4E0E\u786C\u4EF6\u52A0\u901F\u8BBE\u7F6E\u3002
-  \u53EF\u7528 navigator.gpu \u662F\u5426\u5B58\u5728\u5FEB\u901F\u5224\u65AD\u3002`
-    );
-  }
-};
-var CompileError = class extends WgpuKitError {
-  constructor(kernelName, messages, userCodeOffset) {
-    const mapped = messages.map((m) => {
-      const userLine = m.line - userCodeOffset;
-      const where = userLine > 0 ? `\u7528\u6237\u4EE3\u7801\u7B2C ${userLine} \u884C` : `\u751F\u6210\u4EE3\u7801\u7B2C ${m.line} \u884C(\u5E93\u7684\u95EE\u9898,\u6B22\u8FCE\u62A5 issue)`;
-      return `  ${where}: ${m.msg}`;
-    }).join("\n");
-    super(`kernel "${kernelName}" WGSL \u7F16\u8BD1\u5931\u8D25:
-${mapped}`);
-  }
-};
-
-// src/core/context.ts
-var GpuContext = class _GpuContext {
-  device;
-  adapterInfo;
-  constructor(device, adapterInfo) {
-    this.device = device;
-    this.adapterInfo = adapterInfo;
-  }
-  static #singleton = null;
-  static get() {
-    if (!_GpuContext.#singleton) {
-      _GpuContext.#singleton = _GpuContext.#create().catch((e) => {
-        _GpuContext.#singleton = null;
-        throw e;
-      });
-    }
-    return _GpuContext.#singleton;
-  }
-  static async #create() {
-    if (typeof navigator === "undefined" || !("gpu" in navigator) || !navigator.gpu) {
-      throw new WebGPUUnavailableError("navigator.gpu \u4E0D\u5B58\u5728");
-    }
-    const adapter = await navigator.gpu.requestAdapter({ powerPreference: "high-performance" });
-    if (!adapter) throw new WebGPUUnavailableError("requestAdapter() \u8FD4\u56DE null");
-    const info = adapter.info;
-    const label = info ? [info.vendor, info.architecture, info.description].filter(Boolean).join(" / ") || "unknown" : "unknown";
-    const requiredLimits = {};
-    const want = [
-      "maxStorageBuffersPerShaderStage",
-      "maxStorageBuffersInVertexStage",
-      "maxStorageBufferBindingSize"
-    ];
-    for (const key of want) {
-      const supported = adapter.limits[key];
-      if (typeof supported === "number") requiredLimits[key] = supported;
-    }
-    const device = await adapter.requestDevice({ label: "wgpu-kit", requiredLimits });
-    return new _GpuContext(device, label);
-  }
-  /** device lost 时 reject;调用方可 await 做清理/提示 */
-  get lost() {
-    return this.device.lost;
-  }
-  /** 等待队列中已提交的全部 GPU 工作完成(测试/读回前同步用) */
-  async sync() {
-    await this.device.queue.onSubmittedWorkDone();
-  }
-};
-
-// src/packs/image/index.ts
-var OP_IDS = { grayscale: 0, invert: 1, edge: 2, blur: 3, sharpen: 4, brightness: 5, contrast: 6 };
-async function applyImage(source, target, ops) {
-  if (ops.length === 0) throw new Error("applyImage \u9700\u8981\u81F3\u5C11\u4E00\u4E2A\u7B97\u5B50");
-  const width = "naturalWidth" in source ? source.naturalWidth : source.width;
-  const height = "naturalHeight" in source ? source.naturalHeight : source.height;
-  const ctx = await GpuContext.get();
-  const device = ctx.device;
-  const srcTex = device.createTexture({
-    size: [width, height],
-    format: "rgba8unorm",
-    usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST | GPUTextureUsage.RENDER_ATTACHMENT
-  });
-  const c2d = document.createElement("canvas");
-  c2d.width = width;
-  c2d.height = height;
-  const sctx = c2d.getContext("2d", { willReadFrequently: true });
-  sctx.drawImage(source, 0, 0);
-  const imageData = sctx.getImageData(0, 0, width, height);
-  device.queue.writeTexture({ texture: srcTex }, imageData.data, { bytesPerRow: width * 4, rowsPerImage: height }, [width, height]);
-  const pool = [];
-  const temp = () => {
-    const t = pool.pop() ?? device.createTexture({
-      size: [width, height],
-      format: "rgba8unorm",
-      usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.COPY_SRC
-    });
-    return t;
-  };
-  const module = device.createShaderModule({ code: shader(), label: "image-filters" });
-  const info = await module.getCompilationInfo();
-  const errors = info.messages.filter((m) => m.type === "error");
-  if (errors.length > 0) throw new CompileError("image-filters", errors.map((m) => ({ line: m.lineNum, msg: m.message })), 0);
-  const pipeline = device.createRenderPipeline({
-    layout: "auto",
-    vertex: { module, entryPoint: "vs" },
-    fragment: { module, entryPoint: "fs", targets: [{ format: "rgba8unorm" }] },
-    primitive: { topology: "triangle-list" }
-  });
-  const makePass = (input, output, op) => {
-    const p = op;
-    const uniform = device.createBuffer({ size: 32, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST });
-    const ub = new ArrayBuffer(32);
-    const uv = new DataView(ub);
-    uv.setUint32(0, OP_IDS[op.op] ?? 0, true);
-    uv.setUint32(4, Math.max(1, Math.min(4, Math.round(p.radius ?? 1))), true);
-    uv.setFloat32(8, width, true);
-    uv.setFloat32(12, height, true);
-    uv.setFloat32(16, p.amount ?? 1, true);
-    uv.setFloat32(20, p.value ?? 0, true);
-    device.queue.writeBuffer(uniform, 0, ub);
-    const sampler = device.createSampler({ magFilter: "linear", minFilter: "linear" });
-    const bg = device.createBindGroup({
-      layout: pipeline.getBindGroupLayout(0),
-      entries: [
-        { binding: 0, resource: { buffer: uniform } },
-        { binding: 1, resource: sampler },
-        { binding: 2, resource: input.createView() }
-      ]
-    });
-    const enc2 = device.createCommandEncoder();
-    const pass2 = enc2.beginRenderPass({
-      colorAttachments: output ? [{ view: output.createView(), clearValue: { r: 0, g: 0, b: 0, a: 1 }, loadOp: "clear", storeOp: "store" }] : []
-    });
-    pass2.setPipeline(pipeline);
-    pass2.setBindGroup(0, bg);
-    pass2.draw(3);
-    pass2.end();
-    return enc2.finish();
-  };
-  let cur = srcTex;
-  let passes = 0;
-  for (const op of ops) {
-    const out = temp();
-    device.queue.submit([makePass(cur, out, op)]);
-    if (cur !== srcTex) pool.push(cur);
-    cur = out;
-    passes++;
-  }
-  const gpuCtx = target.getContext("webgpu");
-  if (!gpuCtx) throw new Error('\u76EE\u6807 canvas.getContext("webgpu") \u8FD4\u56DE\u7A7A');
-  const format = navigator.gpu.getPreferredCanvasFormat();
-  gpuCtx.configure({ device, format, alphaMode: "opaque" });
-  const blit = device.createShaderModule({
-    code: `
-@group(0) @binding(0) var samp: sampler;
-@group(0) @binding(1) var tex: texture_2d<f32>;
-struct VsOut { @builtin(position) pos: vec4f, @location(0) uv: vec2f };
-@vertex fn vs(@builtin(vertex_index) v: u32) -> VsOut {
-  var p = array<vec2f, 3>(vec2f(-1.0, -1.0), vec2f(3.0, -1.0), vec2f(-1.0, 3.0));
-  var o: VsOut;
-  o.pos = vec4f(p[v], 0.0, 1.0);
-  o.uv = (p[v] + vec2f(1.0)) * 0.5;
-  o.uv.y = 1.0 - o.uv.y;
-  return o;
-}
-@fragment fn fs(i: VsOut) -> @location(0) vec4f {
-  let c = textureSample(tex, samp, i.uv);
-  return vec4f(c.rgb, 1.0);
-}
-`,
-    label: "image-blit"
-  });
-  const blitPipeline = device.createRenderPipeline({
-    layout: "auto",
-    vertex: { module: blit, entryPoint: "vs" },
-    fragment: { module: blit, entryPoint: "fs", targets: [{ format }] },
-    primitive: { topology: "triangle-list" }
-  });
-  const enc = device.createCommandEncoder();
-  const pass = enc.beginRenderPass({
-    colorAttachments: [{ view: gpuCtx.getCurrentTexture().createView(), clearValue: { r: 0, g: 0, b: 0, a: 1 }, loadOp: "clear", storeOp: "store" }]
-  });
-  pass.setPipeline(blitPipeline);
-  pass.setBindGroup(0, device.createBindGroup({
-    layout: blitPipeline.getBindGroupLayout(0),
-    entries: [
-      { binding: 0, resource: device.createSampler({ magFilter: "linear", minFilter: "linear" }) },
-      { binding: 1, resource: cur.createView() }
-    ]
-  }));
-  pass.draw(3);
-  pass.end();
-  device.queue.submit([enc.finish()]);
-  srcTex.destroy();
-  for (const t of pool) t.destroy();
-  const bytesPerRow = Math.ceil(width * 4 / 256) * 256;
-  const staging = device.createBuffer({ size: bytesPerRow * height, usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ });
-  return {
-    width,
-    height,
-    passes,
-    async readback() {
-      const enc2 = device.createCommandEncoder();
-      enc2.copyTextureToBuffer({ texture: cur }, { buffer: staging, bytesPerRow, rowsPerImage: height }, [width, height]);
-      device.queue.submit([enc2.finish()]);
-      await staging.mapAsync(GPUMapMode.READ);
-      const ab = staging.getMappedRange().slice(0);
-      staging.unmap();
-      staging.destroy();
-      return new Uint8Array(ab);
-    }
-  };
-}
-function shader() {
-  return (
-    /* wgsl */
-    `
-struct U {
-  op: u32, radius: u32,
-  w: f32, h: f32, amount: f32, value: f32,
-  _p0: u32, _p1: u32,
-};
-@group(0) @binding(0) var<uniform> u: U;
-@group(0) @binding(1) var samp: sampler;
-@group(0) @binding(2) var tex: texture_2d<f32>;
-
-struct VsOut { @builtin(position) pos: vec4f, @location(0) uv: vec2f };
-
-@vertex
-fn vs(@builtin(vertex_index) v: u32) -> VsOut {
-  var p = array<vec2f, 3>(vec2f(-1.0, -1.0), vec2f(3.0, -1.0), vec2f(-1.0, 3.0));
-  var o: VsOut;
-  o.pos = vec4f(p[v], 0.0, 1.0);
-  o.uv = (p[v] + vec2f(1.0)) * 0.5;
-  o.uv.y = 1.0 - o.uv.y;
-  return o;
-}
-
-fn lum(c: vec3f) -> f32 {
-  return dot(c, vec3f(0.2126, 0.7152, 0.0722));
-}
-
-@fragment
-fn fs(i: VsOut) -> @location(0) vec4f {
-  let c = textureSample(tex, samp, i.uv).rgb;
-  let op = u.op;
-
-  if (op == 0u) { // grayscale
-    let g = vec3f(lum(c));
-    return vec4f(g, 1.0);
-  }
-  if (op == 1u) { // invert
-    return vec4f(1.0 - c, 1.0);
-  }
-  if (op == 2u) { // edge (sobel \u5E45\u503C)
-    let e = 1.0 / vec2f(u.w, u.h);
-    let t00 = textureSample(tex, samp, i.uv + vec2f(-e.x, -e.y)).rgb;
-    let t10 = textureSample(tex, samp, i.uv + vec2f(0.0, -e.y)).rgb;
-    let t20 = textureSample(tex, samp, i.uv + vec2f(e.x, -e.y)).rgb;
-    let t01 = textureSample(tex, samp, i.uv + vec2f(-e.x, 0.0)).rgb;
-    let t21 = textureSample(tex, samp, i.uv + vec2f(e.x, 0.0)).rgb;
-    let t02 = textureSample(tex, samp, i.uv + vec2f(-e.x, e.y)).rgb;
-    let t12 = textureSample(tex, samp, i.uv + vec2f(0.0, e.y)).rgb;
-    let t22 = textureSample(tex, samp, i.uv + vec2f(e.x, e.y)).rgb;
-    let sx = (t22 + 2.0 * t21 + t02) - (t00 + 2.0 * t01 + t20);
-    let sy = (t02 + 2.0 * t12 + t22) - (t00 + 2.0 * t10 + t20);
-    let g = sqrt(vec3f(dot(sx, sx) / 3.0 + dot(sy, sy) / 3.0));
-    return vec4f(clamp(g * u.amount, vec3f(0.0), vec3f(1.0)), 1.0);
-  }
-  if (op == 3u) { // box blur,radius \u6298\u53E0\u6210\u6B65\u957F\u91C7\u6837
-    let r = f32(u.radius);
-    let e = vec2f(1.0) / vec2f(u.w, u.h) * r;
-    var s = vec3f(0.0);
-    var n = 0.0;
-    for (var dy = -2; dy <= 2; dy++) {
-      for (var dx = -2; dx <= 2; dx++) {
-        s = s + textureSample(tex, samp, i.uv + vec2f(f32(dx), f32(dy)) * e * 0.6).rgb;
-        n = n + 1.0;
-      }
-    }
-    return vec4f(s / n, 1.0);
-  }
-  if (op == 4u) { // sharpen(3x3 \u5377\u79EF)
-    let e = vec2f(1.0) / vec2f(u.w, u.h);
-    let c0 = textureSample(tex, samp, i.uv).rgb;
-    let t00 = textureSample(tex, samp, i.uv + vec2f(-e.x, -e.y)).rgb;
-    let t10 = textureSample(tex, samp, i.uv + vec2f(0.0, -e.y)).rgb;
-    let t20 = textureSample(tex, samp, i.uv + vec2f(e.x, -e.y)).rgb;
-    let t01 = textureSample(tex, samp, i.uv + vec2f(-e.x, 0.0)).rgb;
-    let t21 = textureSample(tex, samp, i.uv + vec2f(e.x, 0.0)).rgb;
-    let t02 = textureSample(tex, samp, i.uv + vec2f(-e.x, e.y)).rgb;
-    let t12 = textureSample(tex, samp, i.uv + vec2f(0.0, e.y)).rgb;
-    let t22 = textureSample(tex, samp, i.uv + vec2f(e.x, e.y)).rgb;
-    let k = u.amount;
-    let acc = t00 + t10 + t20 + t01 + t21 + t02 + t12 + t22;
-    return vec4f(clamp(c0 * (1.0 + 8.0 * k) - acc * k, vec3f(0.0), vec3f(1.0)), 1.0);
-  }
-  if (op == 5u) { // brightness
-    return vec4f(clamp(c + vec3f(u.value), vec3f(0.0), vec3f(1.0)), 1.0);
-  }
-  // contrast
-  let g = vec3f(0.5);
-  return vec4f(clamp(g + (c - g) * u.value, vec3f(0.0), vec3f(1.0)), 1.0);
-}
-`
-  );
-}
-export {
-  applyImage
-};