@aardworx/wombat.rendering 0.9.18 → 0.9.19
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/runtime/derivedModes/gpuDispatcher.d.ts +46 -0
- package/dist/runtime/derivedModes/gpuDispatcher.d.ts.map +1 -0
- package/dist/runtime/derivedModes/gpuDispatcher.js +170 -0
- package/dist/runtime/derivedModes/gpuDispatcher.js.map +1 -0
- package/dist/runtime/derivedModes/gpuKernel.d.ts +9 -0
- package/dist/runtime/derivedModes/gpuKernel.d.ts.map +1 -0
- package/dist/runtime/derivedModes/gpuKernel.js +97 -0
- package/dist/runtime/derivedModes/gpuKernel.js.map +1 -0
- package/dist/runtime/derivedModes/rule.d.ts +33 -0
- package/dist/runtime/derivedModes/rule.d.ts.map +1 -1
- package/dist/runtime/derivedModes/rule.js +39 -0
- package/dist/runtime/derivedModes/rule.js.map +1 -1
- package/dist/runtime/heapScene.d.ts.map +1 -1
- package/dist/runtime/heapScene.js +78 -0
- package/dist/runtime/heapScene.js.map +1 -1
- package/dist/runtime/index.d.ts +3 -1
- package/dist/runtime/index.d.ts.map +1 -1
- package/dist/runtime/index.js +3 -1
- package/dist/runtime/index.js.map +1 -1
- package/package.json +1 -1
- package/src/runtime/derivedModes/gpuDispatcher.ts +190 -0
- package/src/runtime/derivedModes/gpuKernel.ts +98 -0
- package/src/runtime/derivedModes/rule.ts +62 -0
- package/src/runtime/heapScene.ts +76 -0
- package/src/runtime/index.ts +9 -0
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import type { CullMode } from "../pipelineCache/index.js";
|
|
2
|
+
export declare class GpuDerivedModesScene {
|
|
3
|
+
readonly device: GPUDevice;
|
|
4
|
+
private readonly entries;
|
|
5
|
+
private liveCount;
|
|
6
|
+
private dirty;
|
|
7
|
+
private inputBuf;
|
|
8
|
+
private outputBuf;
|
|
9
|
+
private stagingBuf;
|
|
10
|
+
private paramsBuf;
|
|
11
|
+
private pipeline;
|
|
12
|
+
private bindGroup;
|
|
13
|
+
private layout;
|
|
14
|
+
/** Capacity in number of u32 entries (= numROs the buffers can hold). */
|
|
15
|
+
private capacity;
|
|
16
|
+
/** Last-known arena buffer, used to detect bind-group invalidation. */
|
|
17
|
+
private arenaBuf;
|
|
18
|
+
/** Last-known declared (for kernel uniform). Common case: all RO
|
|
19
|
+
* rules share the same declared (= the SG scope's value). */
|
|
20
|
+
private declared;
|
|
21
|
+
/** Per-RO last-known output. CPU mirror, populated by finish(). */
|
|
22
|
+
readonly lastOutput: Uint32Array;
|
|
23
|
+
constructor(device: GPUDevice);
|
|
24
|
+
get registered(): number;
|
|
25
|
+
registerRo(drawId: number, modelRef: number, declared: CullMode): void;
|
|
26
|
+
deregisterRo(drawId: number): void;
|
|
27
|
+
markDirty(): void;
|
|
28
|
+
consumeDirty(): boolean;
|
|
29
|
+
private ensureResources;
|
|
30
|
+
/** Upload current per-RO input refs + params; called when dirty. */
|
|
31
|
+
private uploadInputs;
|
|
32
|
+
/**
|
|
33
|
+
* Encode the compute dispatch + the staging copy into `enc`. Caller
|
|
34
|
+
* is responsible for `device.queue.submit([enc.finish()])`. Returns
|
|
35
|
+
* the number of ROs the kernel was dispatched against.
|
|
36
|
+
*/
|
|
37
|
+
dispatch(arenaBuf: GPUBuffer, numROs: number, enc: GPUCommandEncoder): number;
|
|
38
|
+
/**
|
|
39
|
+
* mapAsync the staging buffer + copy values into `lastOutput`.
|
|
40
|
+
* Resolves on the GPU finishing the copy. Caller diffs lastOutput
|
|
41
|
+
* against its own cached values to find changed ROs.
|
|
42
|
+
*/
|
|
43
|
+
finish(numROs: number): Promise<void>;
|
|
44
|
+
dispose(): void;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=gpuDispatcher.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gpuDispatcher.d.ts","sourceRoot":"","sources":["../../../src/runtime/derivedModes/gpuDispatcher.ts"],"names":[],"mappings":"AAwBA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAc1D,qBAAa,oBAAoB;IAC/B,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;IAE3B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA+B;IACvD,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,KAAK,CAAS;IAGtB,OAAO,CAAC,QAAQ,CAA4B;IAC5C,OAAO,CAAC,SAAS,CAA2B;IAC5C,OAAO,CAAC,UAAU,CAA0B;IAC5C,OAAO,CAAC,SAAS,CAA2B;IAC5C,OAAO,CAAC,QAAQ,CAAqC;IACrD,OAAO,CAAC,SAAS,CAA8B;IAC/C,OAAO,CAAC,MAAM,CAAuC;IACrD,yEAAyE;IACzE,OAAO,CAAC,QAAQ,CAAK;IACrB,uEAAuE;IACvE,OAAO,CAAC,QAAQ,CAA0B;IAC1C;kEAC8D;IAC9D,OAAO,CAAC,QAAQ,CAA4B;IAE5C,mEAAmE;IACnE,QAAQ,CAAC,UAAU,EAAE,WAAW,CAAC;gBAErB,MAAM,EAAE,SAAS;IAK7B,IAAI,UAAU,IAAI,MAAM,CAA2B;IAEnD,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAG,IAAI;IAOtE,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAOlC,SAAS,IAAI,IAAI;IACjB,YAAY,IAAI,OAAO;IAEvB,OAAO,CAAC,eAAe;IAmDvB,oEAAoE;IACpE,OAAO,CAAC,YAAY;IAWpB;;;;OAIG;IACH,QAAQ,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,iBAAiB,GAAG,MAAM;IAa7E;;;;OAIG;IACG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS3C,OAAO,IAAI,IAAI;CAMhB"}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
// Scene-level dispatcher for the v1 GPU-evaluated derived-mode rule
|
|
2
|
+
// (flip-cull-by-determinant; see ./gpuKernel.ts).
|
|
3
|
+
//
|
|
4
|
+
// Lifecycle
|
|
5
|
+
// =========
|
|
6
|
+
//
|
|
7
|
+
// - Construct once per heap scene with the device.
|
|
8
|
+
// - registerRo(drawId, modelRef, declaredCull): register an RO that
|
|
9
|
+
// has the GPU flip-cull rule. modelRef = byte offset into the
|
|
10
|
+
// arena where the RO's ModelTrafo mat4 lives.
|
|
11
|
+
// - deregisterRo(drawId): unregister on removeDraw.
|
|
12
|
+
// - markDirty(): caller hints that an input changed; the next
|
|
13
|
+
// `dispatch()` will run the kernel.
|
|
14
|
+
// - dispatch(arenaBuf, enc): if dirty, encode the compute pass into
|
|
15
|
+
// `enc`. Always copies the output buffer into a staging buffer for
|
|
16
|
+
// CPU readback.
|
|
17
|
+
// - finish(): mapAsync the staging buffer and resolve with the
|
|
18
|
+
// per-RO output array. The scene calls this AFTER submit and
|
|
19
|
+
// before deciding rebuckets. The promise resolves on the next
|
|
20
|
+
// microtask after the GPU completes the copy.
|
|
21
|
+
//
|
|
22
|
+
// The buffers grow lazily (pow2) as ROs are added.
|
|
23
|
+
import { GPU_FLIP_CULL_BY_DET_WGSL, CULL_TO_U32 } from "./gpuKernel.js";
|
|
24
|
+
const NO_RULE = 0xFFFFFFFF;
|
|
25
|
+
const WG_SIZE = 64;
|
|
26
|
+
const POW2 = (n) => { let p = 1; while (p < n)
|
|
27
|
+
p <<= 1; return Math.max(64, p); };
|
|
28
|
+
export class GpuDerivedModesScene {
|
|
29
|
+
device;
|
|
30
|
+
entries = [];
|
|
31
|
+
liveCount = 0;
|
|
32
|
+
dirty = false;
|
|
33
|
+
// GPU resources — lazily allocated on first registerRo.
|
|
34
|
+
inputBuf = null;
|
|
35
|
+
outputBuf = null;
|
|
36
|
+
stagingBuf = null;
|
|
37
|
+
paramsBuf = null;
|
|
38
|
+
pipeline = null;
|
|
39
|
+
bindGroup = null;
|
|
40
|
+
layout = null;
|
|
41
|
+
/** Capacity in number of u32 entries (= numROs the buffers can hold). */
|
|
42
|
+
capacity = 0;
|
|
43
|
+
/** Last-known arena buffer, used to detect bind-group invalidation. */
|
|
44
|
+
arenaBuf = null;
|
|
45
|
+
/** Last-known declared (for kernel uniform). Common case: all RO
|
|
46
|
+
* rules share the same declared (= the SG scope's value). */
|
|
47
|
+
declared = CULL_TO_U32.back;
|
|
48
|
+
/** Per-RO last-known output. CPU mirror, populated by finish(). */
|
|
49
|
+
lastOutput;
|
|
50
|
+
constructor(device) {
|
|
51
|
+
this.device = device;
|
|
52
|
+
this.lastOutput = new Uint32Array(0);
|
|
53
|
+
}
|
|
54
|
+
get registered() { return this.liveCount; }
|
|
55
|
+
registerRo(drawId, modelRef, declared) {
|
|
56
|
+
this.entries[drawId] = { modelRef, declaredU32: CULL_TO_U32[declared] };
|
|
57
|
+
this.liveCount++;
|
|
58
|
+
this.dirty = true;
|
|
59
|
+
this.declared = CULL_TO_U32[declared];
|
|
60
|
+
}
|
|
61
|
+
deregisterRo(drawId) {
|
|
62
|
+
if (this.entries[drawId] === undefined)
|
|
63
|
+
return;
|
|
64
|
+
this.entries[drawId] = undefined;
|
|
65
|
+
this.liveCount--;
|
|
66
|
+
this.dirty = true;
|
|
67
|
+
}
|
|
68
|
+
markDirty() { this.dirty = true; }
|
|
69
|
+
consumeDirty() { const was = this.dirty; this.dirty = false; return was; }
|
|
70
|
+
ensureResources(arenaBuf, numROs) {
|
|
71
|
+
const needCap = POW2(Math.max(1, numROs));
|
|
72
|
+
if (this.outputBuf === null || needCap > this.capacity) {
|
|
73
|
+
this.inputBuf?.destroy();
|
|
74
|
+
this.outputBuf?.destroy();
|
|
75
|
+
this.stagingBuf?.destroy();
|
|
76
|
+
const bytes = needCap * 4;
|
|
77
|
+
this.inputBuf = this.device.createBuffer({ size: bytes, usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST, label: "gpuDerivedModes/in" });
|
|
78
|
+
this.outputBuf = this.device.createBuffer({ size: bytes, usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST, label: "gpuDerivedModes/out" });
|
|
79
|
+
this.stagingBuf = this.device.createBuffer({ size: bytes, usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST, label: "gpuDerivedModes/staging" });
|
|
80
|
+
this.lastOutput = new Uint32Array(needCap);
|
|
81
|
+
this.capacity = needCap;
|
|
82
|
+
this.bindGroup = null;
|
|
83
|
+
}
|
|
84
|
+
if (this.paramsBuf === null) {
|
|
85
|
+
this.paramsBuf = this.device.createBuffer({ size: 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST, label: "gpuDerivedModes/params" });
|
|
86
|
+
}
|
|
87
|
+
if (this.layout === null) {
|
|
88
|
+
this.layout = this.device.createBindGroupLayout({
|
|
89
|
+
label: "gpuDerivedModes/bgl",
|
|
90
|
+
entries: [
|
|
91
|
+
{ binding: 0, visibility: GPUShaderStage.COMPUTE, buffer: { type: "read-only-storage" } },
|
|
92
|
+
{ binding: 1, visibility: GPUShaderStage.COMPUTE, buffer: { type: "read-only-storage" } },
|
|
93
|
+
{ binding: 2, visibility: GPUShaderStage.COMPUTE, buffer: { type: "storage" } },
|
|
94
|
+
{ binding: 3, visibility: GPUShaderStage.COMPUTE, buffer: { type: "uniform" } },
|
|
95
|
+
],
|
|
96
|
+
});
|
|
97
|
+
}
|
|
98
|
+
if (this.pipeline === null) {
|
|
99
|
+
const module = this.device.createShaderModule({ code: GPU_FLIP_CULL_BY_DET_WGSL, label: "gpuDerivedModes/module" });
|
|
100
|
+
this.pipeline = this.device.createComputePipeline({
|
|
101
|
+
label: "gpuDerivedModes/pipeline",
|
|
102
|
+
layout: this.device.createPipelineLayout({ bindGroupLayouts: [this.layout] }),
|
|
103
|
+
compute: { module, entryPoint: "evaluate" },
|
|
104
|
+
});
|
|
105
|
+
}
|
|
106
|
+
if (this.bindGroup === null || this.arenaBuf !== arenaBuf) {
|
|
107
|
+
this.bindGroup = this.device.createBindGroup({
|
|
108
|
+
label: "gpuDerivedModes/bindGroup",
|
|
109
|
+
layout: this.layout,
|
|
110
|
+
entries: [
|
|
111
|
+
{ binding: 0, resource: { buffer: arenaBuf } },
|
|
112
|
+
{ binding: 1, resource: { buffer: this.inputBuf } },
|
|
113
|
+
{ binding: 2, resource: { buffer: this.outputBuf } },
|
|
114
|
+
{ binding: 3, resource: { buffer: this.paramsBuf } },
|
|
115
|
+
],
|
|
116
|
+
});
|
|
117
|
+
this.arenaBuf = arenaBuf;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
/** Upload current per-RO input refs + params; called when dirty. */
|
|
121
|
+
uploadInputs(numROs) {
|
|
122
|
+
const refs = new Uint32Array(this.capacity).fill(NO_RULE);
|
|
123
|
+
for (let i = 0; i < numROs; i++) {
|
|
124
|
+
const e = this.entries[i];
|
|
125
|
+
if (e !== undefined)
|
|
126
|
+
refs[i] = e.modelRef;
|
|
127
|
+
}
|
|
128
|
+
this.device.queue.writeBuffer(this.inputBuf, 0, refs.buffer, refs.byteOffset, refs.byteLength);
|
|
129
|
+
const params = new Uint32Array([numROs, this.declared, 0, 0]);
|
|
130
|
+
this.device.queue.writeBuffer(this.paramsBuf, 0, params.buffer, params.byteOffset, params.byteLength);
|
|
131
|
+
}
|
|
132
|
+
/**
|
|
133
|
+
* Encode the compute dispatch + the staging copy into `enc`. Caller
|
|
134
|
+
* is responsible for `device.queue.submit([enc.finish()])`. Returns
|
|
135
|
+
* the number of ROs the kernel was dispatched against.
|
|
136
|
+
*/
|
|
137
|
+
dispatch(arenaBuf, numROs, enc) {
|
|
138
|
+
this.ensureResources(arenaBuf, numROs);
|
|
139
|
+
this.uploadInputs(numROs);
|
|
140
|
+
const pass = enc.beginComputePass({ label: "gpuDerivedModes/pass" });
|
|
141
|
+
pass.setPipeline(this.pipeline);
|
|
142
|
+
pass.setBindGroup(0, this.bindGroup);
|
|
143
|
+
pass.dispatchWorkgroups(Math.ceil(numROs / WG_SIZE));
|
|
144
|
+
pass.end();
|
|
145
|
+
const bytes = numROs * 4;
|
|
146
|
+
enc.copyBufferToBuffer(this.outputBuf, 0, this.stagingBuf, 0, bytes);
|
|
147
|
+
return numROs;
|
|
148
|
+
}
|
|
149
|
+
/**
|
|
150
|
+
* mapAsync the staging buffer + copy values into `lastOutput`.
|
|
151
|
+
* Resolves on the GPU finishing the copy. Caller diffs lastOutput
|
|
152
|
+
* against its own cached values to find changed ROs.
|
|
153
|
+
*/
|
|
154
|
+
async finish(numROs) {
|
|
155
|
+
if (this.stagingBuf === null || numROs === 0)
|
|
156
|
+
return;
|
|
157
|
+
const bytes = numROs * 4;
|
|
158
|
+
await this.stagingBuf.mapAsync(GPUMapMode.READ, 0, bytes);
|
|
159
|
+
const view = new Uint32Array(this.stagingBuf.getMappedRange(0, bytes).slice(0));
|
|
160
|
+
this.stagingBuf.unmap();
|
|
161
|
+
this.lastOutput.set(view, 0);
|
|
162
|
+
}
|
|
163
|
+
dispose() {
|
|
164
|
+
this.inputBuf?.destroy();
|
|
165
|
+
this.outputBuf?.destroy();
|
|
166
|
+
this.stagingBuf?.destroy();
|
|
167
|
+
this.paramsBuf?.destroy();
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
//# sourceMappingURL=gpuDispatcher.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gpuDispatcher.js","sourceRoot":"","sources":["../../../src/runtime/derivedModes/gpuDispatcher.ts"],"names":[],"mappings":"AAAA,oEAAoE;AACpE,kDAAkD;AAClD,EAAE;AACF,YAAY;AACZ,YAAY;AACZ,EAAE;AACF,qDAAqD;AACrD,sEAAsE;AACtE,kEAAkE;AAClE,kDAAkD;AAClD,sDAAsD;AACtD,gEAAgE;AAChE,wCAAwC;AACxC,sEAAsE;AACtE,uEAAuE;AACvE,oBAAoB;AACpB,iEAAiE;AACjE,iEAAiE;AACjE,kEAAkE;AAClE,kDAAkD;AAClD,EAAE;AACF,mDAAmD;AAEnD,OAAO,EAAE,yBAAyB,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAGxE,MAAM,OAAO,GAAG,UAAU,CAAC;AAS3B,MAAM,OAAO,GAAG,EAAE,CAAC;AACnB,MAAM,IAAI,GAAG,CAAC,CAAS,EAAU,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IAAE,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAElG,MAAM,OAAO,oBAAoB;IACtB,MAAM,CAAY;IAEV,OAAO,GAA4B,EAAE,CAAC;IAC/C,SAAS,GAAG,CAAC,CAAC;IACd,KAAK,GAAG,KAAK,CAAC;IAEtB,wDAAwD;IAChD,QAAQ,GAAuB,IAAI,CAAC;IACpC,SAAS,GAAsB,IAAI,CAAC;IACpC,UAAU,GAAqB,IAAI,CAAC;IACpC,SAAS,GAAsB,IAAI,CAAC;IACpC,QAAQ,GAAgC,IAAI,CAAC;IAC7C,SAAS,GAAyB,IAAI,CAAC;IACvC,MAAM,GAAkC,IAAI,CAAC;IACrD,yEAAyE;IACjE,QAAQ,GAAG,CAAC,CAAC;IACrB,uEAAuE;IAC/D,QAAQ,GAAqB,IAAI,CAAC;IAC1C;kEAC8D;IACtD,QAAQ,GAAW,WAAW,CAAC,IAAI,CAAC;IAE5C,mEAAmE;IAC1D,UAAU,CAAc;IAEjC,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,UAAU,KAAa,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAEnD,UAAU,CAAC,MAAc,EAAE,QAAgB,EAAE,QAAkB;QAC7D,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxE,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,SAAS;YAAE,OAAO;QAC/C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,SAAS,KAAW,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC;IACxC,YAAY,KAAc,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;IAE3E,eAAe,CAAC,QAAmB,EAAE,MAAc;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACvD,IAAI,CAAC,QAAU,EAAE,OAAO,EAAE,CAAC;YAC3B,IAAI,CAAC,SAAU,EAAE,OAAO,EAAE,CAAC;YAC3B,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,OAAO,GAAG,CAAC,CAAC;YAC1B,IAAI,CAAC,QAAQ,GAAK,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAClJ,IAAI,CAAC,SAAS,GAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC,CAAC;YAC7K,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,CAAC;YACvJ,IAAoC,CAAC,UAAU,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC;YAC5E,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;YAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC,CAAC;QACpJ,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;gBAC9C,KAAK,EAAE,qBAAqB;gBAC5B,OAAO,EAAE;oBACP,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE;oBACzF,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE;oBACzF,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC/E,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;iBAChF;aACF,CAAC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC,CAAC;YACpH,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;gBAChD,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC7E,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE;aAC5C,CAAC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;gBAC3C,KAAK,EAAE,2BAA2B;gBAClC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE;oBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;oBAC9C,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,QAAS,EAAG,EAAE;oBACrD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,SAAU,EAAE,EAAE;oBACrD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,SAAU,EAAE,EAAE;iBACtD;aACF,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,oEAAoE;IAC5D,YAAY,CAAC,MAAc;QACjC,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,KAAK,SAAS;gBAAE,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,QAAS,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAChG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,SAAU,EAAE,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;IACzG,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,QAAmB,EAAE,MAAc,EAAE,GAAsB;QAClE,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,GAAG,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAS,CAAC,CAAC;QACjC,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,IAAI,CAAC,SAAU,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC;QACzB,GAAG,CAAC,kBAAkB,CAAC,IAAI,CAAC,SAAU,EAAE,CAAC,EAAE,IAAI,CAAC,UAAW,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QACvE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,MAAM,CAAC,MAAc;QACzB,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,MAAM,KAAK,CAAC;YAAE,OAAO;QACrD,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC;QACzB,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACvB,IAAoC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAU,EAAE,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAU,EAAE,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAU,EAAE,OAAO,EAAE,CAAC;IAC7B,CAAC;CACF"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export declare const GPU_FLIP_CULL_BY_DET_WGSL = "\n// Per-RO record. Index i corresponds to drawId i in the scene's\n// flat RO table.\nstruct GpuRuleParams {\n numROs: u32,\n declared: u32, // 0 = none, 1 = front, 2 = back (must match CULL_MODE[] order in bitfield.ts)\n _pad0: u32,\n _pad1: u32,\n};\n\n@group(0) @binding(0) var<storage, read> arena: array<u32>;\n// One per-RO entry: byte offset within the arena of the ROs ModelTrafo\n// allocation header. Data is at offset + 16 (ALLOC_HEADER_PAD_TO).\n// 0xFFFFFFFFu means \"this RO does NOT have a flip-cull-by-det rule\".\n@group(0) @binding(1) var<storage, read> roInputRefs: array<u32>;\n@group(0) @binding(2) var<storage, read_write> roOutputs: array<u32>;\n@group(0) @binding(3) var<uniform> uIn: GpuRuleParams;\n\n// Read a row-major mat4 from the arena at byte offset `refBytes + 16`\n// (header is 16 bytes wide due to ALLOC_HEADER_PAD_TO). Returns the\n// 3\u00D73 upper-left in row-major order: [m00,m01,m02, m10,m11,m12, m20,m21,m22].\nfn loadUpper3x3(refBytes: u32) -> array<f32, 9> {\n let baseU32 = (refBytes + 16u) >> 2u;\n // The arena is a Float32 storage of mat4 in row-major, 16 floats.\n // Read 9 of them.\n return array<f32, 9>(\n bitcast<f32>(arena[baseU32 + 0u]), bitcast<f32>(arena[baseU32 + 1u]), bitcast<f32>(arena[baseU32 + 2u]),\n bitcast<f32>(arena[baseU32 + 4u]), bitcast<f32>(arena[baseU32 + 5u]), bitcast<f32>(arena[baseU32 + 6u]),\n bitcast<f32>(arena[baseU32 + 8u]), bitcast<f32>(arena[baseU32 + 9u]), bitcast<f32>(arena[baseU32 + 10u]),\n );\n}\n\nfn det3x3(m: array<f32, 9>) -> f32 {\n return m[0] * (m[4] * m[8] - m[5] * m[7])\n - m[1] * (m[3] * m[8] - m[5] * m[6])\n + m[2] * (m[3] * m[7] - m[4] * m[6]);\n}\n\n// Mirror of bitfield.ts CULL_MODE[] order: 0=none, 1=front, 2=back.\nfn flipCull(c: u32) -> u32 {\n if (c == 1u) { return 2u; } // front -> back\n if (c == 2u) { return 1u; } // back -> front\n return c; // none\n}\n\n@compute @workgroup_size(64)\nfn evaluate(@builtin(global_invocation_id) gid: vec3<u32>) {\n let i = gid.x;\n if (i >= uIn.numROs) { return; }\n let ref = roInputRefs[i];\n if (ref == 0xFFFFFFFFu) {\n // No rule on this RO; leave existing output untouched. Allows a\n // single kernel dispatch to skip non-participating ROs.\n return;\n }\n let m = loadUpper3x3(ref);\n let d = det3x3(m);\n let declared = uIn.declared;\n let outVal = select(declared, flipCull(declared), d < 0.0);\n roOutputs[i] = outVal;\n}\n";
|
|
2
|
+
/** Map between bitfield.ts's CULL_MODE order and the kernel's u32 enum. */
|
|
3
|
+
export declare const CULL_TO_U32: {
|
|
4
|
+
readonly none: 0;
|
|
5
|
+
readonly front: 1;
|
|
6
|
+
readonly back: 2;
|
|
7
|
+
};
|
|
8
|
+
export declare const U32_TO_CULL: readonly ["none", "front", "back"];
|
|
9
|
+
//# sourceMappingURL=gpuKernel.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gpuKernel.d.ts","sourceRoot":"","sources":["../../../src/runtime/derivedModes/gpuKernel.ts"],"names":[],"mappings":"AAgCA,eAAO,MAAM,yBAAyB,s+EA6DrC,CAAC;AAEF,2EAA2E;AAC3E,eAAO,MAAM,WAAW;;;;CAAgD,CAAC;AACzE,eAAO,MAAM,WAAW,oCAAqC,CAAC"}
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
// GPU-evaluated derived-mode rules — minimum viable kernel.
|
|
2
|
+
//
|
|
3
|
+
// CPU eval (Task 2 Phase 1-3, 0.9.17) runs rule closures in JS. For
|
|
4
|
+
// rules whose inputs live GPU-side (per-RO trafos, etc.) JS evaluation
|
|
5
|
+
// fans out CPU work that's wasted: the trafo bytes are right there in
|
|
6
|
+
// the arena. This module runs the rule on the GPU instead, in one
|
|
7
|
+
// dispatch over all dirty ROs per frame.
|
|
8
|
+
//
|
|
9
|
+
// v1 scope is deliberately narrow: ONE hard-coded WGSL kernel for
|
|
10
|
+
// "flip cullMode by sign of det(upper3x3 of an arena mat4)". Picked
|
|
11
|
+
// because it's the motivating case from docs/derived-modes.md and
|
|
12
|
+
// proves the GPU-eval path end-to-end. Generalizes via a traced IR
|
|
13
|
+
// builder in v2 (see derived-mode-rules-plan.md Phase 5).
|
|
14
|
+
//
|
|
15
|
+
// Architecture
|
|
16
|
+
// ============
|
|
17
|
+
//
|
|
18
|
+
// - One **module-wide** `GpuDerivedModesScene` per heap scene. Holds:
|
|
19
|
+
// * `roOutputBuf` — `array<u32>` sized to scene.numROs.
|
|
20
|
+
// Each entry = the rule's modeKey contribution.
|
|
21
|
+
// * `roInputBuf` — `array<u32>` (one ref-into-arena per RO)
|
|
22
|
+
// telling the kernel where to read the input.
|
|
23
|
+
// * `stagingBuf` — MAP_READ + COPY_DST mirror of roOutputBuf,
|
|
24
|
+
// for the per-frame readback.
|
|
25
|
+
// - Dispatch each frame in `update()`, ONLY when something dirty.
|
|
26
|
+
// - `mapAsync(staging)` resolves on a microtask; the CPU diffs the
|
|
27
|
+
// buffer against `roLastKey[]`. Any change → mark drawId dirty;
|
|
28
|
+
// the existing rebucket flow does the rest.
|
|
29
|
+
//
|
|
30
|
+
// The dirty gate keeps steady-state cost ≈ 0 — when no input aval
|
|
31
|
+
// marks AND no RO add/remove, we skip the dispatch + readback.
|
|
32
|
+
export const GPU_FLIP_CULL_BY_DET_WGSL = /* wgsl */ `
|
|
33
|
+
// Per-RO record. Index i corresponds to drawId i in the scene's
|
|
34
|
+
// flat RO table.
|
|
35
|
+
struct GpuRuleParams {
|
|
36
|
+
numROs: u32,
|
|
37
|
+
declared: u32, // 0 = none, 1 = front, 2 = back (must match CULL_MODE[] order in bitfield.ts)
|
|
38
|
+
_pad0: u32,
|
|
39
|
+
_pad1: u32,
|
|
40
|
+
};
|
|
41
|
+
|
|
42
|
+
@group(0) @binding(0) var<storage, read> arena: array<u32>;
|
|
43
|
+
// One per-RO entry: byte offset within the arena of the ROs ModelTrafo
|
|
44
|
+
// allocation header. Data is at offset + 16 (ALLOC_HEADER_PAD_TO).
|
|
45
|
+
// 0xFFFFFFFFu means "this RO does NOT have a flip-cull-by-det rule".
|
|
46
|
+
@group(0) @binding(1) var<storage, read> roInputRefs: array<u32>;
|
|
47
|
+
@group(0) @binding(2) var<storage, read_write> roOutputs: array<u32>;
|
|
48
|
+
@group(0) @binding(3) var<uniform> uIn: GpuRuleParams;
|
|
49
|
+
|
|
50
|
+
// Read a row-major mat4 from the arena at byte offset \`refBytes + 16\`
|
|
51
|
+
// (header is 16 bytes wide due to ALLOC_HEADER_PAD_TO). Returns the
|
|
52
|
+
// 3×3 upper-left in row-major order: [m00,m01,m02, m10,m11,m12, m20,m21,m22].
|
|
53
|
+
fn loadUpper3x3(refBytes: u32) -> array<f32, 9> {
|
|
54
|
+
let baseU32 = (refBytes + 16u) >> 2u;
|
|
55
|
+
// The arena is a Float32 storage of mat4 in row-major, 16 floats.
|
|
56
|
+
// Read 9 of them.
|
|
57
|
+
return array<f32, 9>(
|
|
58
|
+
bitcast<f32>(arena[baseU32 + 0u]), bitcast<f32>(arena[baseU32 + 1u]), bitcast<f32>(arena[baseU32 + 2u]),
|
|
59
|
+
bitcast<f32>(arena[baseU32 + 4u]), bitcast<f32>(arena[baseU32 + 5u]), bitcast<f32>(arena[baseU32 + 6u]),
|
|
60
|
+
bitcast<f32>(arena[baseU32 + 8u]), bitcast<f32>(arena[baseU32 + 9u]), bitcast<f32>(arena[baseU32 + 10u]),
|
|
61
|
+
);
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
fn det3x3(m: array<f32, 9>) -> f32 {
|
|
65
|
+
return m[0] * (m[4] * m[8] - m[5] * m[7])
|
|
66
|
+
- m[1] * (m[3] * m[8] - m[5] * m[6])
|
|
67
|
+
+ m[2] * (m[3] * m[7] - m[4] * m[6]);
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
// Mirror of bitfield.ts CULL_MODE[] order: 0=none, 1=front, 2=back.
|
|
71
|
+
fn flipCull(c: u32) -> u32 {
|
|
72
|
+
if (c == 1u) { return 2u; } // front -> back
|
|
73
|
+
if (c == 2u) { return 1u; } // back -> front
|
|
74
|
+
return c; // none
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
@compute @workgroup_size(64)
|
|
78
|
+
fn evaluate(@builtin(global_invocation_id) gid: vec3<u32>) {
|
|
79
|
+
let i = gid.x;
|
|
80
|
+
if (i >= uIn.numROs) { return; }
|
|
81
|
+
let ref = roInputRefs[i];
|
|
82
|
+
if (ref == 0xFFFFFFFFu) {
|
|
83
|
+
// No rule on this RO; leave existing output untouched. Allows a
|
|
84
|
+
// single kernel dispatch to skip non-participating ROs.
|
|
85
|
+
return;
|
|
86
|
+
}
|
|
87
|
+
let m = loadUpper3x3(ref);
|
|
88
|
+
let d = det3x3(m);
|
|
89
|
+
let declared = uIn.declared;
|
|
90
|
+
let outVal = select(declared, flipCull(declared), d < 0.0);
|
|
91
|
+
roOutputs[i] = outVal;
|
|
92
|
+
}
|
|
93
|
+
`;
|
|
94
|
+
/** Map between bitfield.ts's CULL_MODE order and the kernel's u32 enum. */
|
|
95
|
+
export const CULL_TO_U32 = { "none": 0, "front": 1, "back": 2 };
|
|
96
|
+
export const U32_TO_CULL = ["none", "front", "back"];
|
|
97
|
+
//# sourceMappingURL=gpuKernel.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gpuKernel.js","sourceRoot":"","sources":["../../../src/runtime/derivedModes/gpuKernel.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,EAAE;AACF,oEAAoE;AACpE,uEAAuE;AACvE,sEAAsE;AACtE,kEAAkE;AAClE,yCAAyC;AACzC,EAAE;AACF,kEAAkE;AAClE,oEAAoE;AACpE,kEAAkE;AAClE,mEAAmE;AACnE,0DAA0D;AAC1D,EAAE;AACF,eAAe;AACf,eAAe;AACf,EAAE;AACF,wEAAwE;AACxE,+DAA+D;AAC/D,yEAAyE;AACzE,oEAAoE;AACpE,uEAAuE;AACvE,sEAAsE;AACtE,uDAAuD;AACvD,oEAAoE;AACpE,qEAAqE;AACrE,oEAAoE;AACpE,gDAAgD;AAChD,EAAE;AACF,kEAAkE;AAClE,+DAA+D;AAE/D,MAAM,CAAC,MAAM,yBAAyB,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6DnD,CAAC;AAEF,2EAA2E;AAC3E,MAAM,CAAC,MAAM,WAAW,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAW,CAAC;AACzE,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAU,CAAC"}
|
|
@@ -17,10 +17,28 @@ export type ModeValue<A extends ModeAxis> = A extends "cull" ? CullMode : A exte
|
|
|
17
17
|
* want to pass declared through unchanged.
|
|
18
18
|
*/
|
|
19
19
|
export type DerivedModeBuilder<A extends ModeAxis> = (u: Record<string, unknown>, declared: ModeValue<A>) => ModeValue<A>;
|
|
20
|
+
/**
|
|
21
|
+
* Opt-in GPU evaluation marker. When present on a rule, the heap
|
|
22
|
+
* scene runs a hard-coded WGSL kernel (see `./gpuKernel.ts`) instead
|
|
23
|
+
* of the CPU closure. The closure is retained for fallback (e.g.
|
|
24
|
+
* MockGPU tests, devices without compute, or before frame 0).
|
|
25
|
+
*
|
|
26
|
+
* v1 supports one kernel: `"flipCullByDeterminant"`. v2 generalizes
|
|
27
|
+
* via a traced IR builder.
|
|
28
|
+
*/
|
|
29
|
+
export interface GpuRuleSpec<A extends ModeAxis = ModeAxis> {
|
|
30
|
+
readonly kernel: "flipCullByDeterminant";
|
|
31
|
+
/** Name of the input uniform the kernel reads (e.g. "ModelTrafo"). */
|
|
32
|
+
readonly inputUniform: string;
|
|
33
|
+
/** Declared (SG-scope) value the kernel mutates by. */
|
|
34
|
+
readonly declared: ModeValue<A>;
|
|
35
|
+
}
|
|
20
36
|
export interface DerivedModeRule<A extends ModeAxis = ModeAxis> {
|
|
21
37
|
readonly __derivedModeRule: true;
|
|
22
38
|
readonly axis: A;
|
|
23
39
|
readonly evaluate: DerivedModeBuilder<A>;
|
|
40
|
+
/** Opt-in GPU evaluation. Absent → CPU closure path. */
|
|
41
|
+
readonly gpu?: GpuRuleSpec<A>;
|
|
24
42
|
/**
|
|
25
43
|
* Conservative over-approximation of the rule's output set. Used by
|
|
26
44
|
* the build-time pre-warm in v2 to enumerate the pipelines a scene
|
|
@@ -56,4 +74,19 @@ export declare function isDerivedModeRule(x: unknown): x is DerivedModeRule;
|
|
|
56
74
|
/** Convenience: flip 'back' ↔ 'front' (and pass through 'none').
|
|
57
75
|
* Useful in `(u, declared) => mirrored ? flipCull(declared) : declared`. */
|
|
58
76
|
export declare function flipCull(c: CullMode): CullMode;
|
|
77
|
+
/**
|
|
78
|
+
* Construct a derived-mode rule that flips cullMode based on the sign
|
|
79
|
+
* of `det(upperLeft3x3(u.<inputUniform>))`. Runs on the GPU when
|
|
80
|
+
* heapScene's compute path is available; the same logic is mirrored
|
|
81
|
+
* in the CPU closure for tests and fallback.
|
|
82
|
+
*
|
|
83
|
+
* const rule = gpuFlipCullByDeterminant("ModelTrafo", "back");
|
|
84
|
+
* spec.modeRules = { cull: rule };
|
|
85
|
+
*
|
|
86
|
+
* Useful for mirrored geometry: scenes that animate a per-RO trafo
|
|
87
|
+
* with potentially-negative scale don't pay CPU per-RO determinant
|
|
88
|
+
* cost — the kernel walks the arena once per frame for ALL ROs that
|
|
89
|
+
* carry this rule.
|
|
90
|
+
*/
|
|
91
|
+
export declare function gpuFlipCullByDeterminant(inputUniform: string, declared?: CullMode): DerivedModeRule<"cull">;
|
|
59
92
|
//# sourceMappingURL=rule.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rule.d.ts","sourceRoot":"","sources":["../../../src/runtime/derivedModes/rule.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAE/E;;;wDAGwD;AACxD,MAAM,MAAM,QAAQ,GAChB,MAAM,GAAG,WAAW,GAAG,UAAU,GACjC,cAAc,GAAG,YAAY,GAC7B,iBAAiB,CAAC;AAEtB,kCAAkC;AAClC,MAAM,MAAM,SAAS,CAAC,CAAC,SAAS,QAAQ,IACtC,CAAC,SAAS,MAAM,GAAa,QAAQ,GACrC,CAAC,SAAS,WAAW,GAAQ,SAAS,GACtC,CAAC,SAAS,UAAU,GAAS,QAAQ,GACrC,CAAC,SAAS,cAAc,GAAK,kBAAkB,GAC/C,CAAC,SAAS,YAAY,GAAO,OAAO,GACpC,CAAC,SAAS,iBAAiB,GAAG,OAAO,GACrC,KAAK,CAAC;AAER;;;;;;;;;GASG;AACH,MAAM,MAAM,kBAAkB,CAAC,CAAC,SAAS,QAAQ,IAAI,CACnD,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC1B,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,KACnB,SAAS,CAAC,CAAC,CAAC,CAAC;AAElB,MAAM,WAAW,eAAe,CAAC,CAAC,SAAS,QAAQ,GAAG,QAAQ;IAC5D,QAAQ,CAAC,iBAAiB,EAAE,IAAI,CAAC;IACjC,QAAQ,CAAC,IAAI,EAAK,CAAC,CAAC;IACpB,QAAQ,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC;IACzC;;;;;OAKG;IACH,QAAQ,CAAC,MAAM,CAAC,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;CAC/C;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,WAAW,CAAC,CAAC,SAAS,QAAQ,EAC5C,IAAI,EAAE,CAAC,EACP,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC,EAC/B,OAAO,GAAE;IAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAA;CAAO,GAC9D,eAAe,CAAC,CAAC,CAAC,CAKpB;AAED;mEACmE;AACnE,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,IAAI,eAAe,CAGlE;AAED;6EAC6E;AAC7E,wBAAgB,QAAQ,CAAC,CAAC,EAAE,QAAQ,GAAG,QAAQ,CAE9C"}
|
|
1
|
+
{"version":3,"file":"rule.d.ts","sourceRoot":"","sources":["../../../src/runtime/derivedModes/rule.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAE/E;;;wDAGwD;AACxD,MAAM,MAAM,QAAQ,GAChB,MAAM,GAAG,WAAW,GAAG,UAAU,GACjC,cAAc,GAAG,YAAY,GAC7B,iBAAiB,CAAC;AAEtB,kCAAkC;AAClC,MAAM,MAAM,SAAS,CAAC,CAAC,SAAS,QAAQ,IACtC,CAAC,SAAS,MAAM,GAAa,QAAQ,GACrC,CAAC,SAAS,WAAW,GAAQ,SAAS,GACtC,CAAC,SAAS,UAAU,GAAS,QAAQ,GACrC,CAAC,SAAS,cAAc,GAAK,kBAAkB,GAC/C,CAAC,SAAS,YAAY,GAAO,OAAO,GACpC,CAAC,SAAS,iBAAiB,GAAG,OAAO,GACrC,KAAK,CAAC;AAER;;;;;;;;;GASG;AACH,MAAM,MAAM,kBAAkB,CAAC,CAAC,SAAS,QAAQ,IAAI,CACnD,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC1B,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,KACnB,SAAS,CAAC,CAAC,CAAC,CAAC;AAElB;;;;;;;;GAQG;AACH,MAAM,WAAW,WAAW,CAAC,CAAC,SAAS,QAAQ,GAAG,QAAQ;IACxD,QAAQ,CAAC,MAAM,EAAE,uBAAuB,CAAC;IACzC,sEAAsE;IACtE,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,uDAAuD;IACvD,QAAQ,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,eAAe,CAAC,CAAC,SAAS,QAAQ,GAAG,QAAQ;IAC5D,QAAQ,CAAC,iBAAiB,EAAE,IAAI,CAAC;IACjC,QAAQ,CAAC,IAAI,EAAK,CAAC,CAAC;IACpB,QAAQ,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC;IACzC,wDAAwD;IACxD,QAAQ,CAAC,GAAG,CAAC,EAAK,WAAW,CAAC,CAAC,CAAC,CAAC;IACjC;;;;;OAKG;IACH,QAAQ,CAAC,MAAM,CAAC,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;CAC/C;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,WAAW,CAAC,CAAC,SAAS,QAAQ,EAC5C,IAAI,EAAE,CAAC,EACP,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC,EAC/B,OAAO,GAAE;IAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAA;CAAO,GAC9D,eAAe,CAAC,CAAC,CAAC,CAKpB;AAED;mEACmE;AACnE,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,IAAI,eAAe,CAGlE;AAED;6EAC6E;AAC7E,wBAAgB,QAAQ,CAAC,CAAC,EAAE,QAAQ,GAAG,QAAQ,CAE9C;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,wBAAwB,CACtC,YAAY,EAAE,MAAM,EACpB,QAAQ,GAAE,QAAiB,GAC1B,eAAe,CAAC,MAAM,CAAC,CAwBzB"}
|
|
@@ -52,4 +52,43 @@ export function isDerivedModeRule(x) {
|
|
|
52
52
|
export function flipCull(c) {
|
|
53
53
|
return c === "back" ? "front" : c === "front" ? "back" : "none";
|
|
54
54
|
}
|
|
55
|
+
/**
|
|
56
|
+
* Construct a derived-mode rule that flips cullMode based on the sign
|
|
57
|
+
* of `det(upperLeft3x3(u.<inputUniform>))`. Runs on the GPU when
|
|
58
|
+
* heapScene's compute path is available; the same logic is mirrored
|
|
59
|
+
* in the CPU closure for tests and fallback.
|
|
60
|
+
*
|
|
61
|
+
* const rule = gpuFlipCullByDeterminant("ModelTrafo", "back");
|
|
62
|
+
* spec.modeRules = { cull: rule };
|
|
63
|
+
*
|
|
64
|
+
* Useful for mirrored geometry: scenes that animate a per-RO trafo
|
|
65
|
+
* with potentially-negative scale don't pay CPU per-RO determinant
|
|
66
|
+
* cost — the kernel walks the arena once per frame for ALL ROs that
|
|
67
|
+
* carry this rule.
|
|
68
|
+
*/
|
|
69
|
+
export function gpuFlipCullByDeterminant(inputUniform, declared = "back") {
|
|
70
|
+
return {
|
|
71
|
+
__derivedModeRule: true,
|
|
72
|
+
axis: "cull",
|
|
73
|
+
domain: ["back", "front", "none"],
|
|
74
|
+
gpu: { kernel: "flipCullByDeterminant", inputUniform, declared },
|
|
75
|
+
evaluate: (u, declaredFromCtx) => {
|
|
76
|
+
const m = u[inputUniform];
|
|
77
|
+
// Accept Trafo3d ({forward: M44d}), M44 ({_data}), or row-major
|
|
78
|
+
// Float32Array. Reads upper-left 3x3.
|
|
79
|
+
const data = m.forward?._data ??
|
|
80
|
+
m._data ??
|
|
81
|
+
m;
|
|
82
|
+
if (data === undefined)
|
|
83
|
+
return declaredFromCtx;
|
|
84
|
+
const m00 = data[0], m01 = data[1], m02 = data[2];
|
|
85
|
+
const m10 = data[4], m11 = data[5], m12 = data[6];
|
|
86
|
+
const m20 = data[8], m21 = data[9], m22 = data[10];
|
|
87
|
+
const det = m00 * (m11 * m22 - m12 * m21)
|
|
88
|
+
- m01 * (m10 * m22 - m12 * m20)
|
|
89
|
+
+ m02 * (m10 * m21 - m11 * m20);
|
|
90
|
+
return det < 0 ? flipCull(declaredFromCtx) : declaredFromCtx;
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
}
|
|
55
94
|
//# sourceMappingURL=rule.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rule.js","sourceRoot":"","sources":["../../../src/runtime/derivedModes/rule.ts"],"names":[],"mappings":"AAAA,kEAAkE;AAClE,oEAAoE;AACpE,iBAAiB;AACjB,EAAE;AACF,mEAAmE;AACnE,kEAAkE;AAClE,oEAAoE;AACpE,oEAAoE;AACpE,iEAAiE;AACjE,iEAAiE;AACjE,sBAAsB;AACtB,EAAE;AACF,iEAAiE;AACjE,oEAAoE;AACpE,4DAA4D;AAC5D,EAAE;AACF,uEAAuE;AACvE,oEAAoE;AACpE,oDAAoD;
|
|
1
|
+
{"version":3,"file":"rule.js","sourceRoot":"","sources":["../../../src/runtime/derivedModes/rule.ts"],"names":[],"mappings":"AAAA,kEAAkE;AAClE,oEAAoE;AACpE,iBAAiB;AACjB,EAAE;AACF,mEAAmE;AACnE,kEAAkE;AAClE,oEAAoE;AACpE,oEAAoE;AACpE,iEAAiE;AACjE,iEAAiE;AACjE,sBAAsB;AACtB,EAAE;AACF,iEAAiE;AACjE,oEAAoE;AACpE,4DAA4D;AAC5D,EAAE;AACF,uEAAuE;AACvE,oEAAoE;AACpE,oDAAoD;AAsEpD;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,WAAW,CACzB,IAAO,EACP,QAA+B,EAC/B,UAA6D,EAAE;IAE/D,MAAM,GAAG,GAAuB,OAAO,CAAC,MAAM,KAAK,SAAS;QAC1D,CAAC,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;QACrE,CAAC,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAChD,OAAO,GAAG,CAAC;AACb,CAAC;AAED;mEACmE;AACnE,MAAM,UAAU,iBAAiB,CAAC,CAAU;IAC1C,OAAO,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI;WAClC,CAAqC,CAAC,iBAAiB,KAAK,IAAI,CAAC;AAC3E,CAAC;AAED;6EAC6E;AAC7E,MAAM,UAAU,QAAQ,CAAC,CAAW;IAClC,OAAO,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;AAClE,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,wBAAwB,CACtC,YAAoB,EACpB,WAAqB,MAAM;IAE3B,OAAO;QACL,iBAAiB,EAAE,IAAI;QACvB,IAAI,EAAE,MAAM;QACZ,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC;QACjC,GAAG,EAAE,EAAE,MAAM,EAAE,uBAAuB,EAAE,YAAY,EAAE,QAAQ,EAAE;QAChE,QAAQ,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE;YAC/B,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAoG,CAAC;YAC7H,gEAAgE;YAChE,sCAAsC;YACtC,MAAM,IAAI,GACP,CAAiD,CAAC,OAAO,EAAE,KAAK;gBAChE,CAAmC,CAAC,KAAK;gBACzC,CAAkC,CAAC;YACtC,IAAI,IAAI,KAAK,SAAS;gBAAE,OAAO,eAAe,CAAC;YAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;YACrD,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;YACrD,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,EAAE,GAAG,GAAG,IAAI,CAAC,EAAE,CAAE,CAAC;YACtD,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC;kBAC7B,GAAG,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC;kBAC7B,GAAG,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;YAC1C,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC;QAC/D,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"heapScene.d.ts","sourceRoot":"","sources":["../../src/runtime/heapScene.ts"],"names":[],"mappings":"AA4CA,OAAO,EAA2B,KAAK,GAAG,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAwB,aAAa,EAAa,MAAM,2BAA2B,CAAC;AAC3F,OAAO,KAAK,EAAE,IAAI,EAAE,IAAI,EAAgD,MAAM,2BAA2B,CAAC;AAC1G,OAAO,KAAK,EAAE,MAAM,EAAkB,MAAM,yBAAyB,CAAC;AACtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAG9D,OAAO,EAEc,KAAK,oBAAoB,EAE7C,MAAM,iBAAiB,CAAC;AAMzB,OAAO,EAEL,KAAK,SAAS,EAAwB,KAAK,SAAS,EACrD,MAAM,6BAA6B,CAAC;
|
|
1
|
+
{"version":3,"file":"heapScene.d.ts","sourceRoot":"","sources":["../../src/runtime/heapScene.ts"],"names":[],"mappings":"AA4CA,OAAO,EAA2B,KAAK,GAAG,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAwB,aAAa,EAAa,MAAM,2BAA2B,CAAC;AAC3F,OAAO,KAAK,EAAE,IAAI,EAAE,IAAI,EAAgD,MAAM,2BAA2B,CAAC;AAC1G,OAAO,KAAK,EAAE,MAAM,EAAkB,MAAM,yBAAyB,CAAC;AACtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAG9D,OAAO,EAEc,KAAK,oBAAoB,EAE7C,MAAM,iBAAiB,CAAC;AAMzB,OAAO,EAEL,KAAK,SAAS,EAAwB,KAAK,SAAS,EACrD,MAAM,6BAA6B,CAAC;AA+DrC;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,SAAS,EAAE,YAAY,CAAC;IACjC,QAAQ,CAAC,OAAO,EAAI,YAAY,CAAC;IACjC,QAAQ,CAAC,OAAO,EAAI,WAAW,CAAC;CACjC;AAgJD,MAAM,WAAW,YAAY;IAC3B;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB;;;;;;;;;;;OAWG;IACH,QAAQ,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,OAAO,CAAA;KAAE,CAAC;IACtE;;;;;;;;;;OAUG;IACH,QAAQ,CAAC,kBAAkB,CAAC,EAAE;QAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,OAAO,CAAA;KAAE,CAAC;IACnF;;;;OAIG;IACH,QAAQ,CAAC,aAAa,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;IAC/C;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,WAAW,CAAC;IAClD;;;;;;OAMG;IACH,QAAQ,CAAC,QAAQ,CAAC,EAAE,cAAc,CAAC;IACnC;;;;;;;;;;;;OAYG;IACH,QAAQ,CAAC,aAAa,CAAC,EAAE,aAAa,CAAC;IACvC;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE;QACnB,QAAQ,CAAC,IAAI,CAAC,EAAY,OAAO,wBAAwB,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QACnF,QAAQ,CAAC,SAAS,CAAC,EAAO,OAAO,wBAAwB,EAAE,eAAe,CAAC,WAAW,CAAC,CAAC;QACxF,QAAQ,CAAC,QAAQ,CAAC,EAAQ,OAAO,wBAAwB,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;QACvF,QAAQ,CAAC,YAAY,CAAC,EAAI,OAAO,wBAAwB,EAAE,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3F,QAAQ,CAAC,UAAU,CAAC,EAAM,OAAO,wBAAwB,EAAE,eAAe,CAAC,YAAY,CAAC,CAAC;QACzF,QAAQ,CAAC,eAAe,CAAC,EAAE,OAAO,wBAAwB,EAAE,eAAe,CAAC,iBAAiB,CAAC,CAAC;KAChG,CAAC;CACH;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,4DAA4D;IAC5D,aAAa,EAAI,MAAM,CAAC;IACxB,eAAe,EAAE,MAAM,CAAC;IACxB,eAAe,EAAE,MAAM,CAAC;IACxB,cAAc,EAAG,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,SAAS;IACxB;;;;;;OAMG;IACH,KAAK,CAAC,WAAW,EAAE,OAAO,kBAAkB,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,GAAG,IAAI,CAAC;IACxF;;;;OAIG;IACH,MAAM,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAAC;IACnC;;;;;;;;OAQG;IACH,cAAc,CAAC,OAAO,EAAE,oBAAoB,GAAG,IAAI,CAAC;IACpD;;;;;;;OAOG;IACH,iBAAiB,CAAC,GAAG,EAAE,iBAAiB,EAAE,KAAK,EAAE,aAAa,GAAG,IAAI,CAAC;IACtE;;;OAGG;IACH,OAAO,CAAC,IAAI,EAAE,YAAY,GAAG,MAAM,CAAC;IACpC,mFAAmF;IACnF,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,QAAQ,CAAC,KAAK,EAAE,cAAc,CAAC;IAC/B,OAAO,IAAI,IAAI,CAAC;CACjB;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,MAAM,cAAc,GACtB;IACE,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC;IAC5B,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;IAC3B,QAAQ,CAAC,WAAW,CAAC,EAAE,cAAc,CAAC;IACtC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;CAC5B,GACD;IACE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC;IAClC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,wDAAwD;IACxD,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC;IACrB,oDAAoD;IACpD,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC;IACnB,mEAAmE;IACnE,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;IAC3B;;;;;OAKG;IACH,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB;;;;;OAKG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC;IAC7B;;;;;;;;;;OAUG;IACH,QAAQ,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrC;;;;;OAKG;IACH,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,KAAK,OAAO,6BAA6B,EAAE,gBAAgB,CAAC;CAChG,CAAC;AAEN;;;;;;;;GAQG;AACH,MAAM,WAAW,qBAAqB;IACpC;;;;;;;;;;OAUG;IACH,QAAQ,CAAC,oBAAoB,CAAC,EAAE,oBAAoB,CAAC;IACrD;;;;;;;OAOG;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC;IAC/B;;;;;;;;OAQG;IACH,QAAQ,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC;IACrC;;;;;;;;;;;OAWG;IACH,QAAQ,CAAC,qBAAqB,CAAC,EAAE,OAAO,CAAC;CAC1C;AAED,wBAAgB,cAAc,CAC5B,MAAM,EAAE,SAAS,EACjB,GAAG,EAAE,OAAO,iCAAiC,EAAE,oBAAoB,EACnE,YAAY,EAAE,SAAS,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,EAC1D,IAAI,GAAE,qBAA0B,GAC/B,SAAS,CAyyGX"}
|
|
@@ -55,6 +55,8 @@ import { GrowBuffer, MIN_BUFFER_BYTES, POW2, ALIGN16, } from "./heapScene/growBu
|
|
|
55
55
|
import { UniformPool, IndexPool, DrawHeap, AttributeArena, IndexAllocator, insertSortedFreeBlock, buildArenaState, arenaBytes, writeAttribute, asAval, isBufferView, asFloat32, ALLOC_HEADER_BYTES, ALLOC_HEADER_PAD_TO, ENC_V3F_TIGHT, SEM_POSITIONS, SEM_NORMALS, } from "./heapScene/pools.js";
|
|
56
56
|
import { encodeModeKey } from "./pipelineCache/index.js";
|
|
57
57
|
import { snapshotDescriptor, ModeKeyTracker } from "./derivedModes/modeKeyCpu.js";
|
|
58
|
+
import { GpuDerivedModesScene } from "./derivedModes/gpuDispatcher.js";
|
|
59
|
+
import { U32_TO_CULL } from "./derivedModes/gpuKernel.js";
|
|
58
60
|
import { addMarkingCallback } from "@aardworx/wombat.adaptive";
|
|
59
61
|
export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
60
62
|
const atlasPool = opts.atlasPool;
|
|
@@ -335,6 +337,13 @@ export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
|
335
337
|
/** Per-RO §7 registration handles, keyed by global drawId.
|
|
336
338
|
* Drained on removeDraw to release slots + records. */
|
|
337
339
|
const derivedByDrawId = new Map();
|
|
340
|
+
// ─── GPU-eval mode rules (Task 2 Phase 5, narrow) ─────────────────
|
|
341
|
+
// Lazily allocated on first registration; one dispatcher per scene.
|
|
342
|
+
let gpuModesScene;
|
|
343
|
+
/** Last per-RO GPU output we routed to a bucket. Diff against
|
|
344
|
+
* newly-readback values to decide rebuckets without re-bucketing
|
|
345
|
+
* un-changed ROs. */
|
|
346
|
+
const gpuModesLastKey = new Map();
|
|
338
347
|
/** The derived-uniform rule for a uniform binding on this RO, if any: an explicit
|
|
339
348
|
* `derivedUniform(...)` value in `spec.inputs`, or a standard trafo recipe by name
|
|
340
349
|
* (recipes apply only when their base-trafo leaves are actually bound on this RO —
|
|
@@ -1410,6 +1419,21 @@ export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
|
1410
1419
|
});
|
|
1411
1420
|
drawIdToModeTracker[drawId] = tracker;
|
|
1412
1421
|
tracker.forEachLeaf((av) => subscribeModeLeaf(av, drawId));
|
|
1422
|
+
// ─── GPU-eval mode rule registration (Task 2 Phase 5) ────────────
|
|
1423
|
+
// If spec.modeRules.cull is GPU-flavoured, register the RO with
|
|
1424
|
+
// the scene-wide dispatcher. Provide the arena byte offset of the
|
|
1425
|
+
// RO's input uniform so the kernel reads from the right slot.
|
|
1426
|
+
{
|
|
1427
|
+
const gpuRule = spec.modeRules?.cull?.gpu;
|
|
1428
|
+
if (gpuRule !== undefined && gpuRule.kernel === "flipCullByDeterminant") {
|
|
1429
|
+
const inputRef = perDrawRefs.get(gpuRule.inputUniform);
|
|
1430
|
+
if (inputRef !== undefined) {
|
|
1431
|
+
if (gpuModesScene === undefined)
|
|
1432
|
+
gpuModesScene = new GpuDerivedModesScene(device);
|
|
1433
|
+
gpuModesScene.registerRo(drawId, inputRef, gpuRule.declared);
|
|
1434
|
+
}
|
|
1435
|
+
}
|
|
1436
|
+
}
|
|
1413
1437
|
// ─── §7 derived-uniforms registration ────────────────────────────
|
|
1414
1438
|
// A uniform binding on this RO is either a value (aval/constant) or a rule —
|
|
1415
1439
|
// collect the rules (explicit `derivedUniform(...)` values + standard trafo
|
|
@@ -1578,6 +1602,10 @@ export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
|
1578
1602
|
}
|
|
1579
1603
|
drawIdToModeTracker[drawId] = undefined;
|
|
1580
1604
|
dirtyModeKeyDrawIds.delete(drawId);
|
|
1605
|
+
if (gpuModesScene !== undefined) {
|
|
1606
|
+
gpuModesScene.deregisterRo(drawId);
|
|
1607
|
+
gpuModesLastKey.delete(drawId);
|
|
1608
|
+
}
|
|
1581
1609
|
stats.totalDraws--;
|
|
1582
1610
|
stats.geometryBytes = arenaBytes(arena);
|
|
1583
1611
|
// GC the bucket if it just emptied (common after reactive
|
|
@@ -1912,6 +1940,55 @@ export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
|
1912
1940
|
* we just swap the bind group + dispatch shape per bucket.
|
|
1913
1941
|
*/
|
|
1914
1942
|
function encodeComputePrep(enc, token) {
|
|
1943
|
+
// GPU-eval derived-mode rules (Task 2 Phase 5): dispatch the
|
|
1944
|
+
// hardcoded determinant-flip-cull kernel + start readback.
|
|
1945
|
+
// Resolved on the next frame's update() — one-frame lag for the
|
|
1946
|
+
// first appearance of a state transition. CPU fallback runs at
|
|
1947
|
+
// addDraw time so the initial bucket assignment is still correct.
|
|
1948
|
+
if (gpuModesScene !== undefined && gpuModesScene.registered > 0) {
|
|
1949
|
+
const numROs = nextDrawId; // upper bound; sparse entries are skipped via 0xFFFFFFFF sentinel
|
|
1950
|
+
gpuModesScene.dispatch(arena.attrs.buffer, numROs, enc);
|
|
1951
|
+
// mapAsync the staging buffer in the background. When it
|
|
1952
|
+
// resolves, diff against gpuModesLastKey to schedule rebuckets.
|
|
1953
|
+
gpuModesScene.finish(numROs).then(() => {
|
|
1954
|
+
const out = gpuModesScene.lastOutput;
|
|
1955
|
+
for (let i = 0; i < numROs; i++) {
|
|
1956
|
+
const cur = out[i] ?? 0xFFFFFFFF;
|
|
1957
|
+
const prev = gpuModesLastKey.get(i);
|
|
1958
|
+
if (cur === 0xFFFFFFFF)
|
|
1959
|
+
continue; // no rule on this RO
|
|
1960
|
+
if (prev === cur)
|
|
1961
|
+
continue;
|
|
1962
|
+
gpuModesLastKey.set(i, cur);
|
|
1963
|
+
// Project the kernel's u32 enum back to a CullMode value,
|
|
1964
|
+
// store on the tracker's modeRules so the next snapshot
|
|
1965
|
+
// picks it up, then schedule a rebucket.
|
|
1966
|
+
const tracker = drawIdToModeTracker[i];
|
|
1967
|
+
const spec = drawIdToSpec[i];
|
|
1968
|
+
if (tracker === undefined || spec === undefined)
|
|
1969
|
+
continue;
|
|
1970
|
+
const cull = U32_TO_CULL[cur];
|
|
1971
|
+
if (cull === undefined)
|
|
1972
|
+
continue;
|
|
1973
|
+
// Inject the GPU result by patching the spec's CPU rule to
|
|
1974
|
+
// return a constant. Slightly hacky but contained: the next
|
|
1975
|
+
// recompute() will read this and route to the right bucket.
|
|
1976
|
+
const patched = {
|
|
1977
|
+
...spec.modeRules,
|
|
1978
|
+
cull: {
|
|
1979
|
+
__derivedModeRule: true, axis: "cull",
|
|
1980
|
+
evaluate: () => cull,
|
|
1981
|
+
},
|
|
1982
|
+
};
|
|
1983
|
+
spec.modeRules = patched;
|
|
1984
|
+
dirtyModeKeyDrawIds.add(i);
|
|
1985
|
+
}
|
|
1986
|
+
}).catch((e) => {
|
|
1987
|
+
// mapAsync can reject if the device was lost or the scene was
|
|
1988
|
+
// disposed mid-readback. Log but don't crash the frame.
|
|
1989
|
+
console.warn("gpuModesScene readback failed:", e);
|
|
1990
|
+
});
|
|
1991
|
+
}
|
|
1915
1992
|
// §7: derived-uniforms dispatch first — writes into per-bucket
|
|
1916
1993
|
// drawHeap regions before the scan reads anything. One dispatcher
|
|
1917
1994
|
// per bucket; constituents are shared so dirty state propagates
|
|
@@ -2043,6 +2120,7 @@ export function buildHeapScene(device, sig, initialDraws, opts = {}) {
|
|
|
2043
2120
|
arena.indices.destroy();
|
|
2044
2121
|
for (const b of buckets)
|
|
2045
2122
|
destroyBucketResources(b);
|
|
2123
|
+
gpuModesScene?.dispose();
|
|
2046
2124
|
}
|
|
2047
2125
|
// Test-only escape hatch for inspecting megacall bucket state. Not
|
|
2048
2126
|
// part of the public API surface — keep cast at use-site.
|