#include <assert.h>

#include "MM7.h"

#include "MapInfo.h"
#include "Game.h"
#include "Player.h"
#include "GUIWindow.h"
#include "GUIFont.h"
#include "GUIProgressBar.h"
#include "Party.h"
#include "AudioPlayer.h"
#include "Outdoor.h"
#include "IndoorCamera.h"
#include "Overlays.h"
#include "Monsters.h"
#include "Arcomage.h"
#include "LOD.h"
#include "Actor.h"
#include "Allocator.h"
#include "Events.h"
#include "Viewport.h"
#include "FrameTableInc.h"
#include "Math.h"
#include "SpriteObject.h"
#include "ObjectList.h"
#include "Chest.h"
#include "PaletteManager.h"
#include "DecorationList.h"
#include "SaveLoad.h"
#include "stru123.h"
#include "Time.h"
#include "IconFrameTable.h"
#include "Awards.h"
#include "Autonotes.h"
#include "stru160.h"
#include "TurnEngine.h"
#include "CastSpellInfo.h"
#include "Weather.h"
#include "stru298.h"
#include "StorylineTextTable.h"
#include "Events2D.h"
#include "texts.h"

#include "mm7_data.h"



//----- (0041F6C1) --------------------------------------------------------
void __cdecl RestUI_Initialize()
{
    Player *v0; // eax@10

    if ( !dword_506F14 )
        pAudioPlayer->StopChannels(-1, -1);
    if ( pCurrentScreen )
        {
        pGUIWindow_CurrentMenu->Release();
        pCurrentScreen = SCREEN_GAME;
        viewparams->bRedrawGameUI = 1;
        }
    pEventTimer->Pause();
    if ( dword_506F14 != 2 )
        GUIWindow::Create(518, 450, 0, 0, WINDOW_PressedButton2, (int)pBtn_Rest, 0);
    _506F18_num_hours_to_sleep = 0;
    dword_506F14 = 0;
    uRestUI_FoodRequiredToRest = 2;
    if ( uCurrentlyLoadedLevelType == LEVEL_Outdoor)
        uRestUI_FoodRequiredToRest = pOutdoor->GetNumFoodRequiredToRestInCurrentPos(pParty->vPosition.x, pParty->vPosition.y, pParty->vPosition.z);
    if ( PartyHasDragon() )
        {
        v0 = pParty->pPlayers;//[0].uClass;
        while ( v0 <= &pParty->pPlayers[3] )
            {
            ++v0;
            if ( v0 > &pParty->pPlayers[3] )
                break;
            }
        if(v0->classType == PLAYER_CLASS_WARLOCK)
            ++uRestUI_FoodRequiredToRest;
        }
    if ( CheckHiredNPCSpeciality(Porter) )
        --uRestUI_FoodRequiredToRest;
    if ( CheckHiredNPCSpeciality(QuarterMaster) )
        uRestUI_FoodRequiredToRest -= 2;
    if ( CheckHiredNPCSpeciality(Gypsy) )
        --uRestUI_FoodRequiredToRest;
    if ( uRestUI_FoodRequiredToRest < 1 )
        uRestUI_FoodRequiredToRest = 1;
    if ( !_strcmpi(pCurrentMapName, "d29.blv") && (unsigned __int16)_449B57_test_bit(pParty->_quest_bits, 98) )
        uRestUI_FoodRequiredToRest = 0;
    ++pIcons_LOD->uTexturePacksCount;
    if ( !pIcons_LOD->uNumPrevLoadedFiles )
        pIcons_LOD->uNumPrevLoadedFiles = pIcons_LOD->uNumLoadedFiles;
    pCurrentScreen = SCREEN_REST;
    _507CD4_RestUI_hourglass_anim_controller = 0;
    uTextureID_RestUI_restmain = pIcons_LOD->LoadTexture("restmain", TEXTURE_16BIT_PALETTE);
    uTextureID_RestUI_restb1 = pIcons_LOD->LoadTexture("restb1", TEXTURE_16BIT_PALETTE);
    uTextureID_RestUI_restb2 = pIcons_LOD->LoadTexture("restb2", TEXTURE_16BIT_PALETTE);
    uTextureID_RestUI_restb3 = pIcons_LOD->LoadTexture("restb3", TEXTURE_16BIT_PALETTE);
    uTextureID_RestUI_restb4 = pIcons_LOD->LoadTexture("restb4", TEXTURE_16BIT_PALETTE);
    uTextureID_RestUI_restexit = pIcons_LOD->LoadTexture("restexit", TEXTURE_16BIT_PALETTE);

    LoadActualSkyFrame();

    pGUIWindow_CurrentMenu = GUIWindow::Create(0, 0, 640u, 480u, WINDOW_Rest, 0, 0);
    pButton_RestUI_Exit          = pGUIWindow_CurrentMenu->CreateButton(280, 297, 154, 37, 1, 0, UIMSG_ExitRest,       0, 0,     "", pIcons_LOD->GetTexture(uTextureID_RestUI_restexit), 0);
    pButton_RestUI_Main          = pGUIWindow_CurrentMenu->CreateButton( 24, 154, 225, 37, 1, 0, UIMSG_Rest8Hour,      0, 0x52u, "", pIcons_LOD->GetTexture(uTextureID_RestUI_restb4), 0);
    pButton_RestUI_WaitUntilDawn = pGUIWindow_CurrentMenu->CreateButton( 61, 232, 154, 33, 1, 0, UIMSG_AlreadyResting, 0, 0x44u, "", pIcons_LOD->GetTexture(uTextureID_RestUI_restb1), 0);
    pButton_RestUI_Wait1Hour     = pGUIWindow_CurrentMenu->CreateButton( 61, 264, 154, 33, 1, 0, UIMSG_Wait1Hour,      0, 0x48u, "", pIcons_LOD->GetTexture(uTextureID_RestUI_restb2), 0);
    pButton_RestUI_Wait5Minutes  = pGUIWindow_CurrentMenu->CreateButton( 61, 296, 154, 33, 1, 0, UIMSG_Wait5Minutes,   0, 0x4Du, "", pIcons_LOD->GetTexture(uTextureID_RestUI_restb3), 0);
}

