#include "CIOWinManager.hpp"
#include "CArchitectureMessage.hpp"

namespace Retro
{

bool CIOWinManager::OnIOWinMessage(const CArchitectureMessage& msg)
{
    switch (msg.GetType())
    {
    case MsgRemoveIOWin:
    {
        const CArchMsgParmVoidPtr& parm = MakeMsg::GetParmDeleteIOWin(msg);
        CIOWin* iow = FindIOWin(*static_cast<const std::string*>(parm.x4_parm1));
        if (iow)
            RemoveIOWin(iow);
        return false;
    }
    case MsgCreateIOWin:
    {
        const CArchMsgParmInt32Int32VoidPtr& parm = MakeMsg::GetParmCreateIOWin(msg);
        rstl::rc_ptr<CIOWin> iow(static_cast<CIOWin*>(parm.xc_parm3));
        AddIOWin(iow, parm.x4_parm1, parm.x8_parm2);
        return false;
    }
    case MsgChangeIOWinPriority:
    {
        const CArchMsgParmInt32Int32VoidPtr& parm = MakeMsg::GetParmChangeIOWinPriority(msg);
        CIOWin* iow = FindIOWin(*static_cast<const std::string*>(parm.xc_parm3));
        if (iow)
            ChangeIOWinPriority(iow, parm.x4_parm1, parm.x8_parm2);
        return false;
    }
    case MsgRemoveAllIOWins:
    {
        RemoveAllIOWins();
        return true;
    }
    default: break;
    }
    return false;
}

void CIOWinManager::Draw() const
{
    IOWinPQNode* node = x0_drawRoot;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        iow->PreDraw();
        if (!iow->GetIsContinueDraw())
            break;
        node = node->x8_next;
    }
    node = x0_drawRoot;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        iow->Draw();
        if (!iow->GetIsContinueDraw())
            break;
        node = node->x8_next;
    }
}

bool CIOWinManager::DistributeOneMessage(const CArchitectureMessage& msg,
                                         CArchitectureQueue& queue)
{
    IOWinPQNode* node = x4_pumpRoot;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        CIOWin::EMessageReturn mret = iow->OnMessage(msg, x8_localGatherQueue);

        while (x8_localGatherQueue)
        {
            CArchitectureMessage msg = x8_localGatherQueue.Pop();
            if (msg.GetTarget() == TargetIOWinManager)
            {
                if (OnIOWinMessage(msg))
                {
                    x8_localGatherQueue.Clear();
                    queue.Clear();
                    return true;
                }
            }
            else
                queue.Push(std::move(msg));
        }

        switch (mret)
        {
        case CIOWin::MsgRetRemoveIOWinAndExit:
        case CIOWin::MsgRetRemoveIOWin:
            RemoveIOWin(iow);
        default: break;
        }

        switch (mret)
        {
        case CIOWin::MsgRetExit:
        case CIOWin::MsgRetRemoveIOWinAndExit:
            return false;
        default: break;
        }

        node = node->x8_next;
    }

    return false;
}

void CIOWinManager::PumpMessages(CArchitectureQueue& queue)
{
    while (queue)
    {
        CArchitectureMessage msg = queue.Pop();
        if (DistributeOneMessage(msg, queue))
            break;
    }
}

CIOWin* CIOWinManager::FindIOWin(const std::string& name)
{
    size_t findHash = std::hash<std::string>()(name);

    IOWinPQNode* node = x4_pumpRoot;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow->GetNameHash() == findHash)
            return iow;
        node = node->x8_next;
    }

    node = x0_drawRoot;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow->GetNameHash() == findHash)
            return iow;
        node = node->x8_next;
    }

    return nullptr;
}

rstl::rc_ptr<CIOWin> CIOWinManager::FindAndShareIOWin(const std::string& name)
{
    size_t findHash = std::hash<std::string>()(name);

    IOWinPQNode* node = x4_pumpRoot;
    while (node)
    {
        rstl::rc_ptr<CIOWin> iow = node->ShareIOWin();
        if (iow->GetNameHash() == findHash)
            return iow;
        node = node->x8_next;
    }

    node = x0_drawRoot;
    while (node)
    {
        rstl::rc_ptr<CIOWin> iow = node->ShareIOWin();
        if (iow->GetNameHash() == findHash)
            return iow;
        node = node->x8_next;
    }

    return rstl::rc_ptr<CIOWin>();
}

