Make Experience Lanterns pull in experience orbs (#288)

Author: Catman06

src/main/java/plus/dragons/createenchantmentindustry/common/fluids/lantern/ExperienceLanternBlockEntity.java

@@ -67,6 +67,9 @@ public void tick() {
         if (!level.isClientSide && level.getGameTime() % 10 == 0) {
             drainExp();
         }
+        if (!level.isClientSide && CEIConfig.fluids().experienceLanternPullToggle.get()) {
+            pullExp();
+        }
     }
 
     public FluidTankBehaviour getTank() {
@@ -128,6 +131,19 @@ protected void drainExp() {
         }
     }
 
+    protected void pullExp() {
+        List<ExperienceOrb> experienceOrbs = level.getEntitiesOfClass(ExperienceOrb.class, effectiveAABB.inflate(CEIConfig.fluids().experienceLanternPullRadius.get()));
+        if (!experienceOrbs.isEmpty()) {
+            for (var orb : experienceOrbs) {
+                if (orb.getDeltaMovement().length() <= .5) {
+                    var pushForce = CEIConfig.fluids().experienceLanternPullForceMultiplier.get() * 1 / orb.position().distanceTo(getBlockPos().getCenter());
+                    var directionToLantern = getBlockPos().getCenter().subtract(orb.position()).normalize().multiply(pushForce, pushForce, pushForce);
+                    orb.push(directionToLantern);
+                }
+            }
+        }
+    }
+
     @Override
     public void addBehaviours(List<BlockEntityBehaviour> behaviours) {
         tank = new FluidTankBehaviour(this, this::createTank);

src/main/java/plus/dragons/createenchantmentindustry/common/fluids/lantern/ExperienceLanternMovementBehavior.java

@@ -28,6 +28,7 @@
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.phys.AABB;
+import net.minecraft.world.phys.Vec3;
 import net.neoforged.neoforge.fluids.FluidStack;
 import net.neoforged.neoforge.fluids.capability.IFluidHandler;
 import plus.dragons.createenchantmentindustry.common.fluids.experience.ExperienceHelper;
@@ -37,11 +38,17 @@
 public class ExperienceLanternMovementBehavior implements MovementBehaviour {
     @Override
     public void tick(MovementContext context) {
+        var effectiveAABB = new AABB(context.position.subtract(0.5d, 0.5d, 0.5d), context.position.add(0.5d, 0.5d, 0.5d)).inflate(0.5);
         if (!context.world.isClientSide && context.world.getGameTime() % 10 == 0) {
             drainExp(context.world,
-                    new AABB(context.position.subtract(0.5d, 0.5d, 0.5d), context.position.add(0.5d, 0.5d, 0.5d)).inflate(0.5),
+                    effectiveAABB,
                     context.contraption.getStorage().getFluids());
         }
+        if (!context.world.isClientSide && CEIConfig.fluids().experienceLanternPullToggle.get()) {
+            pullExp(context.world,
+                    effectiveAABB,
+                    context.position);
+        }
     }
 
     protected void drainExp(Level level, AABB effectiveAABB, MountedFluidStorageWrapper tank) {
@@ -99,4 +106,17 @@ protected void drainExp(Level level, AABB effectiveAABB, MountedFluidStorageWrap
             }
         }
     }
+
+    protected void pullExp(Level level, AABB effectiveAABB, Vec3 position) {
+        List<ExperienceOrb> experienceOrbs = level.getEntitiesOfClass(ExperienceOrb.class, effectiveAABB.inflate(CEIConfig.fluids().experienceLanternPullRadius.get()));
+        if (!experienceOrbs.isEmpty()) {
+            for (var orb : experienceOrbs) {
+                if (orb.getDeltaMovement().length() <= .5) {
+                    var pushForce = CEIConfig.fluids().experienceLanternPullForceMultiplier.get() * 1 / orb.position().distanceTo(position);
+                    var directionToLantern = position.subtract(orb.position()).normalize().multiply(pushForce, pushForce, pushForce);
+                    orb.push(directionToLantern);
+                }
+            }
+        }
+    }
 }

src/main/java/plus/dragons/createenchantmentindustry/config/CEIFluidsConfig.java

@@ -81,6 +81,15 @@ public class CEIFluidsConfig extends ConfigBase {
             "experienceLanternDrainRate",
             Comments.experienceLanternDrainRate,
             RequiresRestart.SERVER.asComment());
+    public final ConfigBool experienceLanternPullToggle = b(true,
+            "experienceLanternPullToggle",
+            Comments.experienceLanternPullToggle);
+    public final ConfigInt experienceLanternPullRadius = i(10, 0,
+            "experienceLanternPullRadius",
+            Comments.experienceLanternPullRadius);
+    public final ConfigFloat experienceLanternPullForceMultiplier = f(.075f, 0.0f, .5f,
+            "experienceLanternPullForceMultiplier",
+            Comments.experienceLanternPullForceMultiplier);
 
     @Override
     public String getName() {
@@ -106,5 +115,8 @@ static class Comments {
         static final String blazeForgerFluidCapacity = "The amount of liquid a Blaze Forger can hold (mB).";
         static final String experienceLanternFluidCapacity = "The amount of liquid an Experience Lantern can hold (mB).";
         static final String experienceLanternDrainRate = "The amount of experience an Experience Lantern can drain from player per 0.5 ticks (mB).";
+        static final String experienceLanternPullToggle = "Whether the Experience Lantern will pull in experience orbs from nearby.";
+        static final String experienceLanternPullRadius = "The range at which experience orbs will be pulled into the lantern.";
+        static final String experienceLanternPullForceMultiplier = "Modifier for the amount of force with which to pull the experience orbs.";
     }
 }