#include "memory.h"

sdv::pointer<uint8_t> CMemoryManager::Allocate(uint32_t uiLength)
{
    return sdv::internal::make_ptr<uint8_t>(this, uiLength);
}

void* CMemoryManager::Alloc(size_t nSize)
{
    void* pAlloc = malloc(nSize);
#ifdef MEMORY_TRACKER
    std::unique_lock<std::mutex> lock(m_mtxTracker);
    if (pAlloc)
        m_mapTracker.insert(std::make_pair(pAlloc, nSize));
#endif
    return pAlloc;
}

void* CMemoryManager::Realloc(void* pData, size_t nSize)
{
#ifdef MEMORY_TRACKER
    std::unique_lock<std::mutex> lock(m_mtxTracker);
    auto itAlloc = m_mapTracker.find(pData);
    if (pData && itAlloc == m_mapTracker.end())
    {
        std::cout << "Illegal request for resizing memory at location 0x" << (void*)pData << std::endl;
        return nullptr;
    }
    lock.unlock();
#endif

    void* pAlloc = realloc(pData, nSize);

#ifdef MEMORY_TRACKER
    if (pAlloc)
    {
        lock.lock();
        m_mapTracker.erase(pData);
        m_mapTracker.insert(std::make_pair(pAlloc, nSize));
    }
#endif

    return pAlloc;
}

void CMemoryManager::Free(void* pData)
{
#ifdef MEMORY_TRACKER
    std::unique_lock<std::mutex> lock(m_mtxTracker);
    auto itAlloc = m_mapTracker.find(pData);
    if (itAlloc == m_mapTracker.end())
    {
        std::cout << "Illegal request for freeing memory at location 0x" << (void*)pData << std::endl;
        return;
    }
    else
        m_mapTracker.erase(itAlloc);
    lock.unlock();
#endif

    free(pData);
}