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 optional = data.get("X").asNumber().result(); Optional optional2 = data.get("Y").asNumber().result(); Optional 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 Typed cast(Type type, Typed 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 TypeRewriteRule typePatcher(Type oldType, Type newType) { RewriteResult 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 Function, Typed> chainAllFilters(Function, Typed>... filters) { return typed -> { for (Function, Typed> function : filters) { typed = (Typed)function.apply(typed); } return typed; }; } public static Dynamic blockState(String blockId, Map properties) { Dynamic dynamic = new Dynamic<>(NbtOps.INSTANCE, new CompoundTag()); Dynamic dynamic2 = dynamic.set("Name", dynamic.createString(blockId)); if (!properties.isEmpty()) { dynamic2 = dynamic2.set( "Properties", dynamic.createMap( (Map, ? 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 fixer) { return data.update(fieldName, dynamic2 -> DataFixUtils.orElse(dynamic2.asString().map(fixer).map(data::createString).result(), dynamic2)); } }