/*
===========================================================================
Copyright (C) 2023 the OpenMoHAA team

This file is part of OpenMoHAA source code.

OpenMoHAA source code is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of the License,
or (at your option) any later version.

OpenMoHAA source code is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with OpenMoHAA source code; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
===========================================================================
*/

#include "../qcommon/localization.h"
#include "ui_local.h"

Event EV_Pulldown_ChildKilled
(
    "_ev_pulldown_child_killed",
    EV_DEFAULT,
    NULL,
    NULL,
    "Signaled when the sub menu is killed for one reason or another"
);

Event EV_Pulldown_HighlightBGColor
(
    "highlight_bgcolor",
    EV_DEFAULT,
    "fff",
    "r g b",
    "Set the background color of the highlighted text when a pulldown is used"
);

Event EV_Pulldown_HighlightFGColor
(
    "highlight_fgcolor",
    EV_DEFAULT,
    "fff",
    "r g b",
    "Set the foreground color of the highlighted text when a pulldown is used"
);

CLASS_DECLARATION(UIWidget, UIPulldownMenu, NULL) {
    {&W_LeftMouseDown,              &UIPulldownMenu::MousePressed    },
    {&W_LeftMouseUp,                &UIPulldownMenu::MouseReleased   },
    {&W_LeftMouseDragged,           &UIPulldownMenu::MouseDragged    },
    {&EV_Pulldown_ChildKilled,      &UIPulldownMenu::ChildKilled     },
    {&EV_Pulldown_HighlightBGColor, &UIPulldownMenu::HighlightBGColor},
    {&EV_Pulldown_HighlightFGColor, &UIPulldownMenu::HighlightFGColor},
    {NULL,                          NULL                             }
};

static UIPopupMenu *MenuFromPoint(UIPoint2D& p)
{
    UIWidget *res = uWinMan.FindResponder(p);
    if (res && res->isSubclassOf(UIPopupMenu)) {
        return (UIPopupMenu *)res;
    }

    return NULL;
}

UIPulldownMenu::UIPulldownMenu()
{
    m_listener           = NULL;
    m_submenu            = -1;
    m_highlightBGColor.r = 0.02f;
    m_highlightBGColor.g = 0.07f;
    m_highlightBGColor.b = 0.005f;
    m_highlightBGColor.a = 1.0f;
    m_highlightFGColor   = (UColor)UHudColor;

    // uninitialized in original game
    m_submenuptr = NULL;

    // Added in OPM
    //  This is so the window manager knows it can be activated
    //  and appropriately handle this as the first responder
    m_canactivate = true;
}

UIPulldownMenu::~UIPulldownMenu()
{
    for (int i = m_desc.NumObjects(); i > 0; i--) {
        uipull_describe *updb = m_desc.ObjectAt(i);
        // FIXME: possible OG bug, memory leak? updb->desc is not cleared
        m_desc.RemoveObjectAt(i);
        delete updb;
    }
}

UIRect2D UIPulldownMenu::getAlignmentRect(UIWidget *parent)
{
    if (!parent) {
        parent = (m_parent == NULL) ? &uWinMan : m_parent;
    }

    UIRect2D parentRect = parent->getClientFrame();
    if (m_bVirtual) {
        m_vVirtualScale[0] = (float)uid.vidWidth / SCREEN_WIDTH;
        m_vVirtualScale[1] = (float)uid.vidHeight / SCREEN_HEIGHT;
    }

    int maxheight = m_font->getHeight(m_bVirtual ? m_vVirtualScale : NULL);
    for (int i = 1; i <= m_desc.NumObjects(); i++) {
        uipull_describe *uipd = m_desc.ObjectAt(i);
        if (uipd && uipd->material) {
            int h = uii.Rend_GetShaderHeight(uipd->material->GetMaterial());
            if (h > maxheight) {
                maxheight = h;
            }
        }
    }

    return UIRect2D(parentRect.pos.x, parentRect.pos.y, parentRect.size.width, (maxheight + 4) * m_vVirtualScale[1]);
}

float UIPulldownMenu::getDescWidth(uipull_describe *desc)
{
    UIReggedMaterial *mat = desc->material;
    if (mat) {
        return uii.Rend_GetShaderWidth(mat->GetMaterial());
    }

    return m_font->getWidth(desc->title, -1) + 8.0f;
}

