94 lines
2.7 KiB
Java
94 lines
2.7 KiB
Java
package carpet.utils;
|
|
|
|
import carpet.CarpetServer;
|
|
import com.google.common.collect.Sets;
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.entity.EntityType;
|
|
import net.minecraft.server.world.ServerWorld;
|
|
import net.minecraft.util.registry.Registry;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.stream.Collectors;
|
|
|
|
public class MobAI
|
|
{
|
|
private static Map<EntityType,Set<TrackingType>> aiTrackers = new HashMap<>();
|
|
|
|
public static void resetTrackers()
|
|
{
|
|
aiTrackers.clear();
|
|
}
|
|
|
|
public static boolean isTracking(Entity e, TrackingType type)
|
|
{
|
|
Set<TrackingType> currentTrackers = aiTrackers.get(e.getType());
|
|
if (currentTrackers == null)
|
|
return false;
|
|
return currentTrackers.contains(type);
|
|
}
|
|
|
|
public static void clearTracking(EntityType etype)
|
|
{
|
|
aiTrackers.remove(etype);
|
|
for(ServerWorld world : CarpetServer.minecraft_server.getWorlds() )
|
|
{
|
|
for (Entity e: world.getEntities(etype, Entity::hasCustomName))
|
|
{
|
|
e.setCustomNameVisible(false);
|
|
e.setCustomName(null);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static void startTracking(EntityType e, TrackingType type)
|
|
{
|
|
aiTrackers.putIfAbsent(e,Sets.newHashSet());
|
|
aiTrackers.get(e).add(type);
|
|
}
|
|
|
|
public static List<String> availbleTypes()
|
|
{
|
|
Set<EntityType> types = new HashSet<>();
|
|
for (TrackingType type: TrackingType.values())
|
|
{
|
|
types.addAll(type.types);
|
|
}
|
|
return types.stream().map(t -> Registry.ENTITY_TYPE.getId(t).getPath()).collect(Collectors.toList());
|
|
}
|
|
|
|
public static List<String> availableFor(EntityType<?> entityType)
|
|
{
|
|
Set<TrackingType> availableOptions = new HashSet<>();
|
|
for (TrackingType type: TrackingType.values())
|
|
for (EntityType etype: type.types)
|
|
if (etype == entityType)
|
|
availableOptions.add(type);
|
|
return availableOptions.stream().map(t -> t.name).collect(Collectors.toList());
|
|
}
|
|
|
|
public enum TrackingType
|
|
{
|
|
IRON_GOLEM_SPAWNING("iron_golem_spawning", Sets.newHashSet(EntityType.VILLAGER));
|
|
public Set<EntityType> types;
|
|
public String name;
|
|
TrackingType(String name, Set<EntityType> applicableTypes)
|
|
{
|
|
this.name = name;
|
|
types = applicableTypes;
|
|
}
|
|
|
|
public static TrackingType byName(String aspect)
|
|
{
|
|
for (TrackingType type: values())
|
|
if (type.name.equalsIgnoreCase(aspect))
|
|
return type;
|
|
return null;
|
|
}
|
|
}
|
|
|
|
}
|