Make definition of ci and projects dynamic (#33)

* Make definition of ci and projects dynamic

* Try defining matrix properly

* Working impl

* Fix target task

* Fix path
This commit is contained in:
Alex
2024-10-27 17:46:28 +01:00
committed by GitHub
parent b583525f2c
commit e40ba85ab9
508 changed files with 29 additions and 47 deletions
+53
View File
@@ -0,0 +1,53 @@
import xyz.wagyourtail.unimined.api.minecraft.task.RemapJarTask
import xyz.wagyourtail.unimined.internal.minecraft.MinecraftProvider
plugins {
id 'xyz.wagyourtail.unimined'
}
unimined.minecraft {
version "1.7"
mappings {
legacyIntermediary()
legacyYarn(514)
devFallbackNamespace "intermediary"
}
runs.config("server") {
javaVersion = JavaVersion.VERSION_21
}
customPatcher(new CustomLegacyFabricMinecraftTransformer(project, delegate as MinecraftProvider)) {
it.loader libs.versions.fabric.loader.get()
}
defaultRemapJar = true
}
dependencies {
implementation project(":common")
}
processResources {
filteringCharset "UTF-8"
filesMatching("fabric.mod.json") {
expand "version": project.version
}
}
tasks.withType(RemapJarTask).configureEach {
onlyIf { false}
}
tasks.withType(JavaCompile).configureEach {
it.options.encoding = "UTF-8"
}
java {
toolchain {
languageVersion.set(JavaLanguageVersion.of(21))
}
}
@@ -0,0 +1,33 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.LanguageAccessor;
import net.minecraft.util.Language;
import java.util.Map;
public class AttributesDataGenerator implements IDataGenerator {
@Override
public String getDataName() {
return "attributes";
}
@Override
public JsonElement generateDataJson() {
Language lang = new Language();
Map<String, String> translations = ((LanguageAccessor) lang).translations();
JsonArray arr = new JsonArray();
for (Map.Entry<String, String> translation : translations.entrySet()) {
String key = translation.getKey();
if (!key.startsWith("attribute.name.")) continue;
JsonObject obj = new JsonObject();
key = key.replace("attribute.name.", "");
obj.addProperty("name", key.split("\\.")[1]);
obj.addProperty("resource", key);
arr.add(obj);
}
return arr;
}
}
@@ -0,0 +1,106 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.BiomeAccessor;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.world.biome.*;
import java.util.Locale;
import java.util.SortedMap;
import java.util.TreeMap;
public class BiomesDataGenerator implements IDataGenerator {
private static String guessBiomeDimensionFromCategory(Biome biome) {
if (biome instanceof NetherBiome) {
return "nether";
} else if (biome instanceof EndBiome) {
return "end";
}
return "overworld";
}
public static JsonObject generateBiomeInfo(Biome biome) {
JsonObject biomeDesc = new JsonObject();
biomeDesc.addProperty("id", Registries.BIOMES.getRawId(biome));
biomeDesc.addProperty("name", String.join("_", ((BiomeAccessor) biome).name().toLowerCase(Locale.ENGLISH).split(" ")));
biomeDesc.addProperty("category", category(biome));
biomeDesc.addProperty("temperature", biome.temperature);
biomeDesc.addProperty("precipitation", precipitation(biome));
biomeDesc.addProperty("depth", biome.depth);
biomeDesc.addProperty("dimension", guessBiomeDimensionFromCategory(biome));
biomeDesc.addProperty("displayName", ((BiomeAccessor) biome).name());
biomeDesc.addProperty("color", biome.getSkyColor(biome.temperature));
biomeDesc.addProperty("rainfall", biome.downfall);
return biomeDesc;
}
private static String category(Biome biome) {
if (biome instanceof ForestBiome) {
return "forest";
} else if (biome instanceof OceanBiome) {
return "ocean";
} else if (biome instanceof PlainsBiome) {
return "plains";
} else if (biome instanceof DesertBiome) {
return "desert";
} else if (biome instanceof ExtremeHillsBiome) {
return "extreme_hills";
} else if (biome instanceof TaigaBiome) {
return "taiga";
} else if (biome instanceof SwampBiome) {
return "swamp";
} else if (biome instanceof RiverBiome) {
return "river";
} else if (biome instanceof NetherBiome) {
return "nether";
} else if (biome instanceof EndBiome) {
return "the_end";
} else if (biome instanceof IceBiome) {
return "icy";
} else if (biome instanceof MushroomBiome) {
return "mushroom";
} else if (biome instanceof BeachBiome) {
return "beach";
} else if (biome instanceof JungleBiome) {
return "jungle";
} else if (biome instanceof SavannaBiome) {
return "savanna";
} else if (biome instanceof MesaBiome) {
return "mesa";
} else if (biome instanceof StoneBeachBiome) {
return "none"; // Should StoneBeachBiome be beach too? this is how it is now in mcdata
}
throw new IllegalStateException("Unable to find biome category for " + biome.getClass().getName());
}
private static String precipitation(Biome biome) {
float rainfall = biome.downfall;
float temperature = biome.temperature;
if (rainfall == 0) {
return "none";
} else if (temperature < 0.2f) {
return "snow";
}
return "rain";
}
@Override
public String getDataName() {
return "biomes";
}
@Override
public JsonArray generateDataJson() {
SortedMap<Integer, JsonObject> biomes = new TreeMap<>();
for (Biome biome : Registries.BIOMES) {
biomes.put(biome.id, generateBiomeInfo(biome));
}
JsonArray biomesArray = new JsonArray();
biomes.values().forEach(biomesArray::add);
return biomesArray;
}
}
@@ -0,0 +1,132 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.block.Block;
import net.minecraft.client.Texture;
import net.minecraft.util.math.Box;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
public class BlockCollisionShapesDataGenerator implements IDataGenerator {
private static final Box ENTITY_BOX = Box.of(0.0D, 0.0D, 0.0D, 1.0D, 2.0D, 1.0D);
private static String nameOf(Block block) {
return Objects.requireNonNull(Registries.BLOCKS.getId(block));
}
private static JsonArray jsonOf(Box box) {
JsonArray arr = new JsonArray();
if (box == null) return arr;
arr.add(new JsonPrimitive(box.minX));
arr.add(new JsonPrimitive(box.minY));
arr.add(new JsonPrimitive(box.minZ));
arr.add(new JsonPrimitive(box.maxX));
arr.add(new JsonPrimitive(box.maxY));
arr.add(new JsonPrimitive(box.maxZ));
return arr;
}
@Override
public String getDataName() {
return "blockCollisionShapes";
}
@Override
public JsonObject generateDataJson() {
ShapeCache shapeCache = new ShapeCache();
JsonObject blocksObject = new JsonObject();
for (Block block : Registries.BLOCKS) {
Object val = shapeCache.addShapesFrom(block);
if (val instanceof JsonArray) {
blocksObject.add(nameOf(block), (JsonElement) val);
} else {
blocksObject.addProperty(nameOf(block), (Integer) val);
}
}
JsonObject resultObject = new JsonObject();
resultObject.add("blocks", blocksObject);
resultObject.add("shapes", shapeCache.toJSON());
return resultObject;
}
public static class ShapeCache {
private final ArrayList<Shapes> shapesCache = new ArrayList<>();
public Object addShapesFrom(Block block) {
List<Integer> indexesOfBoxesInTheShapesCache = new ArrayList<>();
for (Field field : block.getClass().getDeclaredFields()) {
if (!field.getType().getName().equals("[Lnet.minecraft.client.ItemIcon;")) { // ItemIcon[]
continue;
}
try {
field.setAccessible(true);
Texture[] icons = (Texture[]) field.get(block);
return icons.length;
} catch (Exception err) {
err.printStackTrace();
}
}
// old way
// new Throwable("find block damages based on ItemIcon[]").printStackTrace();
// for (BlockState state : block.getStateManager().getBlockStates().reverse()) {
// List<Box> boxes = new ArrayList<>();
// try {
// block.appendCollisionBoxes(DGU.getWorld(), BlockPos.ORIGIN, state, ENTITY_BOX, boxes, null);
// } catch (Exception e) {
// e.printStackTrace();
// }
// Shapes thisBlockStateShapes = new Shapes(boxes);
// int indexOfThisBlockStatesShapes = shapesCache.indexOf(thisBlockStateShapes);
// if (indexOfThisBlockStatesShapes != -1) {
// indexesOfBoxesInTheShapesCache.add(indexOfThisBlockStatesShapes);
// } else {
// shapesCache.add(thisBlockStateShapes);
// indexesOfBoxesInTheShapesCache.add(shapesCache.size() - 1);
// }
// }
// if (indexesOfBoxesInTheShapesCache.stream().distinct().count() < 2) {
// return indexesOfBoxesInTheShapesCache.getFirst();
// } else {
// JsonArray shapeIndexes = new JsonArray();
// indexesOfBoxesInTheShapesCache.forEach(shapeIndex -> shapeIndexes.add(new JsonPrimitive(shapeIndex)));
// return shapeIndexes;
// }
return 1;
}
public JsonObject toJSON() {
JsonObject shapes = new JsonObject();
int i = 0;
for (Shapes s : shapesCache) {
shapes.add(String.valueOf(i++), s.toJSON());
}
return shapes;
}
private record Shapes(List<Box> boxes) {
public JsonArray toJSON() {
JsonArray arr = new JsonArray();
boxes.forEach(box -> arr.add(jsonOf(box)));
return arr;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Shapes shapes = (Shapes) o;
return Objects.equals(boxes, shapes.boxes);
}
}
}
}
@@ -0,0 +1,91 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.BlockAccessor;
import dev.u9g.minecraftdatagenerator.mixin.accessor.MiningToolItemAccessor;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.block.AirBlock;
import net.minecraft.block.Block;
import net.minecraft.block.TransparentBlock;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.item.ToolItem;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
public class BlocksDataGenerator implements IDataGenerator {
private static List<Item> getItemsEffectiveForBlock(Block block) {
List<Item> items = new ArrayList<>();
for (Item item : Registries.ITEMS) {
if (item instanceof ToolItem && ((MiningToolItemAccessor) item).getEffectiveBlocks().contains(block)) {
items.add(item);
}
}
return items;
}
private static List<ItemStack> populateDropsIfPossible(Item firstToolItem) {
return new ArrayList<>();
}
public static JsonObject generateBlock(Block block) {
JsonObject blockDesc = new JsonObject();
String registryKey = Registries.BLOCKS.getId(block);
List<Item> effectiveTools = getItemsEffectiveForBlock(block);
blockDesc.addProperty("id", Registries.BLOCKS.getRawId(block));
blockDesc.addProperty("name", Objects.requireNonNull(registryKey));
if (!block.getTranslatedName().startsWith("tile.")) {
blockDesc.addProperty("displayName", block.getTranslatedName());
}
float hardness = block.method_471(null, 0, 0, 0);
blockDesc.addProperty("hardness", hardness);
blockDesc.addProperty("resistance", ((BlockAccessor) block).getBlastResistance());
blockDesc.addProperty("diggable", hardness != -1.0f && !(block instanceof AirBlock));
JsonObject effTools = new JsonObject();
effectiveTools.forEach(item -> effTools.addProperty(
String.valueOf(Registries.ITEMS.getRawId(item)), // key
item.getMiningSpeedMultiplier(DGU.stackFor(item), block) // value
));
blockDesc.add("effectiveTools", effTools);
blockDesc.addProperty("transparent", block instanceof TransparentBlock);
blockDesc.addProperty("emitLight", block.getLightLevel());
blockDesc.addProperty("filterLight", block.getOpacity());
blockDesc.addProperty("boundingBox", boundingBox(block));
return blockDesc;
}
private static String boundingBox(Block block) {
if (block.getBoundingBox(DGU.getWorld(), 0, 0, 0) == null) {
return "empty";
}
return "block";
}
private static Item getItemFromBlock(Block block) {
return Registries.ITEMS.get(Registries.BLOCKS.getId(block));
}
@Override
public String getDataName() {
return "blocks";
}
@Override
public JsonArray generateDataJson() {
JsonArray resultBlocksArray = new JsonArray();
for (Block block : Registries.BLOCKS) {
resultBlocksArray.add(generateBlock(block));
}
return resultBlocksArray;
}
}
@@ -0,0 +1,39 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.StatusEffectAccessor;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.entity.effect.StatusEffect;
import org.jetbrains.annotations.NotNull;
import java.util.Objects;
public class EffectsDataGenerator implements IDataGenerator {
public static JsonObject generateEffect(StatusEffect statusEffect) {
JsonObject effectDesc = new JsonObject();
@NotNull String name = Objects.requireNonNull(Registries.STATUS_EFFECTS.getId(statusEffect));
effectDesc.addProperty("id", Registries.STATUS_EFFECTS.getRawId(statusEffect));
effectDesc.addProperty("name", name);
effectDesc.addProperty("displayName", DGU.translateText(statusEffect.getTranslationKey()));
effectDesc.addProperty("type", !((StatusEffectAccessor) statusEffect).negative() ? "good" : "bad");
return effectDesc;
}
@Override
public String getDataName() {
return "effects";
}
@Override
public JsonArray generateDataJson() {
JsonArray resultsArray = new JsonArray();
for (StatusEffect effect : Registries.STATUS_EFFECTS) {
resultsArray.add(generateEffect(effect));
}
return resultsArray;
}
}
@@ -0,0 +1,107 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.common.collect.ImmutableMap;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.enchantment.Enchantment;
import net.minecraft.enchantment.EnchantmentTarget;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Objects;
public class EnchantmentsDataGenerator implements IDataGenerator {
private static final ImmutableMap<EnchantmentTarget, String> ENCHANTMENT_TARGET_NAMES = ImmutableMap.<EnchantmentTarget, String>builder()
.put(EnchantmentTarget.ARMOR, "armor")
.put(EnchantmentTarget.ARMOR_FEET, "armor_feet")
.put(EnchantmentTarget.ARMOR_LEGS, "armor_legs")
.put(EnchantmentTarget.ARMOR_TORSO, "armor_chest")
.put(EnchantmentTarget.ARMOR_HEAD, "armor_head")
.put(EnchantmentTarget.WEAPON, "weapon")
.put(EnchantmentTarget.DIGGER, "digger")
.put(EnchantmentTarget.FISHING_ROD, "fishing_rod")
.put(EnchantmentTarget.BREAKABLE, "breakable")
.put(EnchantmentTarget.BOW, "bow")
.build();
public static String getEnchantmentTargetName(EnchantmentTarget target) {
return ENCHANTMENT_TARGET_NAMES.getOrDefault(target, target.name().toLowerCase(Locale.ROOT));
}
// Equation enchantment costs follow is a * level + b, so we can easily retrieve a and b by passing zero level
private static JsonObject generateEnchantmentMinPowerCoefficients(Enchantment enchantment) {
int b = enchantment.getMinimumPower(0);
int a = enchantment.getMinimumPower(1) - b;
JsonObject resultObject = new JsonObject();
resultObject.addProperty("a", a);
resultObject.addProperty("b", b);
return resultObject;
}
private static JsonObject generateEnchantmentMaxPowerCoefficients(Enchantment enchantment) {
int b = enchantment.getMaximumPower(0);
int a = enchantment.getMaximumPower(1) - b;
JsonObject resultObject = new JsonObject();
resultObject.addProperty("a", a);
resultObject.addProperty("b", b);
return resultObject;
}
public static JsonObject generateEnchantment(Enchantment enchantment) {
JsonObject enchantmentDesc = new JsonObject();
String registryKey = Registries.ENCHANTMENTS.getId(enchantment);
enchantmentDesc.addProperty("id", Registries.ENCHANTMENTS.getRawId(enchantment));
enchantmentDesc.addProperty("name", Objects.requireNonNull(registryKey));
enchantmentDesc.addProperty("displayName", DGU.translateText(enchantment.getTranslationKey()));
enchantmentDesc.addProperty("maxLevel", enchantment.getMaximumLevel());
enchantmentDesc.add("minCost", generateEnchantmentMinPowerCoefficients(enchantment));
enchantmentDesc.add("maxCost", generateEnchantmentMaxPowerCoefficients(enchantment));
enchantmentDesc.addProperty("treasureOnly", false); // 1.9 added treasure enchants
enchantmentDesc.addProperty("curse", false); // 1.10 added curse enchants
List<Enchantment> incompatibleEnchantments = new ArrayList<>();
for (Enchantment other : Registries.ENCHANTMENTS) {
if (!enchantment.differs(other) && !other.differs(enchantment) && other != enchantment) {
incompatibleEnchantments.add(other);
}
}
JsonArray excludes = new JsonArray();
for (Enchantment excludedEnchantment : incompatibleEnchantments) {
String otherKey = Registries.ENCHANTMENTS.getId(excludedEnchantment);
excludes.add(new JsonPrimitive(Objects.requireNonNull(otherKey)));
}
enchantmentDesc.add("exclude", excludes);
enchantmentDesc.addProperty("category", getEnchantmentTargetName(enchantment.target));
enchantmentDesc.addProperty("weight", enchantment.getEnchantmentType()); // see AnvilScreenhandler L209
enchantmentDesc.addProperty("tradeable", true); // the first non-tradeable enchant came in 1.16, soul speed
enchantmentDesc.addProperty("discoverable", true); // the first non-enchantable enchant came in 1.16, soul speed
return enchantmentDesc;
}
@Override
public String getDataName() {
return "enchantments";
}
@Override
public JsonArray generateDataJson() {
JsonArray resultsArray = new JsonArray();
for (Enchantment enchantment : Registries.ENCHANTMENTS) {
resultsArray.add(generateEnchantment(enchantment));
}
return resultsArray;
}
}
@@ -0,0 +1,134 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.EntityTypeAccessor;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.entity.*;
import net.minecraft.entity.mob.AmbientEntity;
import net.minecraft.entity.mob.HostileEntity;
import net.minecraft.entity.mob.MobEntity;
import net.minecraft.entity.mob.WaterCreatureEntity;
import net.minecraft.entity.passive.AnimalEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.entity.projectile.FishingBobberEntity;
import net.minecraft.entity.projectile.Projectile;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.Objects;
public class EntitiesDataGenerator implements IDataGenerator {
public static JsonObject generateEntity(Class<? extends Entity> entityClass) {
JsonObject entityDesc = new JsonObject();
String registryKey = Registries.ENTITY_TYPES.getId(entityClass);
int entityRawId = Registries.ENTITY_TYPES.getRawId(entityClass);
@Nullable Entity entity = makeEntity(entityClass);
// FIXME: ENTITY ID IS WRONG
int id = entityId(entity);
entityDesc.addProperty("id", id);
entityDesc.addProperty("internalId", id);
entityDesc.addProperty("name", Objects.requireNonNull(registryKey));
String displayName = entity != null ? entity.getTranslationKey() : null;
if (displayName != null && !displayName.startsWith("entity.")) {
entityDesc.addProperty("displayName", displayName);
}
entityDesc.addProperty("width", entity == null ? 0 : entity.width);
entityDesc.addProperty("height", entity == null ? 0 : entity.height);
String entityTypeString = getEntityTypeForClass(entityClass);
entityDesc.addProperty("type", entityTypeString);
entityDesc.addProperty("category", getCategoryFrom(entityClass));
return entityDesc;
}
private static Entity makeEntity(Class<? extends Entity> type) {
String name = EntityTypeAccessor.CLASS_NAME_MAP().get(type);
return EntityType.createInstanceFromName(name, DGU.getWorld());
}
private static String getCategoryFrom(@NotNull Class<?> entityClass) {
if (entityClass == PlayerEntity.class) return "other"; // fail early for player entities
String packageName = entityClass.getPackage().getName();
return switch (packageName) {
case "net.minecraft.entity.decoration", "net.minecraft.entity.decoration.painting" -> "Immobile";
case "net.minecraft.entity.boss", "net.minecraft.entity.mob", "net.minecraft.entity.boss.dragon" ->
"Hostile mobs";
case "net.minecraft.entity.projectile", "net.minecraft.entity.thrown" -> "Projectiles";
case "net.minecraft.entity.passive" -> "Passive mobs";
case "net.minecraft.entity.vehicle" -> "Vehicles";
case "net.minecraft.entity" -> "other";
default -> throw new IllegalStateException("Unexpected entity type: " + packageName);
};
}
//Honestly, both "type" and "category" fields in the schema and examples do not contain any useful information
//Since category is optional, I will just leave it out, and for type I will assume general entity classification
//by the Entity class hierarchy (which has some weirdness too by the way)
private static String getEntityTypeForClass(Class<? extends Entity> entityClass) {
//Top-level classifications
if (WaterCreatureEntity.class.isAssignableFrom(entityClass)) {
return "water_creature";
}
if (AnimalEntity.class.isAssignableFrom(entityClass)) {
return "animal";
}
if (HostileEntity.class.isAssignableFrom(entityClass)) {
return "hostile";
}
if (AmbientEntity.class.isAssignableFrom(entityClass)) {
return "ambient";
}
//Second level classifications. PathAwareEntity is not included because it
//doesn't really make much sense to categorize by it
if (PathAwareEntity.class.isAssignableFrom(entityClass)) {
return "passive";
}
if (MobEntity.class.isAssignableFrom(entityClass)) {
return "mob";
}
//Other classifications only include living entities and projectiles. everything else is categorized as other
if (LivingEntity.class.isAssignableFrom(entityClass)) {
return "living";
}
if (Projectile.class.isAssignableFrom(entityClass)) {
return "projectile";
}
return "other";
}
private static int entityId(Entity entity) {
if (!DGU.getCurrentlyRunningServer().getVersion().equals("1.7")) {
throw new IllegalStateException("These ids were gotten manually for 1.7, remake for " + DGU.getCurrentlyRunningServer().getVersion());
}
int rawId = Registries.ENTITY_TYPES.getRawId(entity.getClass());
if (rawId == -1) { // see TrackedEntityInstance
if (entity instanceof ItemEntity) {
return 2;
} else if (entity instanceof FishingBobberEntity) {
return 90;
} else {
throw new IllegalStateException("unable to find rawId for entity: " + entity.getClass().getName());
}
}
return rawId;
}
@Override
public String getDataName() {
return "entities";
}
@Override
public JsonArray generateDataJson() {
JsonArray resultArray = new JsonArray();
for (Class<? extends Entity> entityType : Registries.ENTITY_TYPES) {
resultArray.add(generateEntity(entityType));
}
return resultArray;
}
}
@@ -0,0 +1,48 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.item.FoodItem;
import net.minecraft.item.Item;
import java.util.Objects;
public class FoodsDataGenerator implements IDataGenerator {
public static JsonObject generateFoodDescriptor(FoodItem foodItem) {
JsonObject foodDesc = new JsonObject();
String registryKey = Registries.ITEMS.getId(foodItem);
foodDesc.addProperty("id", Registries.ITEMS.getRawId(foodItem));
foodDesc.addProperty("name", Objects.requireNonNull(registryKey).replace("minecraft:", ""));
foodDesc.addProperty("stackSize", foodItem.getMaxCount());
foodDesc.addProperty("displayName", foodItem.getDisplayName(DGU.stackFor(foodItem)));
float foodPoints = foodItem.getHungerPoints(DGU.stackFor(foodItem));
float saturationRatio = foodItem.getSaturation(DGU.stackFor(foodItem)) * 2.0F;
float saturation = foodPoints * saturationRatio;
foodDesc.addProperty("foodPoints", foodPoints);
foodDesc.addProperty("saturation", saturation);
foodDesc.addProperty("effectiveQuality", foodPoints + saturation);
foodDesc.addProperty("saturationRatio", saturationRatio);
return foodDesc;
}
@Override
public String getDataName() {
return "foods";
}
public JsonArray generateDataJson() {
JsonArray resultsArray = new JsonArray();
for (Item item : Registries.ITEMS) {
if (item instanceof FoodItem) {
resultsArray.add(generateFoodDescriptor((FoodItem) item));
}
}
return resultsArray;
}
}
@@ -0,0 +1,36 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import net.minecraft.block.material.Material;
public class InstrumentsDataGenerator implements IDataGenerator {
public static Material currentMaterial = null;
public static int LAST_WORLD_BLOCKDATA = -1;
@Override
public String getDataName() {
return "instruments";
}
@Override
public JsonElement generateDataJson() {
JsonArray array = new JsonArray();
// copied from NoteBlock#syncedBlockAction
array.add(makeInstrumentObj(0, "harp"));
array.add(makeInstrumentObj(1, "bd"));
array.add(makeInstrumentObj(2, "snare"));
array.add(makeInstrumentObj(3, "hat"));
array.add(makeInstrumentObj(4, "bassattack"));
return array;
}
private JsonObject makeInstrumentObj(int i, String s) {
JsonObject obj = new JsonObject();
obj.add("id", new JsonPrimitive(i));
obj.add("name", new JsonPrimitive(s));
return obj;
}
}
@@ -0,0 +1,102 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import dev.u9g.minecraftdatagenerator.mixin.VariantBlockItemAccessor;
import dev.u9g.minecraftdatagenerator.mixin.accessor.ItemAccessor;
import dev.u9g.minecraftdatagenerator.util.DGU;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.enchantment.EnchantmentTarget;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.item.VariantBlockItem;
import java.util.*;
import java.util.stream.Collectors;
public class ItemsDataGenerator implements IDataGenerator {
private static List<Item> calculateItemsToRepairWith(Item sourceItem) {
List<Item> items = new ArrayList<>();
for (Item otherItem : Registries.ITEMS) {
if (sourceItem.canRepair(DGU.stackFor(sourceItem), DGU.stackFor(otherItem))) {
items.add(otherItem);
}
}
return items;
}
private static Set<String> getApplicableEnchantmentTargets(Item sourceItem) {
return Arrays.stream(EnchantmentTarget.values())
.filter(target -> target.isCompatible(sourceItem))
.map(EnchantmentsDataGenerator::getEnchantmentTargetName)
.sorted()
.collect(Collectors.toCollection(LinkedHashSet::new));
}
public static JsonObject generateItem(Item item) {
JsonObject itemDesc = new JsonObject();
itemDesc.addProperty("id", Registries.ITEMS.getRawId(item));
String name = Registries.ITEMS.getId(item);
itemDesc.addProperty("name", name == null ? ((ItemAccessor) item).translationKey() : name.replace("minecraft:", ""));
itemDesc.addProperty("displayName", item.getDisplayName(DGU.stackFor(item)));
itemDesc.addProperty("stackSize", item.getMaxCount());
JsonArray enchantCategoriesArray = new JsonArray();
getApplicableEnchantmentTargets(item).forEach(enchantCategoriesArray::add);
if (!enchantCategoriesArray.isEmpty()) {
itemDesc.add("enchantCategories", enchantCategoriesArray);
}
if (item.isDamageable()) {
List<Item> repairWithItems = calculateItemsToRepairWith(item);
JsonArray fixedWithArray = new JsonArray();
for (Item repairWithItem : repairWithItems) {
String repairWithName = Registries.ITEMS.getId(repairWithItem);
fixedWithArray.add(new JsonPrimitive(Objects.requireNonNull(repairWithName)));
}
if (!fixedWithArray.isEmpty()) {
itemDesc.add("repairWith", fixedWithArray);
}
int maxDurability = item.getMaxDamage();
itemDesc.addProperty("maxDurability", maxDurability);
}
if (item instanceof VariantBlockItem it) {
JsonArray variations = new JsonArray();
int i = 0;
JsonObject obj = new JsonObject();
for (String variant : ((VariantBlockItemAccessor) it).variants()) {
ItemStack stack = new ItemStack(item, 1, i);
obj.add("id", new JsonPrimitive(i));
obj.add("name", new JsonPrimitive(variant));
obj.add("displayName", new JsonPrimitive(DGU.translateText(it.getTranslationKey(stack) + ".name")));
variations.add(obj);
i++;
}
itemDesc.add("variations", variations);
}
return itemDesc;
}
@Override
public String getDataName() {
return "items";
}
@Override
public JsonArray generateDataJson() {
JsonArray resultArray = new JsonArray();
for (Item item : Registries.ITEMS) {
resultArray.add(generateItem(item));
}
return resultArray;
}
}
@@ -0,0 +1,29 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import dev.u9g.minecraftdatagenerator.mixin.accessor.LanguageAccessor;
import dev.u9g.minecraftdatagenerator.util.Registries;
import java.util.Map;
public class LanguageDataGenerator implements IDataGenerator {
@Override
public String getDataName() {
return "language";
}
@Override
public JsonElement generateDataJson() {
try {
JsonObject obj = new JsonObject();
Map<String, String> translations = ((LanguageAccessor) Registries.LANGUAGE).translations();
for (Map.Entry<String, String> entry : translations.entrySet()) {
obj.addProperty(entry.getKey(), entry.getValue());
}
return obj;
} catch (Exception e) {
throw new RuntimeException("Failed to generate language file", e);
}
}
}
@@ -0,0 +1,125 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonElement;
import com.google.gson.JsonNull;
public class RecipeDataGenerator implements IDataGenerator {
@Override
public String getDataName() {
return "recipes";
}
@Override
public JsonElement generateDataJson() {
// JsonObject finalObj = new JsonObject();
// Multimap<Integer, JsonObject> recipes = ArrayListMultimap.create();
// for (Recipe<?> recipe : Objects.requireNonNull(DGU.getWorld()).getRecipeManager().values()) {
// if (recipe instanceof ShapedRecipe sr) {
// var ingredients = sr.getIngredients();
// List<Integer> ingr = new ArrayList<>();
// for (int i = 0; i < 9; i++) {
// if (i >= ingredients.size()) {
// ingr.add(-1);
// continue;
// }
// var stacks = ingredients.get(i);
//// var matching = stacks.getMatchingStacks();
//// if (matching.length == 0) {
//// ingr.add(-1);
//// } else {
//// ingr.add(getRawIdFor(matching[0].getItem()));
//// }
// }
// Lists.reverse(ingr);
//
// JsonArray inShape = new JsonArray();
//
// var iter = ingr.iterator();
// for (int y = 0; y < 3; y++) {
// var jsonRow = new JsonArray();
// int one = iter.next();
// int two = iter.next();
// int three = iter.next();
// if (y > 0 && one == -1 && two == -1 && three == -1) continue;
// jsonRow.add(one);
// jsonRow.add(two);
// jsonRow.add(three);
// inShape.add(jsonRow);
// }
//
// JsonObject finalRecipe = new JsonObject();
// finalRecipe.add("inShape", inShape);
//
// var resultObject = new JsonObject();
// resultObject.addProperty("id", getRawIdFor(sr.getOutput().getItem()));
// resultObject.addProperty("count", sr.getOutput().getCount());
// finalRecipe.add("result", resultObject);
//
// String id = ((Integer) getRawIdFor(sr.getOutput().getItem())).toString();
//
// if (!finalObj.has(id)) {
// finalObj.add(id, new JsonArray());
// }
// finalObj.get(id).getAsJsonArray().add(finalRecipe);
//// var input = new JsonArray();
//// var ingredients = sr.getIngredients().stream().toList();
//// for (int y = 0; y < sr.getHeight(); y++) {
//// var arr = new JsonArray();
//// for (int x = 0; x < sr.getWidth(); x++) {
//// if ((y*3)+x >= ingredients.size()) {
//// arr.add(JsonNull.INSTANCE);
//// continue;
//// }
//// var ingredient = ingredients.get((y*3)+x).getMatchingStacks(); // FIXME: fix when there are more than one matching stack
//// if (ingredient.length == 0) {
//// arr.add(JsonNull.INSTANCE);
//// } else {
//// arr.add(getRawIdFor(ingredient[0].getItem()));
//// }
//// }
//// input.add(arr);
//// }
//// var rootRecipeObject = new JsonObject();
//// rootRecipeObject.add("inShape", input);
//// var resultObject = new JsonObject();
//// resultObject.addProperty("id", getRawIdFor(sr.getOutput().getItem()));
//// resultObject.addProperty("count", sr.getOutput().getCount());
//// rootRecipeObject.add("result", resultObject);
//// recipes.put(getRawIdFor(sr.getOutput().getItem()), rootRecipeObject);
// } else if (recipe instanceof ShapelessRecipe sl) {
// var ingredients = new JsonArray();
// for (Ingredient ingredient : sl.getIngredients()) {
// if (ingredient.isEmpty()) continue;
//// ingredients.add(getRawIdFor(ingredient.getMatchingStacks()[0].getItem()));
// }
// var rootRecipeObject = new JsonObject();
// rootRecipeObject.add("ingredients", ingredients);
// var resultObject = new JsonObject();
// resultObject.addProperty("id", getRawIdFor(sl.getOutput().getItem()));
// resultObject.addProperty("count", sl.getOutput().getCount());
// rootRecipeObject.add("result", resultObject);
// recipes.put(getRawIdFor(sl.getOutput().getItem()), rootRecipeObject);
// }
// }
// recipes.forEach((a, b) -> {
// if (!finalObj.has(a.toString())) {
// finalObj.add(a.toString(), new JsonArray());
// }
// finalObj.get(a.toString()).getAsJsonArray().add(b);
// });
// return finalObj;
return JsonNull.INSTANCE;
}
@Override
public boolean isEnabled() {
return false; // TODO: Implement this
}
//
// private static int getRawIdFor (Item item) {
// return Registry.ITEM.getRawId(item);
// }
}
@@ -0,0 +1,136 @@
package dev.u9g.minecraftdatagenerator.generators;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import dev.u9g.minecraftdatagenerator.mixin.accessor.BiomeAccessor;
import dev.u9g.minecraftdatagenerator.util.Registries;
import net.minecraft.block.Block;
import net.minecraft.block.Blocks;
import net.minecraft.client.color.world.FoliageColors;
import net.minecraft.client.color.world.GrassColors;
import net.minecraft.util.math.MathHelper;
import net.minecraft.world.biome.Biome;
import java.util.*;
public class TintsDataGenerator implements IDataGenerator {
public static BiomeTintColors generateBiomeTintColors() {
BiomeTintColors colors = new BiomeTintColors();
for (Biome biome : Registries.BIOMES) {
double d = MathHelper.clamp(biome.temperature, 0.0f, 1.0f);
double e = MathHelper.clamp(biome.downfall, 0.0f, 1.0f);
int biomeGrassColor = GrassColors.getColor(d, e);
int biomeFoliageColor = FoliageColors.getColor(d, e);
int biomeWaterColor = ((BiomeAccessor) biome).waterColor();
colors.grassColoursMap.computeIfAbsent(biomeGrassColor, k -> new ArrayList<>()).add(biome);
colors.foliageColoursMap.computeIfAbsent(biomeFoliageColor, k -> new ArrayList<>()).add(biome);
colors.waterColourMap.computeIfAbsent(biomeWaterColor, k -> new ArrayList<>()).add(biome);
}
return colors;
}
public static Map<Block, Integer> generateConstantTintColors() {
Map<Block, Integer> resultColors = new LinkedHashMap<>();
// FIXME: ?
// resultColors.put(Blocks.BIRCH_LEAVES, FoliageColors.getBirchColor());
// resultColors.put(Blocks.SPRUCE_LEAVES, FoliageColors.getSpruceColor());
resultColors.put(Registries.BLOCKS.get("waterlily"), Blocks.LILY_PAD.getColor());
// FIXME: ?
// resultColors.put(Blocks.ATTACHED_MELON_STEM, getBlockColor(Blocks.ATTACHED_MELON_STEM));
// resultColors.put(Blocks.ATTACHED_PUMPKIN_STEM, getBlockColor(Blocks.ATTACHED_PUMPKIN_STEM));
//not really constant, depend on the block age, but kinda have to be handled since textures are literally white without them
resultColors.put(Registries.BLOCKS.get("melon_stem"), Blocks.MELON_STEM.getColor());
resultColors.put(Registries.BLOCKS.get("pumpkin_stem"), Blocks.PUMPKIN_STEM.getColor());
return resultColors;
}
private static JsonObject encodeBiomeColorMap(Map<Integer, List<Biome>> colorsMap) {
JsonArray resultColorsArray = new JsonArray();
for (Map.Entry<Integer, List<Biome>> entry : colorsMap.entrySet()) {
JsonObject entryObject = new JsonObject();
JsonArray keysArray = new JsonArray();
for (Biome biome : entry.getValue()) {
keysArray.add(new JsonPrimitive(biome.name));
}
entryObject.add("keys", keysArray);
entryObject.addProperty("color", entry.getKey());
resultColorsArray.add(entryObject);
}
JsonObject resultObject = new JsonObject();
resultObject.add("data", resultColorsArray);
return resultObject;
}
private static JsonObject encodeRedstoneColorMap(Map<Integer, Integer> colorsMap) {
JsonArray resultColorsArray = new JsonArray();
for (Map.Entry<Integer, Integer> entry : colorsMap.entrySet()) {
JsonObject entryObject = new JsonObject();
JsonArray keysArray = new JsonArray();
keysArray.add(new JsonPrimitive(entry.getKey()));
entryObject.add("keys", keysArray);
entryObject.addProperty("color", entry.getValue());
resultColorsArray.add(entryObject);
}
JsonObject resultObject = new JsonObject();
resultObject.add("data", resultColorsArray);
return resultObject;
}
private static JsonObject encodeBlocksColorMap(Map<Block, Integer> colorsMap) {
JsonArray resultColorsArray = new JsonArray();
for (Map.Entry<Block, Integer> entry : colorsMap.entrySet()) {
JsonObject entryObject = new JsonObject();
JsonArray keysArray = new JsonArray();
String registryKey = Registries.BLOCKS.getId(entry.getKey());
keysArray.add(new JsonPrimitive(Objects.requireNonNull(registryKey).replace("minecraft:", "")));
entryObject.add("keys", keysArray);
entryObject.addProperty("color", entry.getValue());
resultColorsArray.add(entryObject);
}
JsonObject resultObject = new JsonObject();
resultObject.add("data", resultColorsArray);
return resultObject;
}
@Override
public String getDataName() {
return "tints";
}
@Override
public JsonObject generateDataJson() {
BiomeTintColors biomeTintColors = generateBiomeTintColors();
Map<Block, Integer> constantTintColors = generateConstantTintColors();
JsonObject resultObject = new JsonObject();
resultObject.add("grass", encodeBiomeColorMap(biomeTintColors.grassColoursMap));
resultObject.add("foliage", encodeBiomeColorMap(biomeTintColors.foliageColoursMap));
resultObject.add("water", encodeBiomeColorMap(biomeTintColors.waterColourMap));
resultObject.add("constant", encodeBlocksColorMap(constantTintColors));
return resultObject;
}
public static class BiomeTintColors {
final Map<Integer, List<Biome>> grassColoursMap = new LinkedHashMap<>();
final Map<Integer, List<Biome>> foliageColoursMap = new LinkedHashMap<>();
final Map<Integer, List<Biome>> waterColourMap = new LinkedHashMap<>();
}
}
@@ -0,0 +1,28 @@
package dev.u9g.minecraftdatagenerator.mixin;
import dev.u9g.minecraftdatagenerator.MinecraftDataGenerator;
import dev.u9g.minecraftdatagenerator.util.DGU;
import net.minecraft.server.dedicated.MinecraftDedicatedServer;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
import java.io.File;
@Mixin(MinecraftDedicatedServer.class)
public class ReadyMixin {
@Inject(method = "setupServer()Z", at = @At("HEAD"))
private void constructor(CallbackInfoReturnable<Boolean> cir) {
((MinecraftDedicatedServer) (Object) this).setServerPort(0);
}
@Inject(method = "setupServer()Z", at = @At("TAIL"))
private void init(CallbackInfoReturnable<Boolean> cir) {
System.setProperty("fabric.development", "false");
MinecraftDataGenerator.start(
DGU.getCurrentlyRunningServer().getVersion(),
(new File(".")).toPath()
);
}
}
@@ -0,0 +1,11 @@
package dev.u9g.minecraftdatagenerator.mixin;
import net.minecraft.item.VariantBlockItem;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
@Mixin(VariantBlockItem.class)
public interface VariantBlockItemAccessor {
@Accessor("field_5448")
String[] variants();
}
@@ -0,0 +1,21 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.world.biome.Biome;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
import java.util.Set;
@Mixin(Biome.class)
public interface BiomeAccessor {
@Accessor("BIOMESET")
static Set<Biome> BIOMESET() {
throw new IllegalStateException();
}
@Accessor("waterColor")
int waterColor();
@Accessor("name")
String name();
}
@@ -0,0 +1,11 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.block.Block;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
@Mixin(Block.class)
public interface BlockAccessor {
@Accessor("blastResistance")
float getBlastResistance();
}
@@ -0,0 +1,13 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.enchantment.Enchantment;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
@Mixin(Enchantment.class)
public interface EnchantmentAccessor {
@Accessor("ALL_ENCHANTMENTS")
static Enchantment[] ALL_ENCHANTMENTS() {
throw new IllegalStateException();
}
}
@@ -0,0 +1,26 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityType;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
import java.util.Map;
@Mixin(EntityType.class)
public interface EntityTypeAccessor {
@Accessor("CLASS_NAME_MAP")
static Map<Class<? extends Entity>, String> CLASS_NAME_MAP() {
throw new IllegalStateException();
}
@Accessor("ID_CLASS_MAP")
static Map<Integer, Class<? extends Entity>> ID_CLASS_MAP() {
throw new IllegalStateException();
}
@Accessor("CLASS_ID_MAP")
static Map<Class<? extends Entity>, Integer> CLASS_ID_MAP() {
throw new IllegalStateException();
}
}
@@ -0,0 +1,11 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.item.Item;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
@Mixin(Item.class)
public interface ItemAccessor {
@Accessor("translationKey")
String translationKey();
}
@@ -0,0 +1,13 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.util.Language;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
import java.util.Map;
@Mixin(Language.class)
public interface LanguageAccessor {
@Accessor("translations")
Map<String, String> translations();
}
@@ -0,0 +1,15 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.block.Block;
import net.minecraft.item.ToolItem;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
import java.util.Set;
@Mixin(ToolItem.class)
public interface MiningToolItemAccessor {
@Accessor
Set<Block> getEffectiveBlocks();
}
@@ -0,0 +1,16 @@
package dev.u9g.minecraftdatagenerator.mixin.accessor;
import net.minecraft.entity.effect.StatusEffect;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
@Mixin(StatusEffect.class)
public interface StatusEffectAccessor {
@Accessor("STATUS_EFFECTS")
static StatusEffect[] STATUS_EFFECTS() {
throw new IllegalStateException();
}
@Accessor("negative")
boolean negative();
}
@@ -0,0 +1,24 @@
package dev.u9g.minecraftdatagenerator.mixin.overwrite;
import net.minecraft.entity.Entity;
import net.minecraft.entity.ItemEntity;
import net.minecraft.world.World;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Overwrite;
@Mixin(ItemEntity.class)
public abstract class ItemEntityOverwrite extends Entity {
public ItemEntityOverwrite(World world) {
super(world);
}
/**
* @author a
* @reason a
*/
@Overwrite
public String getTranslationKey() {
return super.getTranslationKey();
}
}
@@ -0,0 +1,46 @@
package dev.u9g.minecraftdatagenerator.registryview;
import net.minecraft.util.registry.SimpleRegistry;
import org.jetbrains.annotations.NotNull;
import java.util.Iterator;
import java.util.function.Consumer;
public class RegistryBackedRegistryView<V> implements RegistryView<String, V> {
private final SimpleRegistry registry;
public RegistryBackedRegistryView(SimpleRegistry registry) {
this.registry = registry;
}
@Override
public int getRawId(V value) {
return registry.getRawId(value);
}
@Override
public String getId(V value) {
return registry.getId(value);
}
@Override
public V getByRawId(int rawId) {
return (V) registry.getByRawId(rawId);
}
@Override
public V get(String id) {
return (V) registry.get(id);
}
@NotNull
@Override
public Iterator<V> iterator() {
return (Iterator<V>) registry.iterator();
}
@Override
public void forEach(Consumer<? super V> action) {
registry.forEach(value -> action.accept((V) value));
}
}
@@ -0,0 +1,11 @@
package dev.u9g.minecraftdatagenerator.registryview;
public interface RegistryView<K, V> extends Iterable<V> {
int getRawId(V value);
K getId(V value);
V getByRawId(int rawId);
V get(K id);
}
@@ -0,0 +1,75 @@
package dev.u9g.minecraftdatagenerator.registryview;
import com.google.common.collect.HashBasedTable;
import org.jetbrains.annotations.NotNull;
import java.util.Iterator;
import java.util.Map;
import java.util.Objects;
import java.util.function.Consumer;
public class TableBackedRegistryView<K, V> implements RegistryView<K, V> {
private final HashBasedTable<K, Integer, V> table;
protected TableBackedRegistryView(HashBasedTable<K, Integer, V> table) {
this.table = table;
}
@Override
public int getRawId(V value) {
for (Map.Entry<K, Map<Integer, V>> row : table.rowMap().entrySet()) {
Map.Entry<Integer, V> entry = row.getValue().entrySet().iterator().next();
if (Objects.equals(entry.getValue(), value)) {
return entry.getKey();
}
}
throw new IllegalStateException("value not in registry.");
}
@Override
public K getId(V value) {
for (Map.Entry<K, Map<Integer, V>> row : table.rowMap().entrySet()) {
Map.Entry<Integer, V> entry = row.getValue().entrySet().iterator().next();
if (Objects.equals(entry.getValue(), value)) {
return row.getKey();
}
}
throw new IllegalStateException("value not in registry.");
}
@Override
public V getByRawId(int rawId) {
return table.columnMap().get(rawId).entrySet().iterator().next().getValue();
}
@Override
public V get(K id) {
return table.row(id).entrySet().iterator().next().getValue();
}
@NotNull
@Override
public Iterator<V> iterator() {
return table.values().iterator();
}
@Override
public void forEach(Consumer action) {
for (V v : this) {
action.accept(v);
}
}
public static class Builder<K, V> {
private final HashBasedTable<K, Integer, V> table = HashBasedTable.create();
public Builder<K, V> add(K key, int rawId, V value) {
table.put(key, rawId, value);
return this;
}
public TableBackedRegistryView<K, V> build() {
return new TableBackedRegistryView<>(table);
}
}
}
@@ -0,0 +1,26 @@
package dev.u9g.minecraftdatagenerator.util;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.server.MinecraftServer;
import net.minecraft.world.World;
import org.jetbrains.annotations.NotNull;
public class DGU {
public static MinecraftServer getCurrentlyRunningServer() {
return MinecraftServer.getServer();
}
public static String translateText(String translationKey) {
return Registries.LANGUAGE.translate(translationKey);
}
@NotNull
public static World getWorld() {
return getCurrentlyRunningServer().getWorld();
}
public static ItemStack stackFor(Item ic) {
return new ItemStack(ic);
}
}
@@ -0,0 +1,81 @@
package dev.u9g.minecraftdatagenerator.util;
import dev.u9g.minecraftdatagenerator.mixin.accessor.BiomeAccessor;
import dev.u9g.minecraftdatagenerator.mixin.accessor.EnchantmentAccessor;
import dev.u9g.minecraftdatagenerator.mixin.accessor.EntityTypeAccessor;
import dev.u9g.minecraftdatagenerator.mixin.accessor.StatusEffectAccessor;
import dev.u9g.minecraftdatagenerator.registryview.RegistryBackedRegistryView;
import dev.u9g.minecraftdatagenerator.registryview.RegistryView;
import dev.u9g.minecraftdatagenerator.registryview.TableBackedRegistryView;
import net.minecraft.block.Block;
import net.minecraft.enchantment.Enchantment;
import net.minecraft.entity.Entity;
import net.minecraft.entity.effect.StatusEffect;
import net.minecraft.item.Item;
import net.minecraft.util.Language;
import net.minecraft.world.biome.Biome;
import org.apache.commons.lang3.StringUtils;
import java.util.Locale;
import java.util.Map;
public class Registries {
public static final Language LANGUAGE;
public static final RegistryView<String, Biome> BIOMES;
public static final RegistryView<String, Block> BLOCKS;
public static final RegistryView<String, Item> ITEMS;
public static final RegistryView<String, StatusEffect> STATUS_EFFECTS;
public static final RegistryView<String, Enchantment> ENCHANTMENTS;
public static final RegistryView<String, Class<? extends Entity>> ENTITY_TYPES;
static {
LANGUAGE = new Language();
BIOMES = setupBiomeRegistry();
BLOCKS = new RegistryBackedRegistryView<>(Block.REGISTRY);
ITEMS = new RegistryBackedRegistryView<>(Item.REGISTRY);
STATUS_EFFECTS = setupStatusEffectRegistry();
ENCHANTMENTS = setupEnchantmentRegistry();
ENTITY_TYPES = setupEntityTypesRegistry();
}
private static RegistryView<String, Class<? extends Entity>> setupEntityTypesRegistry() {
TableBackedRegistryView.Builder<String, Class<? extends Entity>> registry = new TableBackedRegistryView.Builder<>();
for (Map.Entry<Integer, Class<? extends Entity>> entry : EntityTypeAccessor.ID_CLASS_MAP().entrySet()) {
String name = EntityTypeAccessor.CLASS_NAME_MAP().get(entry.getValue());
if (name.equals("Mob") || name.equals("Monster")) {
continue;
}
registry.add(name, entry.getKey(), entry.getValue());
}
return registry.build();
}
private static RegistryView<String, Enchantment> setupEnchantmentRegistry() {
TableBackedRegistryView.Builder<String, Enchantment> registry = new TableBackedRegistryView.Builder<>();
for (Enchantment enchantment : EnchantmentAccessor.ALL_ENCHANTMENTS()) {
if (enchantment == null) continue;
String translatedName = Registries.LANGUAGE.translate(enchantment.getTranslationKey());
registry.add(String.join("", translatedName.toLowerCase(Locale.ENGLISH).split(" ")), enchantment.id, enchantment);
}
return registry.build();
}
private static RegistryView<String, Biome> setupBiomeRegistry() {
TableBackedRegistryView.Builder<String, Biome> builder = new TableBackedRegistryView.Builder<>();
for (Biome biome : BiomeAccessor.BIOMESET()) {
builder.add(biome.name, biome.id, biome);
}
return builder.build();
}
private static RegistryView<String, StatusEffect> setupStatusEffectRegistry() {
TableBackedRegistryView.Builder<String, StatusEffect> builder = new TableBackedRegistryView.Builder<>();
for (StatusEffect effect : StatusEffectAccessor.STATUS_EFFECTS()) {
if (effect == null) continue;
String[] words = Registries.LANGUAGE.translate(effect.getTranslationKey()).split(" ");
builder.add(StringUtils.join(words, ""), effect.id, effect);
}
return builder.build();
}
}
+20
View File
@@ -0,0 +1,20 @@
{
"schemaVersion": 1,
"id": "minecraft-data-generator",
"version": "${version}",
"name": "Minecraft Data Generator",
"description": "",
"authors": [
"Archengius",
"U9G"
],
"contact": {},
"license": "MIT",
"mixins": [
"minecraft-data-generator.mixins.json"
],
"depends": {
"fabricloader": "*",
"minecraft": "*"
}
}
@@ -0,0 +1,22 @@
{
"required": true,
"minVersion": "0.8",
"package": "dev.u9g.minecraftdatagenerator.mixin",
"compatibilityLevel": "JAVA_17",
"mixins": [
"ReadyMixin",
"VariantBlockItemAccessor",
"accessor.BiomeAccessor",
"accessor.BlockAccessor",
"accessor.EnchantmentAccessor",
"accessor.EntityTypeAccessor",
"accessor.ItemAccessor",
"accessor.LanguageAccessor",
"accessor.MiningToolItemAccessor",
"accessor.StatusEffectAccessor",
"overwrite.ItemEntityOverwrite"
],
"injectors": {
"defaultRequire": 1
}
}