ui_main.cpp

Go to the documentation of this file.

/*
Copyright (C) 2002-2020 UFO: Alien Invasion.

This program 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.

This program 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 this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "ui_main.h"
#include "ui_internal.h"
#include "ui_draw.h"
#include "ui_timer.h"
#include "ui_font.h"
#include "ui_parse.h"
#include "ui_sound.h"
#include "../cl_menu.h"
#include "node/ui_node_abstractnode.h"
#include "ui_lua.h"
#include <vector>
#include <string>

uiGlobal_t ui_global;

memPool_t* ui_dynStringPool;
memPool_t* ui_dynPool;
memPool_t* ui_sysPool;

#ifdef DEBUG
static cvar_t* ui_debug;
#endif

int UI_DebugMode (void)
{
#ifdef DEBUG
    return ui_debug->integer;
#else
    return 0;
#endif
}

#ifdef DEBUG

static void UI_Memory_f (void)
{
    int i;
    const size_t nodeSize = sizeof(uiNode_t);
    size_t size;
    Com_Printf("Allocation:\n");
    Com_Printf("\t-Window allocation: %i/%i\n", ui_global.numWindows, UI_MAX_WINDOWS);
    Com_Printf("\t-Rendering window stack slot: %i\n", UI_MAX_WINDOWSTACK);
    Com_Printf("\t-Action allocation: %i/%i\n", ui_global.numActions, UI_MAX_ACTIONS);
    Com_Printf("\t-Model allocation: %i/%i\n", ui_global.numModels, UI_MAX_MODELS);
    Com_Printf("\t-Node allocation: %i\n", ui_global.numNodes);

    Com_Printf("Memory:\n");
    Com_Printf("\t-Action structure size: " UFO_SIZE_T " B\n", sizeof(uiAction_t));
    Com_Printf("\t-Model structure size: " UFO_SIZE_T " B\n", sizeof(uiModel_t));
    Com_Printf("\t-Node structure size: " UFO_SIZE_T " B x%d\n", sizeof(uiNode_t), ui_global.numNodes);
    for (i = 0; i < UI_GetNodeBehaviourCount(); i++) {
        uiBehaviour_t* b = UI_GetNodeBehaviourByIndex(i);
        Com_Printf("\t -Behaviour %20s structure size: " UFO_SIZE_T " (+" UFO_SIZE_T " B) x%4u\n", b->name, sizeof(uiNode_t) + b->extraDataSize, b->extraDataSize, b->count);
    }

    size = 0;
    for (i = 0; i < UI_GetNodeBehaviourCount(); i++) {
        uiBehaviour_t* b = UI_GetNodeBehaviourByIndex(i);
        size += nodeSize * b->count + b->extraDataSize * b->count;
    }
    Com_Printf("Global memory:\n");
    Com_Printf("\t-System pool: %ui B\n", _Mem_PoolSize(ui_sysPool));
    Com_Printf("\t -AData allocation: " UFO_SIZE_T "/%i B\n", (ptrdiff_t)(ui_global.curadata - ui_global.adata), ui_global.adataize);
    Com_Printf("\t -AData used by nodes: " UFO_SIZE_T " B\n", size);
    Com_Printf("\t-Dynamic node/data pool: %ui B\n", _Mem_PoolSize(ui_dynPool));
    Com_Printf("\t-Dynamic strings pool: %ui B\n", _Mem_PoolSize(ui_dynStringPool));

    size = _Mem_PoolSize(ui_sysPool) + _Mem_PoolSize(ui_dynPool) + _Mem_PoolSize(ui_dynStringPool);
    Com_Printf("\t-Full size: " UFO_SIZE_T " B\n", size);
}
#endif

#define MAX_CONFUNC_SIZE 512

void UI_ExecuteConfunc (const char* fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    Cmd_vExecuteString(fmt, ap);
    va_end(ap);
}

void* UI_AllocHunkMemory (size_t size, int align, bool reset)
{
    byte* memory = (byte*) ALIGN_PTR(ui_global.curadata, align);
    if (memory + size > ui_global.adata + ui_global.adataize)
        return nullptr;
    if (reset)
        memset(memory, 0, size);
    ui_global.curadata = memory + size;
    return memory;
}

static void UI_Restart_f (void)
{
    typedef std::vector<std::string> Names;
    Names names;
    for (int i = 0; i < ui_global.windowStackPos; i++) {
        names.push_back(std::string(ui_global.windowStack[i]->name));
    }

    UI_Shutdown();
    CLMN_Shutdown();
    R_FontShutdown();
    UI_Init();
    R_FontInit();
    Com_Printf("%i ui script files\n", FS_BuildFileList("ufos/ui/*.ufo"));
    FS_NextScriptHeader(nullptr, nullptr, nullptr);
    const char* type, *name, *text;
    text = nullptr;
    while ((type = FS_NextScriptHeader("ufos/*.ufo", &name, &text)) != nullptr) {
        if (Q_streq(type, "font"))
            UI_ParseFont(name, &text);
        else if (Q_streq(type, "menu_model"))
            UI_ParseUIModel(name, &text);
        else if (Q_streq(type, "sprite"))
            UI_ParseSprite(name, &text);
    }
    UI_Reinit();
    FS_NextScriptHeader(nullptr, nullptr, nullptr);
    text = nullptr;
    while ((type = FS_NextScriptHeader("ufos/ui/*.ufo", &name, &text)) != nullptr) {
        if (Q_streq(type, "window"))
            UI_ParseWindow(type, name, &text);
        else if (Q_streq(type, "component"))
            UI_ParseComponent(type, name, &text);
        else if (Q_streq(type, "menu_model"))
            UI_ParseUIModel(name, &text);
        else if (Q_streq(type, "sprite"))
            UI_ParseSprite(name, &text);
        else if (Q_streq(type, "lua"))
            UI_ParseAndLoadLuaScript(name, &text);
    }

    CLMN_Init();

    for (Names::iterator i = names.begin(); i != names.end(); ++i) {
        UI_PushWindow(i->c_str());
    }
}

void UI_Reinit (void)
{
    UI_InitFonts();
    UI_ReleaseInput();
    UI_InvalidateMouse();
}

void UI_Shutdown (void)
{
    /* MN is not yet initialized */
    if (ui_global.adata == nullptr)
        return;

    const uiBehaviour_t* confunc = UI_GetNodeBehaviour("confunc");

    /* remove all confunc commands */
    uiNode_t** nodes[] = {ui_global.windows, ui_global.components};
    for (int nodeType = 0; nodeType < 2; ++nodeType) {
        for (int i = 0; i < ui_global.numWindows; i++) {
            uiNode_t* node = nodes[nodeType][i];
            while (node) {
                /* remove the command */
                if (node->behaviour == confunc) {
                    /* many nodes to one command is allowed */
                    if (Cmd_Exists(node->name)) {
                        Cmd_RemoveCommand(node->name);
                    }
                }

                /* recursive next */
                if (node->firstChild != nullptr) {
                    node = node->firstChild;
                    continue;
                }
                while (node) {
                    if (node->next != nullptr) {
                        node = node->next;
                        break;
                    }
                    node = node->parent;
                }
            }
        }
    }
    UI_ShutdownLua();
    UI_FontShutdown();
    UI_ResetInput();
    UI_ResetTimers();

