From 5eed6c87922736d7c835cfb727f491d0f8b9c216 Mon Sep 17 00:00:00 2001
From: Ellpeck <megamaximal@gmail.com>
Date: Tue, 13 Nov 2018 14:52:19 +0100
Subject: [PATCH] add a passive effect that makes it so that extremely drained
 spots try to balance out with others in the area

---
 .../chunk/effect/DrainSpotEffects.java        |  1 +
 .../chunk/effect/MigrationEffect.java         | 49 +++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 src/main/java/de/ellpeck/naturesaura/chunk/effect/MigrationEffect.java

diff --git a/src/main/java/de/ellpeck/naturesaura/chunk/effect/DrainSpotEffects.java b/src/main/java/de/ellpeck/naturesaura/chunk/effect/DrainSpotEffects.java
index 7f9bbe48..54d7aa3a 100644
--- a/src/main/java/de/ellpeck/naturesaura/chunk/effect/DrainSpotEffects.java
+++ b/src/main/java/de/ellpeck/naturesaura/chunk/effect/DrainSpotEffects.java
@@ -8,5 +8,6 @@ public final class DrainSpotEffects {
         NaturesAuraAPI.DRAIN_SPOT_EFFECTS.put(GrassDieEffect.NAME, GrassDieEffect::new);
         NaturesAuraAPI.DRAIN_SPOT_EFFECTS.put(PlantBoostEffect.NAME, PlantBoostEffect::new);
         NaturesAuraAPI.DRAIN_SPOT_EFFECTS.put(ReplenishingEffect.NAME, ReplenishingEffect::new);
+        NaturesAuraAPI.DRAIN_SPOT_EFFECTS.put(MigrationEffect.NAME, MigrationEffect::new);
     }
 }
diff --git a/src/main/java/de/ellpeck/naturesaura/chunk/effect/MigrationEffect.java b/src/main/java/de/ellpeck/naturesaura/chunk/effect/MigrationEffect.java
new file mode 100644
index 00000000..b78b5761
--- /dev/null
+++ b/src/main/java/de/ellpeck/naturesaura/chunk/effect/MigrationEffect.java
@@ -0,0 +1,49 @@
+package de.ellpeck.naturesaura.chunk.effect;
+
+import de.ellpeck.naturesaura.NaturesAura;
+import de.ellpeck.naturesaura.api.aura.chunk.IAuraChunk;
+import de.ellpeck.naturesaura.api.aura.chunk.IDrainSpotEffect;
+import de.ellpeck.naturesaura.api.aura.type.IAuraType;
+import net.minecraft.util.ResourceLocation;
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.world.World;
+import net.minecraft.world.chunk.Chunk;
+import org.apache.commons.lang3.mutable.MutableInt;
+
+public class MigrationEffect implements IDrainSpotEffect {
+
+    public static final ResourceLocation NAME = new ResourceLocation(NaturesAura.MOD_ID, "migration");
+
+    @Override
+    public void update(World world, Chunk chunk, IAuraChunk auraChunk, BlockPos pos, MutableInt spot) {
+        if (spot.intValue() >= 0)
+            return;
+        int aura = IAuraChunk.getAuraInArea(world, pos, 25);
+        if (aura >= 0)
+            return;
+        int radius = Math.min(80, Math.abs(aura) / 50);
+        if (radius <= 0)
+            return;
+        BlockPos highestPos = IAuraChunk.getHighestSpot(world, pos, radius, null);
+        if (highestPos == null)
+            return;
+        IAuraChunk highestChunk = IAuraChunk.getAuraChunk(world, highestPos);
+        MutableInt highestSpot = highestChunk.getDrainSpot(highestPos);
+        if (highestSpot.intValue() <= 0)
+            return;
+
+        int toTransfer = Math.min(25, highestSpot.intValue());
+        highestChunk.drainAura(highestPos, toTransfer);
+        auraChunk.storeAura(pos, toTransfer);
+    }
+
+    @Override
+    public boolean appliesToType(IAuraType type) {
+        return true;
+    }
+
+    @Override
+    public ResourceLocation getName() {
+        return NAME;
+    }
+}