codon/stdlib/collections.codon

class deque[T]:
    _arr: Array[T]
    _head: int
    _tail: int
    _maxlen: int

    def __init__(self, arr: Array[T], head: int, tail: int, maxlen: int):
        self._arr = arr
        self._head = head
        self._tail = tail
        self._maxlen = maxlen

    def __init__(self):
        self._arr = Array[T](16)
        self._head = 0
        self._tail = 0
        self._maxlen = -1

    def __init__(self, maxlen: int):
        cap = 1
        while cap < maxlen:
            cap *= 2
        self._arr = Array[T](cap)
        self._head = 0
        self._tail = 0
        self._maxlen = maxlen

    def __init__(self, it: Generator[T]):
        self._arr = Array[T](16)
        self._head = 0
        self._tail = 0
        self._maxlen = -1
        for i in it:
            self.append(i)

    @property
    def maxlen(self):
        return self._maxlen

    def _double_cap(self):
        p = self._head
        n = len(self._arr)
        r = n - p
        new_cap = n * 2
        new_arr = Array[T](new_cap)
        for i in range(r):
            new_arr[i] = self._arr[p + i]
        for i in range(p):
            new_arr[i + r] = self._arr[i]
        self._arr = new_arr
        self._head = 0
        self._tail = n

    def _check_not_empty(self):
        if not self:
            raise IndexError("pop from an empty deque")

    def __bool__(self):
        return self._head != self._tail

    def __len__(self):
        return (self._tail - self._head) & (len(self._arr) - 1)

    def appendleft(self, x: T):
        self._head = (self._head - 1) & (len(self._arr) - 1)
        self._arr[self._head] = x
        if self._maxlen >= 0 and len(self) > self._maxlen:
            self.pop()
        if self._head == self._tail:
            self._double_cap()

    def append(self, x: T):
        self._arr[self._tail] = x
        self._tail = (self._tail + 1) & (len(self._arr) - 1)
        if self._maxlen >= 0 and len(self) > self._maxlen:
            self.popleft()
        if self._head == self._tail:
            self._double_cap()

    def popleft(self):
        self._check_not_empty()
        res = self._arr[self._head]
        self._head = (self._head + 1) & (len(self._arr) - 1)
        return res

    def pop(self):
        self._check_not_empty()
        self._tail = (self._tail - 1) & (len(self._arr) - 1)
        return self._arr[self._tail]

    def clear(self):
        self._head = 0
        self._tail = 0

    def __iter__(self):
        i = self._head
        while i != self._tail:
            yield self._arr[i]
            i = (i + 1) & (len(self._arr) - 1)

    def __contains__(self, x: T):
        for i in self:
            if i == x:
                return True
        return False

    def __deepcopy__(self):
        return deque(i.__deepcopy__() for i in self)

    def __copy__(self):
        return deque[T](copy(self._arr), self._head, self._tail, self._maxlen)

    def copy(self):
        return self.__copy__()

    def __repr__(self):
        return repr(List[T](iter(self)))

    def _idx_check(self, idx: int, msg: str):
        if self._head == self._tail or idx >= len(self) or idx < 0:
            raise IndexError(msg)

    @property
    def left(self):
        self._idx_check(0, "list index out of range")
        return self._arr[self._head]

    def __getitem__(self, idx: int):
        if idx < 0:
            idx += len(self)
        self._idx_check(idx, "list index out of range")
        if self._head <= self._tail:
            return self._arr[self._head + idx]
        elif self._head + idx < len(self._arr):
            return self._arr[self._head + idx]
        else:
            idx -= len(self._arr) - self._head
            assert 0 <= idx < self._tail
            return self._arr[idx]

@tuple
class _CounterItem[T]:
    element: T
    count: int

    def __eq__(self, other: _CounterItem[T]):
        return self.count == other.count

    def __ne__(self, other: _CounterItem[T]):
        return self.count != other.count

    def __lt__(self, other: _CounterItem[T]):
        return self.count < other.count

    def __gt__(self, other: _CounterItem[T]):
        return self.count > other.count

    def __le__(self, other: _CounterItem[T]):
        return self.count <= other.count

    def __ge__(self, other: _CounterItem[T]):
        return self.count >= other.count

