#include "Engine/Engine.h"
#include "Engine/Party.h"
#include "Engine/LOD.h"
#include "Engine/Timer.h"
#include "Engine/texts.h"
#include "Engine/Graphics/Render.h"
#include "Engine/Graphics/Viewport.h"

#include "IO/Mouse.h"

#include "GUI/UI/Spellbook.h"
#include "GUI/GUIFont.h"

#include "Media/Audio/AudioPlayer.h"



void InitializeSpellBookTextures();
void OnCloseSpellBookPage();
void OnCloseSpellBook();
void LoadSpellbook(unsigned int spell_school);
void BookUI_Spellbook_DrawCurrentSchoolBackground();


std::array<char *, 9> spellbook_texture_filename_suffices = {{"f", "a", "w", "e", "s", "m", "b", "l", "d"}};

std::array<std::array<unsigned char, 12>, 9> pSpellbookSpellIndices = // 4E2430    from pSpellbookSpellIndices[9][12]
{
    {  //0   1   2   3  4    5   6  7    8  9   10  11
        { 0, 3, 1, 8, 11, 7, 4, 10, 6, 2, 5, 9 },
        { 0, 11, 2, 9, 6, 8, 5, 10, 3, 7, 1, 4 },
        { 0, 4, 8, 9, 1, 10, 3, 11, 7, 6, 2, 5 },
        { 0, 7, 10, 8, 2, 11, 1, 5, 3, 6, 4, 9 },
        { 0, 5, 10, 11, 7, 2, 8, 1, 4, 9, 3, 6 },
        { 0, 5, 9, 8, 3, 7, 6, 4, 1, 11, 2, 10 },
        { 0, 1, 6, 9, 3, 5, 8, 11, 7, 10, 4, 2 },
        { 0, 1, 10, 11, 9, 4, 3, 6, 5, 7, 8, 2 },
        { 0, 9, 3, 7, 1, 5, 2, 10, 11, 8, 6, 4 }
    }
};




GUIWindow_Spellbook::GUIWindow_Spellbook() :
    GUIWindow(0, 0, window->GetWidth(), window->GetHeight(), 0, nullptr)
{
// ------------------------------------
// 004304E7 void Game_EventLoop -- part
    current_screen_type = SCREEN_SPELL_BOOK;
    pEventTimer->Pause();

    InitializeSpellBookTextures();
    OpenSpellbook();

// ------------------------------------
// 004304E7 void Game_EventLoop -- part
    pAudioPlayer->PlaySound(SOUND_48, 0, 0, -1, 0, 0, 0, 0);
    viewparams->field_48 = 1;
}


void GUIWindow_Spellbook::OpenSpellbookPage(int page)
{
// ------------------------------------
// 004304E7 void Game_EventLoop -- part

    OnCloseSpellBookPage();
    pPlayers[uActiveCharacter]->lastOpenedSpellbookPage = page;
    OpenSpellbook();
    pAudioPlayer->PlaySound((SoundID)(SOUND_TurnPageU + rand() % 2), 0, 0, -1, 0, 0, 0, 0);
}

