Tianqi Chen
36 lines
979 B
Python
36 lines
979 B
Python
# pylint: disable=protected-access, no-member
|
|
"""Arithmetic data structure and utility"""
|
|
from __future__ import absolute_import as _abs
|
|
|
|
from ._ctypes._node import NodeBase, register_node
|
|
from . import _api_internal
|
|
|
|
@register_node
|
|
class IntSet(NodeBase):
|
|
"""Represent a set of integer in one dimension."""
|
|
def is_nothing(self):
|
|
"""Whether the set represent nothing"""
|
|
return _api_internal._IntSetIsNothing(self)
|
|
|
|
def is_everything(self):
|
|
"""Whether the set represent everything"""
|
|
return _api_internal._IntSetIsEverything(self)
|
|
|
|
|
|
@register_node
|
|
class IntervalSet(IntSet):
|
|
"""Represent set of continuous interval"""
|
|
def min(self):
|
|
"""get the minimum value"""
|
|
return _api_internal._IntervalSetGetMin(self)
|
|
|
|
def max(self):
|
|
"""get the maximum value"""
|
|
return _api_internal._IntervalSetGetMax(self)
|
|
|
|
|
|
@register_node
|
|
class StrideSet(IntSet):
|
|
"""Represent set of strided integers"""
|
|
pass
|