simono41/fabric-exporter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Another way to fetch world entities (fixes #5)

Browse source
This commit is contained in:
RuscalWorld 2022-07-26 16:16:42 +03:00
parent d06ac8f173
commit 2f6fda00ed
No known key found for this signature in database
GPG key ID: 6FF416979DF2B416
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/main/java/ru/ruscalworld/fabricexporter/metrics/Entities.java
View file

@ -1,8 +1,10 @@
package ru.ruscalworld.fabricexporter.metrics; package ru.ruscalworld.fabricexporter.metrics;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityType; import net.minecraft.entity.EntityType;
import net.minecraft.server.world.ServerWorld; import net.minecraft.server.world.ServerWorld;
import net.minecraft.util.Identifier; import net.minecraft.util.Identifier;
import net.minecraft.util.TypeFilter;
import net.minecraft.util.registry.Registry; import net.minecraft.util.registry.Registry;
import ru.ruscalworld.fabricexporter.FabricExporter; import ru.ruscalworld.fabricexporter.FabricExporter;
import ru.ruscalworld.fabricexporter.util.TextUtil; import ru.ruscalworld.fabricexporter.util.TextUtil;
@ -10,6 +12,19 @@ import ru.ruscalworld.fabricexporter.util.TextUtil;
import java.util.HashMap; import java.util.HashMap;
public class Entities extends Metric { public class Entities extends Metric {
private static final TypeFilter<Entity, ?> ENTITY_FILTER = new TypeFilter<>() {
@Override
public Entity downcast(Entity entity) {
return entity;
}
@Override
public Class<? extends Entity> getBaseClass() {
return Entity.class;
}
};
public Entities() { public Entities() {
super("entities", "Amount of entities in the world", "world", "group", "type"); super("entities", "Amount of entities in the world", "world", "group", "type");
} }
@ -19,7 +34,7 @@ public class Entities extends Metric {
for (ServerWorld world : exporter.getServer().getWorlds()) { for (ServerWorld world : exporter.getServer().getWorlds()) {
HashMap<String, Integer> currentWorldEntities = new HashMap<>(); HashMap<String, Integer> currentWorldEntities = new HashMap<>();
world.iterateEntities().forEach(entity -> { world.getEntitiesByType(ENTITY_FILTER, (entity) -> true).forEach(entity -> {
String name = Registry.ENTITY_TYPE.getId(entity.getType()).getPath(); String name = Registry.ENTITY_TYPE.getId(entity.getType()).getPath();
Integer typeCount = currentWorldEntities.getOrDefault(name, 0); Integer typeCount = currentWorldEntities.getOrDefault(name, 0);
currentWorldEntities.put(name, typeCount + 1); currentWorldEntities.put(name, typeCount + 1);