001/*
002 * This file is part of Baritone.
003 *
004 * Baritone is free software: you can redistribute it and/or modify
005 * it under the terms of the GNU Lesser General Public License as published by
006 * the Free Software Foundation, either version 3 of the License, or
007 * (at your option) any later version.
008 *
009 * Baritone is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
012 * GNU Lesser General Public License for more details.
013 *
014 * You should have received a copy of the GNU Lesser General Public License
015 * along with Baritone.  If not, see <https://www.gnu.org/licenses/>.
016 */
017
018package baritone.api.pathing.goals;
019
020import baritone.api.BaritoneAPI;
021
022public class GoalAxis implements Goal {
023
024    private static final double SQRT_2_OVER_2 = Math.sqrt(2) / 2;
025
026    @Override
027    public boolean isInGoal(int x, int y, int z) {
028        return y == BaritoneAPI.getSettings().axisHeight.value && (x == 0 || z == 0 || Math.abs(x) == Math.abs(z));
029    }
030
031    @Override
032    public double heuristic(int x0, int y, int z0) {
033        int x = Math.abs(x0);
034        int z = Math.abs(z0);
035
036        int shrt = Math.min(x, z);
037        int lng = Math.max(x, z);
038        int diff = lng - shrt;
039
040        double flatAxisDistance = Math.min(x, Math.min(z, diff * SQRT_2_OVER_2));
041
042        return flatAxisDistance * BaritoneAPI.getSettings().costHeuristic.value + GoalYLevel.calculate(BaritoneAPI.getSettings().axisHeight.value, y);
043    }
044
045    @Override
046    public String toString() {
047        return "GoalAxis";
048    }
049}