module Std::Array { extern func bux_alloc(size: uint) -> *void; extern func bux_realloc(ptr: *void, size: uint) -> *void; extern func bux_free(ptr: *void); extern func bux_bounds_check(index: uint, len: uint); /// Growable contiguous buffer of `T` (len + capacity). struct Array { data: *T, len: uint, cap: uint, } /// Create an empty array with the given initial capacity. func Array_New(cap: uint) -> Array { let data = bux_alloc(cap * sizeof(T)) as *T; return Array { data: data, len: 0, cap: cap }; } /// Append `value`, growing capacity if needed. func Array_Push(self: *Array, value: T) { if self.len >= self.cap { self.cap = self.cap * 2; self.data = bux_realloc(self.data as *void, self.cap * sizeof(T)) as *T; } self.data[self.len] = value; self.len = self.len + 1; } /// Element at `index` (bounds-checked unless `@[Release]`). func Array_Get(self: *Array, index: uint) -> T { bux_bounds_check(index, self.len); return self.data[index]; } /// Write `value` at `index` (bounds-checked unless `@[Release]`). func Array_Set(self: *Array, index: uint, value: T) { bux_bounds_check(index, self.len); self.data[index] = value; } /// Number of live elements. func Array_Len(self: *Array) -> uint { return self.len; } /// Free the backing buffer and reset length/capacity to zero. func Array_Free(self: *Array) { bux_free(self.data as *void); self.data = null as *T; self.len = 0; self.cap = 0; } /// Drop trait entry — same as `Array_Free`. func Array_Drop(self: *Array) { Array_Free(self); } func Array_operator_index_get(self: *Array, idx: uint) -> T { return Array_Get(self, idx); } func Array_operator_index_set(self: *Array, idx: uint, value: T) { Array_Set(self, idx, value); } /// True if the array has no elements. func Array_IsEmpty(self: *Array) -> bool { return self.len == 0; } /// Current capacity (not length). func Array_Cap(self: *Array) -> uint { return self.cap; } /// Drop length to zero; keeps allocated capacity. func Array_Clear(self: *Array) { self.len = 0; } /// Ensure capacity is at least `minCap` (does not shrink). func Array_Reserve(self: *Array, minCap: uint) { if minCap <= self.cap { return; } self.cap = minCap; self.data = bux_realloc(self.data as *void, self.cap * sizeof(T)) as *T; } /// First element (bounds-checked if empty). func Array_First(self: *Array) -> T { return Array_Get(self, 0); } /// Last element (bounds-checked if empty). func Array_Last(self: *Array) -> T { return Array_Get(self, self.len - 1); } /// Remove and return the last element (bounds-checked if empty). func Array_Pop(self: *Array) -> T { bux_bounds_check(0, self.len); self.len = self.len - 1; return self.data[self.len]; } /// Linear search: true if `value` is present (uses `==`). func Array_Contains(self: *Array, value: T) -> bool { var i: uint = 0; while i < self.len { if self.data[i] == value { return true; } i = i + 1; } return false; } /// Index of first equal element, or `-1` if not found. func Array_IndexOf(self: *Array, value: T) -> int { var i: uint = 0; while i < self.len { if self.data[i] == value { return i as int; } i = i + 1; } return -1; } /// Append all elements of `other` onto `self`. func Array_Extend(self: *Array, other: *Array) { var i: uint = 0; while i < other.len { Array_Push(self, other.data[i]); i = i + 1; } } /// Remove element at `index`, shifting later elements left. Returns the removed value. func Array_RemoveAt(self: *Array, index: uint) -> T { bux_bounds_check(index, self.len); let val: T = self.data[index]; var i: uint = index; while i + 1 < self.len { self.data[i] = self.data[i + 1]; i = i + 1; } self.len = self.len - 1; return val; } /// Insert `value` at `index` (`0..=len`), shifting later elements right. func Array_Insert(self: *Array, index: uint, value: T) { // allow index == len (append); panic if index > len bux_bounds_check(index, self.len + 1); if self.len >= self.cap { var newCap: uint = self.cap * 2; if newCap == 0 { newCap = 4; } self.cap = newCap; self.data = bux_realloc(self.data as *void, self.cap * sizeof(T)) as *T; } var i: uint = self.len; while i > index { self.data[i] = self.data[i - 1]; i = i - 1; } self.data[index] = value; self.len = self.len + 1; } /// O(1) remove: swap `index` with last, then pop. Does not preserve order. func Array_SwapRemove(self: *Array, index: uint) -> T { bux_bounds_check(index, self.len); let val: T = self.data[index]; self.len = self.len - 1; if index < self.len { self.data[index] = self.data[self.len]; } return val; } /// Shallow clone: new buffer, elements copied by value. func Array_Clone(self: *Array) -> Array { var cap: uint = self.len; if cap == 0 { cap = 1; } var out: Array = Array_New(cap); var i: uint = 0; while i < self.len { Array_Push(&out, self.data[i]); i = i + 1; } return out; } /// Reverse elements in place. func Array_Reverse(self: *Array) { if self.len < 2 { return; } var i: uint = 0; var j: uint = self.len - 1; while i < j { let tmp: T = self.data[i]; self.data[i] = self.data[j]; self.data[j] = tmp; i = i + 1; j = j - 1; } } }