islpy 2025.2.2__cp311-cp311-macosx_11_0_arm64.whl → 2025.2.5__cp311-cp311-macosx_11_0_arm64.whl

islpy/__init__.py

@@ -135,6 +135,7 @@ Alignable: TypeAlias = (
     | Aff | PwAff
 )
 AlignableT = TypeVar("AlignableT", bound=Alignable)
+AlignableT2 = TypeVar("AlignableT2", bound=Alignable)
 
 # }}}
 
@@ -151,11 +152,21 @@ def _get_default_context() -> Context:
 
 def _set_dim_id(obj: AlignableT, dt: dim_type, idx: int, id: Id) -> AlignableT:
     if isinstance(obj, BasicSet):
-        p, = obj.to_set().set_dim_id(dt, idx, id).get_basic_sets()
-        return cast("AlignableT", p)
+        s = obj.to_set().set_dim_id(dt, idx, id)
+        basicsets = s.get_basic_sets()
+        if not basicsets:
+            result = BasicSet.empty(s.space)
+        else:
+            result, = basicsets
+        return cast("AlignableT", result)
     elif isinstance(obj, BasicMap):
-        p, = obj.to_map().set_dim_id(dt, idx, id).get_basic_maps()
-        return cast("AlignableT", p)
+        m = obj.to_map().set_dim_id(dt, idx, id)
+        basicmaps = m.get_basic_maps()
+        if not basicmaps:
+            result = BasicMap.empty(m.space)
+        else:
+            result, = basicmaps
+        return cast("AlignableT", result)
 
     return cast("AlignableT", obj.set_dim_id(dt, idx, id))
 
