80 lines
4.1 KiB
Java
80 lines
4.1 KiB
Java
|
package net.minecraft.core.component.predicates;
|
||
|
|
|
||
|
|
import com.mojang.serialization.Codec;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.function.Predicate;
|
||
|
|
import net.minecraft.advancements.critereon.CollectionPredicate;
|
||
|
|
import net.minecraft.advancements.critereon.MinMaxBounds;
|
||
|
|
import net.minecraft.advancements.critereon.SingleComponentItemPredicate;
|
||
|
|
import net.minecraft.core.HolderSet;
|
||
|
|
import net.minecraft.core.RegistryCodecs;
|
||
|
|
import net.minecraft.core.component.DataComponentType;
|
||
|
|
import net.minecraft.core.component.DataComponents;
|
||
|
|
import net.minecraft.core.registries.Registries;
|
||
|
|
import net.minecraft.resources.ResourceLocation;
|
||
|
|
import net.minecraft.world.entity.EquipmentSlotGroup;
|
||
|
|
import net.minecraft.world.entity.ai.attributes.Attribute;
|
||
|
|
import net.minecraft.world.entity.ai.attributes.AttributeModifier;
|
||
|
|
import net.minecraft.world.item.component.ItemAttributeModifiers;
|
||
|
|
|
||
|
|
public record AttributeModifiersPredicate(Optional<CollectionPredicate<ItemAttributeModifiers.Entry, AttributeModifiersPredicate.EntryPredicate>> modifiers)
|
||
|
|
implements SingleComponentItemPredicate<ItemAttributeModifiers> {
|
||
|
|
public static final Codec<AttributeModifiersPredicate> CODEC = RecordCodecBuilder.create(
|
||
|
|
p_397127_ -> p_397127_.group(
|
||
|
|
CollectionPredicate.<ItemAttributeModifiers.Entry, AttributeModifiersPredicate.EntryPredicate>codec(
|
||
|
|
AttributeModifiersPredicate.EntryPredicate.CODEC
|
||
|
|
)
|
||
|
|
.optionalFieldOf("modifiers")
|
||
|
|
.forGetter(AttributeModifiersPredicate::modifiers)
|
||
|
|
)
|
||
|
|
.apply(p_397127_, AttributeModifiersPredicate::new)
|
||
|
|
);
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public DataComponentType<ItemAttributeModifiers> componentType() {
|
||
|
|
return DataComponents.ATTRIBUTE_MODIFIERS;
|
||
|
|
}
|
||
|
|
|
||
|
|
public boolean matches(ItemAttributeModifiers p_396692_) {
|
||
|
|
return !this.modifiers.isPresent() || this.modifiers.get().test(p_396692_.modifiers());
|
||
|
|
}
|
||
|
|
|
||
|
|
public record EntryPredicate(
|
||
|
|
Optional<HolderSet<Attribute>> attribute,
|
||
|
|
Optional<ResourceLocation> id,
|
||
|
|
MinMaxBounds.Doubles amount,
|
||
|
|
Optional<AttributeModifier.Operation> operation,
|
||
|
|
Optional<EquipmentSlotGroup> slot
|
||
|
|
) implements Predicate<ItemAttributeModifiers.Entry> {
|
||
|
|
public static final Codec<AttributeModifiersPredicate.EntryPredicate> CODEC = RecordCodecBuilder.create(
|
||
|
|
p_391653_ -> p_391653_.group(
|
||
|
|
RegistryCodecs.homogeneousList(Registries.ATTRIBUTE)
|
||
|
|
.optionalFieldOf("attribute")
|
||
|
|
.forGetter(AttributeModifiersPredicate.EntryPredicate::attribute),
|
||
|
|
ResourceLocation.CODEC.optionalFieldOf("id").forGetter(AttributeModifiersPredicate.EntryPredicate::id),
|
||
|
|
MinMaxBounds.Doubles.CODEC
|
||
|
|
.optionalFieldOf("amount", MinMaxBounds.Doubles.ANY)
|
||
|
|
.forGetter(AttributeModifiersPredicate.EntryPredicate::amount),
|
||
|
|
AttributeModifier.Operation.CODEC.optionalFieldOf("operation").forGetter(AttributeModifiersPredicate.EntryPredicate::operation),
|
||
|
|
EquipmentSlotGroup.CODEC.optionalFieldOf("slot").forGetter(AttributeModifiersPredicate.EntryPredicate::slot)
|
||
|
|
)
|
||
|
|
.apply(p_391653_, AttributeModifiersPredicate.EntryPredicate::new)
|
||
|
|
);
|
||
|
|
|
||
|
|
public boolean test(ItemAttributeModifiers.Entry p_396480_) {
|
||
|
|
if (this.attribute.isPresent() && !this.attribute.get().contains(p_396480_.attribute())) {
|
||
|
|
return false;
|
||
|
|
} else if (this.id.isPresent() && !this.id.get().equals(p_396480_.modifier().id())) {
|
||
|
|
return false;
|
||
|
|
} else if (!this.amount.matches(p_396480_.modifier().amount())) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
return this.operation.isPresent() && this.operation.get() != p_396480_.modifier().operation()
|
||
|
|
? false
|
||
|
|
: !this.slot.isPresent() || this.slot.get() == p_396480_.slot();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|