KartAction.cc

#include "KartAction.hh"
 
#include "game/kart/KartMove.hh"
#include "game/kart/KartPhysics.hh"
#include "game/kart/KartState.hh"
 
#include "game/item/ItemDirector.hh"
#include "game/item/KartItem.hh"
 
#include <egg/math/Math.hh>
 
namespace Kart {
 
KartAction::KartAction()
    : m_currentAction(Action::None), m_hitDepth(EGG::Vector3f::zero),
      m_translation(EGG::Vector3f::ez), m_onStart(nullptr), m_onCalc(nullptr), m_onEnd(nullptr),
      m_actionParams(nullptr), m_rotationParams(nullptr), m_priority(0) {}
 
KartAction::~KartAction() = default;
 
void KartAction::init() {
    m_currentAction = Action::None;
    m_flags.makeAllZero();
}
 
void KartAction::calc() {
    if (m_currentAction == Action::None || !m_onCalc) {
        return;
    }
 
    if (calcCurrentAction()) {
        calcEndAction(false);
    }
}
 
void KartAction::calcVehicleSpeed() {
    move()->setSpeed(m_actionParams->calcSpeedMult * move()->speed());
}
 
bool KartAction::start(Action action) {
    ASSERT(action != Action::None);
 
    auto &status = KartObjectProxy::status();
 
    if (status.onBit(eStatus::InRespawn, eStatus::AfterRespawn, eStatus::BeforeRespawn,
                eStatus::InCannon)) {
        return false;
    }
 
    if (status.onBit(eStatus::ZipperStick)) {
        switch (action) {
        case Action::UNK_2:
        case Action::UNK_3:
        case Action::UNK_4:
        case Action::UNK_5:
        case Action::UNK_6:
            action = Action::UNK_1;
            break;
        default:
            break;
        }
    }
 
    size_t actionIdx = static_cast<size_t>(action);
 
    if (m_currentAction != Action::None && s_actionParams[actionIdx].priority <= m_priority) {
        return false;
    }
 
    calcEndAction(true);
    m_currentAction = action;
    m_actionParams = &s_actionParams[actionIdx];
    m_priority = m_actionParams->priority;
    m_onStart = s_onStart[actionIdx];
    m_onCalc = s_onCalc[actionIdx];
    m_onEnd = s_onEnd[actionIdx];
    status.setBit(eStatus::InAction);
    m_frame = 0;
    m_flags.makeAllZero();
    m_up = move()->up();
    move()->clear();
 
    applyStartSpeed();
    (this->*m_onStart)();
    return true;
}
 
void KartAction::startRotation(size_t idx) {
    m_rotation = EGG::Quatf::ident;
 
    EGG::Vector3f dir = move()->dir();
    if (speed() < 0.0f) {
        dir = -dir;
    }
 
    m_rotationDirection = dir.cross(bodyFront()).dot(bodyUp()) > 0.0f ? 1.0f : -1.0f;
    setRotation(idx);
    m_flags.setBit(eFlags::Rotating);
}
 
void KartAction::calcSideFromHitDepth() {
    m_hitDepth.normalise();
    m_side = m_hitDepth.perpInPlane(move()->smoothedUp(), true);
 
    if (m_side.squaredLength() <= std::numeric_limits<f32>::epsilon()) {
        m_side = EGG::Vector3f::ey;
    }
}
 
void KartAction::calcSideFromHitDepthAndTranslation() {
    calcSideFromHitDepth();
 
    EGG::Vector3f cross = m_translation.cross(m_side);
    f32 sign = (cross.y > 0.0f) ? 1.0f : -1.0f;
 
    EGG::Vector3f worldSide = EGG::Vector3f::ey.cross(m_translation);
    worldSide.normalise();
 
    m_side = worldSide.perpInPlane(move()->smoothedUp(), true);
 
    if (m_side.squaredLength() > std::numeric_limits<f32>::epsilon()) {
        m_side *= sign;
    } else {
        m_side = EGG::Vector3f::ey;
    }
}
 
void KartAction::end() {
    status().resetBit(eStatus::InAction, eStatus::LargeFlipHit);
    dynamics()->setForceUpright(true);
 
    m_currentAction = Action::None;
    m_priority = 0;
    m_flags.makeAllZero();
}
 
bool KartAction::calcCurrentAction() {
    ++m_frame;
    return (this->*m_onCalc)();
}
 
void KartAction::calcEndAction(bool endArg) {
    if (m_currentAction == Action::None) {
        return;
    }
 
    if (m_onEnd) {
        (this->*m_onEnd)(endArg);
        end();
    }
}
 
bool KartAction::calcRotation() {
    if (!m_rotationParams) {
        return false;
    }
 
    // Slow the rotation down as we approach the end of the spinout
    if (m_currentAngle > m_finalAngle * m_rotationParams->slowdownThreshold) {
        m_angleIncrement *= m_multiplier;
        if (m_rotationParams->minAngleIncrement > m_angleIncrement) {
            m_angleIncrement = m_rotationParams->minAngleIncrement;
        }
 
        m_multiplier -= m_multiplierDecrement;
        if (m_rotationParams->minMultiplier > m_multiplier) {
            m_multiplier = m_rotationParams->minMultiplier;
        }
    }
 
    m_currentAngle += m_angleIncrement;
    if (m_finalAngle < m_currentAngle) {
        m_currentAngle = m_finalAngle;
        return true;
    }
 
    return false;
}
 
void KartAction::calcUp() {
    m_up += (move()->up() - m_up) * 0.3f;
    m_up.normalise();
}
 
void KartAction::calcLanding() {
    if (m_currentAngle < m_targetRot || status().offBit(eStatus::TouchingGround)) {
        return;
    }
 
    m_flags.setBit(eFlags::Landing);
 
    if (!isBike()) {
        dynamics()->setForceUpright(false);
    }
 
    physics()->composeDecayingExtraRot(m_rotation);
}
 
void KartAction::startLaunch(f32 extVelScalar, f32 extVelKart, f32 extVelBike, f32 numRotations,
        u32 param6) {
    m_targetRot = 360.0f * numRotations;
 
    EGG::Vector3f extVel = EGG::Vector3f::zero;
    extVel.y = isBike() ? extVelBike : extVelKart;
 
    if (param6 == 0) {
        m_hitDepth = move()->dir();
        calcSideFromHitDepth();
    } else if (param6 == 1) {
        calcSideFromHitDepth();
        extVel += extVelScalar * m_side;
    } else if (param6 == 2) {
        calcSideFromHitDepthAndTranslation();
        extVel += extVelScalar * m_side;
    }
 
    setRotation(static_cast<size_t>(numRotations + 3.0f));
    m_groundStartLaunchTimer = 0;
    m_rotAxis = move()->smoothedUp().cross(m_side);
 
    dynamics()->setExtVel(dynamics()->extVel() + extVel);
}
 
void KartAction::activateCrush(u16 timer) {
    move()->activateCrush(m_crushTimer + timer);
    Item::ItemDirector::Instance()->kartItem(0).clear();
}
 
void KartAction::applyStartSpeed() {
    move()->setSpeed(m_actionParams->startSpeedMult * move()->speed());
    if (m_actionParams->startSpeedMult == 0.0f) {
        move()->clearDrift();
    }
}
 
void KartAction::setRotation(size_t idx) {
    ASSERT(idx - 1 < s_rotationParams.size());
    m_rotationParams = &s_rotationParams[--idx];
 
    m_finalAngle = m_rotationParams->finalAngle;
    m_angleIncrement = m_rotationParams->initialAngleIncrement;
    m_multiplierDecrement = m_rotationParams->initialMultiplierDecrement;
    m_currentAngle = 0.0f;
    m_multiplier = 1.0f;
}
 
/* ================================ *
 *     START FUNCTIONS
 * ================================ */
 
void KartAction::startStub() {}
 
void KartAction::startAction1() {
    startRotation(2);
}
 
void KartAction::startAction2() {
    constexpr f32 EXT_VEL_SCALAR = 0.0f;
    constexpr f32 EXT_VEL_KART = 30.0f;
    constexpr f32 EXT_VEL_BIKE = 30.0f;
    constexpr f32 NUM_ROTATIONS = 1.0f;
 
    startLaunch(EXT_VEL_SCALAR, EXT_VEL_KART, EXT_VEL_BIKE, NUM_ROTATIONS, 0);
}
 
void KartAction::startAction3() {
    constexpr f32 EXT_VEL_SCALAR = 25.0f;
    constexpr f32 EXT_VEL_KART = 30.0f;
    constexpr f32 EXT_VEL_BIKE = 30.0f;
    constexpr f32 NUM_ROTATIONS = 1.0f;
 
    startLaunch(EXT_VEL_SCALAR, EXT_VEL_KART, EXT_VEL_BIKE, NUM_ROTATIONS, 1);
    Item::ItemDirector::Instance()->kartItem(0).clear();
}
 
void KartAction::startAction4() {
    constexpr f32 EXT_VEL_SCALAR = 25.0f;
    constexpr f32 EXT_VEL_KART = 30.0f;
    constexpr f32 EXT_VEL_BIKE = 30.0f;
    constexpr f32 NUM_ROTATIONS = 2.0f;
 
    startLaunch(EXT_VEL_SCALAR, EXT_VEL_KART, EXT_VEL_BIKE, NUM_ROTATIONS, 1);
    Item::ItemDirector::Instance()->kartItem(0).clear();
}
 
void KartAction::startAction5() {
    constexpr f32 EXT_VEL_SCALAR = 13.0f;
    constexpr f32 EXT_VEL_KART = 40.0f;
    constexpr f32 EXT_VEL_BIKE = 45.0f;
    constexpr f32 NUM_ROTATIONS = 1.0f;
 
    startLaunch(EXT_VEL_SCALAR, EXT_VEL_KART, EXT_VEL_BIKE, NUM_ROTATIONS, 2);
}
 
void KartAction::startLargeFlipAction() {
    constexpr EGG::Vector3f INIT_VEL = EGG::Vector3f(0.0f, 60.0f, 0.0f);
 
    dynamics()->setExtVel(INIT_VEL);
    dynamics()->setAngVel0(EGG::Vector3f::zero);
 
    if (m_currentAction == Action::UNK_8) {
        calcSideFromHitDepth();
        dynamics()->setExtVel(dynamics()->extVel() + m_side * -20.0f);
    }
 
    Item::ItemDirector::Instance()->kartItem(0).clear();
 
    m_deltaPitch = 0.0f;
    m_pitch = 0.0f;
    m_velPitch = 22.0f;
    m_flipPhase = 0.0f;
    m_framesFlipping = 0;
 
    status().setBit(eStatus::LargeFlipHit);
}
 
void KartAction::startAction9() {
    startRotation(2);
}
 
void KartAction::startLongPressAction() {
    constexpr u32 ACTION_DURATION = 90;
    constexpr u16 CRUSH_DURATION = 480;
 
    m_crushTimer = ACTION_DURATION;
    activateCrush(CRUSH_DURATION);
}
 
void KartAction::startShortPressAction() {
    constexpr u32 ACTION_DURATION = 30;
    constexpr u16 CRUSH_DURATION = 240;
 
    m_crushTimer = ACTION_DURATION;
    activateCrush(CRUSH_DURATION);
}
 
void KartAction::startSpinShrinkAction() {
    startRotation(1);
}
 
/* ================================ *
 *     CALC FUNCTIONS
 * ================================ */
 
bool KartAction::calcStub() {
    return false;
}
 
bool KartAction::calcAction1() {
    calcUp();
    bool finished = calcRotation();
 
    m_rotation.setAxisRotation(DEG2RAD * (m_currentAngle * m_rotationDirection), m_up);
    physics()->composeExtraRot(m_rotation);
    return finished;
}
 
bool KartAction::calcLaunchAction() {
    constexpr u32 ACTION_DURATION = 100;
 
    if (m_flags.offBit(eFlags::Landing)) {
        calcRotation();
        calcLanding();
    }
 
    bool actionEnded = m_frame >= ACTION_DURATION;
 
    if (actionEnded) {
        if (m_flags.offBit(eFlags::Landing)) {
            physics()->composeDecayingExtraRot(m_rotation);
        }
    } else if (m_flags.offBit(eFlags::Landing)) {
        m_rotation.setAxisRotation(DEG2RAD * m_currentAngle, m_rotAxis);
        physics()->composeExtraRot(m_rotation);
    }
 
    return actionEnded;
}
 
bool KartAction::calcAction4() {
    constexpr u32 ACTION_DURATION = 140;
 
    auto &status = state()->status();
    if (status.onBit(eStatus::GroundStart)) {
        if (m_groundStartLaunchTimer++ == 0) {
            EGG::Vector3f extVel = dynamics()->extVel();
            extVel.y = 25.0f;
            dynamics()->setExtVel(extVel);
        }
    }
 
    if (m_flags.offBit(eFlags::Landing)) {
        calcRotation();
        calcLanding();
    }
 
    bool actionEnded = m_frame >= ACTION_DURATION;
 
    if (actionEnded) {
        if (m_flags.offBit(eFlags::Landing)) {
            physics()->composeDecayingExtraRot(m_rotation);
        }
    } else if (m_flags.offBit(eFlags::Landing)) {
        m_rotation.setAxisRotation(DEG2RAD * m_currentAngle, m_rotAxis);
        physics()->composeExtraRot(m_rotation);
    }
 
    return actionEnded;
}
 
bool KartAction::calcLargeFlipAction() {
    constexpr f32 PITCH_DECAY = 0.971f;
    constexpr f32 TOTAL_DELTA_PITCH = 720.0f;
    constexpr f32 PHASE_DELTA = 4.0f;
    constexpr f32 WOBBLE_AMPLITUDE = 18.1f;
    constexpr f32 BOUNCE_FACTOR = 5.0f;
 
    bool decayingRot = false;
    bool stuntRot = false;
 
    if (m_flags.onBit(eFlags::LargeFlip)) {
        ++m_framesFlipping;
    } else {
        if (m_deltaPitch < TOTAL_DELTA_PITCH) {
            m_deltaPitch += m_velPitch;
            m_pitch -= m_velPitch;
            m_velPitch *= (m_velPitch > 1.0f) ? PITCH_DECAY : 1.0f;
        } else {
            m_pitch = 0.0f;
        }
 
        f32 sin;
 
        if (EGG::Mathf::abs(m_flipPhase) < 360.0f) {
            m_flipPhase += PHASE_DELTA;
            sin = EGG::Mathf::SinFIdx(DEG2FIDX * m_flipPhase);
        } else {
            sin = 0.0f;
        }
 
        EGG::Matrix34f mat;
        mat.setAxisRotation(DEG2RAD * (WOBBLE_AMPLITUDE * sin), EGG::Vector3f::ez);
        m_rotation.setAxisRotation(DEG2RAD * m_pitch, mat.ps_multVector(EGG::Vector3f::ex));
 
        stuntRot = true;
    }
 
    bool actionEnded = false;
    auto &status = KartObjectProxy::status();
    bool touchingGround = status.onBit(eStatus::TouchingGround);
 
    if (m_flags.offBit(eFlags::LandingFromFlip) && touchingGround && move()->up().y > 0.0f &&
            m_frame > 50) {
        m_flags.setBit(eFlags::LandingFromFlip);
        dynamics()->setExtVel(move()->up().proj(EGG::Vector3f::ey) * BOUNCE_FACTOR);
    }
 
    if ((m_currentAction != Action::UNK_8 && m_frame < 10) || !touchingGround) {
        dynamics()->setExtVel(EGG::Vector3f(0.0f, dynamics()->extVel().y, 0.0f));
    }
 
    if ((touchingGround && move()->up().dot(EGG::Vector3f::ey) > 0.0f) || m_frame >= 300) {
        if (m_frame >= 40) {
            status.resetBit(eStatus::LargeFlipHit);
        }
 
        if (m_frame <= 120) {
            if (m_flags.onBit(eFlags::LargeFlip)) {
                if (m_framesFlipping > 30) {
                    actionEnded = true;
                }
            } else {
                if (m_frame >= 40 && m_frame <= 80) {
                    decayingRot = true;
                    m_flags.setBit(eFlags::LargeFlip);
                    status.resetBit(eStatus::LargeFlipHit);
                }
            }
        } else {
            actionEnded = true;
        }
    }
 
    if (decayingRot) {
        physics()->composeDecayingStuntRot(m_rotation);
    } else if (stuntRot) {
        physics()->composeStuntRot(m_rotation);
    }
 
    return actionEnded;
}
 
bool KartAction::calcPressAction() {
    EGG::Vector3f extVel = KartObjectProxy::extVel();
    extVel.y = std::min(0.0f, extVel.y);
    dynamics()->setExtVel(extVel);
 
    return m_frame > m_crushTimer;
}
 
/* ================================ *
 *     END FUNCTIONS
 * ================================ */
 
void KartAction::endStub(bool /*arg*/) {}
 
void KartAction::endAction1(bool arg) {
    if (arg) {
        physics()->composeDecayingExtraRot(m_rotation);
    }
}
 
void KartAction::endLaunchAction(bool arg) {
    if (arg) {
        physics()->composeDecayingExtraRot(m_rotation);
    }
}
 
/* ================================ *
 *     ACTION TABLES
 * ================================ */
 
const std::array<KartAction::ActionParams, KartAction::MAX_ACTION> KartAction::s_actionParams = {{
        {0.98f, 0.98f, 1},
        {0.98f, 0.98f, 2},
        {0.96f, 0.96f, 4},
        {0.0f, 0.96f, 4},
        {0.0f, 0.98f, 4},
        {0.0f, 0.96f, 4},
        {0.0f, 0.96f, 4},
        {0.0f, 0.0f, 6},
        {0.0f, 0.99f, 6},
        {0.98f, 0.98f, 3},
        {0.98f, 0.98f, 3},
        {1.0f, 1.0f, 5},
        {0.0f, 0.0f, 3},
        {0.0f, 0.0f, 3},
        {0.0f, 0.0f, 3},
        {0.98f, 0.98f, 3},
        {0.0f, 0.0f, 3},
        {0.98f, 0.98f, 3},
}};
 
const std::array<KartAction::RotationParams, 5> KartAction::s_rotationParams = {{
        {10.0f, 1.5f, 0.9f, 0.005f, 0.6f, 360.0f},
        {11.0f, 1.5f, 0.9f, 0.0028f, 0.7f, 720.0f},
        {11.0f, 1.5f, 0.9f, 0.0028f, 0.8f, 1080.0f},
        {7.0f, 1.5f, 0.9f, 0.005f, 0.6f, 450.0f},
        {9.0f, 1.5f, 0.9f, 0.0028f, 0.7f, 810.0f},
}};
 
const std::array<KartAction::StartActionFunc, KartAction::MAX_ACTION> KartAction::s_onStart = {{
        &KartAction::startStub,
        &KartAction::startAction1,
        &KartAction::startAction2,
        &KartAction::startAction3,
        &KartAction::startAction4,
        &KartAction::startAction5,
        &KartAction::startStub,
        &KartAction::startLargeFlipAction,
        &KartAction::startLargeFlipAction,
        &KartAction::startAction9,
        &KartAction::startStub,
        &KartAction::startStub,
        &KartAction::startLongPressAction,
        &KartAction::startStub,
        &KartAction::startShortPressAction,
        &KartAction::startSpinShrinkAction,
        &KartAction::startStub,
        &KartAction::startStub,
}};
 
const std::array<KartAction::CalcActionFunc, KartAction::MAX_ACTION> KartAction::s_onCalc = {{
        &KartAction::calcStub,
        &KartAction::calcAction1,
        &KartAction::calcLaunchAction,
        &KartAction::calcLaunchAction,
        &KartAction::calcAction4,
        &KartAction::calcLaunchAction,
        &KartAction::calcStub,
        &KartAction::calcLargeFlipAction,
        &KartAction::calcLargeFlipAction,
        &KartAction::calcAction1,
        &KartAction::calcStub,
        &KartAction::calcStub,
        &KartAction::calcPressAction,
        &KartAction::calcStub,
        &KartAction::calcPressAction,
        &KartAction::calcAction1,
        &KartAction::calcStub,
        &KartAction::calcStub,
}};
 
const std::array<KartAction::EndActionFunc, KartAction::MAX_ACTION> KartAction::s_onEnd = {{
        &KartAction::endStub,
        &KartAction::endAction1,
        &KartAction::endLaunchAction,
        &KartAction::endLaunchAction,
        &KartAction::endLaunchAction,
        &KartAction::endLaunchAction,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endAction1,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endStub,
        &KartAction::endAction1,
        &KartAction::endStub,
        &KartAction::endStub,
}};
 
} // namespace Kart