Spaces:
Build error
Build error
File size: 2,297 Bytes
d46f4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
package net.minecraft.world.entity;
import com.google.common.collect.Maps;
import com.mojang.datafixers.util.Either;
import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import net.minecraft.core.registries.Registries;
import net.minecraft.resources.ResourceKey;
import net.minecraft.world.level.storage.loot.LootTable;
public record EquipmentTable(ResourceKey<LootTable> lootTable, Map<EquipmentSlot, Float> slotDropChances) {
public static final Codec<Map<EquipmentSlot, Float>> DROP_CHANCES_CODEC = Codec.either(Codec.FLOAT, Codec.unboundedMap(EquipmentSlot.CODEC, Codec.FLOAT))
.xmap(p_330726_ -> p_330726_.map(EquipmentTable::createForAllSlots, Function.identity()), p_329105_ -> {
boolean flag = p_329105_.values().stream().distinct().count() == 1L;
boolean flag1 = p_329105_.keySet().containsAll(EquipmentSlot.VALUES);
return flag && flag1 ? Either.left(p_329105_.values().stream().findFirst().orElse(0.0F)) : Either.right((Map<EquipmentSlot, Float>)p_329105_);
});
public static final Codec<EquipmentTable> CODEC = RecordCodecBuilder.create(
p_328347_ -> p_328347_.group(
ResourceKey.codec(Registries.LOOT_TABLE).fieldOf("loot_table").forGetter(EquipmentTable::lootTable),
DROP_CHANCES_CODEC.optionalFieldOf("slot_drop_chances", Map.of()).forGetter(EquipmentTable::slotDropChances)
)
.apply(p_328347_, EquipmentTable::new)
);
public EquipmentTable(ResourceKey<LootTable> p_361217_, float p_365263_) {
this(p_361217_, createForAllSlots(p_365263_));
}
private static Map<EquipmentSlot, Float> createForAllSlots(float p_335505_) {
return createForAllSlots(List.of(EquipmentSlot.values()), p_335505_);
}
private static Map<EquipmentSlot, Float> createForAllSlots(List<EquipmentSlot> p_331241_, float p_334411_) {
Map<EquipmentSlot, Float> map = Maps.newHashMap();
for (EquipmentSlot equipmentslot : p_331241_) {
map.put(equipmentslot, p_334411_);
}
return map;
}
} |