/****************************************************************************
*                                                                           *
* Project64 - A Nintendo 64 emulator.                                      *
* http://www.pj64-emu.com/                                                  *
* Copyright (C) 2012 Project64. All rights reserved.                        *
*                                                                           *
* License:                                                                  *
* GNU/GPLv2 http://www.gnu.org/licenses/gpl-2.0.html                        *
*                                                                           *
****************************************************************************/
#include "stdafx.h"
#include "FunctionMapClass.h"
#include <Project64-core/N64System/SystemGlobals.h>
#include <Project64-core/N64System/N64Class.h>
#include <Windows.h>

CFunctionMap::CFunctionMap() :
    m_JumpTable(NULL),
    m_FunctionTable(NULL)
{
}

CFunctionMap::~CFunctionMap()
{
    CleanBuffers();
}

bool CFunctionMap::AllocateMemory()
{
    if (g_System->LookUpMode() == FuncFind_VirtualLookup && m_FunctionTable == NULL)
    {
        m_FunctionTable = (PCCompiledFunc_TABLE *)VirtualAlloc(NULL, 0xFFFFF * sizeof(CCompiledFunc *), MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
        if (m_FunctionTable == NULL)
        {
            WriteTrace(TraceError, __FUNCTION__ ": failed to allocate function table");
            g_Notify->FatalError(MSG_MEM_ALLOC_ERROR);
            return false;
        }
        memset(m_FunctionTable, 0, 0xFFFFF * sizeof(CCompiledFunc *));
    }
    if (g_System->LookUpMode() == FuncFind_PhysicalLookup && m_JumpTable == NULL)
    {
        m_JumpTable = new PCCompiledFunc[g_MMU->RdramSize() >> 2];
        if (m_JumpTable == NULL)
        {
            WriteTrace(TraceError, __FUNCTION__ ": failed to allocate jump table");
            g_Notify->FatalError(MSG_MEM_ALLOC_ERROR);
            return false;
        }
        memset(m_JumpTable, 0, (g_MMU->RdramSize() >> 2) * sizeof(PCCompiledFunc));
    }
    return true;
}

void CFunctionMap::CleanBuffers()
{
    if (m_FunctionTable)
    {
        for (int i = 0, n = 0x100000; i < n; i++)
        {
            if (m_FunctionTable[i] != NULL)
            {
                delete m_FunctionTable[i];
            }
        }
        VirtualFree(m_FunctionTable, 0, MEM_RELEASE);
        m_FunctionTable = NULL;
    }
    if (m_JumpTable)
    {
        delete[] m_JumpTable;
        m_JumpTable = NULL;
    }
}

void CFunctionMap::Reset(bool bAllocate)
{
    CleanBuffers();
    if (bAllocate && (g_System->LookUpMode() == FuncFind_VirtualLookup || g_System->LookUpMode() == FuncFind_PhysicalLookup))
    {
        AllocateMemory();
    }
}