esengine/source/src/ECS/Utils/EntityList.ts

module es {
    export class EntityList {
        public scene: Scene;
        /**
         * 场景中添加的实体列表
         */
        public _entities: Entity[] = [];
        /**
         * 本帧添加的实体列表。用于对实体进行分组，以便我们可以同时处理它们
         */
        public _entitiesToAdded: HashSet<Entity> = new HashSet<Entity>();
        /**
         * 本帧被标记为删除的实体列表。用于对实体进行分组，以便我们可以同时处理它们
         */
        public _entitiesToRemove: HashSet<Entity> = new HashSet<Entity>();
        /**
         * 标志，用于确定我们是否需要在这一帧中对实体进行排序
         */
        public _isEntityListUnsorted: boolean;
        /**
         * 通过标签跟踪实体，便于检索
         */
        public _entityDict: Map<number, Set<Entity>> = new Map<number, Set<Entity>>();
        public _unsortedTags: Set<number> = new Set<number>();

        constructor(scene: Scene) {
            this.scene = scene;
        }

        public get count() {
            return this._entities.length;
        }

        public get buffer() {
            return this._entities;
        }

        public markEntityListUnsorted() {
            this._isEntityListUnsorted = true;
        }

        public markTagUnsorted(tag: number) {
            this._unsortedTags.add(tag);
        }

        /**
         * 将一个实体添加到列表中。所有的生命周期方法将在下一帧中被调用
         * @param entity
         */
        public add(entity: Entity) {
            this._entitiesToAdded.add(entity);
        }

        /**
         * 从列表中删除一个实体。所有的生命周期方法将在下一帧中被调用
         * @param entity
         */
        public remove(entity: Entity) {
            Debug.warnIf(this._entitiesToRemove.contains(entity), `您正在尝试删除已经删除的实体(${entity.name})`);

            // 防止在同一帧中添加或删除实体
            if (this._entitiesToAdded.contains(entity)) {
                this._entitiesToAdded.remove(entity);
                return;
            }

            if (!this._entitiesToRemove.contains(entity))
                this._entitiesToRemove.add(entity);
        }

        /**
         * 从实体列表中删除所有实体
         */
        public removeAllEntities() {
            this._unsortedTags.clear();
            this._entitiesToAdded.clear();
            this._isEntityListUnsorted = false;

            // 为什么我们要在这里更新列表？主要是为了处理在场景切换前被分离的实体。
            // 它们仍然会在_entitiesToRemove列表中，这将由updateLists处理。
            this.updateLists();

            for (let i = 0; i < this._entities.length; i++) {
                this._entities[i]._isDestroyed = true;
                this._entities[i].onRemovedFromScene();
                this._entities[i].scene = null;
            }

            this._entities.length = 0;
            this._entityDict.clear();
        }

        /**
         * 检查实体目前是否由这个EntityList管理
         * @param entity
         */
        public contains(entity: Entity): boolean {
            return new linq.List(this._entities).contains(entity) || this._entitiesToAdded.contains(entity);
        }

        public getTagList(tag: number) {
            let list = this._entityDict.get(tag);
            if (!list) {
                list = new Set();
                this._entityDict.set(tag, list);
            }

            return list;
        }

        public addToTagList(entity: Entity) {
            this.getTagList(entity.tag).add(entity);
            this._unsortedTags.add(entity.tag);
        }

        public removeFromTagList(entity: Entity) {
            let list = this._entityDict.get(entity.tag);
            if (list)
                list.delete(entity);
        }

        public update() {
            for (let entity of this._entities) {
                if (entity.enabled && (entity.updateInterval == 1 || Time.frameCount % entity.updateInterval == 0))
                    entity.update();
            }
        }

        public updateLists() {
            if (this._entitiesToRemove.getCount() > 0) {
                this._entitiesToRemove.toArray().forEach(entity => {
                    // 处理标签列表
                    this.removeFromTagList(entity);

                    // 处理常规实体列表
                    new linq.List(this._entities).remove(entity);
                    entity.onRemovedFromScene();
                    entity.scene = null;

                    this.scene.entityProcessors.onEntityRemoved(entity);
                });
                this._entitiesToRemove.clear();
            }

            if (this._entitiesToAdded.getCount() > 0) {
                this._entitiesToAdded.toArray().forEach(entity => {
                    this._entities.push(entity);
                    entity.scene = this.scene;

                    this.addToTagList(entity);

                    this.scene.entityProcessors.onEntityAdded(entity);
                });

                this._entitiesToAdded.toArray().forEach(entity => {
                    entity.onAddedToScene();
                })

                this._entitiesToAdded.clear();
                this._isEntityListUnsorted = true;
            }

            if (this._isEntityListUnsorted) {
                this._entities.sort(Entity.entityComparer.compare);
                this._isEntityListUnsorted = false;
            }
        }