//----- (00411621) --------------------------------------------------------
void GUIWindow_Spellbook::OpenSpellbook()
{
    Player *pPlayer; // edi@1
    //GUIWindow *pWindow; // esi@1
    //unsigned int v3; // ebp@1
    int v4; // eax@3
    ///GUIButton *result; // eax@25
    int a2; // [sp+10h] [bp-8h]@1
    //int v7; // [sp+14h] [bp-4h]@1

    pPlayer = pPlayers[uActiveCharacter];
    //pWindow = this;
    LoadSpellbook(pPlayer->lastOpenedSpellbookPage);
    //v3 = 0;
    a2 = 0;

    PlayerSpellbookChapter* chapter = &pPlayer->spellbook.pChapters[pPlayer->lastOpenedSpellbookPage];
    for (uint i = 0; i < 11; ++i)
    {
        if (!chapter->bIsSpellAvailable[i])
            continue;
        v4 = pPlayer->lastOpenedSpellbookPage;
        //v4 = (12 * pPlayer->lastOpenedSpellbookPage + pSpellbookSpellIndices[pPlayer->lastOpenedSpellbookPage][i + 1]);
        CreateButton(pViewport->uViewportTL_X + pIconPos[v4][pSpellbookSpellIndices[v4][i + 1]].Xpos,
            pViewport->uViewportTL_Y + pIconPos[v4][pSpellbookSpellIndices[v4][i + 1]].Ypos,  //dword_4E20D0
            SBPageSSpellsTextureList[i + 1]->uTextureWidth,
            SBPageSSpellsTextureList[i + 1]->uTextureHeight,
            1, 79, UIMSG_SelectSpell, i, 0, "", 0);
        ++a2;
        //++v3;
    }
    //while ( (signed int)v3 < 11 );

    CreateButton(0, 0, 0, 0, 1, 0, UIMSG_SpellBook_PressTab, 0, '\t', "", 0);
    if (a2)
        _41D08F_set_keyboard_control_group(a2, 0, 0, 0);

    if (pPlayer->pActiveSkills[PLAYER_SKILL_FIRE])   CreateButton(399, 10, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 0, 0, aSpellSchoolNames[0], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_AIR])    CreateButton(399, 46, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 1, 0, aSpellSchoolNames[1], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_WATER])  CreateButton(399, 83, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 2, 0, aSpellSchoolNames[2], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_EARTH])  CreateButton(399, 121, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 3, 0, aSpellSchoolNames[3], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_SPIRIT]) CreateButton(399, 158, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 4, 0, aSpellSchoolNames[4], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_MIND])   CreateButton(400, 196, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 5, 0, aSpellSchoolNames[5], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_BODY])   CreateButton(400, 234, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 6, 0, aSpellSchoolNames[6], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_LIGHT])  CreateButton(400, 271, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 7, 0, aSpellSchoolNames[7], 0);
    if (pPlayer->pActiveSkills[PLAYER_SKILL_DARK])   CreateButton(400, 307, 50, 36, 1, 0, UIMSG_OpenSpellbookPage, 8, 0, aSpellSchoolNames[8], 0);

    CreateButton(476, 450, pSBClickQuickSpellBtnTextr->uTextureWidth, pSBClickQuickSpellBtnTextr->uTextureHeight, 1, 78, UIMSG_ClickInstallRemoveQuickSpellBtn, 0, 0, "", 0);
    pBtn_InstallRemoveSpell = CreateButton(476, 450, 48, 32, 1, 78, UIMSG_ClickInstallRemoveQuickSpellBtn, 0, 0, "", pSBClickQuickSpellBtnTextr, 0);
    CreateButton(561, 450, pSpellBookClickCloseBtnTextr->uTextureWidth, pSpellBookClickCloseBtnTextr->uTextureHeight, 1, 0, UIMSG_Escape, 0, 0, pGlobalTXT_LocalizationStrings[79], 0);
    pBtn_CloseBook = CreateButton(561, 450, 48, 32, 1, 0, UIMSG_Escape, 0, 0, pGlobalTXT_LocalizationStrings[79], pSpellBookClickCloseBtnTextr, 0);
}

