Commit | Line | Data |
---|---|---|
64ded3fb PP |
1 | /* |
2 | * Copyright (c) 2020 Philippe Proulx <pproulx@efficios.com> | |
3 | * | |
4 | * SPDX-License-Identifier: MIT | |
5 | */ | |
6 | ||
7 | #ifndef BABELTRACE_CPP_COMMON_BT2_INTEGER_RANGE_SET_HPP | |
8 | #define BABELTRACE_CPP_COMMON_BT2_INTEGER_RANGE_SET_HPP | |
9 | ||
10 | #include <cstdint> | |
11 | #include <type_traits> | |
c802cacb | 12 | |
64ded3fb PP |
13 | #include <babeltrace2/babeltrace.h> |
14 | ||
0d218157 | 15 | #include "borrowed-object.hpp" |
8aee46a3 | 16 | #include "common-iterator.hpp" |
c802cacb SM |
17 | #include "exc.hpp" |
18 | #include "integer-range.hpp" | |
64ded3fb | 19 | #include "internal/utils.hpp" |
7f5cdaf0 | 20 | #include "shared-object.hpp" |
64ded3fb PP |
21 | |
22 | namespace bt2 { | |
64ded3fb PP |
23 | namespace internal { |
24 | ||
25 | template <typename LibObjT> | |
26 | struct IntegerRangeSetRefFuncs; | |
27 | ||
28 | template <> | |
29 | struct IntegerRangeSetRefFuncs<const bt_integer_range_set_unsigned> final | |
30 | { | |
c677c492 | 31 | static void get(const bt_integer_range_set_unsigned * const libObjPtr) noexcept |
64ded3fb PP |
32 | { |
33 | bt_integer_range_set_unsigned_get_ref(libObjPtr); | |
34 | } | |
35 | ||
c677c492 | 36 | static void put(const bt_integer_range_set_unsigned * const libObjPtr) noexcept |
64ded3fb | 37 | { |
1796d135 | 38 | bt_integer_range_set_unsigned_put_ref(libObjPtr); |
64ded3fb PP |
39 | } |
40 | }; | |
41 | ||
42 | template <> | |
43 | struct IntegerRangeSetRefFuncs<const bt_integer_range_set_signed> final | |
44 | { | |
c677c492 | 45 | static void get(const bt_integer_range_set_signed * const libObjPtr) noexcept |
64ded3fb PP |
46 | { |
47 | bt_integer_range_set_signed_get_ref(libObjPtr); | |
48 | } | |
49 | ||
c677c492 | 50 | static void put(const bt_integer_range_set_signed * const libObjPtr) noexcept |
64ded3fb | 51 | { |
1796d135 | 52 | bt_integer_range_set_signed_put_ref(libObjPtr); |
64ded3fb PP |
53 | } |
54 | }; | |
55 | ||
56 | template <typename LibObjT> | |
57 | struct CommonIntegerRangeSetSpec; | |
58 | ||
b5f55e9f | 59 | /* Functions specific to unsigned integer range sets */ |
64ded3fb PP |
60 | template <> |
61 | struct CommonIntegerRangeSetSpec<const bt_integer_range_set_unsigned> final | |
62 | { | |
c0b73c63 | 63 | static std::uint64_t length(const bt_integer_range_set_unsigned * const libRangePtr) noexcept |
64ded3fb PP |
64 | { |
65 | return bt_integer_range_set_get_range_count( | |
66 | bt_integer_range_set_unsigned_as_range_set_const(libRangePtr)); | |
67 | } | |
68 | ||
69 | static const bt_integer_range_unsigned * | |
70 | rangeByIndex(const bt_integer_range_set_unsigned * const libRangePtr, | |
71 | const std::uint64_t index) noexcept | |
72 | { | |
73 | return bt_integer_range_set_unsigned_borrow_range_by_index_const(libRangePtr, index); | |
74 | } | |
75 | ||
76 | static bool isEqual(const bt_integer_range_set_unsigned * const libRangePtrA, | |
77 | const bt_integer_range_set_unsigned * const libRangePtrB) noexcept | |
78 | { | |
79 | return static_cast<bool>( | |
80 | bt_integer_range_set_unsigned_is_equal(libRangePtrA, libRangePtrB)); | |
81 | } | |
82 | ||
83 | static bt_integer_range_set_add_range_status | |
84 | addRange(bt_integer_range_set_unsigned * const libRangePtr, const std::uint64_t lower, | |
85 | const std::uint64_t upper) noexcept | |
86 | { | |
87 | return bt_integer_range_set_unsigned_add_range(libRangePtr, lower, upper); | |
88 | } | |
89 | ||
90 | static bt_integer_range_set_unsigned *create() noexcept | |
91 | { | |
92 | return bt_integer_range_set_unsigned_create(); | |
93 | } | |
94 | }; | |
95 | ||
b5f55e9f | 96 | /* Functions specific to signed integer range sets */ |
64ded3fb PP |
97 | template <> |
98 | struct CommonIntegerRangeSetSpec<const bt_integer_range_set_signed> final | |
99 | { | |
c0b73c63 | 100 | static std::uint64_t length(const bt_integer_range_set_signed * const libRangePtr) noexcept |
64ded3fb PP |
101 | { |
102 | return bt_integer_range_set_get_range_count( | |
103 | bt_integer_range_set_signed_as_range_set_const(libRangePtr)); | |
104 | } | |
105 | ||
106 | static const bt_integer_range_signed * | |
107 | rangeByIndex(const bt_integer_range_set_signed * const libRangePtr, | |
108 | const std::uint64_t index) noexcept | |
109 | { | |
110 | return bt_integer_range_set_signed_borrow_range_by_index_const(libRangePtr, index); | |
111 | } | |
112 | ||
113 | static bool isEqual(const bt_integer_range_set_signed * const libRangePtrA, | |
114 | const bt_integer_range_set_signed * const libRangePtrB) noexcept | |
115 | { | |
116 | return static_cast<bool>(bt_integer_range_set_signed_is_equal(libRangePtrA, libRangePtrB)); | |
117 | } | |
118 | ||
119 | static bt_integer_range_set_add_range_status | |
120 | addRange(bt_integer_range_set_signed * const libRangePtr, const std::int64_t lower, | |
121 | const std::int64_t upper) noexcept | |
122 | { | |
123 | return bt_integer_range_set_signed_add_range(libRangePtr, lower, upper); | |
124 | } | |
125 | ||
126 | static bt_integer_range_set_signed *create() noexcept | |
127 | { | |
128 | return bt_integer_range_set_signed_create(); | |
129 | } | |
130 | }; | |
131 | ||
b5f55e9f | 132 | } /* namespace internal */ |
64ded3fb PP |
133 | |
134 | template <typename LibObjT> | |
0d218157 | 135 | class CommonIntegerRangeSet final : public BorrowedObject<LibObjT> |
64ded3fb | 136 | { |
64ded3fb | 137 | private: |
0d218157 PP |
138 | using typename BorrowedObject<LibObjT>::_ThisBorrowedObject; |
139 | using typename BorrowedObject<LibObjT>::_LibObjPtr; | |
64ded3fb PP |
140 | using _ConstLibObjT = typename std::add_const<LibObjT>::type; |
141 | using _RefFuncs = internal::IntegerRangeSetRefFuncs<_ConstLibObjT>; | |
142 | using _Spec = internal::CommonIntegerRangeSetSpec<_ConstLibObjT>; | |
143 | using _ThisCommonIntegerRangeSet = CommonIntegerRangeSet<LibObjT>; | |
144 | ||
145 | public: | |
7f5cdaf0 | 146 | using Shared = SharedObject<_ThisCommonIntegerRangeSet, LibObjT, _RefFuncs>; |
64ded3fb PP |
147 | |
148 | using Range = typename std::conditional< | |
149 | std::is_same<_ConstLibObjT, const bt_integer_range_set_unsigned>::value, | |
150 | ConstUnsignedIntegerRange, ConstSignedIntegerRange>::type; | |
151 | ||
152 | using Value = typename Range::Value; | |
70cfba94 | 153 | using Iterator = CommonIterator<CommonIntegerRangeSet, Range>; |
64ded3fb PP |
154 | |
155 | explicit CommonIntegerRangeSet(const _LibObjPtr libObjPtr) noexcept : | |
0d218157 | 156 | _ThisBorrowedObject {libObjPtr} |
64ded3fb PP |
157 | { |
158 | } | |
159 | ||
160 | static Shared create() | |
161 | { | |
162 | const auto libObjPtr = _Spec::create(); | |
163 | ||
164 | internal::validateCreatedObjPtr(libObjPtr); | |
c9c0b6e2 | 165 | return CommonIntegerRangeSet::Shared::createWithoutRef(libObjPtr); |
64ded3fb PP |
166 | } |
167 | ||
168 | template <typename OtherLibObjT> | |
100fa861 | 169 | CommonIntegerRangeSet(const CommonIntegerRangeSet<OtherLibObjT> rangeSet) noexcept : |
0d218157 | 170 | _ThisBorrowedObject {rangeSet} |
64ded3fb PP |
171 | { |
172 | } | |
173 | ||
174 | template <typename OtherLibObjT> | |
175 | _ThisCommonIntegerRangeSet& | |
100fa861 | 176 | operator=(const CommonIntegerRangeSet<OtherLibObjT> rangeSet) noexcept |
64ded3fb | 177 | { |
0d218157 | 178 | _ThisBorrowedObject::operator=(rangeSet); |
64ded3fb PP |
179 | return *this; |
180 | } | |
181 | ||
328a274a PP |
182 | _ConstLibObjT asConst() const noexcept |
183 | { | |
184 | return _ConstLibObjT {*this}; | |
185 | } | |
186 | ||
64ded3fb | 187 | template <typename OtherLibObjT> |
100fa861 | 188 | bool operator==(const CommonIntegerRangeSet<OtherLibObjT> other) const noexcept |
64ded3fb | 189 | { |
341a67c4 | 190 | return _Spec::isEqual(this->libObjPtr(), other.libObjPtr()); |
64ded3fb PP |
191 | } |
192 | ||
193 | template <typename OtherLibObjT> | |
100fa861 | 194 | bool operator!=(const CommonIntegerRangeSet<OtherLibObjT> other) const noexcept |
64ded3fb PP |
195 | { |
196 | return !(*this == other); | |
197 | } | |
198 | ||
dcb8ae9b | 199 | void addRange(const Value lower, const Value upper) const |
64ded3fb | 200 | { |
5c895f64 PP |
201 | static_assert( |
202 | !std::is_const<LibObjT>::value, | |
203 | "Not available with `bt2::ConstUnsignedIntegerRangeSet` or `bt2::ConstSignedIntegerRangeSet`."); | |
64ded3fb | 204 | |
341a67c4 | 205 | const auto status = _Spec::addRange(this->libObjPtr(), lower, upper); |
64ded3fb PP |
206 | |
207 | if (status == BT_INTEGER_RANGE_SET_ADD_RANGE_STATUS_MEMORY_ERROR) { | |
39278ebc | 208 | throw MemoryError {}; |
64ded3fb PP |
209 | } |
210 | } | |
211 | ||
c0b73c63 | 212 | std::uint64_t length() const noexcept |
64ded3fb | 213 | { |
c0b73c63 | 214 | return _Spec::length(this->libObjPtr()); |
64ded3fb PP |
215 | } |
216 | ||
217 | Range operator[](const std::uint64_t index) const noexcept | |
218 | { | |
341a67c4 | 219 | return Range {_Spec::rangeByIndex(this->libObjPtr(), index)}; |
64ded3fb PP |
220 | } |
221 | ||
70cfba94 FD |
222 | Iterator begin() const noexcept |
223 | { | |
224 | return Iterator {*this, 0}; | |
225 | } | |
226 | ||
227 | Iterator end() const noexcept | |
228 | { | |
c0b73c63 | 229 | return Iterator {*this, this->length()}; |
70cfba94 FD |
230 | } |
231 | ||
64ded3fb PP |
232 | Shared shared() const noexcept |
233 | { | |
c9c0b6e2 | 234 | return Shared::createWithRef(*this); |
64ded3fb PP |
235 | } |
236 | }; | |
237 | ||
238 | using UnsignedIntegerRangeSet = CommonIntegerRangeSet<bt_integer_range_set_unsigned>; | |
239 | using ConstUnsignedIntegerRangeSet = CommonIntegerRangeSet<const bt_integer_range_set_unsigned>; | |
240 | using SignedIntegerRangeSet = CommonIntegerRangeSet<bt_integer_range_set_signed>; | |
241 | using ConstSignedIntegerRangeSet = CommonIntegerRangeSet<const bt_integer_range_set_signed>; | |
242 | ||
4927bae7 PP |
243 | namespace internal { |
244 | ||
245 | struct UnsignedIntegerRangeSetTypeDescr | |
246 | { | |
247 | using Const = ConstUnsignedIntegerRangeSet; | |
248 | using NonConst = UnsignedIntegerRangeSet; | |
249 | }; | |
250 | ||
251 | template <> | |
252 | struct TypeDescr<UnsignedIntegerRangeSet> : public UnsignedIntegerRangeSetTypeDescr | |
253 | { | |
254 | }; | |
255 | ||
256 | template <> | |
257 | struct TypeDescr<ConstUnsignedIntegerRangeSet> : public UnsignedIntegerRangeSetTypeDescr | |
258 | { | |
259 | }; | |
260 | ||
261 | struct SignedIntegerRangeSetTypeDescr | |
262 | { | |
263 | using Const = ConstSignedIntegerRangeSet; | |
264 | using NonConst = SignedIntegerRangeSet; | |
265 | }; | |
266 | ||
267 | template <> | |
268 | struct TypeDescr<SignedIntegerRangeSet> : public SignedIntegerRangeSetTypeDescr | |
269 | { | |
270 | }; | |
271 | ||
272 | template <> | |
273 | struct TypeDescr<ConstSignedIntegerRangeSet> : public SignedIntegerRangeSetTypeDescr | |
274 | { | |
275 | }; | |
276 | ||
277 | } /* namespace internal */ | |
b5f55e9f | 278 | } /* namespace bt2 */ |
64ded3fb | 279 | |
b5f55e9f | 280 | #endif /* BABELTRACE_CPP_COMMON_BT2_INTEGER_RANGE_SET_HPP */ |