ResourceManager support findResourcesByType (#1700)

* feat: ResourceManager support findResourcesByType
galacean · Aug 16, 2023 · ef443a3 · ef443a3
1 parent a933568
commit ef443a3
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 49 deletions.
diff --git a/packages/core/src/asset/ResourceManager.ts b/packages/core/src/asset/ResourceManager.ts
@@ -102,6 +102,23 @@ export class ResourceManager {
     return (this._assetUrlPool[url] as T) ?? null;
   }
 
+  /**
+   * Find the resource by type.
+   * @param type - Resource type
+   * @returns - Resource collection
+   */
+  findResourcesByType<T extends EngineObject>(type: new (...args) => T): T[] {
+    const resources = new Array<T>();
+    const referResourcePool = this._referResourcePool;
+    for (const k in referResourcePool) {
+      const resource = referResourcePool[k];
+      if (resource instanceof type) {
+        resources.push(resource);
+      }
+    }
+    return resources;
+  }
+
   /**
    * Get asset url from instanceId.
    * @param instanceId - Engine instance id

diff --git a/packages/shader-lab/src/parser/tokens/GlslTypes.ts b/packages/shader-lab/src/parser/tokens/GlslTypes.ts
diff --git a/packages/shader-lab/src/parser/tokens/value.ts b/packages/shader-lab/src/parser/tokens/value.ts
diff --git a/tests/src/core/resource/ResourceManager.test.ts b/tests/src/core/resource/ResourceManager.test.ts
@@ -8,7 +8,7 @@ describe("ResourceManager", () => {
     engine = await WebGLEngine.create({ canvas: document.createElement("canvas") });
     engine.run();
   });
-  
+
   beforeEach(() => {
     engine.sceneManager.activeScene.createRootEntity("root");
   });
@@ -28,4 +28,11 @@ describe("ResourceManager", () => {
       expect(getResource).equal(null);
     });
   });
+
+  describe("findResourcesByType", () => {
+    it("findResourcesByType", () => {
+      const textures = engine.resourceManager.findResourcesByType(Texture2D);
+      expect(textures.length).equal(4);
+    });
+  });
 });