//
// J2DTextBoxEx
//

#include "JSystem/J2DGraph/J2DTextBoxEx.h"
#include "JSystem/J2DGraph/J2DPrint.h"
#include "JSystem/JSupport/JSURandomInputStream.h"
#include "JSystem/JUtility/JUTResFont.h"

/* 803071E4-8030751C 301B24 0338+00 0/0 1/1 0/0 .text
 * __ct__12J2DTextBoxExFP7J2DPaneP20JSURandomInputStreamUlP11J2DMaterial */
J2DTextBoxEx::J2DTextBoxEx(J2DPane* p_pane, JSURandomInputStream* p_stream, u32 param_2,
                               J2DMaterial* p_material) {
    J2DTextBoxInfo info;

    mVisibilityAnm = NULL;

    int startPos = p_stream->getPosition();

    J2DTbxBlockHeader header;
    p_stream->read(&header, 8);
    mKind = header.mTag;

    int panHeaderPos = p_stream->getPosition();
    J2DTbxBlockHeader panHeader;
    p_stream->peek(&panHeader, 8);
    makePaneExStream(p_pane, p_stream);
    p_stream->seek(panHeaderPos + panHeader.mSize, JSUStreamSeekFrom_SET);

    p_stream->read(&info, 0x20);
    field_0x13c = info.field_0x2;
    field_0x13e = info.mMaterialNum;
    mMaterial = NULL;
    
    if (field_0x13e != 0xFFFF) {
        mMaterial = &p_material[field_0x13e];
        p_material[field_0x13e].field_0x4 = this;
        rewriteAlpha();

        if (mMaterial != NULL) {
            if (mMaterial->getTevBlock() != NULL) {
                mFont = mMaterial->getTevBlock()->getFont();
            }
        }
    }

    mCharSpacing = info.mCharSpace;
    mLineSpacing = info.mLineSpace;
    mFontSizeX = info.mFontSizeX;
    mFontSizeY = info.mFontSizeY;
    mFlags = (info.mHBind << 2) | info.mVBind;
    mCharColor = JUtility::TColor(info.mCharColor);
    mGradientColor = JUtility::TColor(info.mGradColor);
    setConnectParent(info.mConnected);

    u16 strLength = 0;
    if (!(param_2 & 0x2000000)) {
        strLength = info.field_0x1c;
        if ((s16)info.field_0x1c == -1) {
            strLength = info.field_0x1e + 1;
        }
    }

    mStringLength = 0;
    mStringPtr = NULL;

    if (strLength != 0) {
        mStringPtr = new char[strLength];
    }

    if (mStringPtr != NULL) {
        mStringLength = strLength;
        int temp_r0 = (u16)strLength - 1;
        u16 var_r26_2 = info.field_0x1e;

        if (temp_r0 < var_r26_2) {
            var_r26_2 = (u16)temp_r0;
        }

        p_stream->peek(mStringPtr, var_r26_2);
        mStringPtr[var_r26_2] = 0;
    }

    p_stream->skip(info.field_0x1e);
    p_stream->seek(startPos + header.mSize, JSUStreamSeekFrom_SET);

    field_0x10c = 0.0f;
    field_0x110 = 0.0f;
    mTextFontOwned = false;
    field_0x140 = 0;
}

/* 8030751C-803075AC 301E5C 0090+00 1/0 0/0 0/0 .text            __dt__12J2DTextBoxExFv */
J2DTextBoxEx::~J2DTextBoxEx() {
    if (field_0x140 != 0) {
        delete mMaterial;
    }
}

/* 803075AC-803078AC 301EEC 0300+00 1/0 0/0 0/0 .text            drawSelf__12J2DTextBoxExFffPA3_A4_f
 */
