/* This file is part of Nokia HEIF library
 *
 * Copyright (c) 2015-2025 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.
 *
 * Contact: heif@nokia.com
 *
 * This software, including documentation, is protected by copyright controlled by Nokia Corporation and/ or its
 * subsidiaries. All rights are reserved.
 *
 * Copying, including reproducing, storing, adapting or translating, any or all of this material requires the prior
 * written consent of Nokia.
 */

#include "customallocator.hpp"

#include "../api/common/heifallocator.h"

namespace
{
    class DefaultAllocator : public HEIF::CustomAllocator
    {
    public:
        DefaultAllocator()           = default;
        ~DefaultAllocator() override = default;

        void* allocate(size_t n, size_t size) override
        {
            return malloc(n * size);
        }
        void deallocate(void* ptr) override
        {
            free(ptr);
        }
    };
}  // namespace
static DefaultAllocator defaultAllocator;
static HEIF::CustomAllocator* customAllocator;

bool setCustomAllocator(HEIF::CustomAllocator* customAllocator_)
{
    if (!customAllocator || !customAllocator_)
    {
        customAllocator = customAllocator_;
        return true;
    }
    else
    {
        return false;
    }
}

HEIF::CustomAllocator* getDefaultAllocator()
{
    return &defaultAllocator;
}

HEIF::CustomAllocator* getCustomAllocator()
{
    if (!customAllocator)
    {
        customAllocator = getDefaultAllocator();
    }
    return customAllocator;
}

void* customAllocate(size_t size)
{
    return getCustomAllocator()->allocate(size, 1);
}

void customDeallocate(void* ptr)
{
    getCustomAllocator()->deallocate(ptr);
}