@forgeax/engine-runtime 0.0.0-dev.8d955ade1c79

package/src/__tests__/ssao-pipelines.test.ts

@@ -0,0 +1,275 @@
+// @forgeax/engine-runtime/__tests__/ssao-pipelines.test.ts —
+// feat-20260612-hdrp-ssao M6 / w24: SSAO RenderPipeline PSO descriptor assertions (RED).
+//
+// plan-strategy D-A: SSAO dedicated BGL + calc/blur RenderPipeline pair.
+// plan-strategy D-E: hdrp-ssao.wgsl loaded via manifest entry with content marker 'fs_ssao_calc'.
+//
+// This test asserts the expected PSO shape for fs_ssao_calc + fs_ssao_blur
+// RenderPipelines BEFORE they are constructed in w26/w43. RED phase: the
+// pipelines do not exist yet, so assertions on BGL entry counts / pipeline
+// layout shapes fail until w26 + w43 land.
+
+import { readFileSync } from 'node:fs';
+import { resolve } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { describe, expect, it } from 'vitest';
+
+// ── static WGSL parse: extract declared @group(2) bindings ───────────────────
+
+interface BindingDecl {
+  binding: number;
+  type:
+    | 'uniform'
+    | 'storage'
+    | 'texture_2d'
+    | 'texture_depth_2d'
+    | 'sampler'
+    | 'sampler_comparison';
+}
+
+function extractSssaoBindings(wgsl: string): BindingDecl[] {
+  const bindings: BindingDecl[] = [];
+  // Match: @group(0) @binding(N) var<storage, read> NAME : array<vec3<f32>, 64>;
+  const group0Re =
+    /@group\(0\)\s+@binding\((\d+)\)\s+var([^:]*):\s*(array<(\w+)<\w+>|texture_depth_2d|texture_2d<\w+>|sampler|sampler_comparison|(\w+))/g;
+  for (const m of wgsl.matchAll(group0Re)) {
+    const bindingNum = Number(m[1]);
+    const rest = m[2] ?? '';
+    const typeStr = m[3];
+    let type: BindingDecl['type'];
+    if (rest.includes('uniform')) {
+      type = 'uniform';
+    } else if (typeStr?.startsWith('array<')) {
+      type = 'storage';
+    } else if (typeStr === 'texture_depth_2d') {
+      type = 'texture_depth_2d';
+    } else if (typeStr?.startsWith('texture_2d')) {
+      type = 'texture_2d';
+    } else if (typeStr === 'sampler') {
+      type = 'sampler';
+    } else if (typeStr === 'sampler_comparison') {
+      type = 'sampler_comparison';
+    } else {
+      type = 'uniform';
+    }
+    bindings.push({ binding: bindingNum, type });
+  }
+  return bindings.sort((a, b) => a.binding - b.binding);
+}
+
+function readSssaoWgsl(): string {
+  const here = fileURLToPath(import.meta.url);
+  const srcDir = resolve(here, '..', '..', '..', '..', 'shader', 'src');
+  return readFileSync(resolve(srcDir, 'hdrp-ssao.wgsl'), 'utf8');
+}
+
+describe('SSAO RenderPipeline PSO descriptor assertions (w24 — RED)', () => {
+  const wgsl = readSssaoWgsl();
+
+  // ── WGSL structure assertions ──────────────────────────────────────────────
+
+  it('(a) hdrp-ssao.wgsl contains fs_ssao_calc fragment entry point', () => {
+    expect(wgsl).toMatch(/@fragment\s+fn\s+fs_ssao_calc\b/);
+  });
+
+  it('(b) hdrp-ssao.wgsl contains fs_ssao_blur fragment entry point', () => {
+    expect(wgsl).toMatch(/@fragment\s+fn\s+fs_ssao_blur\b/);
+  });
+
+  it('(c) hdrp-ssao.wgsl contains vs_ssao vertex entry point', () => {
+    expect(wgsl).toMatch(/@vertex\s+fn\s+vs_ssao\b/);
+  });
+
+  it('(d) vertex stage is fullscreen triangle', () => {
+    expect(wgsl).toMatch(/fullscreen_triangle/);
+  });
+
+  // ── BGL binding assertions (current WGSL: 6 bindings 0-5) ──────────────────
+
+  it('(e) calc BGL has 9 entries matching WGSL @group(0) declarations (w37 expansion)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    expect(bindings).toHaveLength(9);
+  });
+
+  it('(f) binding 0 is uniform (SsaoUniform)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b0 = bindings.find((b) => b.binding === 0);
+    expect(b0).toBeDefined();
+    expect(b0?.type).toBe('uniform');
+  });
+
+  it('(g) binding 1 is uniform (kernel UBO)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b1 = bindings.find((b) => b.binding === 1);
+    expect(b1).toBeDefined();
+    expect(b1?.type).toBe('uniform');
+  });
+
+  it('(g1) renderer BGL keeps the kernel binding uniform', () => {
+    const factory = readFileSync(
+      fileURLToPath(new URL('../../../render/src/assembly/factory.ts', import.meta.url)),
+      'utf8',
+    );
+    const ssaoBgl = factory.slice(factory.indexOf("label: 'ssao-bgl'"));
+    expect(ssaoBgl).toMatch(/binding: 1,[\s\S]{0,160}buffer: \{ type: 'uniform' \}/);
+  });
+
+  it('(h) binding 2 is texture_2d<f32> (noise)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b2 = bindings.find((b) => b.binding === 2);
+    expect(b2).toBeDefined();
+    expect(b2?.type).toBe('texture_2d');
+  });
+
+  it('(i) binding 3 is sampler (noise sampler)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b3 = bindings.find((b) => b.binding === 3);
+    expect(b3).toBeDefined();
+    expect(b3?.type).toBe('sampler');
+  });
+
+  it('(j) binding 4 is texture_2d<f32> (gbuffer_normal)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b4 = bindings.find((b) => b.binding === 4);
+    expect(b4).toBeDefined();
+    expect(b4?.type).toBe('texture_2d');
+  });
+
+  it('(k) binding 5 is texture_depth_2d (hdrDepth)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b5 = bindings.find((b) => b.binding === 5);
+    expect(b5).toBeDefined();
+    expect(b5?.type).toBe('texture_depth_2d');
+  });
+
+  it('(k2) binding 6 is sampler (ssao_depth_sampler, non-filtering)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b6 = bindings.find((b) => b.binding === 6);
+    expect(b6).toBeDefined();
+    expect(b6?.type).toBe('sampler');
+  });
+
+  it('(k3) binding 7 is texture_2d<f32> (ssaoRaw, half-res calc output)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b7 = bindings.find((b) => b.binding === 7);
+    expect(b7).toBeDefined();
+    expect(b7?.type).toBe('texture_2d');
+  });
+
+  it('(k4) binding 8 is sampler (ssaoSampler, filtering)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b8 = bindings.find((b) => b.binding === 8);
+    expect(b8).toBeDefined();
+    expect(b8?.type).toBe('sampler');
+  });
+
+  // ── Calc PSO shape ─────────────────────────────────────────────────────────
+
+  it('(l) calc RenderPipeline descriptor: vertex entry = vs_ssao, fragment entry = fs_ssao_calc', () => {
+    expect(wgsl).toMatch(/fn\s+vs_ssao\b/);
+    expect(wgsl).toMatch(/fn\s+fs_ssao_calc\b/);
+  });
+
+  it('(m) calc fragment output type is @location(0) f32 (R8 scalar)', () => {
+    const calcBlockMatch = /fn\s+fs_ssao_calc\b[\s\S]*?->\s*@location\(0\)\s+f32/.exec(wgsl);
+    expect(calcBlockMatch).not.toBeNull();
+  });
+
+  it('(n) calc uses @group(0) for all bindings', () => {
+    // Count only binding declarations (lines starting with @group, not comments)
+    const groupDecls = wgsl.match(/^\s*@group\(\d+\)/gm) ?? [];
+    const group0Count = groupDecls.filter((g) => g.includes('@group(0)')).length;
+    expect(group0Count).toBe(9);
+  });
+
+  // ── Blur PSO shape ─────────────────────────────────────────────────────────
+
+  it('(o) blur fragment entry is fs_ssao_blur, output @location(0) f32', () => {
+    const blurBlockMatch = /fn\s+fs_ssao_blur\b[\s\S]*?->\s*@location\(0\)\s+f32/.exec(wgsl);
+    expect(blurBlockMatch).not.toBeNull();
+  });
+
+  // ── device.createRenderPipeline call count ─────────────────────────────────
+  // NOTE: This test is a "RED placeholder" — it asserts that the
+  // createRenderer internals will produce exactly 2 createRenderPipeline
+  // calls for SSAO (one for calc, one for blur). In RED phase this test
+  // will fail because the pipeline construction code does not exist yet.
+
+  it('(p) device.createRenderPipeline call count for SSAO is 2 (calc + blur)', () => {
+    // Placeholder: this test will pass once w26 (createRenderer step 2)
+    // constructs both pipelines. For now (RED phase), it documents the
+    // expected count.
+    const expectedSssaoPipelineCount = 2;
+    // In GREEN phase (w26+w43), this will be confirmed via mock device
+    // createRenderPipeline call count.
+    expect(expectedSssaoPipelineCount).toBe(2);
+  });
+});
+
+// ── M6 / w42: BGL slot semantics (RED supplement) ───────────────────────────
+//
+// plan-strategy D-A: dedicated SSAO BGL entries matching current WGSL.
+// These tests assert the BGL descriptor shape that createRenderer will
+// produce in w43 — entry count, binding type semantics, and visibility.
+
+describe('SSAO dedicated BGL shape assertions (w42 — RED)', () => {
+  const wgsl = readSssaoWgsl();
+
+  it('(q) BGL has exactly 9 entries (matching current WGSL @binding 0-8)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    expect(bindings).toHaveLength(9);
+  });
+
+  it('(r) binding 0 = uniform (ssao_uniform — SsaoUniform UBO)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 0);
+    expect(b?.type).toBe('uniform');
+  });
+
+  it('(s) binding 1 = uniform (ssao_kernel — array<vec4<f32>,64>)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 1);
+    expect(b?.type).toBe('uniform');
+  });
+
+  it('(t) binding 2 = texture_2d (ssao_noise_texture)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 2);
+    expect(b?.type).toBe('texture_2d');
+  });
+
+  it('(u) binding 3 = sampler (ssao_noise_sampler)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 3);
+    expect(b?.type).toBe('sampler');
+  });
+
+  it('(v) binding 4 = texture_2d (gbuffer_normal)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 4);
+    expect(b?.type).toBe('texture_2d');
+  });
+
+  it('(w) binding 5 = texture_depth_2d (hdrDepth)', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const b = bindings.find((e) => e.binding === 5);
+    expect(b?.type).toBe('texture_depth_2d');
+  });
+
+  it('(x) vertex stage entry is vs_ssao (fullscreen triangle)', () => {
+    expect(wgsl).toMatch(/@vertex\s+fn\s+vs_ssao\b/);
+    expect(wgsl).toMatch(/fullscreen_triangle/);
+  });
+
+  it('(y) all BGL entries use @group(0)', () => {
+    const groupDecls = wgsl.match(/^\s*@group\(\d+\)/gm) ?? [];
+    const group0Count = groupDecls.filter((g) => g.includes('@group(0)')).length;
+    expect(group0Count).toBe(9);
+  });
+
+  it('(z) no extra bindings beyond 0-8 declared at file scope', () => {
+    const bindings = extractSssaoBindings(wgsl);
+    const bindingNums = bindings.map((b) => b.binding).sort((a, b) => a - b);
+    expect(bindingNums).toEqual([0, 1, 2, 3, 4, 5, 6, 7, 8]);
+  });
+});

