#include "StatusDisplay.h"
#include <Quickdraw.h>
#include <Windows.h>
#include <string.h>

const short tableTop = 50;
const short tableLineHeight = 20;
const short tableBaseline = 15;

enum class StatusDisplay::Stat : short
{
    heapSize,
    freeMem,
    fileSize,
    transferred,
    speed,
    timeRemaining,
    transmissionErrors
};

StatusDisplay::StatusDisplay()
{
    statusWindow = GetNewWindow(129, NULL, (WindowPtr) -1);

#if TARGET_API_MAC_CARBON
    Rect bounds;
    GetWindowPortBounds(statusWindow, &bounds);
#else
    Rect bounds = statusWindow->portRect;
#endif

    SetRect(&statusRect, 10, 0, bounds.right-10, 30);
    SetRect(&progressRect, 10, 30, bounds.right-10, 46);

    memset(columnWidths, 0, sizeof(columnWidths));
    columnWidths[1] = columnWidths[4] = 50;
    
#if TARGET_API_MAC_CARBON
    SetPortWindowPort(statusWindow);
#else
    SetPort(statusWindow);
#endif
    TextSize(9);
    TextFace(bold);
    for(int i = 0; i < 2 * nValues; i++)
    {
        Str255 str;
        GetIndString(str, 129, i+1);
        short& col = columnWidths[(i&1)*2 + (i&2)*3/2];
        col = std::max(col, StringWidth(str));
    }

    for(int i = 0; i < nValues; i+=2)
    {
        SetRect(&valueRects[i],
                10 + columnWidths[0],
                tableTop + i/2 * tableLineHeight,
                10 + columnWidths[0] + columnWidths[1],
                tableTop + (i/2+1) * tableLineHeight);
    }
    for(int i = 1; i < nValues; i+=2)
    {
        SetRect(&valueRects[i],
                bounds.right - 10 - columnWidths[5] - columnWidths[4],
                tableTop + i/2 * tableLineHeight,
                bounds.right - 10 - columnWidths[5],
                tableTop + (i/2+1) * tableLineHeight);
    }

    RgnHandle tmp = NewRgn();
    background = NewRgn();
    RectRgn(background, &bounds);
    RectRgn(tmp, &progressRect);
    DiffRgn(background, tmp, background);
    for(int i = 0; i < nValues; i++)
    {
        RectRgn(tmp, &valueRects[i]);
        DiffRgn(background, tmp, background);
    }
    DisposeRgn(tmp);
}

StatusDisplay::~StatusDisplay()
{
    DisposeWindow(statusWindow);
    DisposeRgn(background);
}

void StatusDisplay::Inval(const Rect& r)
{
#if TARGET_API_MAC_CARBON
    InvalWindowRect(statusWindow, &r);
#else
    SetPort(statusWindow);
    InvalRect(&r);
#endif
}

void StatusDisplay::DrawValue(Stat stat, ConstStr255Param str)
{
    Rect& r = valueRects[(short)stat];
    MoveTo(r.right - StringWidth(str), r.top + tableBaseline);
    EraseRect(&r);
    DrawString(str);
}
void StatusDisplay::DrawValue(Stat stat, long val)
{
    Str255 str;
    NumToString(val, str);
    if(val >= 0)
        DrawValue(stat, str);
    else
        DrawValue(stat, "\p--");
}