#ifdef DEBUG
    Cmd_RemoveCommand("debug_uimemory");
#endif
    Cmd_RemoveCommand("ui_restart");

    Mem_Free(ui_global.adata);
    OBJZERO(ui_global);

    /* release pools */
    Mem_FreePool(ui_sysPool);
    Mem_FreePool(ui_dynStringPool);
    Mem_FreePool(ui_dynPool);
    ui_sysPool = nullptr;
    ui_dynStringPool = nullptr;
    ui_dynPool = nullptr;
}

void UI_FinishInit (void)
{
    static bool initialized = false;
    if (initialized)
        return;
    initialized = true;

    UI_FinishWindowsInit();
}

void UI_Init (void)
{
    cvar_t* ui_hunkSize = Cvar_Get("ui_hunksize", "3", 0, "UI memory hunk size in megabytes");

#ifdef DEBUG
    ui_debug = Cvar_Get("debug_ui", "0", CVAR_DEVELOPER, "Prints node names for debugging purposes - valid values are 1 and 2");
#endif

    /* reset global UI structures */
    OBJZERO(ui_global);

    ui_sounds = Cvar_Get("ui_sounds", "1", CVAR_ARCHIVE, "Activates UI sounds");

#ifdef DEBUG
    Cmd_AddCommand("debug_uimemory", UI_Memory_f, "Display info about UI memory allocation");
#endif

    Cmd_AddCommand("ui_restart", UI_Restart_f, "Reload the whole ui");

    ui_sysPool = Mem_CreatePool("Client: UI");
    ui_dynStringPool = Mem_CreatePool("Client: Dynamic string for UI");
    ui_dynPool = Mem_CreatePool("Client: Dynamic memory for UI");

    Com_Printf("Allocate %i megabytes for the ui hunk\n", ui_hunkSize->integer);
    ui_global.adataize = ui_hunkSize->integer * 1024 * 1024;
    ui_global.adata    = Mem_PoolAllocTypeN(byte, ui_global.adataize, ui_sysPool);
    ui_global.curadata = ui_global.adata;

    UI_InitLua();
    UI_InitData();
    UI_InitNodes();
    UI_InitWindows();
    UI_InitDraw();
    UI_InitActions();
}