60 lines
1.7 KiB
Java
60 lines
1.7 KiB
Java
package net.minecraft.world.level.entity;
|
|
|
|
import com.google.common.collect.Iterables;
|
|
import com.google.common.collect.Maps;
|
|
import com.mojang.logging.LogUtils;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectLinkedOpenHashMap;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
import net.minecraft.util.AbortableIterationConsumer;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.slf4j.Logger;
|
|
|
|
public class EntityLookup<T extends EntityAccess> {
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
|
private final Int2ObjectMap<T> byId = new Int2ObjectLinkedOpenHashMap<>();
|
|
private final Map<UUID, T> byUuid = Maps.<UUID, T>newHashMap();
|
|
|
|
public <U extends T> void getEntities(EntityTypeTest<T, U> test, AbortableIterationConsumer<U> consumer) {
|
|
for (T entityAccess : this.byId.values()) {
|
|
U entityAccess2 = (U)test.tryCast(entityAccess);
|
|
if (entityAccess2 != null && consumer.accept(entityAccess2).shouldAbort()) {
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
public Iterable<T> getAllEntities() {
|
|
return Iterables.unmodifiableIterable(this.byId.values());
|
|
}
|
|
|
|
public void add(T entity) {
|
|
UUID uUID = entity.getUUID();
|
|
if (this.byUuid.containsKey(uUID)) {
|
|
LOGGER.warn("Duplicate entity UUID {}: {}", uUID, entity);
|
|
} else {
|
|
this.byUuid.put(uUID, entity);
|
|
this.byId.put(entity.getId(), entity);
|
|
}
|
|
}
|
|
|
|
public void remove(T entity) {
|
|
this.byUuid.remove(entity.getUUID());
|
|
this.byId.remove(entity.getId());
|
|
}
|
|
|
|
@Nullable
|
|
public T getEntity(int id) {
|
|
return this.byId.get(id);
|
|
}
|
|
|
|
@Nullable
|
|
public T getEntity(UUID uuid) {
|
|
return (T)this.byUuid.get(uuid);
|
|
}
|
|
|
|
public int count() {
|
|
return this.byUuid.size();
|
|
}
|
|
}
|