d872598bd559807667b0f362e23630cfb8af2132
[WebKit-https.git] / Source / bmalloc / bmalloc / Algorithm.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 Algorithm_h
27 #define Algorithm_h
28
29 #include "BAssert.h"
30 #include <algorithm>
31 #include <cstdint>
32 #include <cstddef>
33 #include <limits>
34 #include <string.h>
35 #include <type_traits>
36 #include <chrono>
37
38 namespace bmalloc {
39
40 // Versions of min and max that are compatible with compile-time constants.
41 template<typename T> constexpr T max(T a, T b)
42 {
43     return a > b ? a : b;
44 }
45     
46 template<typename T> constexpr T min(T a, T b)
47 {
48     return a < b ? a : b;
49 }
50
51 template<typename T> constexpr T mask(T value, uintptr_t mask)
52 {
53     static_assert(sizeof(T) == sizeof(uintptr_t), "sizeof(T) must be equal to sizeof(uintptr_t).");
54     return static_cast<T>(static_cast<uintptr_t>(value) & mask);
55 }
56
57 template<typename T> inline T* mask(T* value, uintptr_t mask)
58 {
59     return reinterpret_cast<T*>(reinterpret_cast<uintptr_t>(value) & mask);
60 }
61
62 template<typename T> constexpr bool test(T value, uintptr_t mask)
63 {
64     return !!(reinterpret_cast<uintptr_t>(value) & mask);
65 }
66
67 template <typename T>
68 constexpr bool isPowerOfTwo(T size)
69 {
70     static_assert(std::is_integral<T>::value, "");
71     return size && !(size & (size - 1));
72 }
73
74 template<typename T> inline T roundUpToMultipleOf(size_t divisor, T x)
75 {
76     BASSERT(isPowerOfTwo(divisor));
77     static_assert(sizeof(T) == sizeof(uintptr_t), "sizeof(T) must be equal to sizeof(uintptr_t).");
78     return static_cast<T>((static_cast<uintptr_t>(x) + (divisor - 1)) & ~(divisor - 1));
79 }
80
81 template<size_t divisor, typename T> inline T roundUpToMultipleOf(T x)
82 {
83     static_assert(isPowerOfTwo(divisor), "'divisor' must be a power of two.");
84     return roundUpToMultipleOf(divisor, x);
85 }
86
87 template<typename T> inline T* roundUpToMultipleOf(size_t divisor, T* x)
88 {
89     BASSERT(isPowerOfTwo(divisor));
90     return reinterpret_cast<T*>((reinterpret_cast<uintptr_t>(x) + (divisor - 1)) & ~(divisor - 1));
91 }
92
93 template<size_t divisor, typename T> inline T* roundUpToMultipleOf(T* x)
94 {
95     static_assert(isPowerOfTwo(divisor), "'divisor' must be a power of two.");
96     return roundUpToMultipleOf(divisor, x);
97 }
98
99 template<typename T> inline T roundDownToMultipleOf(size_t divisor, T x)
100 {
101     BASSERT(isPowerOfTwo(divisor));
102     static_assert(sizeof(T) == sizeof(uintptr_t), "sizeof(T) must be equal to sizeof(uintptr_t).");
103     return static_cast<T>(mask(static_cast<uintptr_t>(x), ~(divisor - 1ul)));
104 }
105
106 template<typename T> inline T* roundDownToMultipleOf(size_t divisor, T* x)
107 {
108     BASSERT(isPowerOfTwo(divisor));
109     return reinterpret_cast<T*>(mask(reinterpret_cast<uintptr_t>(x), ~(divisor - 1ul)));
110 }
111
112 template<size_t divisor, typename T> constexpr T roundDownToMultipleOf(T x)
113 {
114     static_assert(isPowerOfTwo(divisor), "'divisor' must be a power of two.");
115     return roundDownToMultipleOf(divisor, x);
116 }
117
118 template<typename T> inline void divideRoundingUp(T numerator, T denominator, T& quotient, T& remainder)
119 {
120     // We expect the compiler to emit a single divide instruction to extract both the quotient and the remainder.
121     quotient = numerator / denominator;
122     remainder = numerator % denominator;
123     if (remainder)
124         quotient += 1;
125 }
126
127 template<typename T> inline T divideRoundingUp(T numerator, T denominator)
128 {
129     return (numerator + denominator - 1) / denominator;
130 }
131
132 template<typename T> inline T roundUpToMultipleOfNonPowerOfTwo(size_t divisor, T x)
133 {
134     return divideRoundingUp(x, divisor) * divisor;
135 }
136
137 // Version of sizeof that returns 0 for empty classes.
138
139 template<typename T> constexpr size_t sizeOf()
140 {
141     return std::is_empty<T>::value ? 0 : sizeof(T);
142 }
143
144 template<typename T> constexpr size_t bitCount()
145 {
146     return sizeof(T) * 8;
147 }
148
149 #if BOS(WINDOWS)
150 template<int depth> __forceinline constexpr unsigned long clzl(unsigned long value)
151 {
152     return value & (1UL << (bitCount<unsigned long>() - 1)) ? 0 : 1 + clzl<depth - 1>(value << 1);
153 }
154
155 template<> __forceinline constexpr unsigned long clzl<1>(unsigned long value)
156 {
157     return 0;
158 }
159
160 __forceinline constexpr unsigned long __builtin_clzl(unsigned long value)
161 {
162     return value == 0 ? 32 : clzl<bitCount<unsigned long>()>(value);
163 }
164 #endif
165
166 constexpr unsigned long log2(unsigned long value)
167 {
168     return bitCount<unsigned long>() - 1 - __builtin_clzl(value);
169 }
170
171 #define BOFFSETOF(class, field) (reinterpret_cast<ptrdiff_t>(&(reinterpret_cast<class*>(0x4000)->field)) - 0x4000)
172
173 template<typename T>
174 bool findBitInWord(T word, size_t& index, size_t endIndex, bool value)
175 {
176     static_assert(std::is_unsigned<T>::value, "Type used in findBitInWord must be unsigned");
177     
178     word >>= index;
179     
180     while (index < endIndex) {
181         if ((word & 1) == static_cast<T>(value))
182             return true;
183         index++;
184         word >>= 1;
185     }
186     
187     index = endIndex;
188     return false;
189 }
190
191 } // namespace bmalloc
192
193 #endif // Algorithm_h