void J2DTextBoxEx::drawSelf(f32 param_0, f32 param_1, Mtx* p_mtx) {
    Mtx m;

    JUTFont* font = NULL;
    if (mMaterial != NULL && mMaterial->getTevBlock() != NULL) {
        font = mMaterial->getTevBlock()->getFont();
    }

    J2DPrint print(font, (int)mCharSpacing, (int)mLineSpacing, mCharColor, mGradientColor, mBlackColor,
                   mWhiteColor);
    print.setFontSize((int)mFontSizeX, (int)mFontSizeY);

    if (mMaterial != NULL) {
        mMaterial->setGX();
        MTXConcat(*p_mtx, mGlobalMtx, m);

        GXLoadPosMtxImm(m, GX_PNMTX0);
        GXClearVtxDesc();
        GXSetVtxDesc(GX_VA_POS, GX_DIRECT);
        GXSetVtxDesc(GX_VA_CLR0, GX_DIRECT);
        GXSetVtxDesc(GX_VA_TEX0, GX_DIRECT);

        print.locate(param_0 + mBounds.i.x, param_1 + mBounds.i.y);

        if (mMaterial->isVisible() && mStringPtr != NULL) {
            u8 alpha = 255;

            if (mMaterial->getColorBlock()->getColorChan(1)->getMatSrc() == 1) {
                if (mMaterial->getMaterialAlphaCalc() == 1) {
                    alpha = mColorAlpha;
                }
            } else if (mIsInfluencedAlpha) {
                GXSetChanMatColor(GX_ALPHA0, JUtility::TColor(mColorAlpha));
            }

            print.printReturn(mStringPtr, (int)(mBounds.getWidth() + 0.0001f), (int)mBounds.getHeight(),
                            getHBinding(), getVBinding(), field_0x10c, field_0x110, alpha);
        }
    }
}

/* 803078AC-80307AF0 3021EC 0244+00 1/0 0/0 0/0 .text            draw__12J2DTextBoxExFff */
void J2DTextBoxEx::draw(f32 posX, f32 posY) {
    Mtx m;

    if (isVisible()) {
        JUTFont* font = NULL;
        if (mMaterial != NULL && mMaterial->getTevBlock() != NULL) {
            font = mMaterial->getTevBlock()->getFont();
        }
        
        J2DPrint print(font, mCharSpacing, mLineSpacing, mCharColor, mGradientColor, mBlackColor,
                       mWhiteColor);
        print.setFontSize(mFontSizeX, mFontSizeY);
        mColorAlpha = mAlpha;

        if (mMaterial != NULL) {
            mMaterial->setGX();
            makeMatrix(posX, posY, 0.0f, 0.0f);

            GXLoadPosMtxImm(mPositionMtx, GX_PNMTX0);
            GXSetCurrentMtx(0);

            if (!mMaterial->isVisible()) {
                return;
            }

            GXClearVtxDesc();
            GXSetVtxDesc(GX_VA_POS, GX_DIRECT);
            GXSetVtxDesc(GX_VA_CLR0, GX_DIRECT);
            GXSetVtxDesc(GX_VA_TEX0, GX_DIRECT);

            if (mStringPtr != NULL) {
                u8 alpha = 255;
                if (mMaterial->getMaterialAlphaCalc() == 1) {
                    alpha = mColorAlpha;
                }
                
                print.print(0.0f, 0.0f, alpha, "%s", mStringPtr);
            }

            for (int i = 0; i < 4; i++) {
                GXSetTevSwapModeTable((GXTevSwapSel)i, GX_CH_RED, GX_CH_GREEN, GX_CH_BLUE, GX_CH_ALPHA);
            }

            GXSetNumIndStages(0);

            for (int i = 0; i < 16; i++) {
                GXSetTevDirect((GXTevStageID)i);
            }

            MTXIdentity(m);
            GXLoadPosMtxImm(m, 0);
        }
    }
}

/* 80307AF0-80307D5C 302430 026C+00 1/0 0/0 0/0 .text draw__12J2DTextBoxExFfff18J2DTextBoxHBinding
 */
