/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */


#include <AWSNativeSDKInit/AWSMemoryInterface.h>

namespace AWSNativeSDKInit
{
#if defined(PLATFORM_SUPPORTS_AWS_NATIVE_SDK)
    const char* MemoryManager::AWS_API_ALLOC_TAG = "AwsApi";

    void MemoryManager::Begin() 
    {
        // We can't guarantee that all uses cases have/will Create a SystemAllocator which is a dependency of 
        // the AWSNativeSDKAllocator in SystemAllocator::Create
        if (!AZ::AllocatorInstance<AZ::SystemAllocator>::IsReady())
        {
            AZ::AllocatorInstance<AZ::SystemAllocator>::Create();  
            m_systemAllocatorCreated = true;
        }
        AWSNativeSDKAllocator::Descriptor desc;
        m_allocator.Create(desc);
    }

    void MemoryManager::End() 
    {
        m_allocator.Destroy();

        if (m_systemAllocatorCreated)
        {
            AZ::AllocatorInstance<AZ::SystemAllocator>::Destroy();
        }
    }

    void* MemoryManager::AllocateMemory(std::size_t blockSize, std::size_t alignment, const char* allocationTag) 
    {
        return m_allocator->Allocate(blockSize, alignment, 0, allocationTag);
    }

    void MemoryManager::FreeMemory(void* memoryPtr) 
    {
        m_allocator->DeAllocate(memoryPtr);
    }
#endif
}