mirror of https://github.com/cabaletta/baritone
423 lines
18 KiB
Java
423 lines
18 KiB
Java
/*
|
|
* This file is part of Baritone.
|
|
*
|
|
* Baritone is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Baritone is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with Baritone. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package baritone.process;
|
|
|
|
import baritone.Baritone;
|
|
import baritone.api.pathing.goals.*;
|
|
import baritone.api.process.IMineProcess;
|
|
import baritone.api.process.PathingCommand;
|
|
import baritone.api.process.PathingCommandType;
|
|
import baritone.api.utils.*;
|
|
import baritone.api.utils.input.Input;
|
|
import baritone.cache.CachedChunk;
|
|
import baritone.cache.WorldScanner;
|
|
import baritone.pathing.movement.CalculationContext;
|
|
import baritone.pathing.movement.MovementHelper;
|
|
import baritone.utils.BaritoneProcessHelper;
|
|
import baritone.utils.BlockStateInterface;
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.block.BlockAir;
|
|
import net.minecraft.block.BlockFalling;
|
|
import net.minecraft.block.state.IBlockState;
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.entity.item.EntityItem;
|
|
import net.minecraft.init.Blocks;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.World;
|
|
|
|
import java.util.*;
|
|
import java.util.stream.Collectors;
|
|
|
|
import static baritone.api.pathing.movement.ActionCosts.COST_INF;
|
|
|
|
/**
|
|
* Mine blocks of a certain type
|
|
*
|
|
* @author leijurv
|
|
*/
|
|
public final class MineProcess extends BaritoneProcessHelper implements IMineProcess {
|
|
|
|
private static final int ORE_LOCATIONS_COUNT = 64;
|
|
|
|
private BlockOptionalMetaLookup filter;
|
|
private List<BlockPos> knownOreLocations;
|
|
private List<BlockPos> blacklist; // inaccessible
|
|
private BlockPos branchPoint;
|
|
private GoalRunAway branchPointRunaway;
|
|
private int desiredQuantity;
|
|
private int tickCount;
|
|
|
|
public MineProcess(Baritone baritone) {
|
|
super(baritone);
|
|
}
|
|
|
|
@Override
|
|
public boolean isActive() {
|
|
return filter != null;
|
|
}
|
|
|
|
@Override
|
|
public PathingCommand onTick(boolean calcFailed, boolean isSafeToCancel) {
|
|
if (desiredQuantity > 0) {
|
|
int curr = ctx.player().inventory.mainInventory.stream()
|
|
.filter(stack -> filter.has(stack))
|
|
.mapToInt(ItemStack::getCount).sum();
|
|
System.out.println("Currently have " + curr + " valid items");
|
|
if (curr >= desiredQuantity) {
|
|
logDirect("Have " + curr + " valid items");
|
|
cancel();
|
|
return null;
|
|
}
|
|
}
|
|
if (calcFailed) {
|
|
if (!knownOreLocations.isEmpty() && Baritone.settings().blacklistClosestOnFailure.value) {
|
|
logDirect("Unable to find any path to " + filter + ", blacklisting presumably unreachable closest instance...");
|
|
knownOreLocations.stream().min(Comparator.comparingDouble(ctx.player()::getDistanceSq)).ifPresent(blacklist::add);
|
|
knownOreLocations.removeIf(blacklist::contains);
|
|
} else {
|
|
logDirect("Unable to find any path to " + filter + ", canceling mine");
|
|
cancel();
|
|
return null;
|
|
}
|
|
}
|
|
if (!Baritone.settings().allowBreak.value) {
|
|
logDirect("Unable to mine when allowBreak is false!");
|
|
cancel();
|
|
return null;
|
|
}
|
|
int mineGoalUpdateInterval = Baritone.settings().mineGoalUpdateInterval.value;
|
|
List<BlockPos> curr = new ArrayList<>(knownOreLocations);
|
|
if (mineGoalUpdateInterval != 0 && tickCount++ % mineGoalUpdateInterval == 0) { // big brain
|
|
CalculationContext context = new CalculationContext(baritone, true);
|
|
Baritone.getExecutor().execute(() -> rescan(curr, context));
|
|
}
|
|
if (Baritone.settings().legitMine.value) {
|
|
addNearby();
|
|
}
|
|
Optional<BlockPos> shaft = curr.stream()
|
|
.filter(pos -> pos.getX() == ctx.playerFeet().getX() && pos.getZ() == ctx.playerFeet().getZ())
|
|
.filter(pos -> pos.getY() >= ctx.playerFeet().getY())
|
|
.filter(pos -> !(BlockStateInterface.get(ctx, pos).getBlock() instanceof BlockAir)) // after breaking a block, it takes mineGoalUpdateInterval ticks for it to actually update this list =(
|
|
.min(Comparator.comparingDouble(ctx.player()::getDistanceSq));
|
|
baritone.getInputOverrideHandler().clearAllKeys();
|
|
if (shaft.isPresent() && ctx.player().onGround) {
|
|
BlockPos pos = shaft.get();
|
|
IBlockState state = baritone.bsi.get0(pos);
|
|
if (!MovementHelper.avoidBreaking(baritone.bsi, pos.getX(), pos.getY(), pos.getZ(), state)) {
|
|
Optional<Rotation> rot = RotationUtils.reachable(ctx, pos);
|
|
if (rot.isPresent() && isSafeToCancel) {
|
|
baritone.getLookBehavior().updateTarget(rot.get(), true);
|
|
MovementHelper.switchToBestToolFor(ctx, ctx.world().getBlockState(pos));
|
|
if (ctx.isLookingAt(pos) || ctx.playerRotations().isReallyCloseTo(rot.get())) {
|
|
baritone.getInputOverrideHandler().setInputForceState(Input.CLICK_LEFT, true);
|
|
}
|
|
return new PathingCommand(null, PathingCommandType.REQUEST_PAUSE);
|
|
}
|
|
}
|
|
}
|
|
PathingCommand command = updateGoal();
|
|
if (command == null) {
|
|
// none in range
|
|
// maybe say something in chat? (ahem impact)
|
|
cancel();
|
|
return null;
|
|
}
|
|
return command;
|
|
}
|
|
|
|
@Override
|
|
public void onLostControl() {
|
|
mine(0, (BlockOptionalMetaLookup) null);
|
|
}
|
|
|
|
@Override
|
|
public String displayName0() {
|
|
return "Mine " + filter;
|
|
}
|
|
|
|
private PathingCommand updateGoal() {
|
|
boolean legit = Baritone.settings().legitMine.value;
|
|
List<BlockPos> locs = knownOreLocations;
|
|
if (!locs.isEmpty()) {
|
|
List<BlockPos> locs2 = prune(new CalculationContext(baritone), new ArrayList<>(locs), filter, ORE_LOCATIONS_COUNT, blacklist);
|
|
// can't reassign locs, gotta make a new var locs2, because we use it in a lambda right here, and variables you use in a lambda must be effectively final
|
|
Goal goal = new GoalComposite(locs2.stream().map(loc -> coalesce(loc, locs2)).toArray(Goal[]::new));
|
|
knownOreLocations = locs2;
|
|
return new PathingCommand(goal, legit ? PathingCommandType.FORCE_REVALIDATE_GOAL_AND_PATH : PathingCommandType.REVALIDATE_GOAL_AND_PATH);
|
|
}
|
|
// we don't know any ore locations at the moment
|
|
if (!legit) {
|
|
return null;
|
|
}
|
|
// only in non-Xray mode (aka legit mode) do we do this
|
|
int y = Baritone.settings().legitMineYLevel.value;
|
|
if (branchPoint == null) {
|
|
/*if (!baritone.getPathingBehavior().isPathing() && playerFeet().y == y) {
|
|
// cool, path is over and we are at desired y
|
|
branchPoint = playerFeet();
|
|
branchPointRunaway = null;
|
|
} else {
|
|
return new GoalYLevel(y);
|
|
}*/
|
|
branchPoint = ctx.playerFeet();
|
|
}
|
|
// TODO shaft mode, mine 1x1 shafts to either side
|
|
// TODO also, see if the GoalRunAway with maintain Y at 11 works even from the surface
|
|
if (branchPointRunaway == null) {
|
|
branchPointRunaway = new GoalRunAway(1, y, branchPoint) {
|
|
@Override
|
|
public boolean isInGoal(int x, int y, int z) {
|
|
return false;
|
|
}
|
|
};
|
|
}
|
|
return new PathingCommand(branchPointRunaway, PathingCommandType.REVALIDATE_GOAL_AND_PATH);
|
|
}
|
|
|
|
private void rescan(List<BlockPos> already, CalculationContext context) {
|
|
if (filter == null) {
|
|
return;
|
|
}
|
|
if (Baritone.settings().legitMine.value) {
|
|
return;
|
|
}
|
|
List<BlockPos> locs = searchWorld(context, filter, ORE_LOCATIONS_COUNT, already, blacklist);
|
|
locs.addAll(droppedItemsScan(filter, ctx.world()));
|
|
if (locs.isEmpty()) {
|
|
logDirect("No locations for " + filter + " known, cancelling");
|
|
cancel();
|
|
return;
|
|
}
|
|
knownOreLocations = locs;
|
|
}
|
|
|
|
private boolean internalMiningGoal(BlockPos pos, IPlayerContext ctx, List<BlockPos> locs) {
|
|
// Here, BlockStateInterface is used because the position may be in a cached chunk (the targeted block is one that is kept track of)
|
|
if (locs.contains(pos)) {
|
|
return true;
|
|
}
|
|
IBlockState state = BlockStateInterface.get(ctx, pos);
|
|
if (Baritone.settings().internalMiningAirException.value && state.getBlock() instanceof BlockAir) {
|
|
return true;
|
|
}
|
|
return filter.has(state);
|
|
}
|
|
|
|
private Goal coalesce(BlockPos loc, List<BlockPos> locs) {
|
|
boolean assumeVerticalShaftMine = !(baritone.bsi.get0(loc.up()).getBlock() instanceof BlockFalling);
|
|
if (!Baritone.settings().forceInternalMining.value) {
|
|
if (assumeVerticalShaftMine) {
|
|
// we can get directly below the block
|
|
return new GoalThreeBlocks(loc);
|
|
} else {
|
|
// we need to get feet or head into the block
|
|
return new GoalTwoBlocks(loc);
|
|
}
|
|
}
|
|
boolean upwardGoal = internalMiningGoal(loc.up(), ctx, locs);
|
|
boolean downwardGoal = internalMiningGoal(loc.down(), ctx, locs);
|
|
boolean doubleDownwardGoal = internalMiningGoal(loc.down(2), ctx, locs);
|
|
if (upwardGoal == downwardGoal) { // symmetric
|
|
if (doubleDownwardGoal && assumeVerticalShaftMine) {
|
|
// we have a checkerboard like pattern
|
|
// this one, and the one two below it
|
|
// therefore it's fine to path to immediately below this one, since your feet will be in the doubleDownwardGoal
|
|
// but only if assumeVerticalShaftMine
|
|
return new GoalThreeBlocks(loc);
|
|
} else {
|
|
// this block has nothing interesting two below, but is symmetric vertically so we can get either feet or head into it
|
|
return new GoalTwoBlocks(loc);
|
|
}
|
|
}
|
|
if (upwardGoal) {
|
|
// downwardGoal known to be false
|
|
// ignore the gap then potential doubleDownward, because we want to path feet into this one and head into upwardGoal
|
|
return new GoalBlock(loc);
|
|
}
|
|
// upwardGoal known to be false, downwardGoal known to be true
|
|
if (doubleDownwardGoal && assumeVerticalShaftMine) {
|
|
// this block and two below it are goals
|
|
// path into the center of the one below, because that includes directly below this one
|
|
return new GoalTwoBlocks(loc.down());
|
|
}
|
|
// upwardGoal false, downwardGoal true, doubleDownwardGoal false
|
|
// just this block and the one immediately below, no others
|
|
return new GoalBlock(loc.down());
|
|
}
|
|
|
|
private static class GoalThreeBlocks extends GoalTwoBlocks {
|
|
|
|
public GoalThreeBlocks(BlockPos pos) {
|
|
super(pos);
|
|
}
|
|
|
|
@Override
|
|
public boolean isInGoal(int x, int y, int z) {
|
|
return x == this.x && (y == this.y || y == this.y - 1 || y == this.y - 2) && z == this.z;
|
|
}
|
|
|
|
@Override
|
|
public double heuristic(int x, int y, int z) {
|
|
int xDiff = x - this.x;
|
|
int yDiff = y - this.y;
|
|
int zDiff = z - this.z;
|
|
return GoalBlock.calculate(xDiff, yDiff < -1 ? yDiff + 2 : yDiff == -1 ? 0 : yDiff, zDiff);
|
|
}
|
|
}
|
|
|
|
public static List<BlockPos> droppedItemsScan(BlockOptionalMetaLookup filter, World world) {
|
|
if (!Baritone.settings().mineScanDroppedItems.value) {
|
|
return Collections.emptyList();
|
|
}
|
|
List<BlockPos> ret = new ArrayList<>();
|
|
for (Entity entity : world.loadedEntityList) {
|
|
if (entity instanceof EntityItem) {
|
|
EntityItem ei = (EntityItem) entity;
|
|
if (filter.has(ei.getItem())) {
|
|
ret.add(new BlockPos(entity));
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public static List<BlockPos> searchWorld(CalculationContext ctx, BlockOptionalMetaLookup filter, int max, List<BlockPos> alreadyKnown, List<BlockPos> blacklist) {
|
|
List<BlockPos> locs = new ArrayList<>();
|
|
List<Block> untracked = new ArrayList<>();
|
|
for (BlockOptionalMeta bom : filter.blocks()) {
|
|
Block block = bom.getBlock();
|
|
if (CachedChunk.BLOCKS_TO_KEEP_TRACK_OF.contains(block)) {
|
|
BetterBlockPos pf = ctx.baritone.getPlayerContext().playerFeet();
|
|
|
|
// maxRegionDistanceSq 2 means adjacent directly or adjacent diagonally; nothing further than that
|
|
locs.addAll(ctx.worldData.getCachedWorld().getLocationsOf(
|
|
BlockUtils.blockToString(block),
|
|
Baritone.settings().maxCachedWorldScanCount.value,
|
|
pf.x,
|
|
pf.z,
|
|
2
|
|
));
|
|
} else {
|
|
untracked.add(block);
|
|
}
|
|
}
|
|
|
|
locs = prune(ctx, locs, filter, max, blacklist);
|
|
|
|
if (!untracked.isEmpty() || (Baritone.settings().extendCacheOnThreshold.value && locs.size() < max)) {
|
|
locs.addAll(WorldScanner.INSTANCE.scanChunkRadius(
|
|
ctx.getBaritone().getPlayerContext(),
|
|
filter,
|
|
max,
|
|
10,
|
|
32
|
|
)); // maxSearchRadius is NOT sq
|
|
}
|
|
|
|
locs.addAll(alreadyKnown);
|
|
|
|
return prune(ctx, locs, filter, max, blacklist);
|
|
}
|
|
|
|
private void addNearby() {
|
|
knownOreLocations.addAll(droppedItemsScan(filter, ctx.world()));
|
|
BlockPos playerFeet = ctx.playerFeet();
|
|
BlockStateInterface bsi = new BlockStateInterface(ctx);
|
|
int searchDist = 10;
|
|
double fakedBlockReachDistance = 20; // at least 10 * sqrt(3) with some extra space to account for positioning within the block
|
|
for (int x = playerFeet.getX() - searchDist; x <= playerFeet.getX() + searchDist; x++) {
|
|
for (int y = playerFeet.getY() - searchDist; y <= playerFeet.getY() + searchDist; y++) {
|
|
for (int z = playerFeet.getZ() - searchDist; z <= playerFeet.getZ() + searchDist; z++) {
|
|
// crucial to only add blocks we can see because otherwise this
|
|
// is an x-ray and it'll get caught
|
|
if (filter.has(bsi.get0(x, y, z))) {
|
|
BlockPos pos = new BlockPos(x, y, z);
|
|
if ((Baritone.settings().legitMineIncludeDiagonals.value && knownOreLocations.stream().anyMatch(ore -> ore.distanceSq(pos) <= 2 /* sq means this is pytha dist <= sqrt(2) */)) || RotationUtils.reachable(ctx.player(), pos, fakedBlockReachDistance).isPresent()) {
|
|
knownOreLocations.add(pos);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
knownOreLocations = prune(new CalculationContext(baritone), knownOreLocations, filter, ORE_LOCATIONS_COUNT, blacklist);
|
|
}
|
|
|
|
private static List<BlockPos> prune(CalculationContext ctx, List<BlockPos> locs2, BlockOptionalMetaLookup filter, int max, List<BlockPos> blacklist) {
|
|
List<BlockPos> dropped = droppedItemsScan(filter, ctx.world);
|
|
dropped.removeIf(drop -> {
|
|
for (BlockPos pos : locs2) {
|
|
if (pos.distanceSq(drop) <= 9 && filter.has(ctx.get(pos.getX(), pos.getY(), pos.getZ())) && MineProcess.plausibleToBreak(ctx, pos)) { // TODO maybe drop also has to be supported? no lava below?
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
});
|
|
List<BlockPos> locs = locs2
|
|
.stream()
|
|
.distinct()
|
|
|
|
// remove any that are within loaded chunks that aren't actually what we want
|
|
.filter(pos -> !ctx.bsi.worldContainsLoadedChunk(pos.getX(), pos.getZ()) || filter.has(ctx.get(pos.getX(), pos.getY(), pos.getZ())) || dropped.contains(pos))
|
|
|
|
// remove any that are implausible to mine (encased in bedrock, or touching lava)
|
|
.filter(pos -> MineProcess.plausibleToBreak(ctx, pos))
|
|
|
|
.filter(pos -> !blacklist.contains(pos))
|
|
|
|
.sorted(Comparator.comparingDouble(ctx.getBaritone().getPlayerContext().player()::getDistanceSq))
|
|
.collect(Collectors.toList());
|
|
|
|
if (locs.size() > max) {
|
|
return locs.subList(0, max);
|
|
}
|
|
return locs;
|
|
}
|
|
|
|
public static boolean plausibleToBreak(CalculationContext ctx, BlockPos pos) {
|
|
if (MovementHelper.getMiningDurationTicks(ctx, pos.getX(), pos.getY(), pos.getZ(), ctx.bsi.get0(pos), true) >= COST_INF) {
|
|
return false;
|
|
}
|
|
|
|
// bedrock above and below makes it implausible, otherwise we're good
|
|
return !(ctx.bsi.get0(pos.up()).getBlock() == Blocks.BEDROCK && ctx.bsi.get0(pos.down()).getBlock() == Blocks.BEDROCK);
|
|
}
|
|
|
|
@Override
|
|
public void mineByName(int quantity, String... blocks) {
|
|
mine(quantity, new BlockOptionalMetaLookup(blocks));
|
|
}
|
|
|
|
@Override
|
|
public void mine(int quantity, BlockOptionalMetaLookup filter) {
|
|
this.filter = filter;
|
|
if (filter != null && !Baritone.settings().allowBreak.value) {
|
|
logDirect("Unable to mine when allowBreak is false!");
|
|
filter = null;
|
|
}
|
|
this.desiredQuantity = quantity;
|
|
this.knownOreLocations = new ArrayList<>();
|
|
this.blacklist = new ArrayList<>();
|
|
this.branchPoint = null;
|
|
this.branchPointRunaway = null;
|
|
if (filter != null) {
|
|
rescan(new ArrayList<>(), new CalculationContext(baritone));
|
|
}
|
|
}
|
|
}
|