void GUIWindow_Spellbook::Update()
{
// -----------------------------------
// 004156F0 GUI_UpdateWindows --- part
// {
//     DrawSpellBookContent(pPlayers[uActiveCharacter]);
// }
//
// --------------------------------------------------
// 00412B58 void DrawSpellBookContent(Player *player)
    auto player = pPlayers[uActiveCharacter];

    Texture *PendingTexture; // edi@1
    Texture *pTexture; // edx@5
    int v10; // eax@13
    unsigned int pX_coord; // esi@18
    unsigned int pY_coord; // edi@18
    Texture *pPageTexture; // eax@21
    //  signed int v22; // [sp-4h] [bp-24h]@22
    POINT a2; // [sp+18h] [bp-8h]@13

    static unsigned int texture_tab_coord1[9][2] =
    { { 406, 9 }, { 406, 46 }, { 406, 84 }, { 406, 121 }, { 407, 158 }, { 405, 196 }, { 405, 234 }, { 405, 272 }, { 405, 309 } };

    static unsigned int texture_tab_coord0[9][2] =
    { { 415, 10 }, { 415, 46 }, { 415, 83 }, { 415, 121 }, { 415, 158 }, { 416, 196 }, { 416, 234 }, { 416, 271 }, { 416, 307 } };

    BookUI_Spellbook_DrawCurrentSchoolBackground();

    PendingTexture = pIcons_LOD->GetTexture(pIcons_LOD->FindTextureByName("Pending"));
    pRenderer->ClearZBuffer(0, 479);
    if ((11 * player->lastOpenedSpellbookPage) || ((11 * player->lastOpenedSpellbookPage) + 11))//??? maybe structure need fix
    {
        for (uint i = 1; i <= 11; ++i)
        {
            if (player->_achieved_awards_bits[(11 * player->lastOpenedSpellbookPage) + i + 63])
            {
                if (SBPageSSpellsTextureList[i] != PendingTexture)
                {
                    if (quick_spell_at_page == i)
                        pTexture = SBPageCSpellsTextureList[i];
                    else
                        pTexture = SBPageSSpellsTextureList[i];
                    if (pTexture->pLevelOfDetail0_prolly_alpha_mask)
                    {
                        pX_coord = pViewport->uViewportTL_X + pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][i]].Xpos;
                        pY_coord = pViewport->uViewportTL_Y + pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][i]].Ypos;
                        if (pTexture->pBits & 0x200)
                            pRenderer->DrawTextureIndexedAlpha(pX_coord, pY_coord, pTexture);
                        else
                            pRenderer->DrawTextureIndexed(pX_coord, pY_coord, pTexture);
                        pRenderer->DrawMaskToZBuffer(pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][i]].Xpos,
                            pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][i]].Ypos, pTexture, i);
                    }
                }
            }
        }
    }

    pMouse->GetCursorPos(&a2);
    v10 = pRenderer->pActiveZBuffer[a2.x + pSRZBufferLineOffsets[a2.y]] & 0xFFFF;
    if (v10)
    {
        if (SBPageCSpellsTextureList[v10]->pLevelOfDetail0_prolly_alpha_mask)
        {
            pX_coord = pViewport->uViewportTL_X + pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][v10]].Xpos;
            pY_coord = pViewport->uViewportTL_Y + pIconPos[player->lastOpenedSpellbookPage][pSpellbookSpellIndices[player->lastOpenedSpellbookPage][v10]].Ypos;
            if (SBPageCSpellsTextureList[v10]->pBits & 0x200)
                pRenderer->DrawTextureIndexedAlpha(pX_coord, pY_coord, SBPageCSpellsTextureList[v10]);
            else
                pRenderer->DrawTextureIndexed(pX_coord, pY_coord, SBPageCSpellsTextureList[v10]);
        }
    }
    //pX_coord = (unsigned int)&player->pActiveSkills[PLAYER_SKILL_FIRE];
    //pY_coord = (unsigned int)&player->pActiveSkills[PLAYER_SKILL_FIRE];
    for (uint i = 0; i < 9; i++)
    {
        if (player->pActiveSkills[PLAYER_SKILL_FIRE + i])
        {
            if (player->lastOpenedSpellbookPage == i)
            {
                pPageTexture = pTextures_tabs[i][1];
                pX_coord = texture_tab_coord1[i][0];
                pY_coord = texture_tab_coord1[i][1];
            }
            else
            {
                pPageTexture = pTextures_tabs[i][0];
                pX_coord = texture_tab_coord0[i][0];
                pY_coord = texture_tab_coord0[i][1];
            }
            pRenderer->DrawTextureIndexedAlpha(pX_coord, pY_coord, pPageTexture);
        }
    }
}


void GUIWindow_Spellbook::Release()
{
// -----------------------------------------
// 0041C26A void GUIWindow::Release --- part
    OnCloseSpellBookPage();
    OnCloseSpellBook();

    GUIWindow::Release();
}


//----- (00411300) --------------------------------------------------------
void LoadSpellbook(unsigned int spell_school)
{
  char pContainer[20]; // [sp+Ch] [bp-1Ch]@7

  byte_506550 = 0;
  if ( pPlayers[uActiveCharacter]->uQuickSpell && (unsigned __int8)pPlayers[uActiveCharacter]->uQuickSpell / 11 == spell_school )
    quick_spell_at_page = (unsigned __int8)pPlayers[uActiveCharacter]->uQuickSpell - 11 * spell_school;
  else
    quick_spell_at_page = 0;

  for (uint i = 1; i <= 11; ++i)
  {
    if (pPlayers[uActiveCharacter]->spellbook.pChapters[spell_school].bIsSpellAvailable[i - 1])
    {
      sprintf(pContainer, "SB%sS%02d", spellbook_texture_filename_suffices[spell_school], pSpellbookSpellIndices[spell_school][i]);
      SBPageSSpellsTextureList[i] = pIcons_LOD->LoadTexturePtr(pContainer, TEXTURE_16BIT_PALETTE);

      sprintf(pContainer, "SB%sC%02d", spellbook_texture_filename_suffices[spell_school], pSpellbookSpellIndices[spell_school][i]);
      SBPageCSpellsTextureList[i] = pIcons_LOD->LoadTexturePtr(pContainer, TEXTURE_16BIT_PALETTE);
    }
  }
}