        /**
         * 返回第一个找到的名字为name的实体。如果没有找到则返回null
         * @param name
         */
        public findEntity(name: string) {
            for (let i = 0; i < this._entities.length; i++) {
                if (this._entities[i].name == name)
                    return this._entities[i];
            }

            for (let i = 0; i < this._entitiesToAdded.getCount(); i++) {
                let entity = this._entitiesToAdded.toArray()[i];
                if (entity.name == name)
                    return entity;
            }

            return null;
        }

        /**
         * 返回带有标签的所有实体的列表。如果没有实体有标签，则返回一个空列表。
         * 返回的List可以通过ListPool.free放回池中
         * @param tag
         */
        public entitiesWithTag(tag: number) {
            let list = this.getTagList(tag);

            let returnList = ListPool.obtain<Entity>();
            returnList.length = this._entities.length;
            for (let entity of list) {
                returnList.push(entity);
            }

            return returnList;
        }

        /**
         * 返回一个T类型的所有实体的列表。
         * 返回的List可以通过ListPool.free放回池中。
         * @param type
         */
        public entitiesOfType<T extends Entity>(type): T[] {
            let list = ListPool.obtain<T>();
            for (let i = 0; i < this._entities.length; i++) {
                if (this._entities[i] instanceof type)
                    list.push(this._entities[i] as T);
            }

            for (let i = 0; i < this._entitiesToAdded.getCount(); i++) {
                let entity = this._entitiesToAdded.toArray()[i];
                if (TypeUtils.getType(entity) instanceof type) {
                    list.push(entity as T);
                }
            }

            return list;
        }

        /**
         * 返回在场景中找到的第一个T类型的组件。
         * @param type
         */
        public findComponentOfType<T extends Component>(type): T {
            for (let i = 0; i < this._entities.length; i++) {
                if (this._entities[i].enabled) {
                    let comp = this._entities[i].getComponent<T>(type);
                    if (comp)
                        return comp;
                }
            }

            for (let i = 0; i < this._entitiesToAdded.getCount(); i++) {
                let entity: Entity = this._entitiesToAdded.toArray()[i];
                if (entity.enabled) {
                    let comp = entity.getComponent<T>(type);
                    if (comp)
                        return comp;
                }
            }

            return null;
        }

        /**
         * 返回在场景中找到的所有T类型的组件。
         * 返回的List可以通过ListPool.free放回池中。
         * @param type
         */
        public findComponentsOfType<T extends Component>(type): T[] {
            let comps = ListPool.obtain<T>();
            for (let i = 0; i < this._entities.length; i++) {
                if (this._entities[i].enabled)
                    this._entities[i].getComponents(type, comps);
            }

            for (let i = 0; i < this._entitiesToAdded.getCount(); i++) {
                let entity = this._entitiesToAdded.toArray()[i];
                if (entity.enabled)
                    entity.getComponents(type, comps);
            }

            return comps;
        }

        /**
         * 返回场景中包含特定组件的实体列表
         * @param types 
         * @returns 
         */
        public findEntitesOfComponent(...types): Entity[] {
            let entities = [];
            for (let i = 0; i < this._entities.length; i++) {
                if (this._entities[i].enabled) {
                    let meet = true;
                    for (let type of types) {
                        let hasComp = this._entities[i].hasComponent(type);
                        if (!hasComp) {
                            meet = false;
                            break;
                        }
                    }

                    if (meet) {
                        entities.push(this._entities[i]);
                    }
                }
            }

            for (let i = 0; i < this._entitiesToAdded.getCount(); i++) {
                let entity: Entity = this._entitiesToAdded.toArray()[i];
                if (entity.enabled) {
                    let meet = true;
                    for (let type of types) {
                        let hasComp = entity.hasComponent(type);
                        if (!hasComp) {
                            meet = false;
                            break;
                        }
                    }

                    if (meet) {
                        entities.push(entity);
                    }
                }
            }

            return entities;
        }
    }
}