Adding 30 degree randomness into LEAD behavior targeting.

src/main/java/com/owlmaddie/goals/LeadPlayerGoal.java

@@ -111,7 +111,7 @@ public class LeadPlayerGoal extends PlayerBaseGoal {
                 currentSpeed = MathHelper.stepTowards((float)currentSpeed, (float)this.speed, (float) (0.005 * (this.speed / Math.max(currentSpeed, 0.1))));
 
                 // Apply movement with the adjusted speed towards the target
-                Vec3d newVelocity = new Vec3d(moveDirection.x * currentSpeed, this.entity.getVelocity().y, moveDirection.z * currentSpeed);
+                Vec3d newVelocity = new Vec3d(moveDirection.x * currentSpeed, moveDirection.y * currentSpeed, moveDirection.z * currentSpeed);
 
                 this.entity.setVelocity(newVelocity);
                 this.entity.velocityModified = true;
@@ -123,7 +123,7 @@ public class LeadPlayerGoal extends PlayerBaseGoal {
         // Increment waypoint
         currentWaypoint++;
         LOGGER.info("Waypoint " + currentWaypoint + " / " + this.totalWaypoints);
-        this.currentTarget = RandomTargetFinder.findRandomTarget(this.entity, 0, 24, 36);
+        this.currentTarget = RandomTargetFinder.findRandomTarget(this.entity, 30, 24, 36);
         if (this.currentTarget != null) {
             emitParticleAt(this.currentTarget, ParticleTypes.FLAME);
             emitParticlesAlongRaycast(this.entity.getPos(), this.currentTarget, ParticleTypes.CLOUD, 0.5);

src/main/java/com/owlmaddie/utils/RandomTargetFinder.java

@@ -52,15 +52,21 @@ public class RandomTargetFinder {
     }
 
     private static Vec3d getConstrainedDirection(Vec3d initialDirection, double maxAngleOffset) {
-        double currentAngle = Math.atan2(initialDirection.z, initialDirection.x);
+        double randomYawAngleOffset = (random.nextDouble() * Math.toRadians(maxAngleOffset)) - Math.toRadians(maxAngleOffset / 2);
+        double randomPitchAngleOffset = (random.nextDouble() * Math.toRadians(maxAngleOffset)) - Math.toRadians(maxAngleOffset / 2);
 
-        double randomHorizontalAngleOffset = (random.nextDouble() * Math.toRadians(maxAngleOffset)) - Math.toRadians(maxAngleOffset / 2);
-        double constrainedAngle = currentAngle + randomHorizontalAngleOffset;
+        // Apply the yaw rotation (around the Y axis)
+        double cosYaw = Math.cos(randomYawAngleOffset);
+        double sinYaw = Math.sin(randomYawAngleOffset);
+        double xYaw = initialDirection.x * cosYaw - initialDirection.z * sinYaw;
+        double zYaw = initialDirection.x * sinYaw + initialDirection.z * cosYaw;
 
-        double x = Math.cos(constrainedAngle);
-        double z = Math.sin(constrainedAngle);
-
-        return new Vec3d(x, initialDirection.y, z).normalize();
+        // Apply the pitch rotation (around the X axis)
+        double cosPitch = Math.cos(randomPitchAngleOffset);
+        double sinPitch = Math.sin(randomPitchAngleOffset);
+        double yPitch = initialDirection.y * cosPitch - zYaw * sinPitch;
+        double zPitch = zYaw * cosPitch + initialDirection.y * sinPitch;
+        return new Vec3d(xYaw, yPitch, zPitch).normalize();
     }
 
     private static Vec3d getTargetInDirection(MobEntity entity, Vec3d direction, double minDistance, double maxDistance) {