-# The MIT License (MIT)
+# SPDX-License-Identifier: MIT
#
# Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-from bt2 import native_bt, object, utils
+
import collections.abc
-import bt2
+
+from bt2 import error as bt2_error
+from bt2 import utils as bt2_utils
+from bt2 import object as bt2_object
+from bt2 import native_bt
-class _IntegerRange:
- def __init__(self, lower, upper):
+class _IntegerRangeConst:
+ def __init__(self, lower, upper=None):
self._check_type(lower)
+
+ if upper is None:
+ upper = lower
+
self._check_type(upper)
if lower > upper:
- raise ValueError("range's lower bound ({}) is greater than its upper bound ({})".format(lower, upper))
+ raise ValueError(
+ "range's lower bound ({}) is greater than its upper bound ({})".format(
+ lower, upper
+ )
+ )
self._lower = lower
self._upper = upper
return value >= self._lower and value <= self._upper
def __eq__(self, other):
- if type(other) is not type(self):
+ if not isinstance(other, _IntegerRangeConst):
return False
return self.lower == other.lower and self.upper == other.upper
-class SignedIntegerRange(_IntegerRange):
- _check_type = staticmethod(utils._check_int64)
+class _IntegerRange(_IntegerRangeConst):
+ def __init__(self, lower, upper=None):
+ super().__init__(lower, upper)
-class UnsignedIntegerRange(_IntegerRange):
- _check_type = staticmethod(utils._check_uint64)
+class _SignedIntegerRangeConst(_IntegerRangeConst):
+ _is_type = staticmethod(bt2_utils._is_int64)
+ _check_type = staticmethod(bt2_utils._check_int64)
-class _IntegerRangeSet(object._SharedObject, collections.abc.MutableSet):
- def __init__(self, ranges=None):
- ptr = self._create_range_set()
+class SignedIntegerRange(_SignedIntegerRangeConst, _IntegerRange):
+ pass
- if ptr is None:
- raise bt2.CreationError('cannot create range set object')
- super().__init__(ptr)
+class _UnsignedIntegerRangeConst(_IntegerRangeConst):
+ _is_type = staticmethod(bt2_utils._is_uint64)
+ _check_type = staticmethod(bt2_utils._check_uint64)
- if ranges is not None:
- # will raise if not iterable
- for rg in ranges:
- self.add(rg)
+class UnsignedIntegerRange(_UnsignedIntegerRangeConst, _IntegerRange):
+ pass
+
+
+class _IntegerRangeSetConst(bt2_object._SharedObject, collections.abc.Set):
def __len__(self):
range_set_ptr = self._as_range_set_ptr(self._ptr)
count = native_bt.integer_range_set_get_range_count(range_set_ptr)
def __iter__(self):
for idx in range(len(self)):
- rg_ptr = self._borrow_range_by_index_ptr(self._ptr, idx)
+ rg_ptr = self._borrow_range_ptr_by_index(self._ptr, idx)
assert rg_ptr is not None
lower = self._range_get_lower(rg_ptr)
upper = self._range_get_upper(rg_ptr)
- yield self._range_type(lower, upper)
+ yield self._range_pycls(lower, upper)
def __eq__(self, other):
- if type(other) is not type(self):
+ if not isinstance(other, _IntegerRangeSetConst):
return False
- return self._compare(self._ptr, other._ptr)
+ return self._is_equal(self._ptr, other._ptr)
def contains_value(self, value):
for rg in self:
return False
+
+class _IntegerRangeSet(_IntegerRangeSetConst, collections.abc.MutableSet):
+ def __init__(self, ranges=None):
+ ptr = self._create_range_set()
+
+ if ptr is None:
+ raise bt2_error._MemoryError("cannot create range set object")
+
+ super().__init__(ptr)
+
+ if ranges is not None:
+ # will raise if not iterable
+ for rg in ranges:
+ self.add(rg)
+
def add(self, rg):
- if type(rg) is not self._range_type:
- # assume it's a simple pair (will raise if it's not)
- rg = self._range_type(rg[0], rg[1])
+ if type(rg) is not self._range_pycls:
+ if self._range_pycls._is_type(rg):
+ rg = self._range_pycls(rg)
+ else:
+ # assume it's a simple pair (will raise if it's not)
+ rg = self._range_pycls(rg[0], rg[1])
status = self._add_range(self._ptr, rg.lower, rg.upper)
- utils._handle_func_status(status,
- 'cannot add range to range set object')
+ bt2_utils._handle_func_status(status, "cannot add range to range set object")
def discard(self, rg):
raise NotImplementedError
-class SignedIntegerRangeSet(_IntegerRangeSet):
- _get_ref = staticmethod(native_bt.integer_range_set_signed_get_ref)
- _put_ref = staticmethod(native_bt.integer_range_set_signed_put_ref)
- _as_range_set_ptr = staticmethod(native_bt.integer_range_set_signed_as_range_set_const)
- _create_range_set = staticmethod(native_bt.integer_range_set_signed_create)
- _borrow_range_by_index_ptr = staticmethod(native_bt.integer_range_set_signed_borrow_range_by_index_const)
+class _SignedIntegerRangeSetConst(_IntegerRangeSetConst):
+ @staticmethod
+ def _get_ref(ptr):
+ native_bt.integer_range_set_signed_get_ref(ptr)
+
+ @staticmethod
+ def _put_ref(ptr):
+ native_bt.integer_range_set_signed_put_ref(ptr)
+
+ _as_range_set_ptr = staticmethod(
+ native_bt.integer_range_set_signed_as_range_set_const
+ )
+ _borrow_range_ptr_by_index = staticmethod(
+ native_bt.integer_range_set_signed_borrow_range_by_index_const
+ )
_range_get_lower = staticmethod(native_bt.integer_range_signed_get_lower)
_range_get_upper = staticmethod(native_bt.integer_range_signed_get_upper)
+ _is_equal = staticmethod(native_bt.integer_range_set_signed_is_equal)
+ _range_pycls = _SignedIntegerRangeConst
+
+
+class SignedIntegerRangeSet(_SignedIntegerRangeSetConst, _IntegerRangeSet):
+ _create_range_set = staticmethod(native_bt.integer_range_set_signed_create)
_add_range = staticmethod(native_bt.integer_range_set_signed_add_range)
- _compare = staticmethod(native_bt.integer_range_set_signed_compare)
- _range_type = SignedIntegerRange
+ _range_pycls = SignedIntegerRange
-class UnsignedIntegerRangeSet(_IntegerRangeSet):
- _get_ref = staticmethod(native_bt.integer_range_set_unsigned_get_ref)
- _put_ref = staticmethod(native_bt.integer_range_set_unsigned_put_ref)
- _as_range_set_ptr = staticmethod(native_bt.integer_range_set_unsigned_as_range_set_const)
- _create_range_set = staticmethod(native_bt.integer_range_set_unsigned_create)
- _borrow_range_by_index_ptr = staticmethod(native_bt.integer_range_set_unsigned_borrow_range_by_index_const)
+class _UnsignedIntegerRangeSetConst(_IntegerRangeSetConst):
+ @staticmethod
+ def _get_ref(ptr):
+ native_bt.integer_range_set_unsigned_get_ref(ptr)
+
+ @staticmethod
+ def _put_ref(ptr):
+ native_bt.integer_range_set_unsigned_put_ref(ptr)
+
+ _as_range_set_ptr = staticmethod(
+ native_bt.integer_range_set_unsigned_as_range_set_const
+ )
+ _borrow_range_ptr_by_index = staticmethod(
+ native_bt.integer_range_set_unsigned_borrow_range_by_index_const
+ )
_range_get_lower = staticmethod(native_bt.integer_range_unsigned_get_lower)
_range_get_upper = staticmethod(native_bt.integer_range_unsigned_get_upper)
+ _is_equal = staticmethod(native_bt.integer_range_set_unsigned_is_equal)
+ _range_pycls = _UnsignedIntegerRangeConst
+
+
+class UnsignedIntegerRangeSet(_UnsignedIntegerRangeSetConst, _IntegerRangeSet):
+ _create_range_set = staticmethod(native_bt.integer_range_set_unsigned_create)
_add_range = staticmethod(native_bt.integer_range_set_unsigned_add_range)
- _compare = staticmethod(native_bt.integer_range_set_unsigned_compare)
- _range_type = UnsignedIntegerRange
+ _range_pycls = UnsignedIntegerRange