/**
 * d_a_alink_hvyboots.inc
 * Player Iron Boots action handling
 */

#include "d/actor/d_a_alink.h"
#include "d/actor/d_a_tag_magne.h"

void daAlink_c::concatMagneBootMtx() {
    if (checkMagneBootsOn()) {
        mDoMtx_stack_c::concat(mMagneBootMtx);
    }
}

void daAlink_c::concatMagneBootInvMtx() {
    if (checkMagneBootsOn()) {
        mDoMtx_stack_c::concat(mMagneBootInvMtx);
    }
}

void daAlink_c::multVecMagneBootInvMtx(cXyz* o_dst) {
    if (checkMagneBootsOn()) {
        mDoMtx_multVecSR(mMagneBootInvMtx, o_dst, o_dst);
    }
}

bool daAlink_c::commonMagneLineCheck(cXyz* i_start, cXyz* i_end) {
    mMagneLineChk.Set(i_start, i_end, this);

    if (dComIfG_Bgsp().LineCross(&mMagneLineChk) && daTagMagne_c::checkMagnetCode(mMagneLineChk)) {
        return true;
    }

    return false;
}

BOOL daAlink_c::checkBootsMoveAnime(int param_0) {
    return (checkHeavyStateOn(param_0, 1) && !checkNoResetFlg0(FLG0_WATER_IN_MOVE)) ||
           checkDkCaught2Anime();
}

int daAlink_c::setHeavyBoots(int param_0) {
    if (checkEquipHeavyBoots()) {
        offNoResetFlg0(FLG0_EQUIP_HVY_BOOTS);
        mZ2Link.setLinkBootsType(0);

        if (field_0x06e0 != NULL) {
            field_0x06e0->show();

            if (checkZoraWearMaskDraw()) {
                field_0x06e4->show();
            }
        }

        if (checkModeFlg(0x40000) && !checkHookshotItem(mEquipItem)) {
            deleteEquipItem(TRUE, FALSE);
        }

        if (checkNoResetFlg0(FLG0_WATER_IN_MOVE) && !dComIfGp_checkPlayerStatus1(0, 0x2000000)) {
            return procSwimWaitInit(0);
        }

        if (checkMagneBootsOn()) {
            if (cancelMagneBootsOn()) {
                return procFallInit(1, 5.0f);
            }

            if (mProcID != PROC_BOOTS_EQUIP) {
                procWaitInit();
                m_oldFrameCalc->initOldFrameMorf(3.0f, 0, 35);
                return 1;
            }
        }
    } else if (param_0) {
        onNoResetFlg0(FLG0_EQUIP_HVY_BOOTS);
        mZ2Link.setLinkBootsType(1);

        if (field_0x06e0 != NULL) {
            field_0x06e0->hide();

            if (field_0x06e4 != NULL) {
                field_0x06e4->hide();
            }
        }

        if (checkModeFlg(0x40000)) {
            offNoResetFlg0(FLG0_SWIM_UP);
            mZ2Link.setInWater(true);
        }
    }

    return 0;
}

s16 daAlink_c::getMagneBootsLocalAngleY(s16 param_0, int param_1) {
    if (param_1 == 1 || (param_1 != 2 && field_0x3114 <= 0x4000 && field_0x3114 != -0x8000)) {
        return param_0;
    }

    cXyz vec(cM_ssin(param_0), 0.0f, cM_scos(param_0));
    mDoMtx_stack_c::YrotS(field_0x3116);
    mDoMtx_stack_c::XrotM(-0x8000);
    mDoMtx_stack_c::YrotM(-field_0x3116);
    mDoMtx_stack_c::multVec(&vec, &vec);

    return vec.atan2sX_Z();
}

