Make it possible to use NeighborList without items
This commit is contained in:
parent
7d878fec7b
commit
7d11790d35
@ -37,7 +37,7 @@ class NeighborList:
|
|||||||
WRAP = 1
|
WRAP = 1
|
||||||
RAISE = 2
|
RAISE = 2
|
||||||
|
|
||||||
def __init__(self, items, default=_UNSET, mode=BLOCK):
|
def __init__(self, items=None, default=_UNSET, mode=RAISE):
|
||||||
"""Constructor.
|
"""Constructor.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
@ -49,7 +49,10 @@ class NeighborList:
|
|||||||
RAISE: Raise an IndexError.
|
RAISE: Raise an IndexError.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
self._items = items
|
if items is None:
|
||||||
|
self._items = []
|
||||||
|
else:
|
||||||
|
self._items = items
|
||||||
self._default = default
|
self._default = default
|
||||||
if default is not _UNSET:
|
if default is not _UNSET:
|
||||||
self.idx = self._items.index(default)
|
self.idx = self._items.index(default)
|
||||||
@ -74,6 +77,8 @@ class NeighborList:
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
# FIXME - zooming somehow wraps...
|
# FIXME - zooming somehow wraps...
|
||||||
|
if not self._items:
|
||||||
|
raise IndexError("No items found!")
|
||||||
try:
|
try:
|
||||||
if self.idx + offset > 0:
|
if self.idx + offset > 0:
|
||||||
new = self._items[self.idx + offset]
|
new = self._items[self.idx + offset]
|
||||||
|
Loading…
Reference in New Issue
Block a user