void StatusDisplay::Update()
{
#if TARGET_API_MAC_CARBON
    SetPortWindowPort(statusWindow);
#else
    SetPort(statusWindow);
#endif
    BeginUpdate(statusWindow);
    EraseRgn(background);

    TextSize(12);
    TextFace(normal);

    MoveTo(statusRect.left,statusRect.bottom-10);
    DrawString(statusString);

    Rect r;
    
    if(progressTotal)
    {
        FrameRect(&progressRect);
        InsetRect(&r, 1,1);
        SetRect(&r, progressRect.left+1, progressRect.top+1,
                    progressRect.left+1 + (progressRect.right-progressRect.left-2) * progressDone / progressTotal,
                    progressRect.bottom-1);

#if TARGET_API_MAC_CARBON
        Pattern fg, bg;
        GetQDGlobalsDarkGray(&fg);
        GetQDGlobalsLightGray(&bg);
#else
        const Pattern& fg = qd.dkGray;
        const Pattern& bg = qd.ltGray;
#endif

        FillRect(&r, &fg);
        r.left = r.right;
        r.right = progressRect.right - 1;
        FillRect(&r, &bg);
    }
    else
        EraseRect(&progressRect);

    Str255 str;

    TextSize(9);
    TextFace(bold);

    for(int i = 0; i < nValues; i++)
    {
        GetIndString(str, 129, (i*2)+1);
        MoveTo(valueRects[i].left - StringWidth(str), valueRects[i].top + tableBaseline);
        DrawString(str);
        GetIndString(str, 129, (i*2)+2);
        MoveTo(valueRects[i].right, valueRects[i].top + tableBaseline);
        DrawString(str);
    }

    TextFace(normal);
#if !TARGET_API_MAC_CARBON
    DrawValue(Stat::heapSize, (ApplicationZone()->bkLim - (Ptr)ApplicationZone()) / 1024);
#endif
    DrawValue(Stat::freeMem, freeMem);
    if(progressTotal)
    {
        DrawValue(Stat::fileSize, progressTotal / 1024);
        DrawValue(Stat::transferred, progressDone / 1024);
    }
    else
    {
        DrawValue(Stat::fileSize, "\p--");
        DrawValue(Stat::transferred, "\p--");
    }
    DrawValue(Stat::speed, speed);
    DrawValue(Stat::timeRemaining, timeRemaining);
    DrawValue(Stat::transmissionErrors, errorCount);

    EndUpdate(statusWindow);
}

void StatusDisplay::Idle()
{
    long newFreeMem = FreeMem() / 1024;
    if(newFreeMem != freeMem)
    {
        freeMem = newFreeMem;
        Inval(valueRects[(short)Stat::freeMem]);
    }

    long newTimeRemaining = -1;
    if(status == AppStatus::downloading || status == AppStatus::upgrading)
    {
        long now = TickCount();
        if(now - startTime > 60 && progressDone > 4000)
        {
            long newSpeed = progressDone * 6015LL / (now-startTime) / 100;
            newTimeRemaining = (uint64_t(now-startTime) * (progressTotal - progressDone) * 100
                            / progressDone + 6014) / 6015;
            if(newSpeed != speed)
            {
                speed = newSpeed;
                Inval(valueRects[(short)Stat::speed]);
            }
        }
    }

    if(newTimeRemaining != timeRemaining)
    {
        timeRemaining = newTimeRemaining;
        Inval(valueRects[(short)Stat::timeRemaining]);
    }
}

void StatusDisplay::SetStatus(AppStatus stat)
{
    if(stat != status)
    {
        status = stat;
        if(status == AppStatus::downloading || status == AppStatus::upgrading)
            startTime = TickCount();
        GetIndString(statusString,128,(short)stat);
        Inval(statusRect);
    }
}

void StatusDisplay::SetProgress(int done, int total)
{
    if(done != progressDone || total != progressTotal)
    {
        Inval(progressRect);
        if(total != progressTotal)
            Inval(valueRects[(short)Stat::fileSize]);
        Inval(valueRects[(short)Stat::transferred]);
        progressTotal = total;
        progressDone = done;
    }
}

void StatusDisplay::SetStatus(AppStatus stat, int done, int total)
{
    SetStatus(stat);
    SetProgress(done, total);
}

void StatusDisplay::SetErrorCount(int newErrorCount)
{
    if(newErrorCount != errorCount)
    {
        errorCount = newErrorCount;
        Inval(valueRects[(short)Stat::transmissionErrors]);
    }
}