550 lines
22 KiB
Java
550 lines
22 KiB
Java
|
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.Object2IntMap;
|
||
|
|
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap;
|
||
|
|
import java.util.Collections;
|
||
|
|
import java.util.Comparator;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Locale;
|
||
|
|
import java.util.Map;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.Map.Entry;
|
||
|
|
import java.util.function.Function;
|
||
|
|
import java.util.stream.Collectors;
|
||
|
|
import javax.annotation.Nullable;
|
||
|
|
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.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 org.slf4j.Logger;
|
||
|
|
|
||
|
|
public final class NbtUtils {
|
||
|
|
private static final Comparator<ListTag> YXZ_LISTTAG_INT_COMPARATOR = Comparator.<ListTag>comparingInt(p_389895_ -> p_389895_.getIntOr(1, 0))
|
||
|
|
.thenComparingInt(p_389897_ -> p_389897_.getIntOr(0, 0))
|
||
|
|
.thenComparingInt(p_389901_ -> p_389901_.getIntOr(2, 0));
|
||
|
|
private static final Comparator<ListTag> YXZ_LISTTAG_DOUBLE_COMPARATOR = Comparator.<ListTag>comparingDouble(p_389902_ -> p_389902_.getDoubleOr(1, 0.0))
|
||
|
|
.thenComparingDouble(p_389889_ -> p_389889_.getDoubleOr(0, 0.0))
|
||
|
|
.thenComparingDouble(p_389886_ -> p_389886_.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(",");
|
||
|
|
private static final Splitter COLON_SPLITTER = Splitter.on(':').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 p_129236_, @Nullable Tag p_129237_, boolean p_129238_) {
|
||
|
|
if (p_129236_ == p_129237_) {
|
||
|
|
return true;
|
||
|
|
} else if (p_129236_ == null) {
|
||
|
|
return true;
|
||
|
|
} else if (p_129237_ == null) {
|
||
|
|
return false;
|
||
|
|
} else if (!p_129236_.getClass().equals(p_129237_.getClass())) {
|
||
|
|
return false;
|
||
|
|
} else if (p_129236_ instanceof CompoundTag compoundtag) {
|
||
|
|
CompoundTag compoundtag1 = (CompoundTag)p_129237_;
|
||
|
|
if (compoundtag1.size() < compoundtag.size()) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
for (Entry<String, Tag> entry : compoundtag.entrySet()) {
|
||
|
|
Tag tag2 = entry.getValue();
|
||
|
|
if (!compareNbt(tag2, compoundtag1.get(entry.getKey()), p_129238_)) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
} else if (p_129236_ instanceof ListTag listtag && p_129238_) {
|
||
|
|
ListTag listtag1 = (ListTag)p_129237_;
|
||
|
|
if (listtag.isEmpty()) {
|
||
|
|
return listtag1.isEmpty();
|
||
|
|
} else if (listtag1.size() < listtag.size()) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
for (Tag tag : listtag) {
|
||
|
|
boolean flag = false;
|
||
|
|
|
||
|
|
for (Tag tag1 : listtag1) {
|
||
|
|
if (compareNbt(tag, tag1, p_129238_)) {
|
||
|
|
flag = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!flag) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
return p_129236_.equals(p_129237_);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static BlockState readBlockState(HolderGetter<Block> p_256363_, CompoundTag p_250775_) {
|
||
|
|
Optional<? extends Holder<Block>> optional = p_250775_.read("Name", BLOCK_NAME_CODEC).flatMap(p_256363_::get);
|
||
|
|
if (optional.isEmpty()) {
|
||
|
|
return Blocks.AIR.defaultBlockState();
|
||
|
|
} else {
|
||
|
|
Block block = optional.get().value();
|
||
|
|
BlockState blockstate = block.defaultBlockState();
|
||
|
|
Optional<CompoundTag> optional1 = p_250775_.getCompound("Properties");
|
||
|
|
if (optional1.isPresent()) {
|
||
|
|
StateDefinition<Block, BlockState> statedefinition = block.getStateDefinition();
|
||
|
|
|
||
|
|
for (String s : optional1.get().keySet()) {
|
||
|
|
Property<?> property = statedefinition.getProperty(s);
|
||
|
|
if (property != null) {
|
||
|
|
blockstate = setValueHelper(blockstate, property, s, optional1.get(), p_250775_);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return blockstate;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <S extends StateHolder<?, S>, T extends Comparable<T>> S setValueHelper(
|
||
|
|
S p_129205_, Property<T> p_129206_, String p_129207_, CompoundTag p_129208_, CompoundTag p_129209_
|
||
|
|
) {
|
||
|
|
Optional<T> optional = p_129208_.getString(p_129207_).flatMap(p_129206_::getValue);
|
||
|
|
if (optional.isPresent()) {
|
||
|
|
return p_129205_.setValue(p_129206_, optional.get());
|
||
|
|
} else {
|
||
|
|
LOGGER.warn("Unable to read property: {} with value: {} for blockstate: {}", p_129207_, p_129208_.get(p_129207_), p_129209_);
|
||
|
|
return p_129205_;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static CompoundTag writeBlockState(BlockState p_129203_) {
|
||
|
|
CompoundTag compoundtag = new CompoundTag();
|
||
|
|
compoundtag.putString("Name", BuiltInRegistries.BLOCK.getKey(p_129203_.getBlock()).toString());
|
||
|
|
Map<Property<?>, Comparable<?>> map = p_129203_.getValues();
|
||
|
|
if (!map.isEmpty()) {
|
||
|
|
CompoundTag compoundtag1 = new CompoundTag();
|
||
|
|
|
||
|
|
for (Entry<Property<?>, Comparable<?>> entry : map.entrySet()) {
|
||
|
|
Property<?> property = entry.getKey();
|
||
|
|
compoundtag1.putString(property.getName(), getName(property, entry.getValue()));
|
||
|
|
}
|
||
|
|
|
||
|
|
compoundtag.put("Properties", compoundtag1);
|
||
|
|
}
|
||
|
|
|
||
|
|
return compoundtag;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static CompoundTag writeFluidState(FluidState p_178023_) {
|
||
|
|
CompoundTag compoundtag = new CompoundTag();
|
||
|
|
compoundtag.putString("Name", BuiltInRegistries.FLUID.getKey(p_178023_.getType()).toString());
|
||
|
|
Map<Property<?>, Comparable<?>> map = p_178023_.getValues();
|
||
|
|
if (!map.isEmpty()) {
|
||
|
|
CompoundTag compoundtag1 = new CompoundTag();
|
||
|
|
|
||
|
|
for (Entry<Property<?>, Comparable<?>> entry : map.entrySet()) {
|
||
|
|
Property<?> property = entry.getKey();
|
||
|
|
compoundtag1.putString(property.getName(), getName(property, entry.getValue()));
|
||
|
|
}
|
||
|
|
|
||
|
|
compoundtag.put("Properties", compoundtag1);
|
||
|
|
}
|
||
|
|
|
||
|
|
return compoundtag;
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <T extends Comparable<T>> String getName(Property<T> p_129211_, Comparable<?> p_129212_) {
|
||
|
|
return p_129211_.getName((T)p_129212_);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static String prettyPrint(Tag p_178058_) {
|
||
|
|
return prettyPrint(p_178058_, false);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static String prettyPrint(Tag p_178051_, boolean p_178052_) {
|
||
|
|
return prettyPrint(new StringBuilder(), p_178051_, 0, p_178052_).toString();
|
||
|
|
}
|
||
|
|
|
||
|
|
public static StringBuilder prettyPrint(StringBuilder p_178027_, Tag p_178028_, int p_178029_, boolean p_178030_) {
|
||
|
|
return switch (p_178028_) {
|
||
|
|
case PrimitiveTag primitivetag -> p_178027_.append(primitivetag);
|
||
|
|
case EndTag endtag -> p_178027_;
|
||
|
|
case ByteArrayTag bytearraytag -> {
|
||
|
|
byte[] abyte = bytearraytag.getAsByteArray();
|
||
|
|
int i1 = abyte.length;
|
||
|
|
indent(p_178029_, p_178027_).append("byte[").append(i1).append("] {\n");
|
||
|
|
if (p_178030_) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
|
||
|
|
for (int k1 = 0; k1 < abyte.length; k1++) {
|
||
|
|
if (k1 != 0) {
|
||
|
|
p_178027_.append(',');
|
||
|
|
}
|
||
|
|
|
||
|
|
if (k1 % 16 == 0 && k1 / 16 > 0) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
if (k1 < abyte.length) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
}
|
||
|
|
} else if (k1 != 0) {
|
||
|
|
p_178027_.append(' ');
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append(String.format(Locale.ROOT, "0x%02X", abyte[k1] & 255));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
indent(p_178029_ + 1, p_178027_).append(" // Skipped, supply withBinaryBlobs true");
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append('\n');
|
||
|
|
indent(p_178029_, p_178027_).append('}');
|
||
|
|
yield p_178027_;
|
||
|
|
}
|
||
|
|
case ListTag listtag -> {
|
||
|
|
int l = listtag.size();
|
||
|
|
indent(p_178029_, p_178027_).append("list").append("[").append(l).append("] [");
|
||
|
|
if (l != 0) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
}
|
||
|
|
|
||
|
|
for (int j1 = 0; j1 < l; j1++) {
|
||
|
|
if (j1 != 0) {
|
||
|
|
p_178027_.append(",\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
prettyPrint(p_178027_, listtag.get(j1), p_178029_ + 1, p_178030_);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (l != 0) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
}
|
||
|
|
|
||
|
|
indent(p_178029_, p_178027_).append(']');
|
||
|
|
yield p_178027_;
|
||
|
|
}
|
||
|
|
case IntArrayTag intarraytag -> {
|
||
|
|
int[] aint = intarraytag.getAsIntArray();
|
||
|
|
int l1 = 0;
|
||
|
|
|
||
|
|
for (int i3 : aint) {
|
||
|
|
l1 = Math.max(l1, String.format(Locale.ROOT, "%X", i3).length());
|
||
|
|
}
|
||
|
|
|
||
|
|
int j2 = aint.length;
|
||
|
|
indent(p_178029_, p_178027_).append("int[").append(j2).append("] {\n");
|
||
|
|
if (p_178030_) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
|
||
|
|
for (int k2 = 0; k2 < aint.length; k2++) {
|
||
|
|
if (k2 != 0) {
|
||
|
|
p_178027_.append(',');
|
||
|
|
}
|
||
|
|
|
||
|
|
if (k2 % 16 == 0 && k2 / 16 > 0) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
if (k2 < aint.length) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
}
|
||
|
|
} else if (k2 != 0) {
|
||
|
|
p_178027_.append(' ');
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append(String.format(Locale.ROOT, "0x%0" + l1 + "X", aint[k2]));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
indent(p_178029_ + 1, p_178027_).append(" // Skipped, supply withBinaryBlobs true");
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append('\n');
|
||
|
|
indent(p_178029_, p_178027_).append('}');
|
||
|
|
yield p_178027_;
|
||
|
|
}
|
||
|
|
case CompoundTag compoundtag -> {
|
||
|
|
List<String> list = Lists.newArrayList(compoundtag.keySet());
|
||
|
|
Collections.sort(list);
|
||
|
|
indent(p_178029_, p_178027_).append('{');
|
||
|
|
if (p_178027_.length() - p_178027_.lastIndexOf("\n") > 2 * (p_178029_ + 1)) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
}
|
||
|
|
|
||
|
|
int i2 = list.stream().mapToInt(String::length).max().orElse(0);
|
||
|
|
String s = Strings.repeat(" ", i2);
|
||
|
|
|
||
|
|
for (int j = 0; j < list.size(); j++) {
|
||
|
|
if (j != 0) {
|
||
|
|
p_178027_.append(",\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
String s1 = list.get(j);
|
||
|
|
indent(p_178029_ + 1, p_178027_).append('"').append(s1).append('"').append(s, 0, s.length() - s1.length()).append(": ");
|
||
|
|
prettyPrint(p_178027_, compoundtag.get(s1), p_178029_ + 1, p_178030_);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!list.isEmpty()) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
}
|
||
|
|
|
||
|
|
indent(p_178029_, p_178027_).append('}');
|
||
|
|
yield p_178027_;
|
||
|
|
}
|
||
|
|
case LongArrayTag longarraytag -> {
|
||
|
|
long[] along = longarraytag.getAsLongArray();
|
||
|
|
long i = 0L;
|
||
|
|
|
||
|
|
for (long k : along) {
|
||
|
|
i = Math.max(i, (long)String.format(Locale.ROOT, "%X", k).length());
|
||
|
|
}
|
||
|
|
|
||
|
|
long l2 = along.length;
|
||
|
|
indent(p_178029_, p_178027_).append("long[").append(l2).append("] {\n");
|
||
|
|
if (p_178030_) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
|
||
|
|
for (int j3 = 0; j3 < along.length; j3++) {
|
||
|
|
if (j3 != 0) {
|
||
|
|
p_178027_.append(',');
|
||
|
|
}
|
||
|
|
|
||
|
|
if (j3 % 16 == 0 && j3 / 16 > 0) {
|
||
|
|
p_178027_.append('\n');
|
||
|
|
if (j3 < along.length) {
|
||
|
|
indent(p_178029_ + 1, p_178027_);
|
||
|
|
}
|
||
|
|
} else if (j3 != 0) {
|
||
|
|
p_178027_.append(' ');
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append(String.format(Locale.ROOT, "0x%0" + i + "X", along[j3]));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
indent(p_178029_ + 1, p_178027_).append(" // Skipped, supply withBinaryBlobs true");
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178027_.append('\n');
|
||
|
|
indent(p_178029_, p_178027_).append('}');
|
||
|
|
yield p_178027_;
|
||
|
|
}
|
||
|
|
default -> throw new MatchException(null, null);
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
private static StringBuilder indent(int p_178020_, StringBuilder p_178021_) {
|
||
|
|
int i = p_178021_.lastIndexOf("\n") + 1;
|
||
|
|
int j = p_178021_.length() - i;
|
||
|
|
|
||
|
|
for (int k = 0; k < 2 * p_178020_ - j; k++) {
|
||
|
|
p_178021_.append(' ');
|
||
|
|
}
|
||
|
|
|
||
|
|
return p_178021_;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Component toPrettyComponent(Tag p_178062_) {
|
||
|
|
return new TextComponentTagVisitor("").visit(p_178062_);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static String structureToSnbt(CompoundTag p_178064_) {
|
||
|
|
return new SnbtPrinterTagVisitor().visit(packStructureTemplate(p_178064_));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static CompoundTag snbtToStructure(String p_178025_) throws CommandSyntaxException {
|
||
|
|
return unpackStructureTemplate(TagParser.parseCompoundFully(p_178025_));
|
||
|
|
}
|
||
|
|
|
||
|
|
@VisibleForTesting
|
||
|
|
static CompoundTag packStructureTemplate(CompoundTag p_178068_) {
|
||
|
|
Optional<ListTag> optional = p_178068_.getList("palettes");
|
||
|
|
ListTag listtag;
|
||
|
|
if (optional.isPresent()) {
|
||
|
|
listtag = optional.get().getListOrEmpty(0);
|
||
|
|
} else {
|
||
|
|
listtag = p_178068_.getListOrEmpty("palette");
|
||
|
|
}
|
||
|
|
|
||
|
|
ListTag listtag1 = listtag.compoundStream().map(NbtUtils::packBlockState).map(StringTag::valueOf).collect(Collectors.toCollection(ListTag::new));
|
||
|
|
p_178068_.put("palette", listtag1);
|
||
|
|
if (optional.isPresent()) {
|
||
|
|
ListTag listtag2 = new ListTag();
|
||
|
|
optional.get().stream().flatMap(p_389905_ -> p_389905_.asList().stream()).forEach(p_389894_ -> {
|
||
|
|
CompoundTag compoundtag = new CompoundTag();
|
||
|
|
|
||
|
|
for (int i = 0; i < p_389894_.size(); i++) {
|
||
|
|
compoundtag.putString(listtag1.getString(i).orElseThrow(), packBlockState(p_389894_.getCompound(i).orElseThrow()));
|
||
|
|
}
|
||
|
|
|
||
|
|
listtag2.add(compoundtag);
|
||
|
|
});
|
||
|
|
p_178068_.put("palettes", listtag2);
|
||
|
|
}
|
||
|
|
|
||
|
|
Optional<ListTag> optional1 = p_178068_.getList("entities");
|
||
|
|
if (optional1.isPresent()) {
|
||
|
|
ListTag listtag3 = optional1.get()
|
||
|
|
.compoundStream()
|
||
|
|
.sorted(Comparator.comparing(p_389903_ -> p_389903_.getList("pos"), Comparators.emptiesLast(YXZ_LISTTAG_DOUBLE_COMPARATOR)))
|
||
|
|
.collect(Collectors.toCollection(ListTag::new));
|
||
|
|
p_178068_.put("entities", listtag3);
|
||
|
|
}
|
||
|
|
|
||
|
|
ListTag listtag4 = p_178068_.getList("blocks")
|
||
|
|
.stream()
|
||
|
|
.flatMap(ListTag::compoundStream)
|
||
|
|
.sorted(Comparator.comparing(p_389898_ -> p_389898_.getList("pos"), Comparators.emptiesLast(YXZ_LISTTAG_INT_COMPARATOR)))
|
||
|
|
.peek(p_389885_ -> p_389885_.putString("state", listtag1.getString(p_389885_.getIntOr("state", 0)).orElseThrow()))
|
||
|
|
.collect(Collectors.toCollection(ListTag::new));
|
||
|
|
p_178068_.put("data", listtag4);
|
||
|
|
p_178068_.remove("blocks");
|
||
|
|
return p_178068_;
|
||
|
|
}
|
||
|
|
|
||
|
|
@VisibleForTesting
|
||
|
|
static CompoundTag unpackStructureTemplate(CompoundTag p_178072_) {
|
||
|
|
ListTag listtag = p_178072_.getListOrEmpty("palette");
|
||
|
|
Map<String, Tag> map = listtag.stream()
|
||
|
|
.flatMap(p_389904_ -> p_389904_.asString().stream())
|
||
|
|
.collect(ImmutableMap.toImmutableMap(Function.identity(), NbtUtils::unpackBlockState));
|
||
|
|
Optional<ListTag> optional = p_178072_.getList("palettes");
|
||
|
|
if (optional.isPresent()) {
|
||
|
|
p_178072_.put(
|
||
|
|
"palettes",
|
||
|
|
optional.get()
|
||
|
|
.compoundStream()
|
||
|
|
.map(
|
||
|
|
p_389891_ -> map.keySet()
|
||
|
|
.stream()
|
||
|
|
.map(p_389900_ -> p_389891_.getString(p_389900_).orElseThrow())
|
||
|
|
.map(NbtUtils::unpackBlockState)
|
||
|
|
.collect(Collectors.toCollection(ListTag::new))
|
||
|
|
)
|
||
|
|
.collect(Collectors.toCollection(ListTag::new))
|
||
|
|
);
|
||
|
|
p_178072_.remove("palette");
|
||
|
|
} else {
|
||
|
|
p_178072_.put("palette", map.values().stream().collect(Collectors.toCollection(ListTag::new)));
|
||
|
|
}
|
||
|
|
|
||
|
|
Optional<ListTag> optional1 = p_178072_.getList("data");
|
||
|
|
if (optional1.isPresent()) {
|
||
|
|
Object2IntMap<String> object2intmap = new Object2IntOpenHashMap<>();
|
||
|
|
object2intmap.defaultReturnValue(-1);
|
||
|
|
|
||
|
|
for (int i = 0; i < listtag.size(); i++) {
|
||
|
|
object2intmap.put(listtag.getString(i).orElseThrow(), i);
|
||
|
|
}
|
||
|
|
|
||
|
|
ListTag listtag1 = optional1.get();
|
||
|
|
|
||
|
|
for (int j = 0; j < listtag1.size(); j++) {
|
||
|
|
CompoundTag compoundtag = listtag1.getCompound(j).orElseThrow();
|
||
|
|
String s = compoundtag.getString("state").orElseThrow();
|
||
|
|
int k = object2intmap.getInt(s);
|
||
|
|
if (k == -1) {
|
||
|
|
throw new IllegalStateException("Entry " + s + " missing from palette");
|
||
|
|
}
|
||
|
|
|
||
|
|
compoundtag.putInt("state", k);
|
||
|
|
}
|
||
|
|
|
||
|
|
p_178072_.put("blocks", listtag1);
|
||
|
|
p_178072_.remove("data");
|
||
|
|
}
|
||
|
|
|
||
|
|
return p_178072_;
|
||
|
|
}
|
||
|
|
|
||
|
|
@VisibleForTesting
|
||
|
|
static String packBlockState(CompoundTag p_178076_) {
|
||
|
|
StringBuilder stringbuilder = new StringBuilder(p_178076_.getString("Name").orElseThrow());
|
||
|
|
p_178076_.getCompound("Properties")
|
||
|
|
.ifPresent(
|
||
|
|
p_389888_ -> {
|
||
|
|
String s = p_389888_.entrySet()
|
||
|
|
.stream()
|
||
|
|
.sorted(Entry.comparingByKey())
|
||
|
|
.map(p_389896_ -> p_389896_.getKey() + ":" + p_389896_.getValue().asString().orElseThrow())
|
||
|
|
.collect(Collectors.joining(","));
|
||
|
|
stringbuilder.append('{').append(s).append('}');
|
||
|
|
}
|
||
|
|
);
|
||
|
|
return stringbuilder.toString();
|
||
|
|
}
|
||
|
|
|
||
|
|
@VisibleForTesting
|
||
|
|
static CompoundTag unpackBlockState(String p_178054_) {
|
||
|
|
CompoundTag compoundtag = new CompoundTag();
|
||
|
|
int i = p_178054_.indexOf(123);
|
||
|
|
String s;
|
||
|
|
if (i >= 0) {
|
||
|
|
s = p_178054_.substring(0, i);
|
||
|
|
CompoundTag compoundtag1 = new CompoundTag();
|
||
|
|
if (i + 2 <= p_178054_.length()) {
|
||
|
|
String s1 = p_178054_.substring(i + 1, p_178054_.indexOf(125, i));
|
||
|
|
COMMA_SPLITTER.split(s1).forEach(p_178040_ -> {
|
||
|
|
List<String> list = COLON_SPLITTER.splitToList(p_178040_);
|
||
|
|
if (list.size() == 2) {
|
||
|
|
compoundtag1.putString(list.get(0), list.get(1));
|
||
|
|
} else {
|
||
|
|
LOGGER.error("Something went wrong parsing: '{}' -- incorrect gamedata!", p_178054_);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
compoundtag.put("Properties", compoundtag1);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
s = p_178054_;
|
||
|
|
}
|
||
|
|
|
||
|
|
compoundtag.putString("Name", s);
|
||
|
|
return compoundtag;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static CompoundTag addCurrentDataVersion(CompoundTag p_265050_) {
|
||
|
|
int i = SharedConstants.getCurrentVersion().getDataVersion().getVersion();
|
||
|
|
return addDataVersion(p_265050_, i);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static CompoundTag addDataVersion(CompoundTag p_265534_, int p_265686_) {
|
||
|
|
p_265534_.putInt("DataVersion", p_265686_);
|
||
|
|
return p_265534_;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static int getDataVersion(CompoundTag p_265397_, int p_265399_) {
|
||
|
|
return p_265397_.getIntOr("DataVersion", p_265399_);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static int getDataVersion(Dynamic<?> p_391363_, int p_396363_) {
|
||
|
|
return p_391363_.get("DataVersion").asInt(p_396363_);
|
||
|
|
}
|
||
|
|
}
|