void CIOWinManager::ChangeIOWinPriority(CIOWin* toChange, int pumpPrio, int drawPrio)
{
    IOWinPQNode* node = x4_pumpRoot;
    IOWinPQNode* prevNode = nullptr;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow == toChange)
        {
            if (prevNode)
                prevNode->x8_next = node->x8_next;
            node->x4_prio = pumpPrio;
            IOWinPQNode* testNode = x4_pumpRoot;
            IOWinPQNode* testPrevNode = nullptr;
            while (testNode->x4_prio > pumpPrio)
            {
                testPrevNode = testNode;
                testNode = testNode->x8_next;
            }
            node->x8_next = testNode;
            if (testPrevNode)
                testPrevNode->x8_next = node;
            else
                x4_pumpRoot = node;
            break;
        }
        prevNode = node;
        node = node->x8_next;
    }

    node = x0_drawRoot;
    prevNode = nullptr;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow == toChange)
        {
            if (prevNode)
                prevNode->x8_next = node->x8_next;
            node->x4_prio = drawPrio;
            IOWinPQNode* testNode = x0_drawRoot;
            IOWinPQNode* testPrevNode = nullptr;
            while (testNode->x4_prio > drawPrio)
            {
                testPrevNode = testNode;
                testNode = testNode->x8_next;
            }
            node->x8_next = testNode;
            if (testPrevNode)
                testPrevNode->x8_next = node;
            else
                x0_drawRoot = node;
            break;
        }
        prevNode = node;
        node = node->x8_next;
    }
}

void CIOWinManager::RemoveAllIOWins()
{
    IOWinPQNode* node = x0_drawRoot;
    while (node)
    {
        IOWinPQNode* delNode = node;
        node = node->x8_next;
        delete delNode;
    }
    x0_drawRoot = nullptr;
    node = x4_pumpRoot;
    while (node)
    {
        IOWinPQNode* delNode = node;
        node = node->x8_next;
        delete delNode;
    }
    x4_pumpRoot = nullptr;
}

void CIOWinManager::RemoveIOWin(CIOWin* chIow)
{
    IOWinPQNode* node = x4_pumpRoot;
    IOWinPQNode* prevNode = nullptr;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow == chIow)
        {
            if (prevNode)
                prevNode->x8_next = node->x8_next;
            else
                x4_pumpRoot = node->x8_next;
            delete node;
            break;
        }
        prevNode = node;
        node = node->x8_next;
    }

    node = x0_drawRoot;
    prevNode = nullptr;
    while (node)
    {
        CIOWin* iow = node->GetIOWin();
        if (iow == chIow)
        {
            if (prevNode)
                prevNode->x8_next = node->x8_next;
            else
                x0_drawRoot = node->x8_next;
            delete node;
            break;
        }
        prevNode = node;
        node = node->x8_next;
    }
}

void CIOWinManager::AddIOWin(rstl::ncrc_ptr<CIOWin> chIow, int pumpPrio, int drawPrio)
{
    IOWinPQNode* node = x4_pumpRoot;
    IOWinPQNode* prevNode = nullptr;
    while (node && pumpPrio > node->x4_prio)
    {
        prevNode = node;
        node = node->x8_next;
    }
    IOWinPQNode* newNode = new IOWinPQNode(chIow, pumpPrio, node);
    if (prevNode)
        prevNode->x8_next = newNode;
    else
        x4_pumpRoot = newNode;

    node = x0_drawRoot;
    prevNode = nullptr;
    while (node && drawPrio > node->x4_prio)
    {
        prevNode = node;
        node = node->x8_next;
    }
    newNode = new IOWinPQNode(chIow, drawPrio, node);
    if (prevNode)
        prevNode->x8_next = newNode;
    else
        x0_drawRoot = newNode;
}

}