1// -*- C++ -*-
2//===----------------------------------------------------------------------===//
3//
4// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5// See https://llvm.org/LICENSE.txt for license information.
6// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7//
8//===----------------------------------------------------------------------===//
9
10#ifndef _LIBCPP___SSO_ALLOCATOR
11#define _LIBCPP___SSO_ALLOCATOR
12
13#include <__config>
14#include <memory>
15#include <new>
16#include <type_traits>
17
18#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
19#pragma GCC system_header
20#endif
21
22_LIBCPP_BEGIN_NAMESPACE_STD
23
24template <class _Tp, size_t _Np> class _LIBCPP_HIDDEN __sso_allocator;
25
26template <size_t _Np>
27class _LIBCPP_HIDDEN __sso_allocator<void, _Np>
28{
29public:
30    typedef const void*       const_pointer;
31    typedef void              value_type;
32};
33
34template <class _Tp, size_t _Np>
35class _LIBCPP_HIDDEN __sso_allocator
36{
37    typename aligned_storage<sizeof(_Tp) * _Np>::type buf_;
38    bool __allocated_;
39public:
40    typedef size_t            size_type;
41    typedef _Tp*              pointer;
42    typedef _Tp               value_type;
43
44    _LIBCPP_INLINE_VISIBILITY __sso_allocator() throw() : __allocated_(false) {}
45    _LIBCPP_INLINE_VISIBILITY __sso_allocator(const __sso_allocator&) throw() : __allocated_(false) {}
46    template <class _Up> _LIBCPP_INLINE_VISIBILITY __sso_allocator(const __sso_allocator<_Up, _Np>&) throw()
47         : __allocated_(false) {}
48private:
49    __sso_allocator& operator=(const __sso_allocator&);
50public:
51    _LIBCPP_INLINE_VISIBILITY pointer allocate(size_type __n, typename __sso_allocator<void, _Np>::const_pointer = nullptr)
52    {
53        if (!__allocated_ && __n <= _Np)
54        {
55            __allocated_ = true;
56            return (pointer)&buf_;
57        }
58        return allocator<_Tp>().allocate(__n);
59    }
60    _LIBCPP_INLINE_VISIBILITY void deallocate(pointer __p, size_type __n)
61    {
62        if (__p == (pointer)&buf_)
63            __allocated_ = false;
64        else
65            allocator<_Tp>().deallocate(__p, __n);
66    }
67    _LIBCPP_INLINE_VISIBILITY size_type max_size() const throw() {return size_type(~0) / sizeof(_Tp);}
68
69    _LIBCPP_INLINE_VISIBILITY
70    bool operator==(__sso_allocator& __a) const {return &buf_ == &__a.buf_;}
71    _LIBCPP_INLINE_VISIBILITY
72    bool operator!=(__sso_allocator& __a) const {return &buf_ != &__a.buf_;}
73};
74
75_LIBCPP_END_NAMESPACE_STD
76
77#endif  // _LIBCPP___SSO_ALLOCATOR
78