package net.minecraft.advancements.critereon; import com.mojang.serialization.Codec; import com.mojang.serialization.codecs.RecordCodecBuilder; import java.util.Optional; import net.minecraft.advancements.CriteriaTriggers; import net.minecraft.advancements.Criterion; import net.minecraft.server.level.ServerLevel; import net.minecraft.server.level.ServerPlayer; import net.minecraft.world.entity.Entity; import net.minecraft.world.level.storage.loot.LootContext; import net.minecraft.world.phys.Vec3; import org.jetbrains.annotations.Nullable; public class FallAfterExplosionTrigger extends SimpleCriterionTrigger { @Override public Codec codec() { return FallAfterExplosionTrigger.TriggerInstance.CODEC; } public void trigger(ServerPlayer player, Vec3 pos, @Nullable Entity entity) { Vec3 vec3 = player.position(); LootContext lootContext = entity != null ? EntityPredicate.createContext(player, entity) : null; this.trigger(player, triggerInstance -> triggerInstance.matches(player.serverLevel(), pos, vec3, lootContext)); } public record TriggerInstance( Optional player, Optional startPosition, Optional distance, Optional cause ) implements SimpleCriterionTrigger.SimpleInstance { public static final Codec CODEC = RecordCodecBuilder.create( instance -> instance.group( EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("player").forGetter(FallAfterExplosionTrigger.TriggerInstance::player), LocationPredicate.CODEC.optionalFieldOf("start_position").forGetter(FallAfterExplosionTrigger.TriggerInstance::startPosition), DistancePredicate.CODEC.optionalFieldOf("distance").forGetter(FallAfterExplosionTrigger.TriggerInstance::distance), EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("cause").forGetter(FallAfterExplosionTrigger.TriggerInstance::cause) ) .apply(instance, FallAfterExplosionTrigger.TriggerInstance::new) ); public static Criterion fallAfterExplosion(DistancePredicate distance, EntityPredicate.Builder cause) { return CriteriaTriggers.FALL_AFTER_EXPLOSION .createCriterion( new FallAfterExplosionTrigger.TriggerInstance(Optional.empty(), Optional.empty(), Optional.of(distance), Optional.of(EntityPredicate.wrap(cause))) ); } @Override public void validate(CriterionValidator validator) { SimpleCriterionTrigger.SimpleInstance.super.validate(validator); validator.validateEntity(this.cause(), ".cause"); } public boolean matches(ServerLevel level, Vec3 startPosition, Vec3 endPosition, @Nullable LootContext context) { if (this.startPosition.isPresent() && !((LocationPredicate)this.startPosition.get()).matches(level, startPosition.x, startPosition.y, startPosition.z)) { return false; } else { return this.distance.isPresent() && !((DistancePredicate)this.distance.get()).matches(startPosition.x, startPosition.y, startPosition.z, endPosition.x, endPosition.y, endPosition.z) ? false : !this.cause.isPresent() || context != null && ((ContextAwarePredicate)this.cause.get()).matches(context); } } } }