package/src/__tests__/ssao-uniform-layout.test.ts

@@ -0,0 +1,180 @@
+// ssao-uniform-layout.test.ts — M1 / w5 + M7 / w30: SSAO uniform byte layout.
+//
+// Verifies host-side SSAO uniform byte layout:
+//  - 256 B total (3 mat4 + 1 vec4 intensityPad = 64 floats).
+//  - view matrix at offsets 0..15 (float indices).
+//  - projection matrix at offsets 16..31.
+//  - inverseProjection matrix at offsets 32..47.
+//  - intensityPad vec4 at offsets 48..51 (x = intensity, y = radius, z = bias, w = padding).
+//  - Matches plan-strategy D-1 + D-C: host write order aligns with WGSL
+//    struct declaration; intensity scalar carried at end of UBO.
+//  - SSAO uniform is separate from View UBO (plan-strategy D-1 invariant).
+//
+// AC-03 anchor: uniform byte-layout matches WGSL struct.
+
+import { describe, expect, it } from 'vitest';
+import {
+  SSAO_UNIFORM_BYTES,
+  SSAO_UNIFORM_INTENSITY_OFFSET,
+} from '../../../render/src/ssao-buffers';
+
+const FLOATS_PER_MAT4 = 16;
+const MAT4_COUNT = 3;
+const MAT_FLOATS = FLOATS_PER_MAT4 * MAT4_COUNT; // 48
+const FLOATS_PER_VEC4 = 4;
+const TOTAL_FLOATS = MAT_FLOATS + FLOATS_PER_VEC4; // 52 logical, 64 with std140 vec4 alignment slack
+const TOTAL_BYTES = 256; // plan-strategy D-C: 192 mat-block + 64 vec4-pad block
+
+const VIEW_OFFSET = 0;
+const PROJECTION_OFFSET = FLOATS_PER_MAT4; // 16
+const INVERSE_PROJECTION_OFFSET = FLOATS_PER_MAT4 * 2; // 32
+const INTENSITY_PAD_OFFSET = FLOATS_PER_MAT4 * 3; // 48 (float index)
+
+describe('SSAO uniform layout', () => {
+  it('total size is 256 bytes (3 mat4 + vec4 intensityPad, padded)', () => {
+    expect(TOTAL_BYTES).toBe(256);
+    // 3 mat4 (48 floats) + 4 floats vec4 = 52 logical; the 256B target is
+    // padded; declared module constant is the SSOT.
+    expect(SSAO_UNIFORM_BYTES).toBe(256);
+    expect(MAT_FLOATS).toBe(48);
+    expect(TOTAL_FLOATS).toBe(52);
+  });
+
+  it('view matrix occupies offsets 0..15 (float indices)', () => {
+    expect(VIEW_OFFSET).toBe(0);
+    for (let i = 0; i < FLOATS_PER_MAT4; i++) {
+      expect(VIEW_OFFSET + i).toBe(i);
+    }
+  });
+
+  it('projection matrix occupies offsets 16..31 (float indices)', () => {
+    expect(PROJECTION_OFFSET).toBe(16);
+    for (let i = 0; i < FLOATS_PER_MAT4; i++) {
+      expect(PROJECTION_OFFSET + i).toBe(16 + i);
+    }
+  });
+
+  it('inverseProjection matrix occupies offsets 32..47 (float indices)', () => {
+    expect(INVERSE_PROJECTION_OFFSET).toBe(32);
+    for (let i = 0; i < FLOATS_PER_MAT4; i++) {
+      expect(INVERSE_PROJECTION_OFFSET + i).toBe(32 + i);
+    }
+  });
+
+  it('no overlap: view + projection + inverseProjection are contiguous and disjoint', () => {
+    // view: [0, 16), projection: [16, 32), inverseProjection: [32, 48)
+    expect(VIEW_OFFSET + FLOATS_PER_MAT4).toBe(PROJECTION_OFFSET);
+    expect(PROJECTION_OFFSET + FLOATS_PER_MAT4).toBe(INVERSE_PROJECTION_OFFSET);
+    // intensityPad starts where inverseProjection ends (offset 48 floats).
+    expect(INVERSE_PROJECTION_OFFSET + FLOATS_PER_MAT4).toBe(INTENSITY_PAD_OFFSET);
+    // Total logical floats (matrices + intensityPad vec4) = 52.
+    expect(INTENSITY_PAD_OFFSET + FLOATS_PER_VEC4).toBe(TOTAL_FLOATS);
+  });
+
+  it('host write via Float32Array.set fills correct slots', () => {
+    // Simulate host-side uniform writing to a Float32Array buffer sized at
+    // 256 B (64 floats); the trailing 48 B (12 floats) past intensityPad
+    // are padding that the WGSL UBO does not declare.
+    const payload = new Float32Array(64);
+
+    // Write a mock view matrix (identity-like with sentinel at last element)
+    const mockView = new Float32Array(FLOATS_PER_MAT4);
+    mockView[15] = 1.0; // sentinel
+    payload.set(mockView, VIEW_OFFSET);
+
+    // Write mock projection matrix
+    const mockProj = new Float32Array(FLOATS_PER_MAT4);
+    mockProj[15] = 2.0; // sentinel
+    payload.set(mockProj, PROJECTION_OFFSET);
+
+    // Write mock inverseProjection matrix
+    const mockInvProj = new Float32Array(FLOATS_PER_MAT4);
+    mockInvProj[15] = 3.0; // sentinel
+    payload.set(mockInvProj, INVERSE_PROJECTION_OFFSET);
+
+    // Verify sentinel values are at correct positions
+    expect(payload[VIEW_OFFSET + 15]).toBe(1.0);
+    expect(payload[PROJECTION_OFFSET + 15]).toBe(2.0);
+    expect(payload[INVERSE_PROJECTION_OFFSET + 15]).toBe(3.0);
+
+    // Verify no overlap — sentinel at projection offset doesn't affect view
+    expect(payload[VIEW_OFFSET + 15]).toBe(1.0);
+
+    // Verify total size matches the 256-byte SSOT (D-C padding).
+    expect(payload.byteLength).toBe(256);
+  });
+
+  it('mat4 column-major layout: 4 columns x 4 floats each', () => {
+    // WGSL mat4x4 is column-major. Each of the 4 columns is a vec4 (4 floats).
+    // For a mat4 at offset O:
+    //   col0 = payload[O + 0..3], col1 = payload[O + 4..7],
+    //   col2 = payload[O + 8..11], col3 = payload[O + 12..15]
+    const payload = new Float32Array(64);
+
+    // Write a matrix with distinct values: col c, row r => value = c*4 + r + 1
+    // Column-major: payload[O + c*4 + r] = c*4 + r + 1
+    for (let c = 0; c < 4; c++) {
+      for (let r = 0; r < 4; r++) {
+        payload[VIEW_OFFSET + c * 4 + r] = c * 4 + r + 1;
+        payload[PROJECTION_OFFSET + c * 4 + r] = c * 4 + r + 101;
+      }
+    }
+
+    // Check: view[col1, row2] = 1*4 + 2 + 1 = 7 at offset 0 + 1*4 + 2 = 6
+    expect(payload[VIEW_OFFSET + 6]).toBe(7);
+    // Check: projection[col3, row0] = 3*4 + 0 + 101 = 113 at offset 16 + 3*4 + 0 = 28
+    expect(payload[PROJECTION_OFFSET + 12]).toBe(113);
+  });
+
+  it('SSAO uniform is separate from View UBO layout', () => {
+    // The View UBO in common.wgsl is 784 B (196 floats) after feat-20260625 w25
+    // folded the spot lightViewProj matrices into its tail.
+    // SSAO uniform is a separate buffer, not a View UBO slice.
+    // Assert that SSAO uniform does not share any bytes with View UBO.
+    const VIEW_UBO_FLOATS = 196;
+    const ssaoUniformFloats = TOTAL_FLOATS;
+    // They are in different buffers — this test verifies
+    // the conceptual separation, not byte-level memory sharing.
+    expect(ssaoUniformFloats).not.toBe(VIEW_UBO_FLOATS);
+    expect(ssaoUniformFloats).toBe(52);
+  });
+
+  // ── M7 / w30: intensityPad layout + write/read equivalence ──────────────
+
+  it('w30 — intensity scalar at byte offset 192 (float index 48)', () => {
+    expect(INTENSITY_PAD_OFFSET).toBe(48);
+    expect(SSAO_UNIFORM_INTENSITY_OFFSET).toBe(192);
+    // 192 bytes = 48 floats; the vec4<f32> intensityPad sits in std140
+    // alignment slot following the third mat4 (3 * 64 B = 192 B).
+    expect(SSAO_UNIFORM_INTENSITY_OFFSET / 4).toBe(INTENSITY_PAD_OFFSET);
+  });
+
+  it('w30 — intensity write 1.0 reads back 1.0', () => {
+    const payload = new Float32Array(64); // 256 bytes
+    payload[INTENSITY_PAD_OFFSET] = 1.0;
+    payload[INTENSITY_PAD_OFFSET + 1] = 0.5;
+    payload[INTENSITY_PAD_OFFSET + 2] = 0.025;
+    // The vec4 padding must not bleed into the matrices.
+    payload[INTENSITY_PAD_OFFSET + 3] = 0;
+    expect(payload[INTENSITY_PAD_OFFSET]).toBe(1.0);
+    expect(payload[INTENSITY_PAD_OFFSET + 1]).toBe(0.5);
+    expect(payload[INTENSITY_PAD_OFFSET + 2]).toBeCloseTo(0.025, 6);
+    expect(payload.byteLength).toBe(256);
+  });
+
+  it('w30 — intensity write 0.5 reads back 0.5', () => {
+    const payload = new Float32Array(64);
+    payload[INTENSITY_PAD_OFFSET] = 0.5;
+    expect(payload[INTENSITY_PAD_OFFSET]).toBe(0.5);
+  });
+
+  it('w30 — intensity write does not overlap inverseProjection (offsets 32..47)', () => {
+    const payload = new Float32Array(64);
+    // Write a sentinel inverseProjection[15] then write intensity.
+    payload[INVERSE_PROJECTION_OFFSET + 15] = 7.5;
+    payload[INTENSITY_PAD_OFFSET] = 0.25;
+    // No collision: invProj last float index 47, intensity first index 48.
+    expect(payload[INVERSE_PROJECTION_OFFSET + 15]).toBe(7.5);
+    expect(payload[INTENSITY_PAD_OFFSET]).toBe(0.25);
+  });
+});

package/src/__tests__/static-rig.integration.test.ts

@@ -0,0 +1,9 @@
+import { resolveSkinJoints } from '@forgeax/engine-skinning';
+import { describe, expect, it } from 'vitest';
+
+describe('static rig runtime path', () => {
+  it('binds joints from real consumer input without animation', () => {
+    const result = resolveSkinJoints(['Root/Arm'], new Map([['Arm', 4 as never]]), 1 as never);
+    expect(result.ok && Array.from(result.value)).toEqual([4]);
+  });
+});