void J2DTextBoxEx::draw(f32 posX, f32 posY, f32 param_2, J2DTextBoxHBinding hBind) {
    Mtx m;

    if (isVisible()) {
        JUTFont* font = NULL;
        if (mMaterial != NULL && mMaterial->getTevBlock() != NULL) {
            font = mMaterial->getTevBlock()->getFont();
        }
        
        J2DPrint print(font, mCharSpacing, mLineSpacing, mCharColor, mGradientColor, mBlackColor,
                       mWhiteColor);
        print.setFontSize(mFontSizeX, mFontSizeY);
        mColorAlpha = mAlpha;

        if (mMaterial != NULL) {
            mMaterial->setGX();
            makeMatrix(posX, posY, 0.0f, 0.0f);

            GXLoadPosMtxImm(mPositionMtx, GX_PNMTX0);
            GXSetCurrentMtx(0);

            if (!mMaterial->isVisible()) {
                return;
            }

            GXClearVtxDesc();
            GXSetVtxDesc(GX_VA_POS, GX_DIRECT);
            GXSetVtxDesc(GX_VA_CLR0, GX_DIRECT);
            GXSetVtxDesc(GX_VA_TEX0, GX_DIRECT);

            if (mStringPtr != NULL) {
                u8 alpha = 255;
                if (mMaterial->getMaterialAlphaCalc() == 1) {
                    alpha = mColorAlpha;
                }
                
                print.printReturn(mStringPtr, param_2, 0.0f, hBind, VBIND_TOP, 0.0f, -mFontSizeY,
                                alpha);
            }

            for (int i = 0; i < 4; i++) {
                GXSetTevSwapModeTable((GXTevSwapSel)i, GX_CH_RED, GX_CH_GREEN, GX_CH_BLUE, GX_CH_ALPHA);
            }

            GXSetNumIndStages(0);

            for (int i = 0; i < 16; i++) {
                GXSetTevDirect((GXTevStageID)i);
            }

            MTXIdentity(m);
            GXLoadPosMtxImm(m, 0);
        }
    }
}

/* 80307D5C-80307DC0 30269C 0064+00 1/0 0/0 0/0 .text            setFont__12J2DTextBoxExFP7JUTFont
 */
void J2DTextBoxEx::setFont(JUTFont* p_font) {
    if (p_font != NULL && mMaterial != NULL && mMaterial->getTevBlock() != NULL) {
        mMaterial->getTevBlock()->setFont(p_font);
        mFont = p_font;
    }
}

/* 80307DC0-80307E0C 302700 004C+00 1/0 0/0 0/0 .text            getFont__12J2DTextBoxExCFv */
JUTFont* J2DTextBoxEx::getFont() const {
    if (mMaterial != NULL && mMaterial->getTevBlock() != NULL) {
        return mMaterial->getTevBlock()->getFont();
    }

    return NULL;
}

/* 80307E0C-80307EF0 30274C 00E4+00 1/1 0/0 0/0 .text            setTevOrder__12J2DTextBoxExFb */
void J2DTextBoxEx::setTevOrder(bool param_0) {
    u16 local_18[2];
    if (!param_0) {
        local_18[0] = 4;
        local_18[1] = 0xffff;
    } else {
        local_18[0] = 0xff;
        local_18[1] = 0xff04;
    }
    for (u8 i = 0; i < 2; i++) {
        if (mMaterial->getTevBlock()->getMaxStage() > i) {
            J2DTevOrderInfo info;
            info.mTexCoord = (local_18[i]) >> 8;
            info.mTexMap = (local_18[i]) >> 8;
            info.mColor = local_18[i] & 0xff;
            J2DTevOrder tevOrder = info;
            mMaterial->getTevBlock()->setTevOrder(i, tevOrder);
        }
    }
}

/* 80307EF0-80307F94 302830 00A4+00 1/1 0/0 0/0 .text            setTevStage__12J2DTextBoxExFb */
void J2DTextBoxEx::setTevStage(bool param_0) {
    J2DTevStage* stage = mMaterial->getTevBlock()->getTevStage(0);

    if (!param_0) {
        setStage(stage, STAGE_0);
    } else {
        setStage(stage, STAGE_1);
        setStage(mMaterial->getTevBlock()->getTevStage(1), STAGE_2);
    }
}

/* 80307F94-8030823C 3028D4 02A8+00 1/1 0/0 0/0 .text
 * setStage__12J2DTextBoxExFP11J2DTevStageQ212J2DTextBoxEx10stage_enum */
