94 lines
3.5 KiB
Java
94 lines
3.5 KiB
Java
package net.minecraft.util.datafix.fixes;
|
|
|
|
import com.google.common.collect.Lists;
|
|
import com.google.common.collect.Maps;
|
|
import com.google.common.collect.Streams;
|
|
import com.mojang.datafixers.DSL;
|
|
import com.mojang.datafixers.DataFix;
|
|
import com.mojang.datafixers.OpticFinder;
|
|
import com.mojang.datafixers.TypeRewriteRule;
|
|
import com.mojang.datafixers.Typed;
|
|
import com.mojang.datafixers.schemas.Schema;
|
|
import com.mojang.datafixers.types.Type;
|
|
import com.mojang.datafixers.types.templates.List.ListType;
|
|
import com.mojang.datafixers.util.Pair;
|
|
import com.mojang.serialization.Dynamic;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.function.Function;
|
|
|
|
public class ChunkBedBlockEntityInjecterFix extends DataFix {
|
|
public ChunkBedBlockEntityInjecterFix(Schema outputSchema, boolean changesType) {
|
|
super(outputSchema, changesType);
|
|
}
|
|
|
|
@Override
|
|
public TypeRewriteRule makeRule() {
|
|
Type<?> type = this.getOutputSchema().getType(References.CHUNK);
|
|
Type<?> type2 = type.findFieldType("Level");
|
|
if (!(type2.findFieldType("TileEntities") instanceof ListType<?> listType)) {
|
|
throw new IllegalStateException("Tile entity type is not a list type.");
|
|
} else {
|
|
return this.cap(type2, listType);
|
|
}
|
|
}
|
|
|
|
private <TE> TypeRewriteRule cap(Type<?> levelType, ListType<TE> tileEntityTypes) {
|
|
Type<TE> type = tileEntityTypes.getElement();
|
|
OpticFinder<?> opticFinder = DSL.fieldFinder("Level", levelType);
|
|
OpticFinder<List<TE>> opticFinder2 = DSL.fieldFinder("TileEntities", tileEntityTypes);
|
|
int i = 416;
|
|
return TypeRewriteRule.seq(
|
|
this.fixTypeEverywhere(
|
|
"InjectBedBlockEntityType",
|
|
this.getInputSchema().findChoiceType(References.BLOCK_ENTITY),
|
|
this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY),
|
|
dynamicOps -> pair -> pair
|
|
),
|
|
this.fixTypeEverywhereTyped(
|
|
"BedBlockEntityInjecter",
|
|
this.getOutputSchema().getType(References.CHUNK),
|
|
typed -> {
|
|
Typed<?> typed2 = typed.getTyped(opticFinder);
|
|
Dynamic<?> dynamic = typed2.get(DSL.remainderFinder());
|
|
int ix = dynamic.get("xPos").asInt(0);
|
|
int j = dynamic.get("zPos").asInt(0);
|
|
List<TE> list = Lists.<TE>newArrayList(typed2.getOrCreate(opticFinder2));
|
|
|
|
for (Dynamic<?> dynamic2 : dynamic.get("Sections").asList(Function.identity())) {
|
|
int k = dynamic2.get("Y").asInt(0);
|
|
Streams.mapWithIndex(dynamic2.get("Blocks").asIntStream(), (l, m) -> {
|
|
if (416 == (l & 0xFF) << 4) {
|
|
int n = (int)m;
|
|
int o = n & 15;
|
|
int p = n >> 8 & 15;
|
|
int q = n >> 4 & 15;
|
|
Map<Dynamic<?>, Dynamic<?>> map = Maps.<Dynamic<?>, Dynamic<?>>newHashMap();
|
|
map.put(dynamic2.createString("id"), dynamic2.createString("minecraft:bed"));
|
|
map.put(dynamic2.createString("x"), dynamic2.createInt(o + (ix << 4)));
|
|
map.put(dynamic2.createString("y"), dynamic2.createInt(p + (k << 4)));
|
|
map.put(dynamic2.createString("z"), dynamic2.createInt(q + (j << 4)));
|
|
map.put(dynamic2.createString("color"), dynamic2.createShort((short)14));
|
|
return map;
|
|
} else {
|
|
return null;
|
|
}
|
|
})
|
|
.forEachOrdered(
|
|
map -> {
|
|
if (map != null) {
|
|
list.add(
|
|
((Pair)type.read(dynamic2.createMap(map)).result().orElseThrow(() -> new IllegalStateException("Could not parse newly created bed block entity.")))
|
|
.getFirst()
|
|
);
|
|
}
|
|
}
|
|
);
|
|
}
|
|
|
|
return !list.isEmpty() ? typed.set(opticFinder, typed2.set(opticFinder2, list)) : typed;
|
|
}
|
|
)
|
|
);
|
|
}
|
|
}
|