40 lines
1.6 KiB
Java
40 lines
1.6 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Optional;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.LightningBolt;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public record LightningBoltPredicate(MinMaxBounds.Ints blocksSetOnFire, Optional<EntityPredicate> entityStruck) implements EntitySubPredicate {
|
|
public static final MapCodec<LightningBoltPredicate> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(
|
|
MinMaxBounds.Ints.CODEC.optionalFieldOf("blocks_set_on_fire", MinMaxBounds.Ints.ANY).forGetter(LightningBoltPredicate::blocksSetOnFire),
|
|
EntityPredicate.CODEC.optionalFieldOf("entity_struck").forGetter(LightningBoltPredicate::entityStruck)
|
|
)
|
|
.apply(instance, LightningBoltPredicate::new)
|
|
);
|
|
|
|
public static LightningBoltPredicate blockSetOnFire(MinMaxBounds.Ints blocksSetOnFire) {
|
|
return new LightningBoltPredicate(blocksSetOnFire, Optional.empty());
|
|
}
|
|
|
|
@Override
|
|
public MapCodec<LightningBoltPredicate> codec() {
|
|
return EntitySubPredicates.LIGHTNING;
|
|
}
|
|
|
|
@Override
|
|
public boolean matches(Entity entity, ServerLevel level, @Nullable Vec3 position) {
|
|
return !(entity instanceof LightningBolt lightningBolt)
|
|
? false
|
|
: this.blocksSetOnFire.matches(lightningBolt.getBlocksSetOnFire())
|
|
&& (
|
|
this.entityStruck.isEmpty()
|
|
|| lightningBolt.getHitEntities().anyMatch(entityx -> ((EntityPredicate)this.entityStruck.get()).matches(level, position, entityx))
|
|
);
|
|
}
|
|
}
|