//----- (00412AF9) --------------------------------------------------------
static void BookUI_Spellbook_DrawCurrentSchoolBackground()
{
  int pTexID = 0;
  if ( uActiveCharacter )
    pTexID = pParty->pPlayers[uActiveCharacter - 1].lastOpenedSpellbookPage;
  pRenderer->DrawTextureIndexed(8, 8, pSpellBookPagesTextr[pTexID]);
  pRenderer->DrawTextureIndexed(476, 450, pSBQuickSpellBtnTextr);
  pRenderer->DrawTextureIndexed(561, 450, pSpellBookCloseBtnTextr);
}





//----- (0041192C) --------------------------------------------------------
void InitializeSpellBookTextures()
{
    pAudioPlayer->StopChannels(-1, -1);
    ++pIcons_LOD->uTexturePacksCount;
    if (!pIcons_LOD->uNumPrevLoadedFiles)
        pIcons_LOD->uNumPrevLoadedFiles = pIcons_LOD->uNumLoadedFiles;
    pAudioPlayer->PlaySound(SOUND_openbook, 0, 0, -1, 0, 0, 0, 0);
    pSpellBookPagesTextr_9 = pIcons_LOD->LoadTexturePtr("book", TEXTURE_16BIT_PALETTE);
    pTexture_pagemask = pIcons_LOD->LoadTexturePtr("pagemask", TEXTURE_16BIT_PALETTE);
    pSpellBookCloseBtnTextr = pIcons_LOD->LoadTexturePtr("ib-m5-u", TEXTURE_16BIT_PALETTE);
    pSpellBookClickCloseBtnTextr = pIcons_LOD->LoadTexturePtr("ib-m5-d", TEXTURE_16BIT_PALETTE);
    pSBQuickSpellBtnTextr = pIcons_LOD->LoadTexturePtr("ib-m6-u", TEXTURE_16BIT_PALETTE);

    static const char *texNames[9] = // 004E24EC
    {
        "SBFB00", "SBAB00", "SBWB00", "SBEB00",
        "SBSB00", "SBMB00", "SBBB00", "SBLB00", "SBDB00"
    };

    pSBClickQuickSpellBtnTextr = pIcons_LOD->LoadTexturePtr("ib-m6-d", TEXTURE_16BIT_PALETTE);
    for (uint i = 0; i < 9; ++i)
    {
        pSpellBookPagesTextr[i] = pIcons_LOD->LoadTexturePtr(texNames[i], TEXTURE_16BIT_PALETTE);
        sprintf(pTmpBuf.data(), "tab%da", i + 1);
        pTextures_tabs[i][0] = pIcons_LOD->LoadTexturePtr(pTmpBuf.data(), TEXTURE_16BIT_PALETTE);
        sprintf(pTmpBuf.data(), "tab%db", i + 1);
        pTextures_tabs[i][1] = pIcons_LOD->LoadTexturePtr(pTmpBuf.data(), TEXTURE_16BIT_PALETTE);
    }
}


//----- (00411473) --------------------------------------------------------
void OnCloseSpellBook()
{
    pTexture_pagemask->Release();
    pSpellBookCloseBtnTextr->Release();
    pSBQuickSpellBtnTextr->Release();
    for (uint i = 0; i < 9; ++i)
    {
        pSpellBookPagesTextr[i]->Release();
        pTextures_tabs[i][0]->Release();
        pTextures_tabs[i][1]->Release();
    }
    pAudioPlayer->PlaySound(SOUND_openbook, 0, 0, -1, 0, 0, 0, 0);
    pIcons_LOD->RemoveTexturesPackFromTextureList();
}