//----- (0041FA01) --------------------------------------------------------
void __cdecl RestUI_Draw()
    {
    int live_characters; // esi@1
    unsigned int v3; // eax@15
    //char v4; // al@17
    bool v5; // eax@21
    GUIButton tmp_button; // [sp+8h] [bp-DCh]@19
    //double v7; // [sp+C4h] [bp-20h]@17
    float v8; // [sp+CCh] [bp-18h]@17
    __int64 v9; // [sp+D0h] [bp-14h]@17
    unsigned int am_pm_hours; // [sp+D8h] [bp-Ch]@9
    __int16 shadow_color; // [sp+DCh] [bp-8h]@1
    int text_color; // [sp+E0h] [bp-4h]@1

    live_characters = 0;
    text_color =   TargetColor(10, 0, 0);
    shadow_color = TargetColor(230, 214, 193);
    for(int i=1; i<5; ++i)
        if ( !pPlayers[i]->pConditions[Player::Condition_Dead] && !pPlayers[i]->pConditions[Player::Condition_Eradicated] && pPlayers[i]->sHealth > 0 )
            ++live_characters;

    if ( live_characters )
        {
        pRenderer->DrawTextureIndexed(8, 8, pIcons_LOD->GetTexture(uTextureID_RestUI_restmain));
        am_pm_hours = pParty->uCurrentHour;
        dword_506F1C = pGUIWindow_CurrentMenu->pCurrentPosActiveItem;
        if ( (signed int)pParty->uCurrentHour <= 12 )
            {
            if ( !am_pm_hours )
                am_pm_hours = 12;
            }
        else
            {
            am_pm_hours -= 12;
            }
        pRenderer->DrawTextureIndexed(16u, 26u, pTexture_RestUI_CurrentSkyFrame);
        if ( pTexture_RestUI_CurrentHourglassFrame )
            {
            pTexture_RestUI_CurrentHourglassFrame->Release();
            pIcons_LOD->SyncLoadedFilesCount();
            }
        v3 = pEventTimer->uTimeElapsed + _507CD4_RestUI_hourglass_anim_controller;
        _507CD4_RestUI_hourglass_anim_controller += pEventTimer->uTimeElapsed;
        if ( (unsigned int)_507CD4_RestUI_hourglass_anim_controller >= 512 )
            {
            v3 = 0;
            _507CD4_RestUI_hourglass_anim_controller = 0;
            }
        v9 = v3;
        v8 = (double)v3 / 512.0 * 120.0;
        //v7 = v8 + 6.7553994e15;
        HIDWORD(v9) = floorf(v8 + 0.5f);//LODWORD(v7);
        hourglass_icon_idx = (int)floorf(v8 + 0.5f) % 256 + 1;//LOBYTE(v7) + 1;
        //hourglass_icon_idx = v4;
        if (hourglass_icon_idx >= 120 )
            hourglass_icon_idx = 1;

        sprintf(pTmpBuf, "hglas%03d", hourglass_icon_idx);
        pTexture_RestUI_CurrentHourglassFrame = pIcons_LOD->LoadTexturePtr(pTmpBuf, TEXTURE_16BIT_PALETTE);
        pRenderer->DrawTextureIndexed(267, 159, pTexture_RestUI_CurrentHourglassFrame);
        memset(&tmp_button, 0, sizeof(GUIButton));
        tmp_button.uX = 24;
        tmp_button.uY = 154;

        tmp_button.uZ = 194;
        tmp_button.uW = 190;

        tmp_button.uWidth = 171;
        tmp_button.uHeight = 37;

        tmp_button.pParent = pButton_RestUI_WaitUntilDawn->pParent;
        tmp_button.DrawLabel(pGlobalTXT_LocalizationStrings[183], pFontCreate, text_color, shadow_color);
        tmp_button.pParent = 0;
        sprintf(pTmpBuf, "\r408%d", uRestUI_FoodRequiredToRest);
        pGUIWindow_CurrentMenu->DrawText(pFontCreate, 0, 164, text_color, pTmpBuf, 0, 0, shadow_color);
        pButton_RestUI_WaitUntilDawn->DrawLabel(pGlobalTXT_LocalizationStrings[237], pFontCreate, text_color, shadow_color);
        pButton_RestUI_Wait1Hour->DrawLabel(pGlobalTXT_LocalizationStrings[239], pFontCreate, text_color, shadow_color);
        pButton_RestUI_Wait5Minutes->DrawLabel(pGlobalTXT_LocalizationStrings[238], pFontCreate, text_color, shadow_color);
        pButton_RestUI_Exit->DrawLabel(pGlobalTXT_LocalizationStrings[81], pFontCreate, text_color, shadow_color);
        memset(&tmp_button, 0, sizeof(GUIButton));
        tmp_button.uX = 45;
        tmp_button.uY = 199;

        tmp_button.uZ = 229;
        tmp_button.uW = 228;

        tmp_button.uWidth = 185;
        tmp_button.uHeight = 30;

        tmp_button.pParent = pButton_RestUI_WaitUntilDawn->pParent;
        tmp_button.DrawLabel(pGlobalTXT_LocalizationStrings[236], pFontCreate, text_color, shadow_color);
        tmp_button.pParent = 0;
        v5 = (pParty->uCurrentHour >= 12 && pParty->uCurrentHour < 24)? 1:0;
        sprintf(pTmpBuf, "%d:%02d %s", am_pm_hours, pParty->uCurrentMinute, aAMPMNames[v5]);
        pGUIWindow_CurrentMenu->DrawText(pFontCreate, 368, 168, text_color, pTmpBuf, 0, 0, shadow_color);
        sprintf(pTmpBuf, "%s\r190%d", pGlobalTXT_LocalizationStrings[56], pParty->uDaysPlayed + 1);
        pGUIWindow_CurrentMenu->DrawText(pFontCreate, 350, 190, text_color, pTmpBuf, 0, 0, shadow_color);
        sprintf(pTmpBuf, "%s\r190%d", pGlobalTXT_LocalizationStrings[146], pParty->uCurrentMonth + 1);
        pGUIWindow_CurrentMenu->DrawText(pFontCreate, 350, 222, text_color, pTmpBuf, 0, 0, shadow_color);
        sprintf(pTmpBuf, "%s\r190%d", pGlobalTXT_LocalizationStrings[245], pParty->uCurrentYear);
        pGUIWindow_CurrentMenu->DrawText(pFontCreate, 350, 254, text_color, pTmpBuf, 0, 0, shadow_color);
        if ( dword_506F14 )
            Sleep6Hours();
        }
    else
        {
        GUIWindow::Create(pButton_RestUI_Exit->uX, pButton_RestUI_Exit->uY, 0, 0, WINDOW_CloseRestWindowBtn,
            (int)pButton_RestUI_Exit, (int)pGlobalTXT_LocalizationStrings[81]);
        }
    }