package com.plasma.optimiser.task; import com.plasma.optimiser.PlasmaOptimiser; import com.plasma.optimiser.config.ConfigManager; import org.bukkit.Chunk; import org.bukkit.World; import org.bukkit.entity.*; import org.bukkit.scheduler.BukkitRunnable; import java.util.*; /** * Scans every loaded chunk across all worlds and removes excess non-critical * entities when the per-chunk count exceeds the configured limit. * * Culling priority (removed first → last): * 1. Unnamed, untamed hostile mobs farthest from any player * 2. Unnamed, untamed passive mobs farthest from any player * 3. Item entities and experience orbs * * Protected always: players, tamed entities (if configured), named entities * (if configured), and any type listed in protected-types. */ public class EntityOptimiser extends BukkitRunnable { private final PlasmaOptimiser plugin; private final ConfigManager cfg; public EntityOptimiser(PlasmaOptimiser plugin) { this.plugin = plugin; this.cfg = plugin.getConfigManager(); } @Override public void run() { int totalCulled = 0; int maxPerChunk = cfg.getMaxEntitiesPerChunk(); Set protectedTypes = new HashSet<>(); for (String s : cfg.getProtectedTypes()) protectedTypes.add(s.toUpperCase()); for (World world : plugin.getServer().getWorlds()) { List players = world.getPlayers(); for (Chunk chunk : world.getLoadedChunks()) { Entity[] entities = chunk.getEntities(); if (entities.length <= maxPerChunk) continue; // Separate cullable from protected List cullable = new ArrayList<>(); for (Entity entity : entities) { if (isProtected(entity, protectedTypes)) continue; cullable.add(entity); } int toRemove = entities.length - maxPerChunk; if (toRemove <= 0 || cullable.isEmpty()) continue; // Sort: XP orbs and drops first, then by distance from nearest player (farthest first) cullable.sort((a, b) -> { int priorityA = cullPriority(a); int priorityB = cullPriority(b); if (priorityA != priorityB) return Integer.compare(priorityA, priorityB); // Same priority — remove the one farthest from any player double distA = nearestPlayerDistance(a, players); double distB = nearestPlayerDistance(b, players); return Double.compare(distB, distA); // descending (farthest first) }); int removed = 0; for (Entity entity : cullable) { if (removed >= toRemove) break; entity.remove(); removed++; } totalCulled += removed; } } if (totalCulled > 0) { plugin.getTpsMonitor().recordEntityCull(totalCulled); } } private boolean isProtected(Entity entity, Set protectedTypes) { if (entity instanceof Player) return true; if (protectedTypes.contains(entity.getType().name())) return true; if (cfg.isProtectNamed() && entity.getCustomName() != null) return true; if (cfg.isProtectTamed() && entity instanceof Tameable && ((Tameable) entity).isTamed()) return true; return false; } /** Lower number = removed first. */ private int cullPriority(Entity entity) { if (entity instanceof ExperienceOrb) return 0; if (entity instanceof Item) return 1; if (entity instanceof Monster) return 2; if (entity instanceof Animals) return 3; return 4; } private double nearestPlayerDistance(Entity entity, List players) { if (players.isEmpty()) return Double.MAX_VALUE; double min = Double.MAX_VALUE; for (Player p : players) { if (!p.getWorld().equals(entity.getWorld())) continue; double d = p.getLocation().distanceSquared(entity.getLocation()); if (d < min) min = d; } return min; } public int getCullableCount() { int count = 0; Set protectedTypes = new HashSet<>(); for (String s : cfg.getProtectedTypes()) protectedTypes.add(s.toUpperCase()); for (World world : plugin.getServer().getWorlds()) { for (Chunk chunk : world.getLoadedChunks()) { for (Entity entity : chunk.getEntities()) { if (!isProtected(entity, protectedTypes)) count++; } } } return count; } }