void J2DTextBoxEx::setStage(J2DTevStage* param_0, J2DTextBoxEx::stage_enum param_1) {
    const u8 tevColors[3][4] = {
        {0x0F, 0x08, 0x0A, 0x0F}, {0x02, 0x04, 0x08, 0x0F}, {0x0F, 0x0A, 0x00, 0x0F},
    };
    const u8 tevAlpha[3][4] = {
        {0x07, 0x04, 0x05, 0x07}, {0x01, 0x02, 0x04, 0x07}, {0x07, 0x05, 0x00, 0x07},
    };

    const u8 tevColorOps[3][5] = {
        {0x00, 0x00, 0x00, 0x01, 0x00},
        {0x00, 0x00, 0x00, 0x01, 0x00},
        {0x00, 0x00, 0x00, 0x01, 0x00},
    };

    const u8 tevAlphaOps[3][5] = {
        {0x00, 0x00, 0x00, 0x01, 0x00},
        {0x00, 0x00, 0x00, 0x01, 0x00},
        {0x00, 0x00, 0x00, 0x01, 0x00},
    };

    param_0->setTevColorAB(tevColors[param_1][0], tevColors[param_1][1]);
    param_0->setTevColorCD(tevColors[param_1][2], tevColors[param_1][3]);
    param_0->setTevColorOp(
        tevColorOps[param_1][0], tevColorOps[param_1][1], tevColorOps[param_1][2], tevColorOps[param_1][3], tevColorOps[param_1][4]);
    param_0->setAlphaABCD(
        tevAlpha[param_1][0], tevAlpha[param_1][1], tevAlpha[param_1][2], tevAlpha[param_1][3]);
    param_0->setTevAlphaOp(
        tevAlphaOps[param_1][0], tevAlphaOps[param_1][1], tevAlphaOps[param_1][2], tevAlphaOps[param_1][3], tevAlphaOps[param_1][4]);
}

/* 8030823C-803082C4 302B7C 0088+00 1/0 0/0 0/0 .text setBlack__12J2DTextBoxExFQ28JUtility6TColor
 */
bool J2DTextBoxEx::setBlack(JUtility::TColor black) {
    JUtility::TColor tevBlack;
    JUtility::TColor tevWhite;

    if (!getBlackWhite(&tevBlack, &tevWhite)) {
        return false;
    } else {
        return setBlackWhite(black, tevWhite);
    }
}

/* 803082C4-8030834C 302C04 0088+00 1/0 0/0 0/0 .text setWhite__12J2DTextBoxExFQ28JUtility6TColor
 */
bool J2DTextBoxEx::setWhite(JUtility::TColor white) {
    JUtility::TColor tevBlack;
    JUtility::TColor tevWhite;

    if (!getBlackWhite(&tevBlack, &tevWhite)) {
        return false;
    } else {
        return setBlackWhite(tevBlack, white);
    }
}

/* 8030834C-803084CC 302C8C 0180+00 1/0 0/0 0/0 .text
 * setBlackWhite__12J2DTextBoxExFQ28JUtility6TColorQ28JUtility6TColor */
bool J2DTextBoxEx::setBlackWhite(JUtility::TColor param_0, JUtility::TColor param_1) {
    if (mMaterial == NULL) {
        return false;
    }

    if (mMaterial->getTevBlock() == NULL) {
        return false;
    }

    if (!isSetBlackWhite(param_0, param_1)) {
        return false;
    }

    bool bvar = false;
    if (param_0 != 0 || param_1 != -1) {
        bvar = true;
    }

    mMaterial->getTevBlock()->setTevStageNum(bvar ? 2 : 1);
    setTevOrder(bvar);
    setTevStage(bvar);

    if (bvar) {
        J2DGXColorS10 color;
        color.r = param_0.r;
        color.g = param_0.g;
        color.b = param_0.b;
        color.a = param_0.a;
        mMaterial->getTevBlock()->setTevColor(0, color);

        color.r = param_1.r;
        color.g = param_1.g;
        color.b = param_1.b;
        color.a = param_1.a;
        mMaterial->getTevBlock()->setTevColor(1, color); 
    }

    return true;
}

