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

class EntityList{
    public scene: Scene;
    private _entitiesToRemove: Entity[] = [];
    private _entitiesToAdded: Entity[] = [];
    private _tempEntityList: Entity[] = [];
    private _entities: Entity[] = [];

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

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

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

    public add(entity: Entity){
        this._entitiesToAdded.push(entity);
    }

    public remove(entity: Entity){
        if (this._entitiesToAdded.contains(entity)){
            this._entitiesToAdded.remove(entity);
            return;
        }

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

    public findEntity(name: string){
        for (let i = 0; i < this._entities.length; i ++){
            if (this._entities[i].name == name)
                return this._entities[i];
        }

        return this._entitiesToAdded.firstOrDefault(entity => entity.name == name);
    }

    public update(){
        for (let i = 0; i < this._entities.length; i++){
            let entity = this._entities[i];
            if (entity.enabled)
                entity.update();
        }
    }

    public removeAllEntities(){
        this._entitiesToAdded.length = 0;

        this.updateLists();

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

        this._entities.length = 0;
    }

    public updateLists(){
        if (this._entitiesToRemove.length > 0){
            let temp = this._entitiesToRemove;
            this._entitiesToRemove = this._tempEntityList;
            this._tempEntityList = temp;
            this._tempEntityList.forEach(entity => {
                this._entities.remove(entity);
                entity.scene = null;

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

            this._tempEntityList.length = 0;
        }

        if (this._entitiesToAdded.length > 0){
            let temp = this._entitiesToAdded;
            this._entitiesToAdded = this._tempEntityList;
            this._tempEntityList = temp;
            this._tempEntityList.forEach(entity => {
                this._entities.push(entity);
                entity.scene = this.scene;

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

            this._tempEntityList.length = 0;
        }
    }
}