Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

WebGPURenderer: add BundleGroup #1226

Merged
merged 10 commits into from
Sep 17, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
61 changes: 34 additions & 27 deletions src-testing/changes.patch
Original file line number Diff line number Diff line change
Expand Up @@ -3426,6 +3426,21 @@ index 99ddcb48..e18679b1 100644
const strideLength = 4;

return vectorLength + ((strideLength - (vectorLength % strideLength)) % strideLength);
diff --git a/src-testing/src/renderers/common/BundleGroup.ts b/src-testing/src/renderers/common/BundleGroup.ts
index 41ae6ddc..c872490a 100644
--- a/src-testing/src/renderers/common/BundleGroup.ts
+++ b/src-testing/src/renderers/common/BundleGroup.ts
@@ -1,6 +1,10 @@
import { Group } from '../../objects/Group.js';

class BundleGroup extends Group {
+ readonly isBundleGroup: true;
+
+ readonly type: string;
+
constructor() {
super();

diff --git a/src-testing/src/renderers/common/ChainMap.ts b/src-testing/src/renderers/common/ChainMap.ts
index b17e7080..e2b545b7 100644
--- a/src-testing/src/renderers/common/ChainMap.ts
Expand Down Expand Up @@ -4253,40 +4268,30 @@ index e59e4937..ca572cea 100644
}

diff --git a/src-testing/src/renderers/common/RenderBundles.ts b/src-testing/src/renderers/common/RenderBundles.ts
index 29140365..165ece53 100644
index 29140365..46f14369 100644
--- a/src-testing/src/renderers/common/RenderBundles.ts
+++ b/src-testing/src/renderers/common/RenderBundles.ts
@@ -1,14 +1,18 @@
import ChainMap from './ChainMap.js';
import RenderBundle from './RenderBundle.js';
+import { Object3D } from '../../core/Object3D.js';
+import { Camera } from '../../cameras/Camera.js';
+import BundleGroup from './BundleGroup.js';

class RenderBundles {
+ lists: ChainMap<readonly [Object3D, Camera], RenderBundle>;
+ lists: ChainMap<readonly [BundleGroup, Camera], RenderBundle>;
+
constructor() {
- this.lists = new ChainMap();
+ this.lists = new ChainMap<readonly [Object3D, Camera], RenderBundle>();
this.lists = new ChainMap();
}

- get(scene, camera) {
+ get(scene: Object3D, camera: Camera) {
+ get(scene: BundleGroup, camera: Camera) {
const lists = this.lists;
- const keys = [scene, camera];
+ const keys = [scene, camera] as const;

let list = lists.get(keys);

@@ -21,7 +25,7 @@ class RenderBundles {
}

dispose() {
- this.lists = new ChainMap();
+ this.lists = new ChainMap<readonly [Object3D, Camera], RenderBundle>();
}
}

diff --git a/src-testing/src/renderers/common/RenderContext.ts b/src-testing/src/renderers/common/RenderContext.ts
index 342029ea..ab04452a 100644
--- a/src-testing/src/renderers/common/RenderContext.ts
Expand Down Expand Up @@ -4398,18 +4403,20 @@ index e77308c1..a3556aa1 100644

dispose() {
diff --git a/src-testing/src/renderers/common/RenderList.ts b/src-testing/src/renderers/common/RenderList.ts
index b9db4833..befdccd9 100644
index b9db4833..b64a3247 100644
--- a/src-testing/src/renderers/common/RenderList.ts
+++ b/src-testing/src/renderers/common/RenderList.ts
@@ -1,32 +1,66 @@
@@ -1,32 +1,67 @@
import { LightsNode } from '../../nodes/Nodes.js';
-
-function painterSortStable(a, b) {
+import { Object3D } from '../../core/Object3D.js';
+import { Camera } from '../../cameras/Camera.js';
+import { BufferGeometry, GeometryGroup } from '../../core/BufferGeometry.js';
+import { Material } from '../../materials/Material.js';
+import { Light } from '../../lights/Light.js';
-function painterSortStable(a, b) {
+import BundleGroup from './BundleGroup.js';
+
+function painterSortStable(a: RenderItem, b: RenderItem) {
if (a.groupOrder !== b.groupOrder) {
- return a.groupOrder - b.groupOrder;
Expand Down Expand Up @@ -4448,7 +4455,7 @@ index b9db4833..befdccd9 100644
}

+export interface Bundle {
+ object: Object3D;
+ bundleGroup: BundleGroup;
+ camera: Camera;
+ renderList: RenderList;
+}
Expand Down Expand Up @@ -4480,7 +4487,7 @@ index b9db4833..befdccd9 100644
constructor() {
this.renderItems = [];
this.renderItemsIndex = 0;
@@ -55,7 +89,14 @@ class RenderList {
@@ -55,7 +90,14 @@ class RenderList {
return this;
}

Expand All @@ -4496,7 +4503,7 @@ index b9db4833..befdccd9 100644
let renderItem = this.renderItems[this.renderItemsIndex];

if (renderItem === undefined) {
@@ -87,7 +128,14 @@ class RenderList {
@@ -87,7 +129,14 @@ class RenderList {
return renderItem;
}

Expand All @@ -4512,7 +4519,7 @@ index b9db4833..befdccd9 100644
const renderItem = this.getNextRenderItem(object, geometry, material, groupOrder, z, group);

if (object.occlusionTest === true) this.occlusionQueryCount++;
@@ -95,25 +143,35 @@ class RenderList {
@@ -95,25 +144,35 @@ class RenderList {
(material.transparent === true || material.transmission > 0 ? this.transparent : this.opaque).push(renderItem);
}

Expand Down Expand Up @@ -4864,7 +4871,7 @@ index 0ec34b04..573cae2b 100644

this.vertexProgram = vertexProgram;
diff --git a/src-testing/src/renderers/common/Renderer.ts b/src-testing/src/renderers/common/Renderer.ts
index 98e8a43f..fabd1b2d 100644
index d077a225..ff34bd8e 100644
--- a/src-testing/src/renderers/common/Renderer.ts
+++ b/src-testing/src/renderers/common/Renderer.ts
@@ -35,7 +35,36 @@ import {
Expand Down Expand Up @@ -5188,15 +5195,15 @@ index 98e8a43f..fabd1b2d 100644

- _renderBundle(bundle, sceneRef, lightsNode) {
+ _renderBundle(bundle: Bundle, sceneRef: Scene, lightsNode: LightsNode) {
const { object, camera, renderList } = bundle;
const { bundleGroup, camera, renderList } = bundle;

- const renderContext = this._currentRenderContext;
+ const renderContext = this._currentRenderContext!;

//

- const renderBundle = this._bundles.get(object, camera);
+ const renderBundle = this._bundles!.get(object, camera);
- const renderBundle = this._bundles.get(bundleGroup, camera);
+ const renderBundle = this._bundles!.get(bundleGroup, camera);
const renderBundleData = this.backend.get(renderBundle);

if (renderBundleData.renderContexts === undefined) renderBundleData.renderContexts = new Set();
Expand Down
1 change: 1 addition & 0 deletions src-testing/create-src.js
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ const files = [
'renderers/common/Bindings',
'renderers/common/Buffer',
'renderers/common/BufferUtils',
'renderers/common/BundleGroup',
'renderers/common/ChainMap',
'renderers/common/ClippingContext',
'renderers/common/Color4',
Expand Down
1 change: 1 addition & 0 deletions src-testing/declarations.js
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ const files = [
'renderers/common/Bindings',
'renderers/common/Buffer',
'renderers/common/BufferUtils',
'renderers/common/BundleGroup',
'renderers/common/ChainMap',
'renderers/common/ClippingContext',
'renderers/common/Color4',
Expand Down
1 change: 1 addition & 0 deletions types/three/src/Three.WebGPU.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -191,6 +191,7 @@ export { default as NodeObjectLoader } from "./loaders/nodes/NodeObjectLoader.js
export * from "./materials/nodes/NodeMaterials.js";
export * from "./nodes/Nodes.js";
export * from "./nodes/TSL.js";
export { default as BundleGroup } from "./renderers/common/BundleGroup.js";
export { default as PMREMGenerator } from "./renderers/common/extras/PMREMGenerator.js";
export { default as PostProcessing } from "./renderers/common/PostProcessing.js";
export { default as QuadMesh } from "./renderers/common/QuadMesh.js";
Expand Down
2 changes: 1 addition & 1 deletion types/three/src/core/Object3D.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ export class Object3D<TEventMap extends Object3DEventMap = Object3DEventMap> ext
* Sub-classes will update this value.
* @defaultValue `Object3D`
*/
readonly type: string | "Object3D";
readonly type: string;

/**
* Object's parent in the {@link https://en.wikipedia.org/wiki/Scene_graph | scene graph}.
Expand Down
6 changes: 0 additions & 6 deletions types/three/src/objects/Group.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,4 @@ export class Group<TEventMap extends Object3DEventMap = Object3DEventMap> extend
* @defaultValue `true`
*/
readonly isGroup: true;

/**
* @override
* @defaultValue `Group`
*/
override readonly type: string | "Group";
}
7 changes: 7 additions & 0 deletions types/three/src/renderers/common/BundleGroup.d.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
import { Group } from "../../objects/Group.js";
declare class BundleGroup extends Group {
readonly isBundleGroup: true;
readonly type: string;
constructor();
}
export default BundleGroup;
6 changes: 3 additions & 3 deletions types/three/src/renderers/common/RenderBundles.d.ts
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
import { Camera } from "../../cameras/Camera.js";
import { Object3D } from "../../core/Object3D.js";
import BundleGroup from "./BundleGroup.js";
import ChainMap from "./ChainMap.js";
import RenderBundle from "./RenderBundle.js";
declare class RenderBundles {
lists: ChainMap<readonly [Object3D, Camera], RenderBundle>;
lists: ChainMap<readonly [BundleGroup, Camera], RenderBundle>;
constructor();
get(scene: Object3D, camera: Camera): RenderBundle;
get(scene: BundleGroup, camera: Camera): RenderBundle;
dispose(): void;
}
export default RenderBundles;
3 changes: 2 additions & 1 deletion types/three/src/renderers/common/RenderList.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,9 @@ import { Object3D } from "../../core/Object3D.js";
import { Light } from "../../lights/Light.js";
import { Material } from "../../materials/Material.js";
import { LightsNode } from "../../nodes/Nodes.js";
import BundleGroup from "./BundleGroup.js";
export interface Bundle {
object: Object3D;
bundleGroup: BundleGroup;
camera: Camera;
renderList: RenderList;
}
Expand Down
Loading