float UIPulldownMenu::getDescHeight(uipull_describe *desc)
{
    UIReggedMaterial *mat = desc->material;
    if (mat) {
        return uii.Rend_GetShaderHeight(mat->GetMaterial());
    }

    return m_font->getHeight(m_bVirtual ? m_vVirtualScale : NULL);
}

uipull_describe *UIPulldownMenu::getPulldown(str title)
{
    for (int i = 1; i <= m_desc.NumObjects(); i++) {
        uipull_describe *pd = m_desc.ObjectAt(i);
        if (pd->title == title) {
            return pd;
        }
    }

    return NULL;
}

void UIPulldownMenu::HighlightBGColor(Event *ev)
{
    m_highlightBGColor.r = ev->GetFloat(1);
    m_highlightBGColor.g = ev->GetFloat(2);
    m_highlightBGColor.b = ev->GetFloat(3);
    m_highlightBGColor.a = 1.0f;
}

void UIPulldownMenu::HighlightFGColor(Event *ev)
{
    m_highlightFGColor.r = ev->GetFloat(1);
    m_highlightFGColor.g = ev->GetFloat(2);
    m_highlightFGColor.b = ev->GetFloat(3);
    m_highlightFGColor.a = 1.0f;
}

void UIPulldownMenu::Create(UIWidget *parent, Listener *listener, const UIRect2D& rect)
{
    m_listener = listener;
    InitFrame(parent, rect.pos.x, rect.pos.y, rect.size.width, rect.size.height, 0);
}

void UIPulldownMenu::CreateAligned(UIWidget *parent, Listener *listener)
{
    setFont("verdana-12");
    UIRect2D rect = getAlignmentRect(parent);
    Create(parent, listener, rect);
}

void UIPulldownMenu::MousePressed(Event *ev)
{
    uWinMan.setFirstResponder(this);
    MouseDragged(ev);
}

void UIPulldownMenu::MouseDragged(Event *ev)
{
    int              newSubMenu = -1;
    UIRect2D         subRect;
    uipull_describe *subdesc = NULL;

    UIPoint2D point = MouseEventToClientPoint(ev);
    if (getClientFrame().contains(point)) {
        float atx = m_vVirtualScale[0] * 2;
        for (int i = 1; i <= m_desc.NumObjects(); i++) {
            uipull_describe *desc  = m_desc.ObjectAt(i);
            float            width = getDescWidth(desc) * m_vVirtualScale[0];
            if (atx <= point.x && atx + width > point.x) {
                newSubMenu          = i;
                subdesc             = desc;
                subRect.pos.x       = atx;
                subRect.pos.y       = 0.0f;
                subRect.size.width  = width;
                subRect.size.height = m_frame.size.height;
                break;
            }
            atx += width;
        }
    }

    if (newSubMenu == m_submenu || newSubMenu == -1) {
        if (m_submenu == -1) {
            return;
        }

        UIPoint2D cursorLocation(point.x + m_screenframe.pos.x, point.y + m_screenframe.pos.y);

        UIPopupMenu *menu = MenuFromPoint(cursorLocation);
        if (!menu) {
            UIPopupMenu *submenu = m_submenuptr;
            while (submenu->m_submenu != -1) {
                submenu = submenu->m_submenuptr;
            }

            submenu->MouseMoved(ev);
        } else {
            while (menu->m_parentMenu) {
                menu = menu->m_parentMenu;
            }
            if (menu == m_submenuptr) {
                menu->MouseMoved(ev);
            }
        }

        return;
    }

    if (m_submenu != -1) {
        m_submenuptr->Disconnect(this, W_Destroyed);
        if (m_submenuptr) {
            delete m_submenuptr;
        }
    }

    m_submenu    = newSubMenu;
    m_submenuptr = new UIPopupMenu();
    subRect.pos.x += m_screenframe.pos.x;
    subRect.pos.y += m_screenframe.pos.y;
    m_submenuptr->Create(&subdesc->desc, m_listener, subRect, UIP_WHERE_DOWN, m_bVirtual, -1.0f);
    m_submenuptr->Connect(this, W_Destroyed, EV_Pulldown_ChildKilled);
}

void UIPulldownMenu::MouseReleased(Event *ev)
{
    if (uWinMan.getFirstResponder() == this) {
        uWinMan.setFirstResponder(NULL);
        UIPoint2D    point(ev->GetFloat(1), ev->GetFloat(2));
        UIPopupMenu *menu = MenuFromPoint(point);
        if (menu) {
            menu->MouseReleased(ev);
        }
    }

    if (m_submenu != -1) {
        m_submenuptr->Disconnect(this, W_Destroyed);
        delete m_submenuptr;
        m_submenu = -1;
    }
}

