ObjectWanwan.cc

#include "ObjectWanwan.hh"
 
#include "game/field/CollisionDirector.hh"
#include "game/field/ObjectDirector.hh"
 
#include "game/kart/KartObject.hh"
 
#include "game/system/RaceConfig.hh"
 
#include "game/system/RaceManager.hh"
 
namespace Kinoko::Field {

ObjectWanwan::ObjectWanwan(const System::MapdataGeoObj &params)
    : ObjectCollidable(params), StateManager(this, STATE_ENTRIES), m_pitch(0.0f),
      m_chainLength(static_cast<f32>(params.setting(0))),
      m_attackDistance(4800.0f + static_cast<f32>(params.setting(2))),
      m_attackArcTargetX(10.0f * static_cast<f32>(static_cast<s16>(params.setting(3)))),
      m_attackArcTargetZ(10.0f * static_cast<f32>(static_cast<s16>(params.setting(4)))),
      m_chainAttachMat(EGG::Matrix34f::ident) {
    constexpr EGG::Vector3f ANCHOR_OFFSET = EGG::Vector3f(0.0f, 20.0f, 0.0f);
    constexpr EGG::Vector3f POS_OFFSET_MCWII = EGG::Vector3f(14500.0f, 1300.0f, 44850.0f);
    constexpr EGG::Vector3f POS_OFFSET_RMC = EGG::Vector3f(8012.0f, 1668.0f, -30150.0f);
 
    m_idleDuration = static_cast<u32>(params.setting(5));
    m_attackArc = static_cast<f32>(params.setting(6));
 
    if (m_idleDuration == 0) {
        m_idleDuration = 300;
    }
 
    if (m_attackArc == 0.0f) {
        m_attackArc = 30.0f;
    }
 
    auto *pile = EGG::egg_new<ObjectWanwanPile>(pos(), rot(), scale());
    pile->load();
 
    m_anchor = pos() + ANCHOR_OFFSET;
 
    setScale(SCALE);
 
    f32 sqLen = m_chainLength * m_chainLength + 300.0f * (300.0f * scale().y) * scale().y;
    m_chainCount = static_cast<u32>(EGG::Mathf::sqrt(sqLen) / (CHAIN_LENGTH * scale().y));
    if (m_chainCount > 0) {
        --m_chainCount;
    }
 
    m_initPos = m_anchor + EGG::Vector3f(0.5f * -m_chainLength, 600.0f, 0.5f * -m_chainLength);
 
    auto course = System::RaceConfig::Instance()->raceScenario().course;
    if (course == Course::Mario_Circuit) {
        EGG::Vector3f pos = POS_OFFSET_MCWII - m_anchor;
        pos.y = 0.0f;
        pos.normalise2();
        m_attackArcCenter = pos * m_chainLength + m_anchor;
        m_attackArcCenter.y = POS_OFFSET_MCWII.y;
    } else if (course == Course::GCN_Mario_Circuit) {
        EGG::Vector3f pos = POS_OFFSET_RMC - m_anchor;
        pos.y = 0.0f;
        pos.normalise2();
        m_attackArcCenter = pos * m_chainLength + m_anchor;
        m_attackArcCenter.y = POS_OFFSET_RMC.y;
    } else {
        m_attackArcCenter = m_anchor;
        m_attackArcCenter.z += m_chainLength;
    }
 
    initTransformKeyframes();
}

ObjectWanwan::~ObjectWanwan() = default;

void ObjectWanwan::init() {
    setPos(m_initPos);
    m_chainAttachPos = m_initPos;
    m_vel.setZero();
    m_accel.setZero();
    m_speed = 0.0f;
    m_tangent = EGG::Vector3f::ex;
    m_up = EGG::Vector3f::ey;
    m_targetUp = EGG::Vector3f::ey;
    m_touchingFloor = false;
    m_chainTaut = false;
    m_frame = 0;
    m_target.setZero();
    m_targetDir = EGG::Vector3f::ez;
    m_retarget = false;
    m_wanderTimer = 0;
    m_attackStill = false;
    m_backDir = EGG::Vector3f::ex;
    m_nextStateId = 0;
}

void ObjectWanwan::calc() {
    StateManager::calc();
 
    calcPos();
    calcAttackPos();
    calcCollision();
    calcMat();
    calcChainAttachMat();
    calcChain();
 
    ++m_frame;
 
    if (pos().y < m_anchor.y - 1000.0f) {
        setPos(EGG::Vector3f(pos().x, m_anchor.y + 1000.0f, pos().z));
        m_vel.y = 0.0f;
    }
}

Kart::Reaction ObjectWanwan::onCollision(Kart::KartObject *kartObj, Kart::Reaction reactionOnKart,
        Kart::Reaction /*reactionOnObj*/, EGG::Vector3f & /*hitDepth*/) {
    if (m_currentStateId == 1 && !m_attackStill) {
        return reactionOnKart;
    }
 
    return kartObj->speedRatioCapped() < 0.5f ? Kart::Reaction::WallAllSpeed : reactionOnKart;
}

void ObjectWanwan::enterWait() {
    constexpr f32 ANGLE_RANGE = 0.33f * 60.0f;
    constexpr f32 ANGLE_NORMALIZATION = 0.66f * 60.0f;
    m_retarget = false;
    m_vel.x = 0.0f;
    m_vel.z = 0.0f;
    m_accel.x = 0.0f;
    m_accel.z = 0.0f;
    m_speed = 0.0f;
 
    f32 randAngle =
            System::RaceManager::Instance()->random().getF32(ANGLE_RANGE) + ANGLE_NORMALIZATION;
 
    if (CrossXZ(pos() + m_tangent, pos(), m_anchor) >= 0.0f) {
        randAngle *= -1.0f;
    }
 
    EGG::Vector3f vStack_40 = m_anchor - EGG::Vector3f(pos().x, m_anchor.y, pos().z);
    vStack_40.normalise2();
 
    EGG::Vector3f vStack_4c = RotateXZByYaw(DEG2RAD * randAngle, vStack_40);
    f32 fVar2 = m_chainLength < 3000.0f ? 0.5f : 0.7f;
    m_target = vStack_4c * m_chainLength * fVar2 + m_anchor;
}

void ObjectWanwan::enterAttack() {
    m_vel.x = 0.0f;
    m_vel.z = 0.0f;
    m_accel.x = 0.0f;
    m_accel.z = 0.0f;
    m_speed = 0.0f;
    m_pitch = 0.0f;
    m_wanderTimer = 0;
    m_attackStill = false;
    m_chainTaut = false;
    m_anchor.y += 30.0f;
 
    calcRandomTarget();
}

void ObjectWanwan::enterBack() {
    m_vel.x = 0.0f;
    m_vel.z = 0.0f;
    m_accel.x = 0.0f;
    m_accel.z = 0.0f;
    m_speed = 15.0f;
    m_pitch = -ObjectDirector::Instance()->WanwanMaxPitch();
    m_backDir = m_anchor - pos();
    m_backDir.y = 0.0f;
    m_backDir.normalise2();
    m_attackStill = false;
    m_chainTaut = false;
    m_anchor.y -= 30.0f;
}

void ObjectWanwan::calcWait() {
    constexpr f32 ANGLE_RANGE = 0.33f * 0.5f * 60.0f;
    constexpr f32 ANGLE_NORMALIZATION = 0.66f * 0.5f * 60.0f;
 
    EGG::Vector3f targetDir = m_target - pos();
    targetDir.y = 0.0f;
 
    // In the base game, if the dot product is less than epsilon, then r31 contains F_PI
    f32 distFromTarget = F_PI;
    if (targetDir.squaredLength() > std::numeric_limits<f32>::epsilon()) {
        distFromTarget = targetDir.normalise();
    }
 
    if (!m_retarget && distFromTarget < 0.55f * m_chainLength) {
        EGG::Vector3f relTarget = m_target - m_anchor;
        auto &rand = System::RaceManager::Instance()->random();
        f32 angle = rand.getF32(ANGLE_RANGE) + ANGLE_NORMALIZATION;
        if (CrossXZ(m_target, m_attackArcCenter, m_anchor) >= 0.0f) {
            angle *= -1.0f;
        }
 
        m_target = RotateXZByYaw(angle * DEG2RAD, relTarget) + m_anchor;
        m_retarget = true;
    }
 
    m_targetDir = targetDir;
 
    calcTangent(0.04f);
    calcUp(0.1f);
 
    calcSpeed();
    calcBounce();
 
    calcWanderTimer();
}

void ObjectWanwan::calcAttack() {
    constexpr u32 ATTACK_DURATION = 120;
    constexpr f32 PITCH_STEP = 25.0f;
 
    if (m_currentFrame > ATTACK_DURATION) {
        m_nextStateId = 2;
    }
 
    f32 maxPitch = ObjectDirector::Instance()->WanwanMaxPitch();
    if (EGG::Mathf::abs(m_pitch) < EGG::Mathf::abs(maxPitch)) {
        m_pitch -= maxPitch / PITCH_STEP;
    }
 
    calcTangent(10.0f * 0.04f);
    calcUp(0.1f);
 
    if (m_chainTaut || m_attackStill) {
        if (!m_attackStill) {
            m_target = m_anchor + (m_chainAttachPos - m_anchor) * 2.0f;
            m_targetDir = m_target - pos();
            m_targetDir.y = 0.0f;
            m_targetDir.normalise2();
            EGG::Vector3f posOffset = pos() - m_chainAttachPos;
            posOffset.y = 0.0f;
            f32 radius = posOffset.length();
            EGG::Vector3f chainDir = m_chainAttachPos - m_anchor;
            chainDir.y = 0.0f;
            chainDir.normalise2();
            EGG::Vector3f nextPos = m_chainAttachPos + chainDir * radius;
            setPos(EGG::Vector3f(nextPos.x, pos().y, nextPos.z));
            EGG::Vector3f tangent = m_tangent + chainDir;
            tangent.y = 0.0f;
            if (tangent.squaredLength() > std::numeric_limits<f32>::epsilon()) {
                tangent.normalise2();
            }
 
            m_tangent = tangent;
        }
 
        m_attackStill = true;
        m_vel.y = 0.0f;
        m_vel *= -0.85f;
    } else {
        m_vel.x = m_tangent.x * 120.0f;
        m_vel.z = m_tangent.z * 120.0f;
    }
}

void ObjectWanwan::calcBack() {
    if (EGG::Mathf::abs(m_pitch) > 2.0f) {
        m_pitch += ObjectDirector::Instance()->WanwanMaxPitch() / 15.0f;
    }
 
    m_vel.x = m_backDir.x * m_speed * 1.5f;
    m_vel.z = m_backDir.z * m_speed * 1.5f;
 
    if (m_touchingFloor) {
        m_vel.y = 0.0f;
        m_accel += EGG::Vector3f::ey * 12.0f;
    } else {
        m_accel.y = 0.0f;
    }
 
    if (m_currentFrame > 90) {
        m_nextStateId = 0;
    }
}

void ObjectWanwan::calcPos() {
    m_vel += m_accel - GRAVITY;
    addPos(m_vel);
    m_accel.setZero();
}

void ObjectWanwan::calcCollision() {
    constexpr EGG::Vector3f POS_OFFSET = EGG::Vector3f(0.0f, -570.0f, 0.0f);
 
    m_touchingFloor = false;
    CollisionInfo info;
    KCLTypeMask mask;
    EGG::Vector3f colPos = pos() + POS_OFFSET;
    auto *colDir = CollisionDirector::Instance();
 
    bool hasCol = colDir->checkSphereFullPush(30.0f, colPos, EGG::Vector3f::inf, KCL_TYPE_FLOOR,
            &info, &mask, 0);
 
    if (!hasCol) {
        return;
    }
 
    m_touchingFloor = true;
    EGG::Vector3f local_84 = info.tangentOff;
    f32 scale = local_84.normalise();
    addPos(EGG::Vector3f::ey * scale);
 
    if (info.floorDist > -std::numeric_limits<f32>::min()) {
        m_targetUp = info.floorNrm;
    }
 
    m_accel += GRAVITY;
}

void ObjectWanwan::calcMat() {
    EGG::Matrix34f mat;
    if (m_currentStateId == 1) {
        SetRotTangentHorizontal(mat, EGG::Vector3f::ey, m_tangent);
    } else {
        SetRotTangentHorizontal(mat, m_up, m_tangent);
    }
    mat.setBase(3, EGG::Vector3f::zero);
 
    EGG::Vector3f rot = EGG::Vector3f(m_pitch * DEG2RAD, 0.0f, 0.0f);
    EGG::Matrix34f rtMat;
    rtMat.makeRT(rot, EGG::Vector3f::zero);
    mat = mat.multiplyTo(rtMat);
    mat.setBase(3, pos());
    setTransform(mat);
}

void ObjectWanwan::calcChainAttachMat() {
    u32 idx = m_currentStateId == 1 ? 0 : m_frame % 15;
 
    calcTransform();
 
    EGG::Matrix34f mat = transform();
    mat.setBase(3, EGG::Vector3f::zero);
    EGG::Vector3f keyFramePos = m_transformKeyframes[idx].base(3);
    EGG::Vector3f posOffset = mat.ps_multVector(keyFramePos) * 2.0f;
    mat.setBase(3, posOffset + pos());
 
    m_chainAttachMat = mat;
}

void ObjectWanwan::calcSpeed() {
    if (m_speed < 8.0f) {
        m_speed += 0.5f;
        m_accel.x = m_tangent.x * 0.5f;
        m_accel.z = m_tangent.z * 0.5f;
    } else {
        m_speed = 8.0f;
        m_accel.x = 0.0f;
        m_accel.z = 0.0f;
        m_vel.x = m_tangent.x * 8.0f;
        m_vel.z = m_tangent.z * 8.0f;
    }
}

void ObjectWanwan::calcBounce() {
    if (m_touchingFloor) {
        m_vel.y = 0.0f;
        m_accel += EGG::Vector3f::ey * 12.0f;
    } else {
        m_accel.y = 0.0f;
    }
}

void ObjectWanwan::calcTangent(f32 t) {
    m_tangent = Interpolate(t, m_tangent, m_targetDir);
    if (m_tangent.squaredLength() > std::numeric_limits<f32>::epsilon()) {
        m_tangent.normalise2();
    } else {
        m_tangent = m_targetDir;
    }
}

void ObjectWanwan::calcUp(f32 t) {
    m_up = Interpolate(t, m_up, m_targetUp);
    if (m_up.squaredLength() > std::numeric_limits<f32>::epsilon()) {
        m_up.normalise2();
    } else {
        m_up = EGG::Vector3f::ey;
    }
}

void ObjectWanwan::calcRandomTarget() {
    f32 angle = System::RaceManager::Instance()->random().getF32(m_attackArc * 2.0f);
    EGG::Vector3f attackArcTarget = EGG::Vector3f(m_attackArcTargetX, 0.0f, m_attackArcTargetZ);
    EGG::Vector3f attackArcDir = attackArcTarget - m_anchor;
    attackArcDir.y = 0.0f;
    attackArcDir.normalise2();
    EGG::Vector3f attackTargetDir = RotateXZByYaw((angle - m_attackArc) * DEG2RAD, attackArcDir);
    m_target = m_anchor + attackTargetDir * m_attackDistance;
    m_targetDir = m_target - pos();
    m_targetDir.y = 0.0f;
    m_targetDir.normalise2();
}

void ObjectWanwan::initTransformKeyframes() {
    std::array<f32, 15> xRot;
    SampleHermiteInterp(0.0f, 10.0f, 2.0f, -1.111111f, std::span(xRot.begin(), 6));
    SampleHermiteInterp(10.0f, -10.0f, -1.111111f, -1.111111f, std::span(xRot.begin() + 5, 5));
    SampleHermiteInterp(-10.0f, 0.0f, -1.111111f, 2.0f, std::span(xRot.begin() + 9, 6));
 
    std::array<f32, 15> yPos;
    SampleHermiteInterp(0.0f, -27.35f, -9.1166658f, 3.75f, std::span(yPos.begin(), 4));
    SampleHermiteInterp(-27.35f, 33.75f, 3.75f, 2.4863639f, std::span(yPos.begin() + 3, 7));
    SampleHermiteInterp(33.75f, 0.0f, 2.4863639f, -6.75f, std::span(yPos.begin() + 9, 6));
 
    std::array<f32, 15> zPos;
    SampleHermiteInterp(0.0f, -33.75f, -6.75f, -4.8214278f, std::span(zPos.begin(), 6));
    SampleHermiteInterp(-33.75f, -33.75f, -4.8214278f, 8.4375f, std::span(zPos.begin() + 5, 3));
    SampleHermiteInterp(-33.75f, 0.0f, 8.4375f, 14.464999f, std::span(zPos.begin() + 7, 3));
    SampleHermiteInterp(0.0f, 24.11f, 14.464999f, 0.0f, std::span(zPos.begin() + 9, 3));
    SampleHermiteInterp(24.11f, 0.0f, 0.0f, -8.0366669f, std::span(zPos.begin() + 11, 4));
 
    for (u8 i = 0; i < m_transformKeyframes.size(); ++i) {
        EGG::Matrix34f mat;
        mat.makeR(EGG::Vector3f(xRot[i] * DEG2RAD, 0.0f, 0.0f));
        mat.setBase(3, EGG::Vector3f(0.0f, yPos[i], zPos[i]));
        m_transformKeyframes[i] = mat;
    }
}

void ObjectWanwan::calcAttackPos() {
    constexpr f32 SCALED_CHAIN_LENGTH = CHAIN_LENGTH * SCALE;
 
    if (m_currentStateId != 1) {
        return;
    }
 
    calcMat();
    calcTransform();
    calcChainAttachPos(transform());
 
    EGG::Vector3f dir = m_chainAttachPos - m_anchor;
    f32 dist = dir.normalise();
    dist -= SCALED_CHAIN_LENGTH * static_cast<f32>(m_chainCount);
 
    if (dist > 0.0f || m_attackStill) {
        m_chainTaut = true;
        subPos(dir * dist);
        addPos(dir * 35.0f);
    }
}
 
void ObjectWanwan::calcChainAttachPos(EGG::Matrix34f mat) {
    EGG::Vector3f pos = mat.base(3);
    pos -= mat.base(2) * 250.0f * scale().x;
    mat.setBase(3, pos);
 
    EGG::Vector3f backOffset = mat.base(2) * 140.0f * scale().x;
    EGG::Vector3f verticalOffset = mat.base(1) * 20.0f * scale().x;
    m_chainAttachPos = pos - backOffset + verticalOffset;
}

void ObjectWanwan::SampleHermiteInterp(f32 start, f32 end, f32 startTangent, f32 endTangent,
        std::span<f32> dst) {
    dst.front() = start;
    dst.back() = end;
 
    f32 scalar = 1.0f / static_cast<f32>(dst.size() - 1);
 
    for (u8 i = 1; i < dst.size() - 1; ++i) {
        dst[i] = EGG::Mathf::Hermite(start, startTangent, end, endTangent,
                scalar * static_cast<f32>(i));
    }
}
 
} // namespace Kinoko::Field