alex/valence
1
0
Fork 0
mirror of https://github.com/italicsjenga/valence.git synced 2025-01-11 07:11:30 +11:00
Code Issues Packages Projects Releases Wiki Activity

Add entity extractor

Browse source
This commit is contained in:
Ryan 2022-07-20 01:12:13 -07:00
parent f9be05ee4a
commit b9fca3503c
1 changed files with 59 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
extractor/src/main/java/dev/_00a/valence_extractor/Extractor.java
View file

@ -2,6 +2,10 @@ package dev._00a.valence_extractor;
import com.google.gson.*; import com.google.gson.*;
import net.fabricmc.api.ModInitializer; import net.fabricmc.api.ModInitializer;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityType;
import net.minecraft.entity.data.TrackedData;
import net.minecraft.entity.data.TrackedDataHandlerRegistry;
import net.minecraft.util.math.BlockPos; import net.minecraft.util.math.BlockPos;
import net.minecraft.util.registry.Registry; import net.minecraft.util.registry.Registry;
import net.minecraft.world.EmptyBlockView; import net.minecraft.world.EmptyBlockView;
@ -10,10 +14,13 @@ import org.slf4j.LoggerFactory;
import java.io.FileWriter; import java.io.FileWriter;
import java.io.IOException; import java.io.IOException;
import java.lang.reflect.ParameterizedType;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.util.HashSet;
import java.util.Locale;
public class Extractor implements ModInitializer { public class Extractor implements ModInitializer {
public static final String MOD_ID = "valence_extractor"; public static final String MOD_ID = "valence_extractor";
@ -27,7 +34,7 @@ public class Extractor implements ModInitializer {
try { try {
outputDirectory = Files.createDirectories(Paths.get("valence_extractor_output")); outputDirectory = Files.createDirectories(Paths.get("valence_extractor_output"));
gson = new GsonBuilder().setPrettyPrinting().disableHtmlEscaping().create(); gson = new GsonBuilder().setPrettyPrinting().disableHtmlEscaping().serializeNulls().create();
extractBlocks(); extractBlocks();
extractEntities(); extractEntities();
@ -102,13 +109,59 @@ public class Extractor implements ModInitializer {
writeJsonFile("blocks.json", blocksJson); writeJsonFile("blocks.json", blocksJson);
} }
void extractEntities() throws IOException { @SuppressWarnings("unchecked")
void extractEntities() throws IOException, IllegalAccessException {
var entitiesJson = new JsonArray(); var entitiesJson = new JsonArray();
for (var entity : Registry.ENTITY_TYPE) { var entityClasses = new HashSet<Class<? extends Entity>>();
var entityJson = new JsonObject();
entityJson.addProperty("translation_key", entity.getTranslationKey());
entitiesJson.add(entityJson); for (var f : EntityType.class.getFields()) {
if (f.getType().equals(EntityType.class)) {
var entityType = (EntityType<?>) f.get(null);
var entityClass = (Class<? extends Entity>) ((ParameterizedType) f.getGenericType()).getActualTypeArguments()[0];
var entityJson = new JsonObject();
while (entityClasses.add(entityClass)) {
entityJson.addProperty("class", entityClass.getSimpleName());
if (entityType != null) {
entityJson.addProperty("translation_key", entityType.getTranslationKey());
} else {
entityJson.add("translation_key", null);
}
var fieldsJson = new JsonArray();
for (var entityField : entityClass.getDeclaredFields()) {
if (entityField.getType().equals(TrackedData.class)) {
entityField.setAccessible(true);
var data = (TrackedData<? extends Entity>) entityField.get(null);
var fieldJson = new JsonObject();
fieldJson.addProperty("name", entityField.getName().toLowerCase(Locale.ROOT));
fieldJson.addProperty("index", data.getId());
fieldJson.addProperty("type_id", TrackedDataHandlerRegistry.getId(data.getType()));
fieldsJson.add(fieldJson);
}
}
entityJson.add("fields", fieldsJson);
var parent = entityClass.getSuperclass();
if (parent == null || !Entity.class.isAssignableFrom(parent)) {
entityJson.add("parent", null);
break;
}
entityJson.addProperty("parent", parent.getSimpleName());
entityClass = (Class<? extends Entity>) parent;
entityType = null;
}
if (entityJson.size() > 0) {
entitiesJson.add(entityJson);
}
}
} }
writeJsonFile("entities.json", entitiesJson); writeJsonFile("entities.json", entitiesJson);