void UIPulldownMenu::ChildKilled(Event *ev)
{
    if (m_submenu != -1) {
        m_submenu = -1;
    }

    if (uWinMan.getFirstResponder() == this) {
        uWinMan.setFirstResponder(NULL);
    }
}

void UIPulldownMenu::AddUIPopupDescribe(const char *title, uipopup_describe *d)
{
    if (!title) {
        return;
    }

    uipull_describe *pd = getPulldown(title);
    if (!pd) {
        pd = new uipull_describe(title, NULL, NULL);
        m_desc.AddObject(pd);
    }

    pd->desc.AddObject(d);
}

void UIPulldownMenu::setHighlightFGColor(UColor c)
{
    m_highlightFGColor = c;
}

void UIPulldownMenu::setHighlightBGColor(UColor c)
{
    m_highlightBGColor = c;
}

void UIPulldownMenu::setPopupHighlightFGColor(str menu, UColor c)
{
    uipull_describe *up = getPulldown(menu);
    if (up) {
        up->highlightFGColor = c;
    }
}

void UIPulldownMenu::setPopupHighlightBGColor(str menu, UColor c)
{
    uipull_describe *up = getPulldown(menu);
    if (up) {
        up->highlightBGColor = c;
    }
}

void UIPulldownMenu::setPopupFGColor(str menu, UColor c)
{
    uipull_describe *up = getPulldown(menu);
    if (up) {
        up->FGColor = c;
    }
}

void UIPulldownMenu::setPopupBGColor(str menu, UColor c)
{
    uipull_describe *up = getPulldown(menu);
    if (up) {
        up->BGColor = c;
    }
}

void UIPulldownMenu::setPulldownShader(const char *title, UIReggedMaterial *mat)
{
    if (!title) {
        return;
    }

    uipull_describe *pd = getPulldown(title);
    if (!pd) {
        pd = new uipull_describe(title, NULL, NULL);
        m_desc.AddObject(pd);
    }

    pd->material = mat;
    Realign();
}

void UIPulldownMenu::setSelectedPulldownShader(const char *title, UIReggedMaterial *mat)
{
    if (!title) {
        return;
    }

    uipull_describe *pd = getPulldown(title);
    if (!pd) {
        pd = new uipull_describe(title, NULL, NULL);
        m_desc.AddObject(pd);
    }

    pd->selected_material = mat;
    Realign();
}

void UIPulldownMenu::Realign(void)
{
    UIRect2D rect = getAlignmentRect(NULL);
    setFrame(rect);
}

void UIPulldownMenu::Draw(void)
{
    float atx = 0.0f;
    for (int i = 1; i <= m_desc.NumObjects(); i++) {
        uipull_describe *desc   = m_desc.ObjectAt(i);
        float            width  = getDescWidth(desc) * m_vVirtualScale[0];
        float            height = getDescHeight(desc) * m_vVirtualScale[1];

        // Set font color based on whether the current item is selected
        if (m_submenu == i) {
            // Selected item
            m_font->setColor(m_highlightFGColor);
            if (desc->selected_material) {
                uii.Rend_DrawPicStretched(
                    atx, 0.0f, width, height, 0.0f, 0.0f, 1.0f, 1.0f, desc->selected_material->GetMaterial()
                );
            } else {
                DrawBox(atx, 0.0f, width, m_frame.size.height, m_highlightBGColor, 1.0f);
            }
        } else {
            // Unselected item
            m_font->setColor(m_foreground_color);
        }

        if (desc->material) {
            if (m_submenu != i) {
                uii.Rend_DrawPicStretched(
                    atx, 0.0f, width, height, 0.0f, 0.0f, 1.0f, 1.0f, desc->material->GetMaterial()
                );
            }
        } else {
            const char *text      = Sys_LV_CL_ConvertString(desc->title);
            float       text_xpos = m_vVirtualScale[0] * 4.0f + atx;
            float       text_ypos = m_vVirtualScale[1] * 2.0f;
            m_font->Print(text_xpos, text_ypos, text, -1, m_bVirtual ? m_vVirtualScale : NULL);
        }

        atx += width;
    }
}