void daAlink_c::setMagneBootsMtx(cBgS_PolyInfo* i_polyinfo, int param_1) {
    cM3dGPla poly;
    dComIfG_Bgsp().GetTriPla(*i_polyinfo, &poly);

    s16 var_r28 = cM_atan2s(poly.mNormal.absXZ(), poly.mNormal.y);
    s16 var_r26 = field_0x3116;
    int temp_r3_2 = abs(var_r28);
    if (temp_r3_2 < 500) {
        var_r28 = 0;
    } else if (temp_r3_2 > 0x7E0B) {
        var_r28 = -0x8000;
    }

    if (var_r28 != 0 && var_r28 != -0x8000) {
        field_0x3116 = poly.mNormal.atan2sX_Z();
    } else if (param_1) {
        field_0x3116 = 0;
    }

    if (param_1) {
        if (mProcID == PROC_MAGNE_BOOTS_FLY) {
            cLib_addCalcAngleS(&field_0x3114, var_r28, 5, 0x1000, 0x100);
        } else {
            field_0x3114 = var_r28;
        }
    } else {
        // D_MN54 is the E3 Goron Mines Demo map
        if (checkStageName("D_MN54") && getMoveBGActorName(*i_polyinfo, TRUE) == fpcNm_Obj_MagneArm_e) {
            fopAcM_onSwitch(this, 0x93);
        }

        field_0x3114 = var_r28;

        f32 temp_f1 = mMagneBootsTopVec.inprod(poly.mNormal);
        if (!(fabsf(temp_f1) > 0.9999f)) {
            cXyz sp44 = mMagneBootsTopVec.outprod(poly.mNormal);
            sp44.normalizeZP();

            f32 temp_f1_2 = cM_atan2f(JMAFastSqrt(1.0f - temp_f1 * temp_f1), temp_f1);
            f32 var_f2 = temp_f1_2 * 0.2f;

            if (var_f2 > 0.4f) {
                var_f2 = 0.4f;
            } else if (var_f2 < 0.025f) {
                var_f2 = 0.025f;
                if (var_f2 > temp_f1_2) {
                    var_f2 = temp_f1_2;
                }
            }

            cXyz sp40;
            mDoMtx_stack_c::rotAxisRadS(&sp44, var_f2);
            mDoMtx_stack_c::concat(mMagneBootMtx);
            mDoMtx_copy(mDoMtx_stack_c::get(), mMagneBootMtx);
            mDoMtx_multVec(mMagneBootMtx, &cXyz::BaseY, &sp40);

            field_0x3114 = cM_atan2s(sp40.absXZ(), sp40.y);

            temp_r3_2 = abs(field_0x3114);
            if (temp_r3_2 < 500) {
                field_0x3114 = 0;
            } else if (temp_r3_2 > 0x7E0B) {
                field_0x3114 = -0x8000;
            }

            if (field_0x3114 != 0 && field_0x3114 != -0x8000) {
                field_0x3116 = sp40.atan2sX_Z();
            } else {
                field_0x3116 = 0;
            }

            mDoMtx_stack_c::YrotS(field_0x3116);
            mDoMtx_stack_c::XrotM(-field_0x3114);
            mDoMtx_stack_c::YrotM(-field_0x3116);
            mDoMtx_stack_c::concat(mMagneBootMtx);
            mDoMtx_stack_c::multVecSR(&cXyz::BaseZ, &sp40);

            s16 angle = sp40.atan2sX_Z();
            shape_angle.y += angle;
            current.angle.y += angle;
            mPrevAngleY += angle;
            mCameraAngleY += angle;
        }
    }

    mDoMtx_YrotS(mMagneBootMtx, field_0x3116);
    mDoMtx_XrotM(mMagneBootMtx, field_0x3114);
    mDoMtx_YrotM(mMagneBootMtx, -field_0x3116);
    mDoMtx_inverse(mMagneBootMtx, mMagneBootInvMtx);
    mDoMtx_multVec(mMagneBootMtx, &cXyz::BaseY, &mMagneBootsTopVec);
}

int daAlink_c::cancelMagneBootsOn() {
    offNoResetFlg0(FLG0_MAGNE_BOOTS_ON);
    mZ2Link.setMagnetized(false);

    if (field_0x3114 == -0x8000) {
        setOldRootQuaternion(field_0x3114, 0, 0);
    } else {
        s16 var_r31 = shape_angle.y - field_0x3116;
        setOldRootQuaternion(0, var_r31, 0);
        setOldRootQuaternion(field_0x3114, 0, 0);
        setOldRootQuaternion(0, -var_r31, 0);
    }

    field_0x3114 = 0;

    if (!cBgW_CheckBGround(mMagneBootsTopVec.y)) {
        current.pos += mMagneBootsTopVec * 100.0f;
        mLinkGndChk.SetPos(&current.pos);

        f32 ground_y = dComIfG_Bgsp().GroundCross(&mLinkGndChk);
        current.pos.y -= 100.0f;
        if (ground_y > current.pos.y) {
            current.pos.y = ground_y;
        }

        old.pos = current.pos;
        field_0x3798 = current.pos;
        mMagneBootsTopVec = cXyz::BaseY;

        if (checkUpperReadyThrowAnime()) {
            resetUpperAnime(UPPER_2, 3.0f);
        }

        return 1;
    }

    mMagneBootsTopVec = cXyz::BaseY;
    return 0;
}

int daAlink_c::checkMagneBootsFly() {
    if (field_0x2fb8 != 0 && checkEquipHeavyBoots() && !checkDeadHP() &&
        mProcID != PROC_MAGNE_BOOTS_FLY && !checkEventRun())
    {
        cXyz vec(field_0x369c);
        vec.normalizeZP();

        if (mMagneBootsTopVec.getDotProduct(vec) > -0.5f && procMagneBootsFlyInit()) {
            return 1;
        }
    }

    return 0;
}

int daAlink_c::procBootsEquipInit() {
    commonProcInit(PROC_BOOTS_EQUIP);
    setSingleAnimeParam(ANM_HVY_BOOTS_PUT_ON, &mpHIO->mItem.mIronBoots.m.mEquipAnm);

    if (checkUpperReadyThrowAnime()) {
        resetUpperAnime(UPPER_2, -1.0f);
        dComIfGp_clearPlayerStatus0(0, ~0x400030);
    }

    current.angle.y = shape_angle.y;
    mNormalSpeed = 0.0f;

    return 1;
}

