114 lines
6.2 KiB
Java
114 lines
6.2 KiB
Java
package com.atsuishio.superbwarfare.tools;
|
|
|
|
import com.atsuishio.superbwarfare.config.server.VehicleConfig;
|
|
import com.atsuishio.superbwarfare.entity.ClaymoreEntity;
|
|
import com.atsuishio.superbwarfare.entity.projectile.C4Entity;
|
|
import com.atsuishio.superbwarfare.entity.vehicle.base.MobileVehicleEntity;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.world.entity.AreaEffectCloud;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.ExperienceOrb;
|
|
import net.minecraft.world.entity.decoration.ArmorStand;
|
|
import net.minecraft.world.entity.decoration.HangingEntity;
|
|
import net.minecraft.world.entity.item.ItemEntity;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.entity.projectile.Projectile;
|
|
import net.minecraft.world.level.ClipContext;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.phys.HitResult;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import net.minecraftforge.registries.ForgeRegistries;
|
|
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
import java.util.stream.StreamSupport;
|
|
|
|
public class SeekTool {
|
|
|
|
public static List<Entity> getVehicleWithinRange(Player player, Level level, double range) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> e.position().distanceTo(player.getEyePosition()) <= range
|
|
&& e instanceof MobileVehicleEntity)
|
|
.toList();
|
|
}
|
|
|
|
public static Entity seekEntity(Entity entity, Level level, double seekRange, double seekAngle) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> {
|
|
if (e.distanceTo(entity) <= seekRange && calculateAngle(e, entity) < seekAngle
|
|
&& e != entity
|
|
&& baseFilter(e)
|
|
&& e.getVehicle() == null
|
|
) {
|
|
return level.clip(new ClipContext(entity.getEyePosition(), e.getEyePosition(),
|
|
ClipContext.Block.COLLIDER, ClipContext.Fluid.NONE, entity)).getType() != HitResult.Type.BLOCK;
|
|
}
|
|
return false;
|
|
}).min(Comparator.comparingDouble(e -> calculateAngle(e, entity))).orElse(null);
|
|
}
|
|
|
|
public static Entity seekLivingEntity(Entity entity, Level level, double seekRange, double seekAngle) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> {
|
|
if (e.distanceTo(entity) <= seekRange && calculateAngle(e, entity) < seekAngle
|
|
&& e != entity
|
|
&& baseFilter(e)
|
|
&& e.getVehicle() == null
|
|
&& (!e.isAlliedTo(entity) || e.getTeam() == null || e.getTeam().getName().equals("TDM"))) {
|
|
return level.clip(new ClipContext(entity.getEyePosition(), e.getEyePosition(),
|
|
ClipContext.Block.COLLIDER, ClipContext.Fluid.NONE, entity)).getType() != HitResult.Type.BLOCK;
|
|
}
|
|
return false;
|
|
}).min(Comparator.comparingDouble(e -> calculateAngle(e, entity))).orElse(null);
|
|
}
|
|
|
|
public static List<Entity> seekLivingEntities(Entity entity, Level level, double seekRange, double seekAngle) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> {
|
|
if (e.distanceTo(entity) <= seekRange && calculateAngle(e, entity) < seekAngle
|
|
&& e != entity
|
|
&& baseFilter(e)
|
|
&& e.getVehicle() == null
|
|
&& (!e.isAlliedTo(entity) || e.getTeam() == null || e.getTeam().getName().equals("TDM"))) {
|
|
return level.clip(new ClipContext(entity.getEyePosition(), e.getEyePosition(),
|
|
ClipContext.Block.COLLIDER, ClipContext.Fluid.NONE, entity)).getType() != HitResult.Type.BLOCK;
|
|
}
|
|
return false;
|
|
}).toList();
|
|
}
|
|
|
|
public static List<Entity> seekLivingEntitiesThroughWall(Entity entity, Level level, double seekRange, double seekAngle) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> e.distanceTo(entity) <= seekRange && calculateAngle(e, entity) < seekAngle
|
|
&& e != entity
|
|
&& baseFilter(e)
|
|
&& e.getVehicle() == null
|
|
&& (!e.isAlliedTo(entity) || e.getTeam() == null || e.getTeam().getName().equals("TDM"))).toList();
|
|
}
|
|
|
|
public static List<Entity> getEntitiesWithinRange(BlockPos pos, Level level, double range) {
|
|
return StreamSupport.stream(EntityFindUtil.getEntities(level).getAll().spliterator(), false)
|
|
.filter(e -> e.distanceToSqr(pos.getX(), pos.getY(), pos.getZ()) <= range * range
|
|
&& baseFilter(e))
|
|
.toList();
|
|
}
|
|
|
|
private static double calculateAngle(Entity entityA, Entity entityB) {
|
|
Vec3 start = new Vec3(entityA.getX() - entityB.getX(), entityA.getY() - entityB.getY(), entityA.getZ() - entityB.getZ());
|
|
Vec3 end = entityB.getLookAngle();
|
|
return VectorTool.calculateAngle(start, end);
|
|
}
|
|
|
|
public static boolean baseFilter(Entity entity) {
|
|
return entity.isAlive()
|
|
&& !(entity instanceof ItemEntity || entity instanceof ExperienceOrb || entity instanceof HangingEntity || entity instanceof Projectile || entity instanceof ArmorStand || entity instanceof ClaymoreEntity || entity instanceof C4Entity || entity instanceof AreaEffectCloud)
|
|
&& !(entity instanceof Player player && player.isSpectator())
|
|
|| includedByConfig(entity);
|
|
}
|
|
|
|
public static boolean includedByConfig(Entity entity) {
|
|
var type = ForgeRegistries.ENTITY_TYPES.getKey(entity.getType());
|
|
if (type == null) return false;
|
|
return VehicleConfig.COLLISION_ENTITY_WHITELIST.get().contains(type.toString());
|
|
}
|
|
}
|