130 lines
4.7 KiB
Java
130 lines
4.7 KiB
Java
package net.minecraft.util.datafix;
|
|
|
|
import com.mojang.datafixers.DataFixUtils;
|
|
import com.mojang.datafixers.OpticFinder;
|
|
import com.mojang.datafixers.RewriteResult;
|
|
import com.mojang.datafixers.TypeRewriteRule;
|
|
import com.mojang.datafixers.Typed;
|
|
import com.mojang.datafixers.View;
|
|
import com.mojang.datafixers.functions.PointFreeRule;
|
|
import com.mojang.datafixers.types.Type;
|
|
import com.mojang.serialization.Dynamic;
|
|
import com.mojang.serialization.DynamicOps;
|
|
import java.util.BitSet;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.function.Function;
|
|
import java.util.function.UnaryOperator;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.IntStream;
|
|
import net.minecraft.Util;
|
|
import net.minecraft.nbt.CompoundTag;
|
|
import net.minecraft.nbt.NbtOps;
|
|
import net.minecraft.nbt.Tag;
|
|
|
|
public class ExtraDataFixUtils {
|
|
public static Dynamic<?> fixBlockPos(Dynamic<?> data) {
|
|
Optional<Number> optional = data.get("X").asNumber().result();
|
|
Optional<Number> optional2 = data.get("Y").asNumber().result();
|
|
Optional<Number> optional3 = data.get("Z").asNumber().result();
|
|
return !optional.isEmpty() && !optional2.isEmpty() && !optional3.isEmpty()
|
|
? createBlockPos(data, ((Number)optional.get()).intValue(), ((Number)optional2.get()).intValue(), ((Number)optional3.get()).intValue())
|
|
: data;
|
|
}
|
|
|
|
public static Dynamic<?> fixInlineBlockPos(Dynamic<?> data, String xField, String yField, String zField, String newPosField) {
|
|
Optional<Number> optional = data.get(xField).asNumber().result();
|
|
Optional<Number> optional2 = data.get(yField).asNumber().result();
|
|
Optional<Number> optional3 = data.get(zField).asNumber().result();
|
|
return !optional.isEmpty() && !optional2.isEmpty() && !optional3.isEmpty()
|
|
? data.remove(xField)
|
|
.remove(yField)
|
|
.remove(zField)
|
|
.set(newPosField, createBlockPos(data, ((Number)optional.get()).intValue(), ((Number)optional2.get()).intValue(), ((Number)optional3.get()).intValue()))
|
|
: data;
|
|
}
|
|
|
|
public static Dynamic<?> createBlockPos(Dynamic<?> data, int x, int y, int z) {
|
|
return data.createIntList(IntStream.of(new int[]{x, y, z}));
|
|
}
|
|
|
|
public static <T, R> Typed<R> cast(Type<R> type, Typed<T> data) {
|
|
return new Typed<>(type, data.getOps(), (R)data.getValue());
|
|
}
|
|
|
|
public static <T> Typed<T> cast(Type<T> type, Object data, DynamicOps<?> ios) {
|
|
return new Typed<>(type, ios, (T)data);
|
|
}
|
|
|
|
public static Type<?> patchSubType(Type<?> type, Type<?> oldSubType, Type<?> newSubType) {
|
|
return type.all(typePatcher(oldSubType, newSubType), true, false).view().newType();
|
|
}
|
|
|
|
private static <A, B> TypeRewriteRule typePatcher(Type<A> oldType, Type<B> newType) {
|
|
RewriteResult<A, B> rewriteResult = RewriteResult.create(View.create("Patcher", oldType, newType, dynamicOps -> object -> {
|
|
throw new UnsupportedOperationException();
|
|
}), new BitSet());
|
|
return TypeRewriteRule.everywhere(TypeRewriteRule.ifSame(oldType, rewriteResult), PointFreeRule.nop(), true, true);
|
|
}
|
|
|
|
@SafeVarargs
|
|
public static <T> Function<Typed<?>, Typed<?>> chainAllFilters(Function<Typed<?>, Typed<?>>... filters) {
|
|
return typed -> {
|
|
for (Function<Typed<?>, Typed<?>> function : filters) {
|
|
typed = (Typed)function.apply(typed);
|
|
}
|
|
|
|
return typed;
|
|
};
|
|
}
|
|
|
|
public static Dynamic<?> blockState(String blockId, Map<String, String> properties) {
|
|
Dynamic<Tag> dynamic = new Dynamic<>(NbtOps.INSTANCE, new CompoundTag());
|
|
Dynamic<Tag> dynamic2 = dynamic.set("Name", dynamic.createString(blockId));
|
|
if (!properties.isEmpty()) {
|
|
dynamic2 = dynamic2.set(
|
|
"Properties",
|
|
dynamic.createMap(
|
|
(Map<? extends Dynamic<?>, ? extends Dynamic<?>>)properties.entrySet()
|
|
.stream()
|
|
.collect(Collectors.toMap(entry -> dynamic.createString((String)entry.getKey()), entry -> dynamic.createString((String)entry.getValue())))
|
|
)
|
|
);
|
|
}
|
|
|
|
return dynamic2;
|
|
}
|
|
|
|
public static Dynamic<?> blockState(String blockId) {
|
|
return blockState(blockId, Map.of());
|
|
}
|
|
|
|
public static Dynamic<?> fixStringField(Dynamic<?> data, String fieldName, UnaryOperator<String> fixer) {
|
|
return data.update(fieldName, dynamic2 -> DataFixUtils.orElse(dynamic2.asString().map(fixer).map(data::createString).result(), dynamic2));
|
|
}
|
|
|
|
public static String dyeColorIdToName(int colorId) {
|
|
return switch (colorId) {
|
|
case 1 -> "orange";
|
|
case 2 -> "magenta";
|
|
case 3 -> "light_blue";
|
|
case 4 -> "yellow";
|
|
case 5 -> "lime";
|
|
case 6 -> "pink";
|
|
case 7 -> "gray";
|
|
case 8 -> "light_gray";
|
|
case 9 -> "cyan";
|
|
case 10 -> "purple";
|
|
case 11 -> "blue";
|
|
case 12 -> "brown";
|
|
case 13 -> "green";
|
|
case 14 -> "red";
|
|
case 15 -> "black";
|
|
default -> "white";
|
|
};
|
|
}
|
|
|
|
public static <T> Typed<?> readAndSet(Typed<?> typed, OpticFinder<T> optic, Dynamic<?> data) {
|
|
return typed.set(optic, Util.readTypedOrThrow(optic.type(), data, true));
|
|
}
|
|
}
|