77 lines
2.5 KiB
Java
77 lines
2.5 KiB
Java
package net.minecraft.world.entity;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import io.netty.buffer.ByteBuf;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.function.IntFunction;
|
|
import java.util.function.Predicate;
|
|
import net.minecraft.network.codec.ByteBufCodecs;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.util.ByIdMap;
|
|
import net.minecraft.util.StringRepresentable;
|
|
|
|
public enum EquipmentSlotGroup implements StringRepresentable, Iterable<EquipmentSlot> {
|
|
ANY(0, "any", equipmentSlot -> true),
|
|
MAINHAND(1, "mainhand", EquipmentSlot.MAINHAND),
|
|
OFFHAND(2, "offhand", EquipmentSlot.OFFHAND),
|
|
HAND(3, "hand", equipmentSlot -> equipmentSlot.getType() == EquipmentSlot.Type.HAND),
|
|
FEET(4, "feet", EquipmentSlot.FEET),
|
|
LEGS(5, "legs", EquipmentSlot.LEGS),
|
|
CHEST(6, "chest", EquipmentSlot.CHEST),
|
|
HEAD(7, "head", EquipmentSlot.HEAD),
|
|
ARMOR(8, "armor", EquipmentSlot::isArmor),
|
|
BODY(9, "body", EquipmentSlot.BODY),
|
|
SADDLE(10, "saddle", EquipmentSlot.SADDLE);
|
|
|
|
public static final IntFunction<EquipmentSlotGroup> BY_ID = ByIdMap.continuous(
|
|
equipmentSlotGroup -> equipmentSlotGroup.id, values(), ByIdMap.OutOfBoundsStrategy.ZERO
|
|
);
|
|
public static final Codec<EquipmentSlotGroup> CODEC = StringRepresentable.fromEnum(EquipmentSlotGroup::values);
|
|
public static final StreamCodec<ByteBuf, EquipmentSlotGroup> STREAM_CODEC = ByteBufCodecs.idMapper(BY_ID, equipmentSlotGroup -> equipmentSlotGroup.id);
|
|
private final int id;
|
|
private final String key;
|
|
private final Predicate<EquipmentSlot> predicate;
|
|
private final List<EquipmentSlot> slots;
|
|
|
|
private EquipmentSlotGroup(final int id, final String key, final Predicate<EquipmentSlot> predicate) {
|
|
this.id = id;
|
|
this.key = key;
|
|
this.predicate = predicate;
|
|
this.slots = EquipmentSlot.VALUES.stream().filter(predicate).toList();
|
|
}
|
|
|
|
private EquipmentSlotGroup(final int id, final String key, final EquipmentSlot slot) {
|
|
this(id, key, equipmentSlot2 -> equipmentSlot2 == slot);
|
|
}
|
|
|
|
public static EquipmentSlotGroup bySlot(EquipmentSlot slot) {
|
|
return switch (slot) {
|
|
case MAINHAND -> MAINHAND;
|
|
case OFFHAND -> OFFHAND;
|
|
case FEET -> FEET;
|
|
case LEGS -> LEGS;
|
|
case CHEST -> CHEST;
|
|
case HEAD -> HEAD;
|
|
case BODY -> BODY;
|
|
case SADDLE -> SADDLE;
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public String getSerializedName() {
|
|
return this.key;
|
|
}
|
|
|
|
public boolean test(EquipmentSlot slot) {
|
|
return this.predicate.test(slot);
|
|
}
|
|
|
|
public List<EquipmentSlot> slots() {
|
|
return this.slots;
|
|
}
|
|
|
|
public Iterator<EquipmentSlot> iterator() {
|
|
return this.slots.iterator();
|
|
}
|
|
}
|