55 lines
2.1 KiB
Java
55 lines
2.1 KiB
Java
package net.minecraft.util.datafix.fixes;
|
|
|
|
import com.mojang.datafixers.DSL;
|
|
import com.mojang.datafixers.DataFix;
|
|
import com.mojang.datafixers.TypeRewriteRule;
|
|
import com.mojang.datafixers.schemas.Schema;
|
|
import com.mojang.serialization.Dynamic;
|
|
import com.mojang.serialization.OptionalDynamic;
|
|
import java.util.List;
|
|
|
|
public class DropChancesFormatFix extends DataFix {
|
|
private static final List<String> ARMOR_SLOT_NAMES = List.of("feet", "legs", "chest", "head");
|
|
private static final List<String> HAND_SLOT_NAMES = List.of("mainhand", "offhand");
|
|
private static final float DEFAULT_CHANCE = 0.085F;
|
|
|
|
public DropChancesFormatFix(Schema outputSchema) {
|
|
super(outputSchema, false);
|
|
}
|
|
|
|
@Override
|
|
protected TypeRewriteRule makeRule() {
|
|
return this.fixTypeEverywhereTyped(
|
|
"DropChancesFormatFix", this.getInputSchema().getType(References.ENTITY), typed -> typed.update(DSL.remainderFinder(), dynamic -> {
|
|
List<Float> list = parseDropChances(dynamic.get("ArmorDropChances"));
|
|
List<Float> list2 = parseDropChances(dynamic.get("HandDropChances"));
|
|
float f = (Float)dynamic.get("body_armor_drop_chance").asNumber().result().map(Number::floatValue).orElse(0.085F);
|
|
dynamic = dynamic.remove("ArmorDropChances").remove("HandDropChances").remove("body_armor_drop_chance");
|
|
Dynamic<?> dynamic2 = dynamic.emptyMap();
|
|
dynamic2 = addSlotChances(dynamic2, list, ARMOR_SLOT_NAMES);
|
|
dynamic2 = addSlotChances(dynamic2, list2, HAND_SLOT_NAMES);
|
|
if (f != 0.085F) {
|
|
dynamic2 = dynamic2.set("body", dynamic.createFloat(f));
|
|
}
|
|
|
|
return !dynamic2.equals(dynamic.emptyMap()) ? dynamic.set("drop_chances", dynamic2) : dynamic;
|
|
})
|
|
);
|
|
}
|
|
|
|
private static Dynamic<?> addSlotChances(Dynamic<?> tag, List<Float> chances, List<String> names) {
|
|
for (int i = 0; i < names.size() && i < chances.size(); i++) {
|
|
String string = (String)names.get(i);
|
|
float f = (Float)chances.get(i);
|
|
if (f != 0.085F) {
|
|
tag = tag.set(string, tag.createFloat(f));
|
|
}
|
|
}
|
|
|
|
return tag;
|
|
}
|
|
|
|
private static List<Float> parseDropChances(OptionalDynamic<?> data) {
|
|
return data.asStream().map(dynamic -> dynamic.asFloat(0.085F)).toList();
|
|
}
|
|
}
|