Fix .gitignore and add mixin.ai.task.run to repository

.gitignore

@@ -22,8 +22,10 @@ bin/
 
 # fabric
 
-run/
-neoforge/runs
+/run/
+/common/run/
+/fabric/run/
+/neoforge/run/
 # macOS
 
 .DS_Store

common/src/main/java/net/caffeinemc/mods/lithium/mixin/ai/task/run/long_jump_weighted_choice/LongJumpTaskMixin.java

@@ -0,0 +1,74 @@
+package net.caffeinemc.mods.lithium.mixin.ai.task.run.long_jump_weighted_choice;
+
+import com.llamalad7.mixinextras.sugar.Local;
+import net.caffeinemc.mods.lithium.common.util.collections.LongJumpChoiceList;
+import net.minecraft.core.BlockPos;
+import net.minecraft.server.level.ServerLevel;
+import net.minecraft.util.RandomSource;
+import net.minecraft.util.random.WeightedRandom;
+import net.minecraft.world.entity.Mob;
+import net.minecraft.world.entity.ai.behavior.LongJumpToRandomPos;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.Redirect;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.function.Consumer;
+
+@Mixin(LongJumpToRandomPos.class)
+public class LongJumpTaskMixin<E extends Mob> {
+
+    @Shadow
+    protected List<LongJumpToRandomPos.PossibleJump> jumpCandidates;
+
+    @Shadow
+    @Final
+    protected int maxLongJumpWidth;
+
+    @Shadow
+    @Final
+    protected int maxLongJumpHeight;
+
+    @Inject(
+            method = "start(Lnet/minecraft/server/level/ServerLevel;Lnet/minecraft/world/entity/Mob;J)V",
+            at = @At(
+                    value = "INVOKE",
+                    target = "Lnet/minecraft/core/BlockPos;betweenClosedStream(IIIIII)Ljava/util/stream/Stream;"
+            ),
+            cancellable = true
+    )
+    private void setTargets(ServerLevel serverWorld, E mobEntity, long l, CallbackInfo ci, @Local BlockPos centerPos) {
+        if (this.maxLongJumpWidth < 128 && this.maxLongJumpHeight < 128) {
+            this.jumpCandidates = LongJumpChoiceList.forCenter(centerPos, (byte) this.maxLongJumpWidth, (byte) this.maxLongJumpHeight);
+            ci.cancel();
+        }
+    }
+
+    @Redirect(
+            method = "getJumpCandidate(Lnet/minecraft/server/level/ServerLevel;)Ljava/util/Optional;",
+            at = @At(value = "INVOKE", target = "Lnet/minecraft/util/random/WeightedRandom;getRandomItem(Lnet/minecraft/util/RandomSource;Ljava/util/List;)Ljava/util/Optional;")
+    )
+    private Optional<LongJumpToRandomPos.PossibleJump> getRandomFast(RandomSource random, List<LongJumpToRandomPos.PossibleJump> pool) {
+        if (pool instanceof LongJumpChoiceList longJumpChoiceList) {
+            return Optional.ofNullable(longJumpChoiceList.removeRandomWeightedByDistanceSq(random));
+        } else {
+            return WeightedRandom.getRandomItem(random, pool);
+        }
+    }
+
+    @SuppressWarnings("OptionalUsedAsFieldOrParameterType")
+    @Redirect(
+            method = "getJumpCandidate(Lnet/minecraft/server/level/ServerLevel;)Ljava/util/Optional;",
+            at = @At(value = "INVOKE", target = "Ljava/util/Optional;ifPresent(Ljava/util/function/Consumer;)V")
+    )
+    private void skipRemoveIfAlreadyRemoved(Optional<LongJumpToRandomPos.PossibleJump> result, Consumer<? super LongJumpToRandomPos.PossibleJump> removeAction) {
+        if (!(this.jumpCandidates instanceof LongJumpChoiceList)) {
+            result.ifPresent(removeAction);
+        }
+    }
+}

common/src/main/java/net/caffeinemc/mods/lithium/mixin/ai/task/run/long_jump_weighted_choice/package-info.java

@@ -0,0 +1,4 @@
+@MixinConfigOption(description = "Speed up the weighted random choice of long jump target positions.")
+package net.caffeinemc.mods.lithium.mixin.ai.task.run.long_jump_weighted_choice;
+
+import net.caffeinemc.gradle.MixinConfigOption;

common/src/main/java/net/caffeinemc/mods/lithium/mixin/ai/task/run/package-info.java

@@ -0,0 +1,4 @@
+@MixinConfigOption(description = "Various optimizations inside AI tasks")
+package net.caffeinemc.mods.lithium.mixin.ai.task.run;
+
+import net.caffeinemc.gradle.MixinConfigOption;