libcruft-util/alloc/raw/malloc.hpp

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2018 Danny Robson <danny@nerdcruft.net>
 */

#ifndef CRUFT_UTIL_ALLOC_RAW_MALLOC_HPP
#define CRUFT_UTIL_ALLOC_RAW_MALLOC_HPP

#include "../../view.hpp"

#include <cstddef>


namespace cruft::alloc::raw {
    class malloc {
    public:
        template <typename T>
        cruft::view<T*> allocate (size_t count)
        {
            return {
                reinterpret_cast<T*> (malloc (sizeof (T) * count)),
                count
            };
        }

        template <typename T>
        cruft::view<T*>
        allocate (size_t count, size_t align)
        {
            void* ptr;
            posix_memalign (&ptr, align, sizeof (T) * count);
            if (!ptr)
                throw std::bad_alloc ();

            return { reinterpret_cast<T*> (ptr), count };
        }

        template <typename T>
        void
        deallocate (cruft::view<std::byte*> ptr)
        {
            ::free (ptr.data ());
        }
    };
}


#endif