KnitFlow/assets/core_tgx/util/ResMgr.ts

import {  instantiate,  Node, NodePool, Prefab, Vec3 } from "cc"


/**
 * @Description: resources manager and pool manager
 * @return {*}
 */
export class ResMgr {
 
    private _dictPool: { [key: string]: NodePool } = {}
    private _dictPrefab: { [key: string]: Prefab } = {}
    private _loadStemp = null;
    private _loadTime = 0;
    private _totalTime = 0
    public loadingRate = 0;
    public debug = false;

    /**
     * @description: get the node from the pool
     * @param {Prefab} prefab
     * @param {Node} parent
     * @param {Vec3} pos
     * @return {*}
     */
    // 节点池 NodePool ,用于处理频繁创建和销毁的物体
    public getNode (prefab: Prefab | string, parent?: Node, pos?: Vec3): Node {
        let tempPre;
        let name;
        if (typeof prefab === 'string') {
            tempPre = this._dictPrefab[prefab];
            name = prefab;
            if (!tempPre) {
                console.log("Pool invalid prefab name = ", name);
               
                return null;
            }
        }
        else {
            tempPre = prefab;
            name = prefab.data.name;
            
        }
        let node = null;
        if (this._dictPool.hasOwnProperty(name)) {
            //own this pool
            let pool = this._dictPool[name];
            if (pool.size() > 0) {
                node = pool.get();
            } else {
                node = instantiate(tempPre);
            }
        } else {
            //create new pool
            let pool = new NodePool();
            this._dictPool[name] = pool;
            node = instantiate(tempPre);
        }

        if (parent) {
            node.parent = parent;
            node.active = true;
            if (pos) node.position = pos;
        }

        return node;
    }


    /**
     * @description: put the node into the pool
     * @param {Node} node
     * @param {*} isActive
     * @return {*}
     */
    public putNode (node: Node | null, isActive = false) {
        if (!node) {
            return;
        }
        let name = node.name;
        let pool = null;
        // node.active = isActive
        if (this._dictPool.hasOwnProperty(name)) {
            pool = this._dictPool[name];
        } else {
            pool = new NodePool();
            this._dictPool[name] = pool;
        }
        pool.put(node);
    }

    /**
     * @description: clear the pool based on name
     * @param {string} name
     * @return {*}
     */
    public clearPool (name: string) {
        if (this._dictPool.hasOwnProperty(name)) {
            let pool = this._dictPool[name];
            pool.clear();
        }
    }

    public clearAllPool () {
       let keys = Object.keys(this._dictPool)
       keys.forEach(it=>{
            this.clearPool(it)
       })
     
        
    }

    public clearAll () {
        let keys = Object.keys(this._dictPool)
        keys.forEach(it=>{
             this.clearPool(it)
        })
      this.clearDict()
         
     }

    public setPrefab (name: string, prefab: Prefab): void {
        if (!this._dictPrefab[name]) {
            this._dictPrefab[name] = prefab;
        }
    }

    public clearDict () {
        this._dictPrefab = {};
    }

    printTimer (name: string = "", end = false) {
        this._loadTime = Date.now() - this._loadStemp;
        this._loadStemp = Date.now();
        this._totalTime += this._loadTime
        console.log(name + "，load time===", this._loadTime, "ms")
        if (end) {
            console.log("Load finish, total time===", this._totalTime, "ms")
        }

    } 
}

// const resMgr = new ResMgr;

// export default resMgr;