677f5de80267eea9a2c7a941b154f7b2811c46fe
[WebKit-https.git] / Source / bmalloc / bmalloc / SmallAllocator.h
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #ifndef SmallAllocator_h
27 #define SmallAllocator_h
28
29 #include "BAssert.h"
30 #include "SmallChunk.h"
31 #include "SmallLine.h"
32
33 namespace bmalloc {
34
35 // Helper object for allocating small objects.
36
37 class SmallAllocator {
38 public:
39     SmallAllocator();
40     SmallAllocator(size_t);
41     
42     bool isNull() { return !m_ptr; }
43     SmallLine* line();
44
45     bool canAllocate() { return !!m_remaining; }
46     void* allocate();
47
48     unsigned short objectCount();
49     unsigned char derefCount();
50
51     void refill(SmallLine*);
52     void clear();
53
54 private:
55     char* m_ptr;
56     unsigned short m_size;
57     unsigned char m_remaining;
58     unsigned char m_maxObjectCount;
59 };
60
61 inline SmallAllocator::SmallAllocator()
62     : m_ptr()
63     , m_size()
64     , m_remaining()
65     , m_maxObjectCount()
66 {
67 }
68
69 inline SmallAllocator::SmallAllocator(size_t size)
70     : m_ptr()
71     , m_size(size)
72     , m_remaining()
73     , m_maxObjectCount(smallLineSize / size)
74 {
75 }
76
77 inline SmallLine* SmallAllocator::line()
78 {
79     return SmallLine::get(canAllocate() ? m_ptr : m_ptr - 1);
80 }
81
82 inline void* SmallAllocator::allocate()
83 {
84     BASSERT(m_remaining);
85     BASSERT(m_size >= SmallLine::minimumObjectSize);
86
87     --m_remaining;
88     char* result = m_ptr;
89     m_ptr += m_size;
90     BASSERT(isSmall(result));
91     return result;
92 }
93
94 inline unsigned short SmallAllocator::objectCount()
95 {
96     return m_maxObjectCount - m_remaining;
97 }
98
99 inline unsigned char SmallAllocator::derefCount()
100 {
101     return SmallLine::maxRefCount - objectCount();
102 }
103
104 inline void SmallAllocator::refill(SmallLine* line)
105 {
106     BASSERT(!canAllocate());
107     line->concurrentRef(SmallLine::maxRefCount);
108     m_ptr = line->begin();
109     m_remaining = m_maxObjectCount;
110 }
111
112 inline void SmallAllocator::clear()
113 {
114     m_ptr = nullptr;
115     m_remaining = 0;
116 }
117
118 } // namespace bmalloc
119
120 #endif // SmallAllocator_h