105 lines
4 KiB
Java
105 lines
4 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.stream.Stream;
|
|
import net.minecraft.advancements.CriteriaTriggers;
|
|
import net.minecraft.advancements.Criterion;
|
|
import net.minecraft.advancements.critereon.InventoryChangeTrigger.TriggerInstance.Slots;
|
|
import net.minecraft.advancements.critereon.SimpleCriterionTrigger.SimpleInstance;
|
|
import net.minecraft.core.HolderSet;
|
|
import net.minecraft.server.level.ServerPlayer;
|
|
import net.minecraft.world.entity.player.Inventory;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.ItemLike;
|
|
|
|
public class InventoryChangeTrigger extends SimpleCriterionTrigger<InventoryChangeTrigger.TriggerInstance> {
|
|
@Override
|
|
public Codec<InventoryChangeTrigger.TriggerInstance> codec() {
|
|
return InventoryChangeTrigger.TriggerInstance.CODEC;
|
|
}
|
|
|
|
public void trigger(ServerPlayer player, Inventory inventory, ItemStack stack) {
|
|
int i = 0;
|
|
int j = 0;
|
|
int k = 0;
|
|
|
|
for (int l = 0; l < inventory.getContainerSize(); l++) {
|
|
ItemStack itemStack = inventory.getItem(l);
|
|
if (itemStack.isEmpty()) {
|
|
j++;
|
|
} else {
|
|
k++;
|
|
if (itemStack.getCount() >= itemStack.getMaxStackSize()) {
|
|
i++;
|
|
}
|
|
}
|
|
}
|
|
|
|
this.trigger(player, inventory, stack, i, j, k);
|
|
}
|
|
|
|
private void trigger(ServerPlayer player, Inventory inventory, ItemStack stack, int full, int empty, int occupied) {
|
|
this.trigger(player, triggerInstance -> triggerInstance.matches(inventory, stack, full, empty, occupied));
|
|
}
|
|
|
|
public record TriggerInstance(Optional<ContextAwarePredicate> player, Slots slots, List<ItemPredicate> items) implements SimpleInstance {
|
|
public static final Codec<InventoryChangeTrigger.TriggerInstance> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("player").forGetter(InventoryChangeTrigger.TriggerInstance::player),
|
|
Slots.CODEC.optionalFieldOf("slots", Slots.ANY).forGetter(InventoryChangeTrigger.TriggerInstance::slots),
|
|
ItemPredicate.CODEC.listOf().optionalFieldOf("items", List.of()).forGetter(InventoryChangeTrigger.TriggerInstance::items)
|
|
)
|
|
.apply(instance, InventoryChangeTrigger.TriggerInstance::new)
|
|
);
|
|
|
|
public static Criterion<InventoryChangeTrigger.TriggerInstance> hasItems(ItemPredicate.Builder... items) {
|
|
return hasItems((ItemPredicate[])Stream.of(items).map(ItemPredicate.Builder::build).toArray(ItemPredicate[]::new));
|
|
}
|
|
|
|
public static Criterion<InventoryChangeTrigger.TriggerInstance> hasItems(ItemPredicate... items) {
|
|
return CriteriaTriggers.INVENTORY_CHANGED.createCriterion(new InventoryChangeTrigger.TriggerInstance(Optional.empty(), Slots.ANY, List.of(items)));
|
|
}
|
|
|
|
public static Criterion<InventoryChangeTrigger.TriggerInstance> hasItems(ItemLike... items) {
|
|
ItemPredicate[] itemPredicates = new ItemPredicate[items.length];
|
|
|
|
for (int i = 0; i < items.length; i++) {
|
|
itemPredicates[i] = new ItemPredicate(
|
|
Optional.of(HolderSet.direct(items[i].asItem().builtInRegistryHolder())), MinMaxBounds.Ints.ANY, DataComponentMatchers.ANY
|
|
);
|
|
}
|
|
|
|
return hasItems(itemPredicates);
|
|
}
|
|
|
|
public boolean matches(Inventory inventory, ItemStack stack, int full, int empty, int occupied) {
|
|
if (!this.slots.matches(full, empty, occupied)) {
|
|
return false;
|
|
} else if (this.items.isEmpty()) {
|
|
return true;
|
|
} else if (this.items.size() != 1) {
|
|
List<ItemPredicate> list = new ObjectArrayList<>(this.items);
|
|
int i = inventory.getContainerSize();
|
|
|
|
for (int j = 0; j < i; j++) {
|
|
if (list.isEmpty()) {
|
|
return true;
|
|
}
|
|
|
|
ItemStack itemStack = inventory.getItem(j);
|
|
if (!itemStack.isEmpty()) {
|
|
list.removeIf(itemPredicate -> itemPredicate.test(itemStack));
|
|
}
|
|
}
|
|
|
|
return list.isEmpty();
|
|
} else {
|
|
return !stack.isEmpty() && ((ItemPredicate)this.items.get(0)).test(stack);
|
|
}
|
|
}
|
|
}
|
|
}
|