int daAlink_c::procBootsEquip() {
    daPy_frameCtrl_c* frameCtrl_p = mUnderFrameCtrl;
    onEndResetFlg0(ERFLG0_UNK_8000000);

    if (checkAnmEnd(frameCtrl_p)) {
        checkNextAction(0);
    } else if (frameCtrl_p->checkPass(10.0f)) {
        setHeavyBoots(1);
    } else if (frameCtrl_p->checkPass(16.0f)) {
        if (checkNoResetFlg0(FLG0_EQUIP_HVY_BOOTS)) {
            dComIfGp_getVibration().StartShock(3, 1, cXyz(0.0f, 1.0f, 0.0f));
        }

        offModeFlg(2);
    } else if (frameCtrl_p->getFrame() > mpHIO->mItem.mIronBoots.m.mEquipAnm.mCancelFrame) {
        onModeFlg(4);
        checkNextAction(1);
    } else if (frameCtrl_p->getFrame() >= 5.0f) {
        mLeftHandIndex = 1;
        mRightHandIndex = 6;
    }

    return 1;
}

int daAlink_c::procMagneBootsFlyInit() {
    cXyz sp44 = field_0x369c.normZP();
    cM3dGPla sp50(&sp44, -sp44.inprod(field_0x3834));
    sp44 = field_0x36b4 + field_0x369c;

    cXyz sp38;
    if (!sp50.crossInfLin(field_0x36b4, sp44, sp38)) {
        return 0;
    }

    sp44 = sp38 + (field_0x369c * 2.0f);

    if (!commonMagneLineCheck(&sp38, &sp44)) {
        return 0;
    }

    if (checkMagneBootsOn()) {
        offNoResetFlg0(FLG0_MAGNE_BOOTS_ON);
        mZ2Link.setMagnetized(true);
    }

    commonProcInit(PROC_MAGNE_BOOTS_FLY);

    dComIfG_Bgsp().GetTriPla(mMagneLineChk, &sp50);
    mPolyInfo2.SetPolyInfo(mMagneLineChk);
    field_0x37c8 = mMagneLineChk.GetCross();

    setSpecialGravity(0.0f, maxFallSpeed, 0);
    speed.y = 0.0f;
    mNormalSpeed = 0.0f;

    setSingleAnimeBase(ANM_MG_KICK);
    if (mEquipItem != 0x103 && !checkBowItem(mEquipItem)) {
        deleteEquipItem(FALSE, TRUE);
    } else {
        offKandelaarModel();
    }

    field_0x3478 = 0.0f;
    mMaxSpeed = mpHIO->mItem.mIronBoots.m.mMaxMagneFlySpeed;
    mProcVar2.field_0x300c = 0;
    voiceStart(Z2SE_AL_V_MAGNET_CAUGHT);
    return 1;
}

int daAlink_c::procMagneBootsFly() {
    field_0x3478 += mpHIO->mItem.mIronBoots.m.mMagneFlyAccelRate;
    if (field_0x3478 > mMaxSpeed) {
        field_0x3478 = mMaxSpeed;
    }

    /* !@bug - If iron boots are unequipped mid-flight, player will still stick to magnetic poly
     * Fixed in GCN and Wii KOR versions by adding a checkEquipHeavyBoots check
     */
    if (dComIfG_Bgsp().ChkPolySafe(mPolyInfo2)
        #if PLATFORM_GCN || VERSION == VERSION_WII_KOR
        && checkEquipHeavyBoots()
        #endif
        )
    {
        if (dComIfG_Bgsp().ChkMoveBG(mPolyInfo2)) {
            cXyz old_37c8(field_0x37c8);
            dComIfG_Bgsp().MoveBgTransPos(mPolyInfo2, true, &field_0x37c8, NULL, NULL);
            current.pos += field_0x37c8 - old_37c8;
        }

        setMagneBootsMtx(&mPolyInfo2, 1);
    } else {
        cancelMagneBootsOn();
        return procFallInit(1, 5.0f);
    }

    cLib_chasePos(&current.pos, field_0x37c8, field_0x3478);
    if (field_0x37c8.abs(current.pos) < 50.0f) {
        current.pos = field_0x37c8;
        onNoResetFlg0(FLG0_MAGNE_BOOTS_ON);
        mZ2Link.setMagnetized(true);
        procLandDamageInit(0);
    } else {
        cXyz sp24;
        mDoMtx_multVecZero(mpLinkModel->getAnmMtx(0), &sp24);
        sp24 = field_0x37c8 - sp24;

        cLib_addCalcAngleS(&shape_angle.x, sp24.atan2sY_XZ() + -0x4000, 5, 0x1000, 0x100);
        cLib_addCalcAngleS(&shape_angle.y, sp24.atan2sX_Z(), 5, 0x1000, 0x100);
        ANGLE_ADD(mProcVar2.field_0x300c, 0x1C00);
    }

    return 1;
}