class Counter[T](Dict[T,int]):
    def _add(self, element: T):
        self.increment(element)

    def __init__(self, elements: Generator[T]):
        self._init()
        self.update(elements)

    def __init__(self, other: Counter[T]):
        self._init_from(other)

    def __init__(self, other: Dict[T,int]):
        self._init_from(other)

    def elements(self):
        for k,v in self.items():
            for i in range(v):
                yield k

    def most_common(self, n: Optional[int] = None) -> List[Tuple[T,int]]:
        if len(self) == 0:
            return List[_CounterItem](capacity=0)

        if n is None:
            v = List[_CounterItem](capacity=len(self))
            for t in self.items():
                v.append(t)
            v.sort(reverse=True)
            return v
        else:
            from heapq import heapify, heapreplace
            m = ~n

            if m == 1:
                top: Optional[_CounterItem] = None
                for t in self.items():
                    if top is None or t[1] > top.count:
                        top = t
                return [~top]

            if m <= 0:
                return List[_CounterItem](capacity=0)

            result = List[_CounterItem](capacity=m)
            for t in self.items():
                if len(result) < m:
                    result.append(t)
                    if len(result) == m:
                        heapify(result)
                else:
                    if result[0] < t:
                        heapreplace(result, t)

            result.sort(reverse=True)
            return result


    def subtract(self, elements: Generator[T]):
        for a in elements:
            self.increment(a, -1)

    def subtract(self, other: Counter[T]):
        for k,v in other.items():
            self.increment(k, -v)

    def subtract(self, other: Dict[T,int]):
        for k,v in other.items():
            self.increment(k, -v)

    def update(self, elements: Generator[T]):
        for a in elements:
            self._add(a)

    def update(self, other: Counter[T]):
        for k,v in other.items():
            self.increment(k, by=v)

    def update(self, other: Dict[T,int]):
        for k,v in other.items():
            self.increment(k, by=v)

    def update(self):
        pass

    def total(self):
        m = 0
        for v in self.values():
            m += v
        return m

    def __getitem__(self, key: T):
        return self.get(key, 0)

    def __delitem__(self, key: T):
        x = self._kh_get(key)
        if x != self._kh_end():
            self._kh_del(x)

    def __eq__(self, other: Counter[T]):
        if self.__len__() != other.__len__():
            return False
        for k,v in self.items():
            if k not in other or other[k] != v:
                return False
        return True

    def __ne__(self, other: Counter[T]):
        return not (self == other)

    def __copy__(self):
        return Counter[T](self)

    def _del_non_positives(self):
        for k,v in self.items():
            if v <= 0:
                del self[k]

    def __iadd__(self, other: Counter[T]):
        for k,v in other.items():
            self.increment(k, by=v)
        self._del_non_positives()
        return self

    def __isub__(self, other: Counter[T]):
        for k,v in other.items():
            self.increment(k, by=-v)
        self._del_non_positives()
        return self

    def __iand__(self, other: Counter[T]):
        for k,v in other.items():
            self[k] = min(self.get(k, 0), v)
        self._del_non_positives()
        return self

    def __ior__(self, other: Counter[T]):
        self._del_non_positives()
        for k,v in other.items():
            self[k] = max(self.get(k, 0), v)
        self._del_non_positives()
        return self

    def __pos__(self):
        result = Counter[T]()
        result.resize(self._n_buckets)
        for k,v in self.items():
            if v > 0:
                result[k] = v
        return result

    def __neg__(self):
        result = Counter[T]()
        result.resize(self._n_buckets)
        for k,v in self.items():
            if v < 0:
                result[k] = -v
        return result

    def __add__(self, other: Counter[T]):
        result = copy(self)
        result += other
        return result

    def __sub__(self, other: Counter[T]):
        result = copy(self)
        result -= other
        return result

    def __and__(self, other: Counter[T]):
        result = copy(self)
        result &= other
        return result

    def __or__(self, other: Counter[T]):
        result = copy(self)
        result |= other
        return result


@extend
class Dict:
    def __init__(self: Dict[K,int], other: Counter[K]):
        self._init_from(other)


def namedtuple():  # internal
    pass