96 lines
3.4 KiB
Plaintext
96 lines
3.4 KiB
Plaintext
import t, c
|
||
import memhub
|
||
import string
|
||
import stdio
|
||
from stdint import *
|
||
|
||
# [LISTDBG] 诊断（用后移除）：list.__init__ 打印初始化结果、append 首次
|
||
# (Count==0) 打印 self/Data，配合 CTORDBG 定位二代编 SlotsBugTest cargs(list)
|
||
# 头覆盖。默认静默——由命令行 --enable-debug-log 开启（memhub.set_debug_log）
|
||
|
||
|
||
class list[T]:
|
||
"""Heap-allocated dynamic list container backed by memhub.MemManager polymorphic allocator.
|
||
|
||
Usage:
|
||
pool: memhub.MemManager = memhub.MemBuddy(arena, arena_size)
|
||
nums: list[int] = list[int](pool)
|
||
nums.append(42)
|
||
x: int = nums.get(0)
|
||
|
||
For pointer types (e.g. AST | t.CPtr), pass elem_size=8 to avoid object slicing:
|
||
lst: list[AST | t.CPtr] = list[AST | t.CPtr](pool, 8)
|
||
"""
|
||
__data__: t.CPtr
|
||
__count__: t.CSizeT
|
||
__capacity__: t.CSizeT
|
||
__pool__: memhub.MemManager | t.CPtr
|
||
__elem_size__: t.CSizeT
|
||
__iter_index__: t.CSizeT
|
||
|
||
def __new__(self, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT = 0):
|
||
ret = pool.alloc(48)
|
||
return ret
|
||
|
||
def __init__(self, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT = 0):
|
||
self.__pool__ = pool
|
||
self.__count__ = 0
|
||
self.__capacity__ = 8
|
||
if elem_size == 0:
|
||
self.__elem_size__ = T.__sizeof__()
|
||
else:
|
||
self.__elem_size__ = elem_size
|
||
self.__data__ = pool.alloc(self.__capacity__ * self.__elem_size__)
|
||
self.__iter_index__ = 0
|
||
|
||
def __len__(self) -> t.CSizeT:
|
||
return self.__count__
|
||
|
||
def append(self, item: T):
|
||
if self.__count__ >= self.__capacity__:
|
||
new_cap: t.CSizeT = self.__capacity__ * 2
|
||
new_data: t.CPtr = self.__pool__.alloc(new_cap * self.__elem_size__)
|
||
if new_data == None: return
|
||
string.memcpy(new_data, self.__data__, self.__count__ * self.__elem_size__)
|
||
self.__data__ = new_data
|
||
self.__capacity__ = new_cap
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + self.__count__ * self.__elem_size__)
|
||
elem_ptr[0] = item
|
||
self.__count__ += 1
|
||
|
||
def get(self, index: t.CSizeT) -> T:
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__)
|
||
return elem_ptr[0]
|
||
|
||
def __getitem__(self, index: t.CSizeT) -> T:
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__)
|
||
return elem_ptr[0]
|
||
|
||
def set(self, index: t.CSizeT, value: T):
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__)
|
||
elem_ptr[0] = value
|
||
|
||
def __setitem__(self, index: t.CSizeT, value: T):
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__)
|
||
elem_ptr[0] = value
|
||
|
||
def pop(self) -> T:
|
||
self.__count__ -= 1
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + self.__count__ * self.__elem_size__)
|
||
return elem_ptr[0]
|
||
|
||
def clear(self):
|
||
self.__count__ = 0
|
||
|
||
def __iter__(self) -> list[T] | t.CPtr:
|
||
self.__iter_index__ = 0
|
||
return self
|
||
|
||
def __next__(self) -> T:
|
||
if self.__iter_index__ >= self.__count__:
|
||
raise StopIteration
|
||
idx: t.CSizeT = self.__iter_index__
|
||
self.__iter_index__ = idx + 1
|
||
elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + idx * self.__elem_size__)
|
||
return elem_ptr[0]
|