84 lines
3.1 KiB
Java
84 lines
3.1 KiB
Java
package net.minecraft.util.datafix;
|
|
|
|
import com.mojang.datafixers.DataFixUtils;
|
|
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 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.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()
|
|
? data.createIntList(
|
|
IntStream.of(new int[]{((Number)optional.get()).intValue(), ((Number)optional2.get()).intValue(), ((Number)optional3.get()).intValue()})
|
|
)
|
|
: data;
|
|
}
|
|
|
|
public static <T, R> Typed<R> cast(Type<R> type, Typed<T> data) {
|
|
return new Typed<>(type, data.getOps(), (R)data.getValue());
|
|
}
|
|
|
|
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));
|
|
}
|
|
}
|