@@ -304,8 +315,8 @@ def align_spaces(
 
 def align_two(
             obj1: AlignableT,
-            obj2: AlignableT,
-        ) -> tuple[AlignableT, AlignableT]:
+            obj2: AlignableT2,
+        ) -> tuple[AlignableT, AlignableT2]:
     """Align the spaces of two objects, potentially modifying both of them.
 
     See also :func:`align_spaces`.

islpy/_isl.pyi

@@ -1976,6 +1976,8 @@ class MultiVal:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -2558,6 +2560,8 @@ class Aff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -2900,6 +2904,8 @@ class PwAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -3048,6 +3054,8 @@ class UnionPwAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -3122,6 +3130,8 @@ class MultiId:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -3506,6 +3516,8 @@ class MultiAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -3826,6 +3838,8 @@ class PwMultiAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -3999,6 +4013,8 @@ class UnionPwMultiAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -4238,6 +4254,8 @@ class MultiPwAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -4429,6 +4447,8 @@ class MultiUnionPwAff:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -4557,6 +4577,8 @@ class Constraint:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -4926,6 +4948,8 @@ class LocalSpace:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -5461,13 +5485,13 @@ class BasicSet:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.set_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.set_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.set_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.set_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -5483,21 +5507,23 @@ class BasicSet:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.bset_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.bset_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.set_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.set_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.set_sub
 
     get_constraints = islpy._monkeypatch.basic_obj_get_constraints
 
@@ -6222,13 +6248,13 @@ class BasicMap:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.map_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.map_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.map_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.map_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -6242,21 +6268,23 @@ class BasicMap:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.bmap_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.bmap_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.map_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.map_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.map_sub
 
     get_constraints = islpy._monkeypatch.basic_obj_get_constraints
 
@@ -6726,13 +6754,13 @@ class Set:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.set_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.set_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.set_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.set_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -6748,21 +6776,23 @@ class Set:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.set_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.set_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.set_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.set_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.set_sub
 
     get_basic_sets = islpy._monkeypatch.set_get_basic_sets
 
@@ -7405,13 +7435,13 @@ class Map:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.map_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.map_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.map_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.map_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -7425,21 +7455,23 @@ class Map:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.map_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.map_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.map_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.map_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.map_sub
 
     get_basic_maps = islpy._monkeypatch.map_get_basic_maps
 
@@ -7624,6 +7656,8 @@ class UnionSet:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -7940,6 +7974,8 @@ class UnionMap:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -7988,6 +8024,8 @@ class Point:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -8162,6 +8200,8 @@ class QPolynomial:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -8333,6 +8373,8 @@ class PwQPolynomial:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -8391,6 +8433,8 @@ class QPolynomialFold:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -8522,6 +8566,8 @@ class PwQPolynomialFold:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str
@@ -8626,6 +8672,8 @@ class UnionPwQPolynomialFold:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __str__ = islpy._monkeypatch.generic_str
 
     __repr__ = islpy._monkeypatch.generic_repr
@@ -8743,6 +8791,8 @@ class UnionPwQPolynomial:
 
     get_var_names = islpy._monkeypatch.obj_get_var_names
 
+    get_var_names_not_none = islpy._monkeypatch.obj_get_var_names_not_none
+
     __reduce__ = islpy._monkeypatch.generic_reduce
 
     __str__ = islpy._monkeypatch.generic_str

islpy/_monkeypatch.py

@@ -5,7 +5,6 @@ from functools import update_wrapper
 from sys import intern
 from typing import (
     TYPE_CHECKING,
-    Any,
     ClassVar,
     Concatenate,
     Literal,
@@ -43,6 +42,10 @@ _CHECK_DIM_TYPES: tuple[_isl.dim_type, ...] = (
 
 # {{{ typing helpers
 
+T = TypeVar("T")
+P = ParamSpec("P")
+ResultT = TypeVar("ResultT")
+
 SelfT = TypeVar("SelfT")
 
 BasicT = TypeVar("BasicT", _isl.BasicSet, _isl.BasicMap)
@@ -54,36 +57,28 @@ ExprLikeT = TypeVar("ExprLikeT", _isl.Aff, _isl.PwAff,
                    _isl.QPolynomial, _isl.PwQPolynomial
                )
 SetLikeT = TypeVar("SetLikeT", bound=_isl.BasicSet | _isl.Set)
+
+SetOrBasic: TypeAlias = _isl.BasicSet | _isl.Set
+SetOrBasicT = TypeVar("SetOrBasicT", bound=SetOrBasic)
+
+MapOrBasic: TypeAlias = _isl.BasicMap | _isl.Map
+MapOrBasicT = TypeVar("MapOrBasicT", bound=MapOrBasic)
+
 SetOrMap: TypeAlias = _isl.BasicSet | _isl.Set | _isl.BasicMap | _isl.Map
-SetOrMapT = TypeVar("SetOrMapT", bound=SetOrMap)
+SetOrMapT = TypeVar("SetOrMapT", _isl.BasicSet, _isl.Set, _isl.BasicMap, _isl.Map)
 
 HasSpace: TypeAlias = (
     _isl.Space
-    | _isl.Aff
-    | _isl.BasicMap
-    | _isl.BasicSet
     | _isl.Constraint
     | _isl.LocalSpace
-    | _isl.Map
+    | _isl.Aff
     | _isl.MultiAff
-    | _isl.MultiId
-    | _isl.MultiPwAff
-    | _isl.MultiUnionPwAff
-    | _isl.MultiVal
-    | _isl.Point
     | _isl.PwAff
     | _isl.PwMultiAff
-    | _isl.PwQPolynomial
-    | _isl.PwQPolynomialFold
-    | _isl.QPolynomial
-    | _isl.QPolynomialFold
+    | _isl.BasicMap
+    | _isl.BasicSet
     | _isl.Set
-    | _isl.UnionMap
-    | _isl.UnionPwAff
-    | _isl.UnionPwMultiAff
-    | _isl.UnionPwQPolynomial
-    | _isl.UnionPwQPolynomialFold
-    | _isl.UnionSet
+    | _isl.Map
     )
 
 
@@ -101,34 +96,30 @@ class IslObject(Protocol):
 
 # {{{ copied verbatim from pytools to avoid numpy/pytools dependency
 
-F = TypeVar("F", bound=Callable[..., Any])
-
-
 class _HasKwargs:
     pass
 
 
-def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
+def _memoize_on_first_arg(
+        function: Callable[Concatenate[T, P], ResultT], *,
+        cache_dict_name: str | None = None) -> Callable[Concatenate[T, P], ResultT]:
     """Like :func:`memoize_method`, but for functions that take the object
     in which do memoization information is stored as first argument.
 
     Supports cache deletion via ``function_name.clear_cache(self)``.
     """
-    from sys import intern
 
     if cache_dict_name is None:
         cache_dict_name = intern(
                 f"_memoize_dic_{function.__module__}{function.__name__}"
                 )
 
-    def wrapper(obj, *args, **kwargs):
-        if kwargs:
-            key = (_HasKwargs, frozenset(kwargs.items()), *args)
-        else:
-            key = args
+    def wrapper(obj: T, *args: P.args, **kwargs: P.kwargs) -> ResultT:
+        key = (_HasKwargs, frozenset(kwargs.items()), *args) if kwargs else args
 
+        assert cache_dict_name is not None
         try:
-            return getattr(obj, cache_dict_name)[key]
+            return cast("ResultT", getattr(obj, cache_dict_name)[key])
         except AttributeError:
             attribute_error = True
         except KeyError:
@@ -138,11 +129,10 @@ def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
         if attribute_error:
             object.__setattr__(obj, cache_dict_name, {key: result})
             return result
-        else:
-            getattr(obj, cache_dict_name)[key] = result
-            return result
+        getattr(obj, cache_dict_name)[key] = result
+        return result
 
-    def clear_cache(obj):
+    def clear_cache(obj: object):
         object.__delattr__(obj, cache_dict_name)
 
     from functools import update_wrapper
@@ -152,7 +142,8 @@ def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
     # into the function's dict is moderately sketchy.
     new_wrapper.clear_cache = clear_cache  # type: ignore[attr-defined]
 
-    return cast("F", new_wrapper)
+    return new_wrapper
+
 
 # }}}
 
@@ -190,7 +181,7 @@ def context_ne(self: object, other: object) -> bool:
     return not self.__eq__(other)
 
 
-def generic_reduce(self: IslObject):
+def generic_reduce(self: HasSpace):
     ctx = self.get_ctx()
     prn = _isl.Printer.to_str(ctx)
     prn = getattr(prn, f"print_{self._base_name}")(self)
@@ -331,21 +322,100 @@ def space_create_from_names(
     return result
 
 
-def obj_or(self: SetOrMapT, other: object) -> SetOrMapT:
+def set_or(
+            self: _isl.Set,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    try:
+        return self.union(other)
+    except TypeError:
+        return NotImplemented
+
+
+def bset_and(
+            self: _isl.BasicSet,
+            other: SetOrBasicT,
+        ) -> SetOrBasicT:
+    if isinstance(other, _isl.Set):
+        try:
+            return self.to_set().intersect(other)
+        except TypeError:
+            return NotImplemented
+    else:
+        try:
+            return self.intersect(other)
+        except TypeError:
+            return NotImplemented
+
+
+def set_and(
+            self: _isl.Set,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    if isinstance(self, _isl.BasicSet):
+        self = self.to_set()
+    try:
+        return self.intersect(other)
+    except TypeError:
+        return NotImplemented
+
+
+def set_sub(
+            self: _isl.Set | _isl.BasicSet,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    if isinstance(self, _isl.BasicSet):
+        self = self.to_set()
+    try:
+        return self.subtract(other)
+    except TypeError:
+        return NotImplemented
+
+
+def map_or(
+            self: _isl.Map | _isl.BasicMap,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
+    if isinstance(self, _isl.BasicMap):
+        self = self.to_map()
     try:
         return self.union(other)
     except TypeError:
         return NotImplemented
 
 
-def obj_and(self: SetOrMapT, other: object) -> SetOrMapT:
+def bmap_and(
+            self: _isl.BasicMap,
+            other: MapOrBasicT,
+        ) -> MapOrBasicT:
+    if isinstance(other, _isl.Map):
+        try:
+            return self.to_map().intersect(other)
+        except TypeError:
+            return NotImplemented
+    else:
+        try:
+            return self.intersect(other)
+        except TypeError:
+            return NotImplemented
+
+
+def map_and(
+            self: _isl.Map,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
     try:
         return self.intersect(other)
     except TypeError:
         return NotImplemented
 
 
-def obj_sub(self: SetOrMapT, other: object) -> SetOrMapT:
+def map_sub(
+            self: _isl.Map | _isl.BasicMap,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
+    if isinstance(self, _isl.BasicMap):
+        self = self.to_map()
     try:
         return self.subtract(other)
     except TypeError:
@@ -538,16 +608,42 @@ def obj_get_var_dict(
 def obj_get_var_ids(
             self: HasSpace,
             dimtype: _isl.dim_type
-        ) -> Sequence[str]:
+        ) -> Sequence[str | None]:
     """Return a list of :class:`Id` instances for :class:`dim_type` *dimtype*."""
-    return [self.get_dim_name(dimtype, i) for i in range(self.dim(dimtype))]
+    return [
+        self.get_dim_name(dimtype, i)
+        for i in range(self.dim(dimtype))]
+
+
+@_memoize_on_first_arg
+def obj_get_var_names_not_none(
+            self: HasSpace,
+            dimtype: _isl.dim_type,
+        ) -> Sequence[str]:
+    """Return a list of dim names (in order) for :class:`dim_type` *dimtype*.
+
+    Raise :exc:`ValueError` if any of the names is *None*.
+
+    .. versionadded:: 2025.2.5
+    """
+    ndim = self.dim(dimtype)
+    res = [n
+        for i in range(ndim)
+        if (n := self.get_dim_name(dimtype, i)) is not None]
+    if len(res) != ndim:
+        raise ValueError("None encountered in dim names")
+    return res
 
 
 @_memoize_on_first_arg
-def obj_get_var_names(self: HasSpace, dimtype: _isl.dim_type) -> Sequence[str]:
-    """Return a list of dim names (in order) for :class:`dim_type` *dimtype*."""
+def obj_get_var_names(
+            self: HasSpace,
+            dimtype: _isl.dim_type,
+        ) -> Sequence[str | None]:
+    """Return a list of dim names (in order) for :class:`dim_type` *dimtype*.
+    """
     return [self.get_dim_name(dimtype, i)
-        for i in range(self.dim(dimtype))]
+            for i in range(self.dim(dimtype))]
 
 
 def pwaff_get_pieces(self: _isl.PwAff | _isl.Aff) -> list[tuple[_isl.Set, _isl.Aff]]:
@@ -585,7 +681,7 @@ def pw_get_aggregate_domain(self: _isl.PwAff | _isl.PwQPolynomial) -> _isl.Set:
 
     result = _isl.Set.empty(self.get_domain_space())
     for dom, _ in self.get_pieces():
-        result = result.union(cast("_isl.Set", dom))
+        result = result.union(dom)
 
     return result
 
@@ -657,7 +753,7 @@ def expr_like_add(self: ExprLikeT, other: ExprLikeT | int | _isl.Val) -> ExprLik
         return NotImplemented
 
 
-def expr_like_sub(self: ExprLikeT, other: ExprLikeT | int | _isl.Val):
+def expr_like_sub(self: ExprLikeT, other: ExprLikeT | int | _isl.Val) -> ExprLikeT:
     if not isinstance(other, ExprLike):
         other = _number_to_expr_like(self, other)
 
@@ -727,29 +823,45 @@ for cls in ALL_CLASSES:
         cls.__ne__ = obj_ne
 
 
-def obj_lt(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_lt(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return self.is_strict_subset(other)
 
 
-def obj_le(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_le(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return self.is_subset(other)
 
 
-def obj_gt(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_gt(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return other.is_strict_subset(self)
 
 
-def obj_ge(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_ge(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
+    return other.is_subset(self)
+
+
+def map_lt(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return self.is_strict_subset(other)
+
+
+def map_le(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return self.is_subset(other)
+
+
+def map_gt(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return other.is_strict_subset(self)
+
+
+def map_ge(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
     return other.is_subset(self)
 
 
 # {{{ project_out_except
 
 def obj_project_out_except(
-            obj: SetLikeT,
+            obj: SetOrMapT,
             names: Collection[str],
             types: Collection[_isl.dim_type]
-        ) -> SetLikeT:
+        ) -> SetOrMapT:
     """
     :param types: list of :class:`dim_type` determining
         the types of axes to project out
