2025-11-24 22:52:51 +03:00

530 lines
24 KiB
Java

/*
* Decompiled with CFR 0.152.
*
* Could not load the following classes:
* com.google.common.annotations.VisibleForTesting
* com.google.common.base.Splitter
* com.google.common.base.Strings
* com.google.common.collect.Comparators
* com.google.common.collect.ImmutableMap
* com.google.common.collect.Lists
* com.mojang.brigadier.exceptions.CommandSyntaxException
* com.mojang.logging.LogUtils
* com.mojang.serialization.Codec
* com.mojang.serialization.Dynamic
* it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap
* org.jspecify.annotations.Nullable
* org.slf4j.Logger
*/
package net.minecraft.nbt;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Splitter;
import com.google.common.base.Strings;
import com.google.common.collect.Comparators;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.logging.LogUtils;
import com.mojang.serialization.Codec;
import com.mojang.serialization.Dynamic;
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap;
import java.lang.runtime.SwitchBootstraps;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.function.Function;
import java.util.stream.Collectors;
import net.minecraft.SharedConstants;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.core.registries.Registries;
import net.minecraft.nbt.ByteArrayTag;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.EndTag;
import net.minecraft.nbt.IntArrayTag;
import net.minecraft.nbt.ListTag;
import net.minecraft.nbt.LongArrayTag;
import net.minecraft.nbt.PrimitiveTag;
import net.minecraft.nbt.SnbtPrinterTagVisitor;
import net.minecraft.nbt.StringTag;
import net.minecraft.nbt.Tag;
import net.minecraft.nbt.TagParser;
import net.minecraft.nbt.TextComponentTagVisitor;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceKey;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.StateDefinition;
import net.minecraft.world.level.block.state.StateHolder;
import net.minecraft.world.level.block.state.properties.Property;
import net.minecraft.world.level.material.FluidState;
import net.minecraft.world.level.storage.ValueOutput;
import org.jspecify.annotations.Nullable;
import org.slf4j.Logger;
public final class NbtUtils {
private static final Comparator<ListTag> YXZ_LISTTAG_INT_COMPARATOR = Comparator.comparingInt(list -> list.getIntOr(1, 0)).thenComparingInt(list -> list.getIntOr(0, 0)).thenComparingInt(list -> list.getIntOr(2, 0));
private static final Comparator<ListTag> YXZ_LISTTAG_DOUBLE_COMPARATOR = Comparator.comparingDouble(list -> list.getDoubleOr(1, 0.0)).thenComparingDouble(list -> list.getDoubleOr(0, 0.0)).thenComparingDouble(list -> list.getDoubleOr(2, 0.0));
private static final Codec<ResourceKey<Block>> BLOCK_NAME_CODEC = ResourceKey.codec(Registries.BLOCK);
public static final String SNBT_DATA_TAG = "data";
private static final char PROPERTIES_START = '{';
private static final char PROPERTIES_END = '}';
private static final String ELEMENT_SEPARATOR = ",";
private static final char KEY_VALUE_SEPARATOR = ':';
private static final Splitter COMMA_SPLITTER = Splitter.on((String)",");
private static final Splitter COLON_SPLITTER = Splitter.on((char)':').limit(2);
private static final Logger LOGGER = LogUtils.getLogger();
private static final int INDENT = 2;
private static final int NOT_FOUND = -1;
private NbtUtils() {
}
@VisibleForTesting
public static boolean compareNbt(@Nullable Tag expected, @Nullable Tag actual, boolean partialListMatches) {
if (expected == actual) {
return true;
}
if (expected == null) {
return true;
}
if (actual == null) {
return false;
}
if (!expected.getClass().equals(actual.getClass())) {
return false;
}
if (expected instanceof CompoundTag) {
CompoundTag expectedCompound = (CompoundTag)expected;
CompoundTag actualCompound = (CompoundTag)actual;
if (actualCompound.size() < expectedCompound.size()) {
return false;
}
for (Map.Entry<String, Tag> entry : expectedCompound.entrySet()) {
Tag tag = entry.getValue();
if (NbtUtils.compareNbt(tag, actualCompound.get(entry.getKey()), partialListMatches)) continue;
return false;
}
return true;
}
if (expected instanceof ListTag) {
ListTag expectedList = (ListTag)expected;
if (partialListMatches) {
ListTag actualList = (ListTag)actual;
if (expectedList.isEmpty()) {
return actualList.isEmpty();
}
if (actualList.size() < expectedList.size()) {
return false;
}
for (Tag tag : expectedList) {
boolean found = false;
for (Tag value : actualList) {
if (!NbtUtils.compareNbt(tag, value, partialListMatches)) continue;
found = true;
break;
}
if (found) continue;
return false;
}
return true;
}
}
return expected.equals(actual);
}
public static BlockState readBlockState(HolderGetter<Block> blocks, CompoundTag tag) {
Optional blockHolder = tag.read("Name", BLOCK_NAME_CODEC).flatMap(blocks::get);
if (blockHolder.isEmpty()) {
return Blocks.AIR.defaultBlockState();
}
Block block = (Block)((Holder)blockHolder.get()).value();
BlockState result = block.defaultBlockState();
Optional<CompoundTag> properties = tag.getCompound("Properties");
if (properties.isPresent()) {
StateDefinition<Block, BlockState> definition = block.getStateDefinition();
for (String key : properties.get().keySet()) {
Property<?> property = definition.getProperty(key);
if (property == null) continue;
result = NbtUtils.setValueHelper(result, property, key, properties.get(), tag);
}
}
return result;
}
private static <S extends StateHolder<?, S>, T extends Comparable<T>> S setValueHelper(S result, Property<T> property, String key, CompoundTag properties, CompoundTag tag) {
Optional value = properties.getString(key).flatMap(property::getValue);
if (value.isPresent()) {
return (S)((StateHolder)result.setValue(property, (Comparable)((Comparable)value.get())));
}
LOGGER.warn("Unable to read property: {} with value: {} for blockstate: {}", new Object[]{key, properties.get(key), tag});
return result;
}
public static CompoundTag writeBlockState(BlockState state) {
CompoundTag tag = new CompoundTag();
tag.putString("Name", BuiltInRegistries.BLOCK.getKey(state.getBlock()).toString());
Map<Property<?>, Comparable<?>> values = state.getValues();
if (!values.isEmpty()) {
CompoundTag properties = new CompoundTag();
for (Map.Entry<Property<?>, Comparable<?>> entry : values.entrySet()) {
Property<?> key = entry.getKey();
properties.putString(key.getName(), NbtUtils.getName(key, entry.getValue()));
}
tag.put("Properties", properties);
}
return tag;
}
public static CompoundTag writeFluidState(FluidState state) {
CompoundTag tag = new CompoundTag();
tag.putString("Name", BuiltInRegistries.FLUID.getKey(state.getType()).toString());
Map<Property<?>, Comparable<?>> values = state.getValues();
if (!values.isEmpty()) {
CompoundTag properties = new CompoundTag();
for (Map.Entry<Property<?>, Comparable<?>> entry : values.entrySet()) {
Property<?> key = entry.getKey();
properties.putString(key.getName(), NbtUtils.getName(key, entry.getValue()));
}
tag.put("Properties", properties);
}
return tag;
}
private static <T extends Comparable<T>> String getName(Property<T> key, Comparable<?> value) {
return key.getName(value);
}
public static String prettyPrint(Tag tag) {
return NbtUtils.prettyPrint(tag, false);
}
public static String prettyPrint(Tag tag, boolean withBinaryBlobs) {
return NbtUtils.prettyPrint(new StringBuilder(), tag, 0, withBinaryBlobs).toString();
}
public static StringBuilder prettyPrint(StringBuilder builder, Tag input, int indent, boolean withBinaryBlobs) {
Tag tag = input;
Objects.requireNonNull(tag);
Tag tag2 = tag;
int n = 0;
return switch (SwitchBootstraps.typeSwitch("typeSwitch", new Object[]{PrimitiveTag.class, EndTag.class, ByteArrayTag.class, ListTag.class, IntArrayTag.class, CompoundTag.class, LongArrayTag.class}, (Object)tag2, n)) {
default -> throw new MatchException(null, null);
case 0 -> {
PrimitiveTag primitive = (PrimitiveTag)tag2;
yield builder.append(primitive);
}
case 1 -> {
EndTag ignored = (EndTag)tag2;
yield builder;
}
case 2 -> {
ByteArrayTag tag = (ByteArrayTag)tag2;
byte[] array = tag.getAsByteArray();
int length = array.length;
NbtUtils.indent(indent, builder).append("byte[").append(length).append("] {\n");
if (withBinaryBlobs) {
NbtUtils.indent(indent + 1, builder);
for (int i = 0; i < array.length; ++i) {
if (i != 0) {
builder.append(',');
}
if (i % 16 == 0 && i / 16 > 0) {
builder.append('\n');
if (i < array.length) {
NbtUtils.indent(indent + 1, builder);
}
} else if (i != 0) {
builder.append(' ');
}
builder.append(String.format(Locale.ROOT, "0x%02X", array[i] & 0xFF));
}
} else {
NbtUtils.indent(indent + 1, builder).append(" // Skipped, supply withBinaryBlobs true");
}
builder.append('\n');
NbtUtils.indent(indent, builder).append('}');
yield builder;
}
case 3 -> {
ListTag tag = (ListTag)tag2;
int size = tag.size();
NbtUtils.indent(indent, builder).append("list").append("[").append(size).append("] [");
if (size != 0) {
builder.append('\n');
}
for (int i = 0; i < size; ++i) {
if (i != 0) {
builder.append(",\n");
}
NbtUtils.indent(indent + 1, builder);
NbtUtils.prettyPrint(builder, tag.get(i), indent + 1, withBinaryBlobs);
}
if (size != 0) {
builder.append('\n');
}
NbtUtils.indent(indent, builder).append(']');
yield builder;
}
case 4 -> {
IntArrayTag tag = (IntArrayTag)tag2;
int[] array = tag.getAsIntArray();
int size = 0;
for (int i : array) {
size = Math.max(size, String.format(Locale.ROOT, "%X", i).length());
}
int length = array.length;
NbtUtils.indent(indent, builder).append("int[").append(length).append("] {\n");
if (withBinaryBlobs) {
NbtUtils.indent(indent + 1, builder);
for (int i = 0; i < array.length; ++i) {
if (i != 0) {
builder.append(',');
}
if (i % 16 == 0 && i / 16 > 0) {
builder.append('\n');
if (i < array.length) {
NbtUtils.indent(indent + 1, builder);
}
} else if (i != 0) {
builder.append(' ');
}
builder.append(String.format(Locale.ROOT, "0x%0" + size + "X", array[i]));
}
} else {
NbtUtils.indent(indent + 1, builder).append(" // Skipped, supply withBinaryBlobs true");
}
builder.append('\n');
NbtUtils.indent(indent, builder).append('}');
yield builder;
}
case 5 -> {
CompoundTag tag = (CompoundTag)tag2;
ArrayList keys = Lists.newArrayList(tag.keySet());
Collections.sort(keys);
NbtUtils.indent(indent, builder).append('{');
if (builder.length() - builder.lastIndexOf("\n") > 2 * (indent + 1)) {
builder.append('\n');
NbtUtils.indent(indent + 1, builder);
}
int paddingLength = keys.stream().mapToInt(String::length).max().orElse(0);
String padding = Strings.repeat((String)" ", (int)paddingLength);
for (int i = 0; i < keys.size(); ++i) {
if (i != 0) {
builder.append(",\n");
}
String key = (String)keys.get(i);
NbtUtils.indent(indent + 1, builder).append('\"').append(key).append('\"').append(padding, 0, padding.length() - key.length()).append(": ");
NbtUtils.prettyPrint(builder, tag.get(key), indent + 1, withBinaryBlobs);
}
if (!keys.isEmpty()) {
builder.append('\n');
}
NbtUtils.indent(indent, builder).append('}');
yield builder;
}
case 6 -> {
LongArrayTag tag = (LongArrayTag)tag2;
long[] array = tag.getAsLongArray();
long size = 0L;
for (long i : array) {
size = Math.max(size, (long)String.format(Locale.ROOT, "%X", i).length());
}
long length = array.length;
NbtUtils.indent(indent, builder).append("long[").append(length).append("] {\n");
if (withBinaryBlobs) {
NbtUtils.indent(indent + 1, builder);
for (int i = 0; i < array.length; ++i) {
if (i != 0) {
builder.append(',');
}
if (i % 16 == 0 && i / 16 > 0) {
builder.append('\n');
if (i < array.length) {
NbtUtils.indent(indent + 1, builder);
}
} else if (i != 0) {
builder.append(' ');
}
builder.append(String.format(Locale.ROOT, "0x%0" + size + "X", array[i]));
}
} else {
NbtUtils.indent(indent + 1, builder).append(" // Skipped, supply withBinaryBlobs true");
}
builder.append('\n');
NbtUtils.indent(indent, builder).append('}');
yield builder;
}
};
}
private static StringBuilder indent(int indent, StringBuilder builder) {
int index = builder.lastIndexOf("\n") + 1;
int len = builder.length() - index;
for (int i = 0; i < 2 * indent - len; ++i) {
builder.append(' ');
}
return builder;
}
public static Component toPrettyComponent(Tag tag) {
return new TextComponentTagVisitor("").visit(tag);
}
public static String structureToSnbt(CompoundTag structure) {
return new SnbtPrinterTagVisitor().visit(NbtUtils.packStructureTemplate(structure));
}
public static CompoundTag snbtToStructure(String snbt) throws CommandSyntaxException {
return NbtUtils.unpackStructureTemplate(TagParser.parseCompoundFully(snbt));
}
@VisibleForTesting
static CompoundTag packStructureTemplate(CompoundTag snbt) {
Optional<ListTag> oldEntities;
Optional<ListTag> palettes = snbt.getList("palettes");
ListTag palette = palettes.isPresent() ? palettes.get().getListOrEmpty(0) : snbt.getListOrEmpty("palette");
ListTag deflatedPalette = palette.compoundStream().map(NbtUtils::packBlockState).map(StringTag::valueOf).collect(Collectors.toCollection(ListTag::new));
snbt.put("palette", deflatedPalette);
if (palettes.isPresent()) {
ListTag newPalettes = new ListTag();
palettes.get().stream().flatMap(tag -> tag.asList().stream()).forEach(oldPalette -> {
CompoundTag newPalette = new CompoundTag();
for (int i = 0; i < oldPalette.size(); ++i) {
newPalette.putString(deflatedPalette.getString(i).orElseThrow(), NbtUtils.packBlockState(oldPalette.getCompound(i).orElseThrow()));
}
newPalettes.add(newPalette);
});
snbt.put("palettes", newPalettes);
}
if ((oldEntities = snbt.getList("entities")).isPresent()) {
ListTag newEntities = oldEntities.get().compoundStream().sorted(Comparator.comparing(tag -> tag.getList("pos"), Comparators.emptiesLast(YXZ_LISTTAG_DOUBLE_COMPARATOR))).collect(Collectors.toCollection(ListTag::new));
snbt.put("entities", newEntities);
}
ListTag blockData = snbt.getList("blocks").stream().flatMap(ListTag::compoundStream).sorted(Comparator.comparing(tag -> tag.getList("pos"), Comparators.emptiesLast(YXZ_LISTTAG_INT_COMPARATOR))).peek(block -> block.putString("state", deflatedPalette.getString(block.getIntOr("state", 0)).orElseThrow())).collect(Collectors.toCollection(ListTag::new));
snbt.put(SNBT_DATA_TAG, blockData);
snbt.remove("blocks");
return snbt;
}
@VisibleForTesting
static CompoundTag unpackStructureTemplate(CompoundTag template) {
ListTag packedPalette = template.getListOrEmpty("palette");
Map palette = (Map)packedPalette.stream().flatMap(tag -> tag.asString().stream()).collect(ImmutableMap.toImmutableMap(Function.identity(), NbtUtils::unpackBlockState));
Optional<ListTag> oldPalettes = template.getList("palettes");
if (oldPalettes.isPresent()) {
template.put("palettes", oldPalettes.get().compoundStream().map(oldPalette -> palette.keySet().stream().map(key -> oldPalette.getString((String)key).orElseThrow()).map(NbtUtils::unpackBlockState).collect(Collectors.toCollection(ListTag::new))).collect(Collectors.toCollection(ListTag::new)));
template.remove("palette");
} else {
template.put("palette", palette.values().stream().collect(Collectors.toCollection(ListTag::new)));
}
Optional<ListTag> maybeBlocks = template.getList(SNBT_DATA_TAG);
if (maybeBlocks.isPresent()) {
Object2IntOpenHashMap paletteToId = new Object2IntOpenHashMap();
paletteToId.defaultReturnValue(-1);
for (int i = 0; i < packedPalette.size(); ++i) {
paletteToId.put((Object)packedPalette.getString(i).orElseThrow(), i);
}
ListTag blocks = maybeBlocks.get();
for (int i = 0; i < blocks.size(); ++i) {
CompoundTag block = blocks.getCompound(i).orElseThrow();
String stateName = block.getString("state").orElseThrow();
int stateId = paletteToId.getInt((Object)stateName);
if (stateId == -1) {
throw new IllegalStateException("Entry " + stateName + " missing from palette");
}
block.putInt("state", stateId);
}
template.put("blocks", blocks);
template.remove(SNBT_DATA_TAG);
}
return template;
}
@VisibleForTesting
static String packBlockState(CompoundTag compound) {
StringBuilder builder = new StringBuilder(compound.getString("Name").orElseThrow());
compound.getCompound("Properties").ifPresent(properties -> {
String keyValues = properties.entrySet().stream().sorted(Map.Entry.comparingByKey()).map(entry -> (String)entry.getKey() + ":" + ((Tag)entry.getValue()).asString().orElseThrow()).collect(Collectors.joining(ELEMENT_SEPARATOR));
builder.append('{').append(keyValues).append('}');
});
return builder.toString();
}
@VisibleForTesting
static CompoundTag unpackBlockState(String compound) {
String name;
CompoundTag tag = new CompoundTag();
int openIndex = compound.indexOf(123);
if (openIndex >= 0) {
name = compound.substring(0, openIndex);
CompoundTag properties = new CompoundTag();
if (openIndex + 2 <= compound.length()) {
String values = compound.substring(openIndex + 1, compound.indexOf(125, openIndex));
COMMA_SPLITTER.split((CharSequence)values).forEach(keyValue -> {
List parts = COLON_SPLITTER.splitToList((CharSequence)keyValue);
if (parts.size() == 2) {
properties.putString((String)parts.get(0), (String)parts.get(1));
} else {
LOGGER.error("Something went wrong parsing: '{}' -- incorrect gamedata!", (Object)compound);
}
});
tag.put("Properties", properties);
}
} else {
name = compound;
}
tag.putString("Name", name);
return tag;
}
public static CompoundTag addCurrentDataVersion(CompoundTag tag) {
int version = SharedConstants.getCurrentVersion().dataVersion().version();
return NbtUtils.addDataVersion(tag, version);
}
public static CompoundTag addDataVersion(CompoundTag tag, int version) {
tag.putInt("DataVersion", version);
return tag;
}
public static Dynamic<Tag> addCurrentDataVersion(Dynamic<Tag> tag) {
int version = SharedConstants.getCurrentVersion().dataVersion().version();
return NbtUtils.addDataVersion(tag, version);
}
public static Dynamic<Tag> addDataVersion(Dynamic<Tag> tag, int version) {
return tag.set("DataVersion", tag.createInt(version));
}
public static void addCurrentDataVersion(ValueOutput output) {
int version = SharedConstants.getCurrentVersion().dataVersion().version();
NbtUtils.addDataVersion(output, version);
}
public static void addDataVersion(ValueOutput output, int version) {
output.putInt("DataVersion", version);
}
public static int getDataVersion(CompoundTag tag) {
return NbtUtils.getDataVersion(tag, -1);
}
public static int getDataVersion(CompoundTag tag, int _default) {
return tag.getIntOr("DataVersion", _default);
}
public static int getDataVersion(Dynamic<?> dynamic, int _default) {
return dynamic.get("DataVersion").asInt(_default);
}
}