4jcraft/targets/minecraft/world/entity/ai/control/MoveControl.cpp

#include "MoveControl.h"

#include <math.h>

#include <numbers>

#include "minecraft/util/Mth.h"
#include "minecraft/world/entity/Mob.h"
#include "minecraft/world/entity/ai/attributes/AttributeInstance.h"
#include "minecraft/world/entity/ai/control/JumpControl.h"
#include "minecraft/world/entity/monster/SharedMonsterAttributes.h"
#include "minecraft/world/phys/AABB.h"

const float MoveControl::MIN_SPEED = 0.0005f;
const float MoveControl::MIN_SPEED_SQR = MIN_SPEED * MIN_SPEED;

MoveControl::MoveControl(Mob* mob) {
    this->mob = mob;
    wantedX = mob->x;
    wantedY = mob->y;
    wantedZ = mob->z;

    speedModifier = 0.0;

    _hasWanted = false;
}

bool MoveControl::hasWanted() { return _hasWanted; }

double MoveControl::getSpeedModifier() { return speedModifier; }

void MoveControl::setWantedPosition(double x, double y, double z,
                                    double speedModifier) {
    wantedX = x;
    wantedY = y;
    wantedZ = z;
    this->speedModifier = speedModifier;
    _hasWanted = true;
}

void MoveControl::tick() {
    mob->setYya(0);
    if (!_hasWanted) return;
    _hasWanted = false;

    int yFloor = floor(mob->bb.y0 + .5f);

    double xd = wantedX - mob->x;
    double zd = wantedZ - mob->z;
    double yd = wantedY - yFloor;
    double dd = xd * xd + yd * yd + zd * zd;
    if (dd < MIN_SPEED_SQR) return;

    float yRotD = (float)(atan2(zd, xd) * 180 / std::numbers::pi) - 90;

    mob->yRot = rotlerp(mob->yRot, yRotD, MAX_TURN);
    mob->setSpeed(
        (float)(speedModifier *
                mob->getAttribute(SharedMonsterAttributes::MOVEMENT_SPEED)
                    ->getValue()));

    if (yd > 0 && xd * xd + zd * zd < 1) mob->getJumpControl()->jump();
}

float MoveControl::rotlerp(float a, float b, float max) {
    float diff = Mth::wrapDegrees(b - a);
    if (diff > max) {
        diff = max;
    }
    if (diff < -max) {
        diff = -max;
    }
    return a + diff;
}