/* 803084CC-80308668 302E0C 019C+00 4/4 0/0 0/0 .text
 * getBlackWhite__12J2DTextBoxExCFPQ28JUtility6TColorPQ28JUtility6TColor */
bool J2DTextBoxEx::getBlackWhite(JUtility::TColor* param_0, JUtility::TColor* param_1) const {
    if (mMaterial == NULL) {
        return false;
    }

    if (mMaterial->getTevBlock() == NULL) {
        return false;
    }

    bool tevStageNum = mMaterial->getTevBlock()->getTevStageNum() != 1;
    *param_0 = JUtility::TColor(0);
    *param_1 = JUtility::TColor(0xffffffff);
    if (tevStageNum) {
        J2DGXColorS10* local_30 = mMaterial->getTevBlock()->getTevColor(0);
        s16 color0r = local_30->r;
        s16 color0g = local_30->g;
        s16 color0b = local_30->b;
        s16 color0a = local_30->a;
        J2DGXColorS10* local_38 = mMaterial->getTevBlock()->getTevColor(1);
        s16 color1r = local_38->r;
        s16 color1g = local_38->g;
        s16 color1b = local_38->b;
        s16 color1a = local_38->a;
        *param_0 = JUtility::TColor(
            (((u8)color0r) << 24) | (((u8)color0g) << 16) | (((u8)color0b) << 8) |
                ((u8)color0a));
        *param_1 = JUtility::TColor(
            (((u8)color1r) << 24) | (((u8)color1g) << 16) | (((u8)color1b) << 8) |
                ((u8)color1a));
    }
    return true;
}

/* 80308668-803086FC 302FA8 0094+00 1/1 0/0 0/0 .text
 * isSetBlackWhite__12J2DTextBoxExCFQ28JUtility6TColorQ28JUtility6TColor */
bool J2DTextBoxEx::isSetBlackWhite(JUtility::TColor param_0, JUtility::TColor param_1) const {
    if ((u32)param_0 == 0 && (u32)param_1 == 0xffffffff) {
        return 1;
    }
    mMaterial->getTevBlock()->getTevStageNum();
    if (mMaterial->getTevBlock()->getMaxStage() == 1) {
        return 0;
    } 
    return 1;
}

/* 803086FC-8030875C 30303C 0060+00 1/0 0/0 0/0 .text            getBlack__12J2DTextBoxExCFv */
JUtility::TColor J2DTextBoxEx::getBlack() const {
    JUtility::TColor black;
    JUtility::TColor white;
    if (getBlackWhite(&black, &white) == 0) {
        return JUtility::TColor(0);
    }
    return black;
}

/* 8030875C-803087BC 30309C 0060+00 1/0 0/0 0/0 .text            getWhite__12J2DTextBoxExCFv */
JUtility::TColor J2DTextBoxEx::getWhite() const {
    JUtility::TColor black;
    JUtility::TColor white;
    if (getBlackWhite(&black, &white) == 0) {
        return JUtility::TColor(0xffffffff);
    }
    return white;
}

/* 803087BC-803087DC 3030FC 0020+00 1/0 0/0 0/0 .text            setAlpha__12J2DTextBoxExFUc */
void J2DTextBoxEx::setAlpha(u8 alpha) {
    mAlpha = alpha;

    if (mMaterial == NULL) {
        return;
    }

    if (mMaterial->getColorBlock()->getMatColor(0) == NULL) {
        return;
    }

    mMaterial->getColorBlock()->getMatColor(0)->a = alpha;
}

/* 803087DC-80308810 30311C 0034+00 1/0 0/0 0/0 .text setCullBack__12J2DTextBoxExF11_GXCullMode */
void J2DTextBoxEx::setCullBack(GXCullMode mode) {
    mCullMode = mode;

    if (mMaterial != NULL) {
        mMaterial->getColorBlock()->setCullMode(mode);
    }

    J2DPane::setCullBack(mode);
}

/* 80308810-80308828 303150 0018+00 1/0 0/0 0/0 .text            rewriteAlpha__12J2DTextBoxExFv */
void J2DTextBoxEx::rewriteAlpha() {
    if (mMaterial == NULL) {
        return;
    }

    mAlpha = mMaterial->getColorBlock()->getMatColor(0)->a;
}

