engine add WebGPU support (#18196)

cocos · Jan 16, 2025 · 037cbe8 · 037cbe8
1 parent ae9a255
commit 037cbe8
Show file tree

Hide file tree

Showing 7 changed files with 302 additions and 82 deletions.
diff --git a/cocos/rendering/custom/define.ts b/cocos/rendering/custom/define.ts
@@ -428,7 +428,8 @@ export function getDescBindingFromName (bindingName: string): number {
     let currDesData: DescriptorSetData;
     for (const i of vertIds) {
         const layout = layoutGraph.getLayout(i);
-        for (const [k, descData] of layout.descriptorSets) {
+        const sets = layout.getSets();
+        for (const [k, descData] of sets) {
             const layoutData = descData.descriptorSetLayoutData;
             const blocks = layoutData.descriptorBlocks;
             for (const b of blocks) {
@@ -493,15 +494,15 @@ export function getDescriptorSetDataFromLayout (layoutName: string): DescriptorS
     const webPip = cclegacy.director.root.pipeline as WebPipeline;
     const stageId = webPip.layoutGraph.locateChild(webPip.layoutGraph.N, layoutName);
     const layout = webPip.layoutGraph.getLayout(stageId);
-    const layoutData = layout.descriptorSets.get(UpdateFrequency.PER_PASS);
+    const layoutData = layout.getSet(UpdateFrequency.PER_PASS);
     layouts.set(layoutName, layoutData!);
     return layoutData;
 }
 
 export function getDescriptorSetDataFromLayoutId (id: number): DescriptorSetData | undefined {
     const webPip = cclegacy.director.root.pipeline as WebPipeline;
     const layout = webPip.layoutGraph.getLayout(id);
-    const layoutData = layout.descriptorSets.get(UpdateFrequency.PER_PASS);
+    const layoutData = layout.getSet(UpdateFrequency.PER_PASS);
     return layoutData;
 }
 
@@ -514,7 +515,7 @@ function getUniformBlock (block: string, layoutName: string): UniformBlock | und
     const lg = webPip.layoutGraph;
     const nodeId = lg.locateChild(0xFFFFFFFF, layoutName);
     const ppl = lg.getLayout(nodeId);
-    const layout = ppl.descriptorSets.get(UpdateFrequency.PER_PASS)!.descriptorSetLayoutData;
+    const layout = ppl.getSet(UpdateFrequency.PER_PASS)!.descriptorSetLayoutData;
     const nameID: number = lg.attributeIndex.get(block)!;
     return layout.uniformBlocks.get(nameID);
 }