com.artemis.managers.UuidEntityManager Maven / Gradle / Ivy

Go to download
package com.artemis.managers;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

import com.artemis.Entity;
import com.artemis.Manager;
import com.artemis.utils.Bag;

public class UuidEntityManager extends Manager {
	private final Map uuidToEntity;
	private final Bag entityToUuid;

	private Entity flyweight;

	public UuidEntityManager() {
		this.uuidToEntity = new HashMap();
		this.entityToUuid = new Bag();
	}

	@Override
	protected void initialize() {
		flyweight = Entity.createFlyweight(world);
	}

	@Override
	public void deleted(int entityId) {
		flyweight.id = entityId;
		UUID uuid = entityToUuid.safeGet(flyweight.getId());
		if (uuid == null)
			return;

		Entity oldEntity = uuidToEntity.get(uuid);
		if (oldEntity != null && oldEntity.id == flyweight.id)
			uuidToEntity.remove(uuid);

		entityToUuid.set(flyweight.getId(), null);
	}
	
	public void updatedUuid(Entity e, UUID newUuid) {
		setUuid(e, newUuid);
	}
	
	public Entity getEntity(UUID uuid) {
		return uuidToEntity.get(uuid);
	}

	public UUID getUuid(Entity e) {
		UUID uuid = entityToUuid.safeGet(e.getId());
		if (uuid == null) {
			uuid = UUID.randomUUID();
			setUuid(e, uuid);
		}
		
		return uuid;
	}
	
	public void setUuid(Entity e, UUID newUuid) {
		UUID oldUuid = entityToUuid.safeGet(e.getId());
		if (oldUuid != null)
			uuidToEntity.remove(oldUuid);
		
		uuidToEntity.put(newUuid, e);
		entityToUuid.set(e.getId(), newUuid);
	}
}