@@ -786,10 +898,10 @@ def obj_project_out_except(
 # {{{ eliminate_except
 
 def obj_eliminate_except(
-            obj: SetLikeT,
+            obj: SetOrMapT,
             names: Collection[str],
             types: Collection[_isl.dim_type]
-        ) -> SetLikeT:
+        ) -> SetOrMapT:
     """
     :param types: list of :class:`dim_type` determining
         the types of axes to eliminate
@@ -871,12 +983,23 @@ def _add_functionality() -> None:
 
     # {{{ Python set-like behavior
 
-    for cls in [_isl.BasicSet, _isl.BasicMap, _isl.Set, _isl.Map]:
-        cls.__or__ = obj_or
-        cls.__ror__ = obj_or
-        cls.__and__ = obj_and
-        cls.__rand__ = obj_and
-        cls.__sub__ = obj_sub
+    _isl.BasicSet.__and__ = bset_and
+    _isl.BasicSet.__rand__ = bset_and
+    _isl.Set.__and__ = set_and
+    _isl.Set.__rand__ = set_and
+    for cls in [_isl.BasicSet, _isl.Set]:
+        cls.__or__ = set_or
+        cls.__ror__ = set_or
+        cls.__sub__ = set_sub
+
+    _isl.BasicMap.__and__ = bmap_and
+    _isl.BasicMap.__rand__ = bmap_and
+    _isl.Map.__and__ = map_and
+    _isl.Map.__rand__ = map_and
+    for cls in [_isl.BasicMap, _isl.Map]:
+        cls.__or__ = map_or
+        cls.__ror__ = map_or
+        cls.__sub__ = map_sub
 
     # }}}
 
@@ -938,6 +1061,7 @@ for cls in ALL_CLASSES:
         cls.get_var_dict = obj_get_var_dict
         cls.get_var_ids = obj_get_var_ids
         cls.get_var_names = obj_get_var_names
+        cls.get_var_names_not_none = obj_get_var_names_not_none
 
     # }}}
 
@@ -1005,11 +1129,17 @@ for cls in ALL_CLASSES:
 
     # {{{ rich comparisons
 
-    for cls in [_isl.BasicSet, _isl.BasicMap, _isl.Set, _isl.Map]:
-        cls.__lt__ = obj_lt
-        cls.__le__ = obj_le
-        cls.__gt__ = obj_gt
-        cls.__ge__ = obj_ge
+    for cls in [_isl.BasicSet, _isl.Set]:
+        cls.__lt__ = set_lt
+        cls.__le__ = set_le
+        cls.__gt__ = set_gt
+        cls.__ge__ = set_ge
+
+    for cls in [_isl.BasicMap, _isl.Map]:
+        cls.__lt__ = map_lt
+        cls.__le__ = map_le
+        cls.__gt__ = map_gt
+        cls.__ge__ = map_ge
 
     # }}}
 
@@ -1024,10 +1154,6 @@ for cls in ALL_CLASSES:
 _add_functionality()
 
 
-P = ParamSpec("P")
-ResultT = TypeVar("ResultT")
-
-
 _DOWNCAST_RE = re.compile(
           r"Downcast from :class:`([A-Za-z]+)` to :class:`([A-Za-z]+)`.")
 

{islpy-2025.2.2.dist-info → islpy-2025.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: islpy
-Version: 2025.2.2
+Version: 2025.2.5
 Summary: Wrapper around isl, an integer set library
 Author-Email: Andreas Kloeckner <inform@tiker.net>
 License-Expression: MIT

islpy-2025.2.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+islpy/version.py,sha256=9KVEg7OX9uCaA6ElUMeiAh-zDrxsy3u2H6VQm_-LTUE,301
+islpy/__init__.py,sha256=0AdPhWQMf1DUB5DDisLP39rN0djxueftEWqfX4_R5Hk,13363
+islpy/_isl.pyi,sha256=c9l3pXn3DEtGO0WscxXu6RsfK1WseF0hdE7oFncXUnA,262025
+islpy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+islpy/_isl.cpython-311-darwin.so,sha256=iI6GktRqAvFJUPu5v3kMLW_sEdqLFMmdCzXpNQSEE4I,7910776
+islpy/_monkeypatch.py,sha256=3ji2nD8gYxY59qwjJiasmeKOI0oSbrxEiXbswjmJXoU,34881
+islpy-2025.2.5.dist-info/RECORD,,
+islpy-2025.2.5.dist-info/WHEEL,sha256=7CikrUOKtiy-2ivl-sZ5UBq05y8ywDmf9_HO2E_GAYY,141
+islpy-2025.2.5.dist-info/METADATA,sha256=dFfe1_neHbSBhU6nwxnsMoO3bNIt1zonZaKdTbVDIbw,2966

islpy-2025.2.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-islpy/version.py,sha256=9KVEg7OX9uCaA6ElUMeiAh-zDrxsy3u2H6VQm_-LTUE,301
-islpy/__init__.py,sha256=di9XuiCE0E17rYvJF2odx710bD4X07gsKGWDW6oSgH0,13021
-islpy/_isl.pyi,sha256=_QvIK3y21H8sud47dr3bBOfxkSONwZzxV1x09WIxtRQ,260121
-islpy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-islpy/_isl.cpython-311-darwin.so,sha256=iI6GktRqAvFJUPu5v3kMLW_sEdqLFMmdCzXpNQSEE4I,7910776
-islpy/_monkeypatch.py,sha256=J_EIwyHBinfCo_rBu1V9o1OzG9R3dE34lJ0XIMOJSQ4,31309
-islpy-2025.2.2.dist-info/RECORD,,
-islpy-2025.2.2.dist-info/WHEEL,sha256=7CikrUOKtiy-2ivl-sZ5UBq05y8ywDmf9_HO2E_GAYY,141
-islpy-2025.2.2.dist-info/METADATA,sha256=Us2hmRlX7V2rNYGp3pg3ZEqYgbSbuzNYT7IIHd9dbT4,2966