Make it possible to use NeighborList without items

This commit is contained in:
Florian Bruhin 2014-02-20 20:45:30 +01:00
parent 7d878fec7b
commit 7d11790d35

View File

@ -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]