[314] | 1 | from typing import TypeVar, Generic, List
|
---|
| 2 | from tudelft.utilities.immutablelist.AbstractImmutableList import AbstractImmutableList
|
---|
| 3 | from tudelft.utilities.immutablelist.ImmutableList import ImmutableList
|
---|
| 4 |
|
---|
| 5 | E = TypeVar('E')
|
---|
| 6 |
|
---|
| 7 | class JoinedList(Generic[E], AbstractImmutableList[E]):
|
---|
| 8 | '''
|
---|
| 9 | Creates conjunction of lists. Eg Join of [1,2],[3,4],[5] -> [1,2,3,4,5]
|
---|
| 10 |
|
---|
| 11 | @param <E> type of the elements
|
---|
| 12 | '''
|
---|
| 13 |
|
---|
| 14 | def __init__(self, items: List[ImmutableList[E]]):
|
---|
| 15 | self._lists = items.copy()
|
---|
| 16 | self._size= sum([l.size() for l in items ] )
|
---|
| 17 |
|
---|
| 18 |
|
---|
| 19 | #Override
|
---|
| 20 | def get(self , index:int) ->E :
|
---|
| 21 | remaining = index
|
---|
| 22 |
|
---|
| 23 | for list in self._lists:
|
---|
| 24 | s = list.size()
|
---|
| 25 | if remaining < s:
|
---|
| 26 | return list.get(remaining)
|
---|
| 27 | remaining = remaining-s
|
---|
| 28 | raise IndexError("" + str(index))
|
---|
| 29 |
|
---|
| 30 | def With( self, list:ImmutableList[E]) -> "JoinedList[E]" :
|
---|
| 31 | '''
|
---|
| 32 | @param list the list to add
|
---|
| 33 | @return new JoinedList with the given list added at end of existing
|
---|
| 34 | lists.
|
---|
| 35 | '''
|
---|
| 36 | newlists = self._lists.copy()
|
---|
| 37 | newlists.append(list)
|
---|
| 38 | return JoinedList(newlists)
|
---|
| 39 |
|
---|
| 40 | #Override
|
---|
| 41 | def size(self) ->int:
|
---|
| 42 | return self._size
|
---|
| 43 |
|
---|
| 44 | #@Override
|
---|
| 45 | def __hash__(self):
|
---|
| 46 | return hash( (self._size, tuple(self._lists)))
|
---|
| 47 |
|
---|
| 48 | #Override
|
---|
| 49 | def __eq__(self, other):
|
---|
| 50 | return isinstance(other, self.__class__) \
|
---|
| 51 | and self._lists == other._lists
|
---|
| 52 |
|
---|