87 lines
3.3 KiB
Java
87 lines
3.3 KiB
Java
package net.minecraft.world.entity;
|
|
|
|
import com.google.common.base.Predicates;
|
|
import java.util.function.Predicate;
|
|
import net.minecraft.world.Container;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.scores.Team;
|
|
|
|
public final class EntitySelector {
|
|
/**
|
|
* Selects only entities which are alive
|
|
*/
|
|
public static final Predicate<Entity> ENTITY_STILL_ALIVE = Entity::isAlive;
|
|
/**
|
|
* Selects only entities which are LivingEntities and alive
|
|
*/
|
|
public static final Predicate<Entity> LIVING_ENTITY_STILL_ALIVE = entity -> entity.isAlive() && entity instanceof LivingEntity;
|
|
/**
|
|
* Selects only entities which are neither ridden by anything nor ride on anything
|
|
*/
|
|
public static final Predicate<Entity> ENTITY_NOT_BEING_RIDDEN = entity -> entity.isAlive() && !entity.isVehicle() && !entity.isPassenger();
|
|
/**
|
|
* Selects only entities which are container entities
|
|
*/
|
|
public static final Predicate<Entity> CONTAINER_ENTITY_SELECTOR = entity -> entity instanceof Container && entity.isAlive();
|
|
/**
|
|
* Selects entities which are neither creative-mode players nor spectator-players
|
|
*/
|
|
public static final Predicate<Entity> NO_CREATIVE_OR_SPECTATOR = entity -> !(entity instanceof Player player && (entity.isSpectator() || player.isCreative()));
|
|
/**
|
|
* Selects entities which are either not players or players that are not spectating
|
|
*/
|
|
public static final Predicate<Entity> NO_SPECTATORS = entity -> !entity.isSpectator();
|
|
/**
|
|
* Selects entities which are collidable with and aren't spectators
|
|
*/
|
|
public static final Predicate<Entity> CAN_BE_COLLIDED_WITH = NO_SPECTATORS.and(entity -> entity.canBeCollidedWith(null));
|
|
public static final Predicate<Entity> CAN_BE_PICKED = NO_SPECTATORS.and(Entity::isPickable);
|
|
|
|
private EntitySelector() {
|
|
}
|
|
|
|
public static Predicate<Entity> withinDistance(double x, double y, double z, double range) {
|
|
double d = range * range;
|
|
return entity -> entity != null && entity.distanceToSqr(x, y, z) <= d;
|
|
}
|
|
|
|
public static Predicate<Entity> pushableBy(Entity entity) {
|
|
Team team = entity.getTeam();
|
|
Team.CollisionRule collisionRule = team == null ? Team.CollisionRule.ALWAYS : team.getCollisionRule();
|
|
return (Predicate<Entity>)(collisionRule == Team.CollisionRule.NEVER
|
|
? Predicates.alwaysFalse()
|
|
: NO_SPECTATORS.and(
|
|
entity2 -> {
|
|
if (!entity2.isPushable()) {
|
|
return false;
|
|
} else if (!entity.level().isClientSide || entity2 instanceof Player player && player.isLocalPlayer()) {
|
|
Team team2 = entity2.getTeam();
|
|
Team.CollisionRule collisionRule2 = team2 == null ? Team.CollisionRule.ALWAYS : team2.getCollisionRule();
|
|
if (collisionRule2 == Team.CollisionRule.NEVER) {
|
|
return false;
|
|
} else {
|
|
boolean bl = team != null && team.isAlliedTo(team2);
|
|
return (collisionRule == Team.CollisionRule.PUSH_OWN_TEAM || collisionRule2 == Team.CollisionRule.PUSH_OWN_TEAM) && bl
|
|
? false
|
|
: collisionRule != Team.CollisionRule.PUSH_OTHER_TEAMS && collisionRule2 != Team.CollisionRule.PUSH_OTHER_TEAMS || bl;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
));
|
|
}
|
|
|
|
public static Predicate<Entity> notRiding(Entity entity) {
|
|
return entity2 -> {
|
|
while (entity2.isPassenger()) {
|
|
entity2 = entity2.getVehicle();
|
|
if (entity2 == entity) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
};
|
|
}
|
|
}
|