package com.kodari.livingworld.herd; import com.kodari.livingworld.LivingWorldPlugin; import org.bukkit.Bukkit; import org.bukkit.Location; import org.bukkit.World; import org.bukkit.entity.Entity; import org.bukkit.entity.LivingEntity; import org.bukkit.entity.Mob; import org.bukkit.entity.EntityType; import org.bukkit.scheduler.BukkitTask; import org.bukkit.util.Vector; import java.util.*; public class HerdManager { private final LivingWorldPlugin plugin; private final Map herds = new HashMap<>(); private final Map entityToHerd = new HashMap<>(); private BukkitTask tickTask; public HerdManager(LivingWorldPlugin plugin) { this.plugin = plugin; } public void startTicking() { int interval = plugin.getConfig().getInt("herd.tick-interval", 40); tickTask = Bukkit.getScheduler().runTaskTimer(plugin, this::tick, interval, interval); } public void stopTicking() { if (tickTask != null) tickTask.cancel(); } private void tick() { List toRemove = new ArrayList<>(); for (Herd herd : herds.values()) { herd.getMembers().removeIf(uid -> { Entity e = Bukkit.getEntity(uid); if (e == null || !e.isValid() || e.isDead()) { entityToHerd.remove(uid); return true; } return false; }); if (herd.isEmpty()) { toRemove.add(herd.getId()); continue; } updateCenter(herd); tickHerd(herd); } toRemove.forEach(herds::remove); } private void updateCenter(Herd herd) { List members = getLivingMembers(herd); if (members.isEmpty()) return; double sx = 0, sy = 0, sz = 0; World world = null; for (LivingEntity e : members) { sx += e.getLocation().getX(); sy += e.getLocation().getY(); sz += e.getLocation().getZ(); world = e.getWorld(); } if (world != null) { herd.setCenter(new Location(world, sx / members.size(), sy / members.size(), sz / members.size())); } } private void tickHerd(Herd herd) { List members = getLivingMembers(herd); if (members.isEmpty()) return; switch (herd.getState()) { case FLEEING -> tickFlee(herd, members); case REGROUPING -> tickRegroup(herd, members); case MIGRATING -> tickMigrating(herd, members); default -> tickIdle(herd, members); } enforceCohesion(herd, members); } private void tickIdle(Herd herd, List members) { herd.setStateTimer(herd.getStateTimer() + 1); if (herd.getStateTimer() >= 12 || herd.getTarget() == null) { herd.setStateTimer(0); if (herd.getCenter() != null) { Location c = herd.getCenter(); Random r = new Random(); double angle = r.nextDouble() * 2 * Math.PI; double dist = 6 + r.nextDouble() * 12; Location target = c.clone().add(Math.cos(angle) * dist, 0, Math.sin(angle) * dist); target.setY(c.getWorld().getHighestBlockYAt(target)); herd.setTarget(target); } } if (herd.getTarget() != null && !members.isEmpty()) { LivingEntity leader = members.get(0); if (leader instanceof Mob mob) { mob.getPathfinder().moveTo(herd.getTarget(), 0.75); } } } private void tickFlee(Herd herd, List members) { herd.setFleeTimer(herd.getFleeTimer() - 1); if (herd.getFleeTimer() <= 0) { herd.setState(HerdState.REGROUPING); herd.setStateTimer(plugin.getConfig().getInt("herd.regroup-duration-ticks", 100)); return; } if (herd.getThreatLocation() == null) return; for (LivingEntity entity : members) { if (entity instanceof Mob mob) { Location flee = fleeLocation(entity.getLocation(), herd.getThreatLocation(), 22); mob.getPathfinder().moveTo(flee, 1.6); } } } private void tickRegroup(Herd herd, List members) { herd.setStateTimer(herd.getStateTimer() - 1); if (herd.getStateTimer() <= 0) { herd.setState(HerdState.GRAZING); herd.setThreatLocation(null); return; } if (herd.getCenter() == null) return; for (LivingEntity entity : members) { if (entity instanceof Mob mob) { mob.getPathfinder().moveTo(herd.getCenter(), 1.1); } } } private void tickMigrating(Herd herd, List members) { if (herd.getTarget() == null) { herd.setState(HerdState.GRAZING); return; } for (LivingEntity entity : members) { if (entity instanceof Mob mob) { mob.getPathfinder().moveTo(herd.getTarget(), 1.2); } } if (herd.getCenter() != null && herd.getCenter().distanceSquared(herd.getTarget()) < 225) { herd.setState(HerdState.GRAZING); herd.setTarget(null); } } private void enforceCohesion(Herd herd, List members) { if (herd.getCenter() == null) return; double maxSpread = plugin.getConfig().getDouble("herd.max-spread", 22.0); for (LivingEntity member : members) { if (member.getLocation().distanceSquared(herd.getCenter()) > maxSpread * maxSpread) { if (member instanceof Mob mob) { mob.getPathfinder().moveTo(herd.getCenter(), 1.35); } } } } private Location fleeLocation(Location from, Location threat, double distance) { Vector dir = from.toVector().subtract(threat.toVector()); if (dir.lengthSquared() < 0.001) dir = new Vector(1, 0, 0); dir.normalize().multiply(distance); Location flee = from.clone().add(dir); flee.setY(from.getWorld().getHighestBlockYAt(flee)); return flee; } public void triggerFlee(UUID entityId, Location threatLocation) { UUID herdId = entityToHerd.get(entityId); if (herdId == null) return; Herd herd = herds.get(herdId); if (herd == null) return; herd.setState(HerdState.FLEEING); herd.setThreatLocation(threatLocation); herd.setFleeTimer(plugin.getConfig().getInt("herd.flee-duration-ticks", 200)); } public void registerEntity(UUID entityId, Herd herd) { entityToHerd.put(entityId, herd.getId()); herd.addMember(entityId); } public void removeEntity(UUID entityId) { UUID herdId = entityToHerd.remove(entityId); if (herdId == null) return; Herd herd = herds.get(herdId); if (herd != null) herd.removeMember(entityId); } public boolean isHerdMember(UUID entityId) { return entityToHerd.containsKey(entityId); } public Herd findNearbyHerd(Location location, EntityType type, double range) { for (Herd herd : herds.values()) { if (herd.getType() != type || herd.getCenter() == null) continue; if (!herd.getCenter().getWorld().equals(location.getWorld())) continue; if (herd.getCenter().distanceSquared(location) <= range * range) return herd; } return null; } public Herd createHerd(Location location, EntityType type, int size) { Herd herd = new Herd(type); herd.setCenter(location.clone()); herds.put(herd.getId(), herd); Random r = new Random(); World world = location.getWorld(); for (int i = 0; i < size; i++) { double angle = (2 * Math.PI * i) / size + r.nextDouble() * 0.5; double radius = 2 + r.nextDouble() * 4; Location spawnLoc = location.clone().add(Math.cos(angle) * radius, 0, Math.sin(angle) * radius); spawnLoc.setY(world.getHighestBlockYAt(spawnLoc) + 0.5); Entity entity = world.spawnEntity(spawnLoc, type); registerEntity(entity.getUniqueId(), herd); } return herd; } public void addToNearbyHerdOrCreate(LivingEntity entity) { EntityType type = entity.getType(); int maxSize = plugin.getConfig().getInt("herd.max-size", 8); Herd nearby = findNearbyHerd(entity.getLocation(), type, 32); if (nearby != null && nearby.size() < maxSize) { registerEntity(entity.getUniqueId(), nearby); return; } if (nearby != null) return; Herd newHerd = new Herd(type); newHerd.setCenter(entity.getLocation().clone()); herds.put(newHerd.getId(), newHerd); registerEntity(entity.getUniqueId(), newHerd); int extraMembers = plugin.getConfig().getInt("herd.min-size", 3) - 1; Random r = new Random(); World world = entity.getWorld(); for (int i = 0; i < extraMembers; i++) { double angle = r.nextDouble() * 2 * Math.PI; double radius = 3 + r.nextDouble() * 6; Location spawnLoc = entity.getLocation().clone().add(Math.cos(angle) * radius, 0, Math.sin(angle) * radius); spawnLoc.setY(world.getHighestBlockYAt(spawnLoc) + 0.5); Entity spawned = world.spawnEntity(spawnLoc, type); registerEntity(spawned.getUniqueId(), newHerd); } } public int getHerdCount() { return herds.size(); } public int getTotalMembers() { return herds.values().stream().mapToInt(Herd::size).sum(); } private List getLivingMembers(Herd herd) { List result = new ArrayList<>(); for (UUID uid : new ArrayList<>(herd.getMembers())) { Entity e = Bukkit.getEntity(uid); if (e instanceof LivingEntity le && e.isValid() && !e.isDead()) { result.add(le); } } return result; } }