/* 80308828-803088B4 303168 008C+00 1/0 0/0 0/0 .text            isUsed__12J2DTextBoxExFPC7ResFONT
 */
bool J2DTextBoxEx::isUsed(ResFONT const* p_font) {
    if (getFont() != NULL && getFont()->getResFont() == p_font) {
        return true;
    }

    return J2DPane::isUsed(p_font);
}

/* 803088B4-803088E0 3031F4 002C+00 1/0 0/0 0/0 .text setAnimation__12J2DTextBoxExFP11J2DAnmColor
 */
void J2DTextBoxEx::setAnimation(J2DAnmColor* anm) {
    if (mMaterial != NULL) {
        mMaterial->setAnimation(anm);
    }
}

/* 803088E0-8030890C 303220 002C+00 1/0 0/0 0/0 .text
 * setAnimation__12J2DTextBoxExFP19J2DAnmTextureSRTKey          */
void J2DTextBoxEx::setAnimation(J2DAnmTextureSRTKey* anm) {
    if (mMaterial != NULL) {
        mMaterial->setAnimation(anm);
    }
}

/* 8030890C-80308938 30324C 002C+00 1/0 0/0 0/0 .text
 * setAnimation__12J2DTextBoxExFP16J2DAnmTexPattern             */
void J2DTextBoxEx::setAnimation(J2DAnmTexPattern* anm) {
    if (mMaterial != NULL) {
        mMaterial->setAnimation(anm);
    }
}

/* 80308938-80308964 303278 002C+00 1/0 0/0 0/0 .text
 * setAnimation__12J2DTextBoxExFP15J2DAnmTevRegKey              */
void J2DTextBoxEx::setAnimation(J2DAnmTevRegKey* anm) {
    if (mMaterial != NULL) {
        mMaterial->setAnimation(anm);
    }
}

/* 80308964-8030896C -00001 0008+00 0/0 0/0 0/0 .text
 * setAnimation__12J2DTextBoxExFP20J2DAnmVisibilityFull         */
void J2DTextBoxEx::setAnimation(J2DAnmVisibilityFull* anm) {
    mVisibilityAnm = anm;
}

/* 8030896C-803089EC 3032AC 0080+00 1/0 0/0 0/0 .text
 * animationPane__12J2DTextBoxExFPC15J2DAnmTransform            */
const J2DAnmTransform* J2DTextBoxEx::animationPane(J2DAnmTransform const* param_0) {
    if (mVisibilityAnm != NULL && field_0x13c != 0xffff) {
        u8 visibility;
        mVisibilityAnm->getVisibility(field_0x13c, &visibility);
        if (visibility != 0) {
            show();
        } else {
            hide();
        }
    }
    return J2DPane::animationPane(param_0);
}

/* 803089EC-80308A28 30332C 003C+00 1/0 0/0 0/0 .text            setCullBack__12J2DTextBoxExFb */
void J2DTextBoxEx::setCullBack(bool param_0) {
    setCullBack(param_0 ? GX_CULL_BACK : GX_CULL_NONE);
}

/* 80308A28-80308A48 303368 0020+00 1/0 0/0 0/0 .text            isUsed__12J2DTextBoxExFPC7ResTIMG
 */
bool J2DTextBoxEx::isUsed(ResTIMG const* p_timg) {
    return J2DPane::isUsed(p_timg);
}

/* 80308A48-80308A4C 303388 0004+00 1/0 0/0 0/0 .text
 * setAnimation__12J2DTextBoxExFP14J2DAnmVtxColor               */
void J2DTextBoxEx::setAnimation(J2DAnmVtxColor* param_0) {
    /* empty function */
}

/* 80308A4C-80308A6C 30338C 0020+00 1/0 0/0 0/0 .text setAnimation__12J2DTextBoxExFP10J2DAnmBase
 */
void J2DTextBoxEx::setAnimation(J2DAnmBase* anm) {
    J2DPane::setAnimation(anm);
}

/* 803A1DF0-803A1DF0 02E450 0000+00 0/0 0/0 0/0 .rodata          @stringBase0 */