mirror of
https://github.com/Ed94/Odin.git
synced 2026-07-24 16:37:54 +00:00
Merge branch 'odin-lang:master' into haiku
This commit is contained in:
@@ -0,0 +1,678 @@
|
||||
/*
|
||||
package avl implements an AVL tree.
|
||||
|
||||
The implementation is non-intrusive, and non-recursive.
|
||||
*/
|
||||
package container_avl
|
||||
|
||||
import "base:intrinsics"
|
||||
import "base:runtime"
|
||||
import "core:slice"
|
||||
|
||||
_ :: intrinsics
|
||||
_ :: runtime
|
||||
|
||||
// Originally based on the CC0 implementation by Eric Biggers
|
||||
// See: https://github.com/ebiggers/avl_tree/
|
||||
|
||||
// Direction specifies the traversal direction for a tree iterator.
|
||||
Direction :: enum i8 {
|
||||
// Backward is the in-order backwards direction.
|
||||
Backward = -1,
|
||||
// Forward is the in-order forwards direction.
|
||||
Forward = 1,
|
||||
}
|
||||
|
||||
// Ordering specifies order when inserting/finding values into the tree.
|
||||
Ordering :: slice.Ordering
|
||||
|
||||
// Tree is an AVL tree.
|
||||
Tree :: struct($Value: typeid) {
|
||||
// user_data is a parameter that will be passed to the on_remove
|
||||
// callback.
|
||||
user_data: rawptr,
|
||||
// on_remove is an optional callback that can be called immediately
|
||||
// after a node is removed from the tree.
|
||||
on_remove: proc(value: Value, user_data: rawptr),
|
||||
|
||||
_root: ^Node(Value),
|
||||
_node_allocator: runtime.Allocator,
|
||||
_cmp_fn: proc(a, b: Value) -> Ordering,
|
||||
_size: int,
|
||||
}
|
||||
|
||||
// Node is an AVL tree node.
|
||||
//
|
||||
// WARNING: It is unsafe to mutate value if the node is part of a tree
|
||||
// if doing so will alter the Node's sort position relative to other
|
||||
// elements in the tree.
|
||||
Node :: struct($Value: typeid) {
|
||||
value: Value,
|
||||
|
||||
_parent: ^Node(Value),
|
||||
_left: ^Node(Value),
|
||||
_right: ^Node(Value),
|
||||
_balance: i8,
|
||||
}
|
||||
|
||||
// Iterator is a tree iterator.
|
||||
//
|
||||
// WARNING: It is unsafe to modify the tree while iterating, except via
|
||||
// the iterator_remove method.
|
||||
Iterator :: struct($Value: typeid) {
|
||||
_tree: ^Tree(Value),
|
||||
_cur: ^Node(Value),
|
||||
_next: ^Node(Value),
|
||||
_direction: Direction,
|
||||
_called_next: bool,
|
||||
}
|
||||
|
||||
// init initializes a tree.
|
||||
init :: proc {
|
||||
init_ordered,
|
||||
init_cmp,
|
||||
}
|
||||
|
||||
// init_cmp initializes a tree.
|
||||
init_cmp :: proc(
|
||||
t: ^$T/Tree($Value),
|
||||
cmp_fn: proc(a, b: Value) -> Ordering,
|
||||
node_allocator := context.allocator,
|
||||
) {
|
||||
t._root = nil
|
||||
t._node_allocator = node_allocator
|
||||
t._cmp_fn = cmp_fn
|
||||
t._size = 0
|
||||
}
|
||||
|
||||
// init_ordered initializes a tree containing ordered items, with
|
||||
// a comparison function that results in an ascending order sort.
|
||||
init_ordered :: proc(
|
||||
t: ^$T/Tree($Value),
|
||||
node_allocator := context.allocator,
|
||||
) where intrinsics.type_is_ordered_numeric(Value) {
|
||||
init_cmp(t, slice.cmp_proc(Value), node_allocator)
|
||||
}
|
||||
|
||||
// destroy de-initializes a tree.
|
||||
destroy :: proc(t: ^$T/Tree($Value), call_on_remove: bool = true) {
|
||||
iter := iterator(t, Direction.Forward)
|
||||
for _ in iterator_next(&iter) {
|
||||
iterator_remove(&iter, call_on_remove)
|
||||
}
|
||||
}
|
||||
|
||||
// len returns the number of elements in the tree.
|
||||
len :: proc "contextless" (t: ^$T/Tree($Value)) -> int {
|
||||
return t._size
|
||||
}
|
||||
|
||||
// first returns the first node in the tree (in-order) or nil iff
|
||||
// the tree is empty.
|
||||
first :: proc "contextless" (t: ^$T/Tree($Value)) -> ^Node(Value) {
|
||||
return tree_first_or_last_in_order(t, Direction.Backward)
|
||||
}
|
||||
|
||||
// last returns the last element in the tree (in-order) or nil iff
|
||||
// the tree is empty.
|
||||
last :: proc "contextless" (t: ^$T/Tree($Value)) -> ^Node(Value) {
|
||||
return tree_first_or_last_in_order(t, Direction.Forward)
|
||||
}
|
||||
|
||||
// find finds the value in the tree, and returns the corresponding
|
||||
// node or nil iff the value is not present.
|
||||
find :: proc(t: ^$T/Tree($Value), value: Value) -> ^Node(Value) {
|
||||
cur := t._root
|
||||
descend_loop: for cur != nil {
|
||||
switch t._cmp_fn(value, cur.value) {
|
||||
case .Less:
|
||||
cur = cur._left
|
||||
case .Greater:
|
||||
cur = cur._right
|
||||
case .Equal:
|
||||
break descend_loop
|
||||
}
|
||||
}
|
||||
|
||||
return cur
|
||||
}
|
||||
|
||||
// find_or_insert attempts to insert the value into the tree, and returns
|
||||
// the node, a boolean indicating if the value was inserted, and the
|
||||
// node allocator error if relevant. If the value is already
|
||||
// present, the existing node is returned un-altered.
|
||||
find_or_insert :: proc(
|
||||
t: ^$T/Tree($Value),
|
||||
value: Value,
|
||||
) -> (
|
||||
n: ^Node(Value),
|
||||
inserted: bool,
|
||||
err: runtime.Allocator_Error,
|
||||
) {
|
||||
n_ptr := &t._root
|
||||
for n_ptr^ != nil {
|
||||
n = n_ptr^
|
||||
switch t._cmp_fn(value, n.value) {
|
||||
case .Less:
|
||||
n_ptr = &n._left
|
||||
case .Greater:
|
||||
n_ptr = &n._right
|
||||
case .Equal:
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
parent := n
|
||||
n = new(Node(Value), t._node_allocator) or_return
|
||||
n.value = value
|
||||
n._parent = parent
|
||||
n_ptr^ = n
|
||||
tree_rebalance_after_insert(t, n)
|
||||
|
||||
t._size += 1
|
||||
inserted = true
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// remove removes a node or value from the tree, and returns true iff the
|
||||
// removal was successful. While the node's value will be left intact,
|
||||
// the node itself will be freed via the tree's node allocator.
|
||||
remove :: proc {
|
||||
remove_value,
|
||||
remove_node,
|
||||
}
|
||||
|
||||
// remove_value removes a value from the tree, and returns true iff the
|
||||
// removal was successful. While the node's value will be left intact,
|
||||
// the node itself will be freed via the tree's node allocator.
|
||||
remove_value :: proc(t: ^$T/Tree($Value), value: Value, call_on_remove: bool = true) -> bool {
|
||||
n := find(t, value)
|
||||
if n == nil {
|
||||
return false
|
||||
}
|
||||
return remove_node(t, n, call_on_remove)
|
||||
}
|
||||
|
||||
// remove_node removes a node from the tree, and returns true iff the
|
||||
// removal was successful. While the node's value will be left intact,
|
||||
// the node itself will be freed via the tree's node allocator.
|
||||
remove_node :: proc(t: ^$T/Tree($Value), node: ^Node(Value), call_on_remove: bool = true) -> bool {
|
||||
if node._parent == node || (node._parent == nil && t._root != node) {
|
||||
return false
|
||||
}
|
||||
defer {
|
||||
if call_on_remove && t.on_remove != nil {
|
||||
t.on_remove(node.value, t.user_data)
|
||||
}
|
||||
free(node, t._node_allocator)
|
||||
}
|
||||
|
||||
parent: ^Node(Value)
|
||||
left_deleted: bool
|
||||
|
||||
t._size -= 1
|
||||
if node._left != nil && node._right != nil {
|
||||
parent, left_deleted = tree_swap_with_successor(t, node)
|
||||
} else {
|
||||
child := node._left
|
||||
if child == nil {
|
||||
child = node._right
|
||||
}
|
||||
parent = node._parent
|
||||
if parent != nil {
|
||||
if node == parent._left {
|
||||
parent._left = child
|
||||
left_deleted = true
|
||||
} else {
|
||||
parent._right = child
|
||||
left_deleted = false
|
||||
}
|
||||
if child != nil {
|
||||
child._parent = parent
|
||||
}
|
||||
} else {
|
||||
if child != nil {
|
||||
child._parent = parent
|
||||
}
|
||||
t._root = child
|
||||
node_reset(node)
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
for {
|
||||
if left_deleted {
|
||||
parent = tree_handle_subtree_shrink(t, parent, +1, &left_deleted)
|
||||
} else {
|
||||
parent = tree_handle_subtree_shrink(t, parent, -1, &left_deleted)
|
||||
}
|
||||
if parent == nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
node_reset(node)
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
// iterator returns a tree iterator in the specified direction.
|
||||
iterator :: proc "contextless" (t: ^$T/Tree($Value), direction: Direction) -> Iterator(Value) {
|
||||
it: Iterator(Value)
|
||||
it._tree = transmute(^Tree(Value))t
|
||||
it._direction = direction
|
||||
|
||||
iterator_first(&it)
|
||||
|
||||
return it
|
||||
}
|
||||
|
||||
// iterator_from_pos returns a tree iterator in the specified direction,
|
||||
// spanning the range [pos, last] (inclusive).
|
||||
iterator_from_pos :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
pos: ^Node(Value),
|
||||
direction: Direction,
|
||||
) -> Iterator(Value) {
|
||||
it: Iterator(Value)
|
||||
it._tree = transmute(^Tree(Value))t
|
||||
it._direction = direction
|
||||
it._next = nil
|
||||
it._called_next = false
|
||||
|
||||
if it._cur = pos; pos != nil {
|
||||
it._next = node_next_or_prev_in_order(it._cur, it._direction)
|
||||
}
|
||||
|
||||
return it
|
||||
}
|
||||
|
||||
// iterator_get returns the node currently pointed to by the iterator,
|
||||
// or nil iff the node has been removed, the tree is empty, or the end
|
||||
// of the tree has been reached.
|
||||
iterator_get :: proc "contextless" (it: ^$I/Iterator($Value)) -> ^Node(Value) {
|
||||
return it._cur
|
||||
}
|
||||
|
||||
// iterator_remove removes the node currently pointed to by the iterator,
|
||||
// and returns true iff the removal was successful. Semantics are the
|
||||
// same as the Tree remove.
|
||||
iterator_remove :: proc(it: ^$I/Iterator($Value), call_on_remove: bool = true) -> bool {
|
||||
if it._cur == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
ok := remove_node(it._tree, it._cur, call_on_remove)
|
||||
if ok {
|
||||
it._cur = nil
|
||||
}
|
||||
|
||||
return ok
|
||||
}
|
||||
|
||||
// iterator_next advances the iterator and returns the (node, true) or
|
||||
// or (nil, false) iff the end of the tree has been reached.
|
||||
//
|
||||
// Note: The first call to iterator_next will return the first node instead
|
||||
// of advancing the iterator.
|
||||
iterator_next :: proc "contextless" (it: ^$I/Iterator($Value)) -> (^Node(Value), bool) {
|
||||
// This check is needed so that the first element gets returned from
|
||||
// a brand-new iterator, and so that the somewhat contrived case where
|
||||
// iterator_remove is called before the first call to iterator_next
|
||||
// returns the correct value.
|
||||
if !it._called_next {
|
||||
it._called_next = true
|
||||
|
||||
// There can be the contrived case where iterator_remove is
|
||||
// called before ever calling iterator_next, which needs to be
|
||||
// handled as an actual call to next.
|
||||
//
|
||||
// If this happens it._cur will be nil, so only return the
|
||||
// first value, if it._cur is valid.
|
||||
if it._cur != nil {
|
||||
return it._cur, true
|
||||
}
|
||||
}
|
||||
|
||||
if it._next == nil {
|
||||
return nil, false
|
||||
}
|
||||
|
||||
it._cur = it._next
|
||||
it._next = node_next_or_prev_in_order(it._cur, it._direction)
|
||||
|
||||
return it._cur, true
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_first_or_last_in_order :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
direction: Direction,
|
||||
) -> ^Node(Value) {
|
||||
first, sign := t._root, i8(direction)
|
||||
if first != nil {
|
||||
for {
|
||||
tmp := node_get_child(first, +sign)
|
||||
if tmp == nil {
|
||||
break
|
||||
}
|
||||
first = tmp
|
||||
}
|
||||
}
|
||||
|
||||
return first
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_replace_child :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
parent, old_child, new_child: ^Node(Value),
|
||||
) {
|
||||
if parent != nil {
|
||||
if old_child == parent._left {
|
||||
parent._left = new_child
|
||||
} else {
|
||||
parent._right = new_child
|
||||
}
|
||||
} else {
|
||||
t._root = new_child
|
||||
}
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_rotate :: proc "contextless" (t: ^$T/Tree($Value), a: ^Node(Value), sign: i8) {
|
||||
b := node_get_child(a, -sign)
|
||||
e := node_get_child(b, +sign)
|
||||
p := a._parent
|
||||
|
||||
node_set_child(a, -sign, e)
|
||||
a._parent = b
|
||||
|
||||
node_set_child(b, +sign, a)
|
||||
b._parent = p
|
||||
|
||||
if e != nil {
|
||||
e._parent = a
|
||||
}
|
||||
|
||||
tree_replace_child(t, p, a, b)
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_double_rotate :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
b, a: ^Node(Value),
|
||||
sign: i8,
|
||||
) -> ^Node(Value) {
|
||||
e := node_get_child(b, +sign)
|
||||
f := node_get_child(e, -sign)
|
||||
g := node_get_child(e, +sign)
|
||||
p := a._parent
|
||||
e_bal := e._balance
|
||||
|
||||
node_set_child(a, -sign, g)
|
||||
a_bal := -e_bal
|
||||
if sign * e_bal >= 0 {
|
||||
a_bal = 0
|
||||
}
|
||||
node_set_parent_balance(a, e, a_bal)
|
||||
|
||||
node_set_child(b, +sign, f)
|
||||
b_bal := -e_bal
|
||||
if sign * e_bal <= 0 {
|
||||
b_bal = 0
|
||||
}
|
||||
node_set_parent_balance(b, e, b_bal)
|
||||
|
||||
node_set_child(e, +sign, a)
|
||||
node_set_child(e, -sign, b)
|
||||
node_set_parent_balance(e, p, 0)
|
||||
|
||||
if g != nil {
|
||||
g._parent = a
|
||||
}
|
||||
|
||||
if f != nil {
|
||||
f._parent = b
|
||||
}
|
||||
|
||||
tree_replace_child(t, p, a, e)
|
||||
|
||||
return e
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_handle_subtree_growth :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
node, parent: ^Node(Value),
|
||||
sign: i8,
|
||||
) -> bool {
|
||||
old_balance_factor := parent._balance
|
||||
if old_balance_factor == 0 {
|
||||
node_adjust_balance_factor(parent, sign)
|
||||
return false
|
||||
}
|
||||
|
||||
new_balance_factor := old_balance_factor + sign
|
||||
if new_balance_factor == 0 {
|
||||
node_adjust_balance_factor(parent, sign)
|
||||
return true
|
||||
}
|
||||
|
||||
if sign * node._balance > 0 {
|
||||
tree_rotate(t, parent, -sign)
|
||||
node_adjust_balance_factor(parent, -sign)
|
||||
node_adjust_balance_factor(node, -sign)
|
||||
} else {
|
||||
tree_double_rotate(t, node, parent, -sign)
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_rebalance_after_insert :: proc "contextless" (t: ^$T/Tree($Value), inserted: ^Node(Value)) {
|
||||
node, parent := inserted, inserted._parent
|
||||
switch {
|
||||
case parent == nil:
|
||||
return
|
||||
case node == parent._left:
|
||||
node_adjust_balance_factor(parent, -1)
|
||||
case:
|
||||
node_adjust_balance_factor(parent, +1)
|
||||
}
|
||||
|
||||
if parent._balance == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
for done := false; !done; {
|
||||
node = parent
|
||||
if parent = node._parent; parent == nil {
|
||||
return
|
||||
}
|
||||
|
||||
if node == parent._left {
|
||||
done = tree_handle_subtree_growth(t, node, parent, -1)
|
||||
} else {
|
||||
done = tree_handle_subtree_growth(t, node, parent, +1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_swap_with_successor :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
x: ^Node(Value),
|
||||
) -> (
|
||||
^Node(Value),
|
||||
bool,
|
||||
) {
|
||||
ret: ^Node(Value)
|
||||
left_deleted: bool
|
||||
|
||||
y := x._right
|
||||
if y._left == nil {
|
||||
ret = y
|
||||
} else {
|
||||
q: ^Node(Value)
|
||||
|
||||
for {
|
||||
q = y
|
||||
if y = y._left; y._left == nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if q._left = y._right; q._left != nil {
|
||||
q._left._parent = q
|
||||
}
|
||||
y._right = x._right
|
||||
x._right._parent = y
|
||||
ret = q
|
||||
left_deleted = true
|
||||
}
|
||||
|
||||
y._left = x._left
|
||||
x._left._parent = y
|
||||
|
||||
y._parent = x._parent
|
||||
y._balance = x._balance
|
||||
|
||||
tree_replace_child(t, x._parent, x, y)
|
||||
|
||||
return ret, left_deleted
|
||||
}
|
||||
|
||||
@(private)
|
||||
tree_handle_subtree_shrink :: proc "contextless" (
|
||||
t: ^$T/Tree($Value),
|
||||
parent: ^Node(Value),
|
||||
sign: i8,
|
||||
left_deleted: ^bool,
|
||||
) -> ^Node(Value) {
|
||||
old_balance_factor := parent._balance
|
||||
if old_balance_factor == 0 {
|
||||
node_adjust_balance_factor(parent, sign)
|
||||
return nil
|
||||
}
|
||||
|
||||
node: ^Node(Value)
|
||||
new_balance_factor := old_balance_factor + sign
|
||||
if new_balance_factor == 0 {
|
||||
node_adjust_balance_factor(parent, sign)
|
||||
node = parent
|
||||
} else {
|
||||
node = node_get_child(parent, sign)
|
||||
if sign * node._balance >= 0 {
|
||||
tree_rotate(t, parent, -sign)
|
||||
if node._balance == 0 {
|
||||
node_adjust_balance_factor(node, -sign)
|
||||
return nil
|
||||
}
|
||||
node_adjust_balance_factor(parent, -sign)
|
||||
node_adjust_balance_factor(node, -sign)
|
||||
} else {
|
||||
node = tree_double_rotate(t, node, parent, -sign)
|
||||
}
|
||||
}
|
||||
|
||||
parent := parent
|
||||
if parent = node._parent; parent != nil {
|
||||
left_deleted^ = node == parent._left
|
||||
}
|
||||
return parent
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_reset :: proc "contextless" (n: ^Node($Value)) {
|
||||
// Mostly pointless as n will be deleted after this is called, but
|
||||
// attempt to be able to catch cases of n not being in the tree.
|
||||
n._parent = n
|
||||
n._left = nil
|
||||
n._right = nil
|
||||
n._balance = 0
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_set_parent_balance :: #force_inline proc "contextless" (
|
||||
n, parent: ^Node($Value),
|
||||
balance: i8,
|
||||
) {
|
||||
n._parent = parent
|
||||
n._balance = balance
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_get_child :: #force_inline proc "contextless" (n: ^Node($Value), sign: i8) -> ^Node(Value) {
|
||||
if sign < 0 {
|
||||
return n._left
|
||||
}
|
||||
return n._right
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_next_or_prev_in_order :: proc "contextless" (
|
||||
n: ^Node($Value),
|
||||
direction: Direction,
|
||||
) -> ^Node(Value) {
|
||||
next, tmp: ^Node(Value)
|
||||
sign := i8(direction)
|
||||
|
||||
if next = node_get_child(n, +sign); next != nil {
|
||||
for {
|
||||
tmp = node_get_child(next, -sign)
|
||||
if tmp == nil {
|
||||
break
|
||||
}
|
||||
next = tmp
|
||||
}
|
||||
} else {
|
||||
tmp, next = n, n._parent
|
||||
for next != nil && tmp == node_get_child(next, +sign) {
|
||||
tmp, next = next, next._parent
|
||||
}
|
||||
}
|
||||
return next
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_set_child :: #force_inline proc "contextless" (
|
||||
n: ^Node($Value),
|
||||
sign: i8,
|
||||
child: ^Node(Value),
|
||||
) {
|
||||
if sign < 0 {
|
||||
n._left = child
|
||||
} else {
|
||||
n._right = child
|
||||
}
|
||||
}
|
||||
|
||||
@(private)
|
||||
node_adjust_balance_factor :: #force_inline proc "contextless" (n: ^Node($Value), amount: i8) {
|
||||
n._balance += amount
|
||||
}
|
||||
|
||||
@(private)
|
||||
iterator_first :: proc "contextless" (it: ^Iterator($Value)) {
|
||||
// This is private because behavior when the user manually calls
|
||||
// iterator_first followed by iterator_next is unintuitive, since
|
||||
// the first call to iterator_next MUST return the first node
|
||||
// instead of advancing so that `for node in iterator_next(&next)`
|
||||
// works as expected.
|
||||
|
||||
switch it._direction {
|
||||
case .Forward:
|
||||
it._cur = tree_first_or_last_in_order(it._tree, .Backward)
|
||||
case .Backward:
|
||||
it._cur = tree_first_or_last_in_order(it._tree, .Forward)
|
||||
}
|
||||
|
||||
it._next = nil
|
||||
it._called_next = false
|
||||
|
||||
if it._cur != nil {
|
||||
it._next = node_next_or_prev_in_order(it._cur, it._direction)
|
||||
}
|
||||
}
|
||||
+121
-24
@@ -147,16 +147,30 @@ aprintln :: proc(args: ..any, sep := " ", allocator := context.allocator) -> str
|
||||
// *Allocates Using Context's Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - fmt: A format string with placeholders for the provided arguments.
|
||||
// - args: A variadic list of arguments to be formatted.
|
||||
// - newline: Whether the string should end with a newline. (See `aprintfln`.)
|
||||
//
|
||||
// Returns: A formatted string. The returned string must be freed accordingly.
|
||||
//
|
||||
aprintf :: proc(fmt: string, args: ..any, allocator := context.allocator, newline := false) -> string {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, allocator)
|
||||
sbprintf(&str, fmt, ..args, newline=newline)
|
||||
return strings.to_string(str)
|
||||
}
|
||||
// Creates a formatted string using a format string and arguments, followed by a newline.
|
||||
//
|
||||
// *Allocates Using Context's Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - fmt: A format string with placeholders for the provided arguments.
|
||||
// - args: A variadic list of arguments to be formatted.
|
||||
//
|
||||
// Returns: A formatted string. The returned string must be freed accordingly.
|
||||
//
|
||||
aprintf :: proc(fmt: string, args: ..any, allocator := context.allocator) -> string {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, allocator)
|
||||
sbprintf(&str, fmt, ..args)
|
||||
return strings.to_string(str)
|
||||
aprintfln :: proc(fmt: string, args: ..any, allocator := context.allocator) -> string {
|
||||
return aprintf(fmt, ..args, allocator=allocator, newline=true)
|
||||
}
|
||||
// Creates a formatted string
|
||||
//
|
||||
@@ -195,16 +209,30 @@ tprintln :: proc(args: ..any, sep := " ") -> string {
|
||||
// *Allocates Using Context's Temporary Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - fmt: A format string with placeholders for the provided arguments.
|
||||
// - args: A variadic list of arguments to be formatted.
|
||||
// - newline: Whether the string should end with a newline. (See `tprintfln`.)
|
||||
//
|
||||
// Returns: A formatted string.
|
||||
//
|
||||
tprintf :: proc(fmt: string, args: ..any, newline := false) -> string {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, context.temp_allocator)
|
||||
sbprintf(&str, fmt, ..args, newline=newline)
|
||||
return strings.to_string(str)
|
||||
}
|
||||
// Creates a formatted string using a format string and arguments, followed by a newline.
|
||||
//
|
||||
// *Allocates Using Context's Temporary Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - fmt: A format string with placeholders for the provided arguments.
|
||||
// - args: A variadic list of arguments to be formatted.
|
||||
//
|
||||
// Returns: A formatted string.
|
||||
//
|
||||
tprintf :: proc(fmt: string, args: ..any) -> string {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, context.temp_allocator)
|
||||
sbprintf(&str, fmt, ..args)
|
||||
return strings.to_string(str)
|
||||
tprintfln :: proc(fmt: string, args: ..any) -> string {
|
||||
return tprintf(fmt, ..args, newline=true)
|
||||
}
|
||||
// Creates a formatted string using a supplied buffer as the backing array. Writes into the buffer.
|
||||
//
|
||||
@@ -238,12 +266,25 @@ bprintln :: proc(buf: []byte, args: ..any, sep := " ") -> string {
|
||||
// - buf: The backing buffer
|
||||
// - fmt: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
// - newline: Whether the string should end with a newline. (See `bprintfln`.)
|
||||
//
|
||||
// Returns: A formatted string
|
||||
//
|
||||
bprintf :: proc(buf: []byte, fmt: string, args: ..any) -> string {
|
||||
bprintf :: proc(buf: []byte, fmt: string, args: ..any, newline := false) -> string {
|
||||
sb := strings.builder_from_bytes(buf)
|
||||
return sbprintf(&sb, fmt, ..args)
|
||||
return sbprintf(&sb, fmt, ..args, newline=newline)
|
||||
}
|
||||
// Creates a formatted string using a supplied buffer as the backing array, followed by a newline. Writes into the buffer.
|
||||
//
|
||||
// Inputs:
|
||||
// - buf: The backing buffer
|
||||
// - fmt: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
//
|
||||
// Returns: A formatted string
|
||||
//
|
||||
bprintfln :: proc(buf: []byte, fmt: string, args: ..any) -> string {
|
||||
return bprintf(buf, fmt, ..args, newline=true)
|
||||
}
|
||||
// Runtime assertion with a formatted message
|
||||
//
|
||||
@@ -294,17 +335,31 @@ panicf :: proc(fmt: string, args: ..any, loc := #caller_location) -> ! {
|
||||
// Inputs:
|
||||
// - format: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
// - newline: Whether the string should end with a newline. (See `caprintfln`.)
|
||||
//
|
||||
// Returns: A formatted C string
|
||||
//
|
||||
caprintf :: proc(format: string, args: ..any) -> cstring {
|
||||
caprintf :: proc(format: string, args: ..any, newline := false) -> cstring {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str)
|
||||
sbprintf(&str, format, ..args)
|
||||
sbprintf(&str, format, ..args, newline=newline)
|
||||
strings.write_byte(&str, 0)
|
||||
s := strings.to_string(str)
|
||||
return cstring(raw_data(s))
|
||||
}
|
||||
// Creates a formatted C string, followed by a newline.
|
||||
//
|
||||
// *Allocates Using Context's Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - format: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
//
|
||||
// Returns: A formatted C string
|
||||
//
|
||||
caprintfln :: proc(format: string, args: ..any) -> cstring {
|
||||
return caprintf(format, ..args, newline=true)
|
||||
}
|
||||
// Creates a formatted C string
|
||||
//
|
||||
// *Allocates Using Context's Temporary Allocator*
|
||||
@@ -312,16 +367,30 @@ caprintf :: proc(format: string, args: ..any) -> cstring {
|
||||
// Inputs:
|
||||
// - format: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
// - newline: Whether the string should end with a newline. (See `ctprintfln`.)
|
||||
//
|
||||
// Returns: A formatted C string
|
||||
//
|
||||
ctprintf :: proc(format: string, args: ..any, newline := false) -> cstring {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, context.temp_allocator)
|
||||
sbprintf(&str, format, ..args, newline=newline)
|
||||
strings.write_byte(&str, 0)
|
||||
s := strings.to_string(str)
|
||||
return cstring(raw_data(s))
|
||||
}
|
||||
// Creates a formatted C string, followed by a newline.
|
||||
//
|
||||
// *Allocates Using Context's Temporary Allocator*
|
||||
//
|
||||
// Inputs:
|
||||
// - format: A format string with placeholders for the provided arguments
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
//
|
||||
// Returns: A formatted C string
|
||||
//
|
||||
ctprintf :: proc(format: string, args: ..any) -> cstring {
|
||||
str: strings.Builder
|
||||
strings.builder_init(&str, context.temp_allocator)
|
||||
sbprintf(&str, format, ..args)
|
||||
strings.write_byte(&str, 0)
|
||||
s := strings.to_string(str)
|
||||
return cstring(raw_data(s))
|
||||
ctprintfln :: proc(format: string, args: ..any) -> cstring {
|
||||
return ctprintf(format, ..args, newline=true)
|
||||
}
|
||||
// Formats using the default print settings and writes to the given strings.Builder
|
||||
//
|
||||
@@ -355,13 +424,25 @@ sbprintln :: proc(buf: ^strings.Builder, args: ..any, sep := " ") -> string {
|
||||
// - buf: A pointer to a strings.Builder buffer
|
||||
// - fmt: The format string
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
// - newline: Whether a trailing newline should be written. (See `sbprintfln`.)
|
||||
//
|
||||
// Returns: The resulting formatted string
|
||||
//
|
||||
sbprintf :: proc(buf: ^strings.Builder, fmt: string, args: ..any) -> string {
|
||||
wprintf(strings.to_writer(buf), fmt, ..args, flush=true)
|
||||
sbprintf :: proc(buf: ^strings.Builder, fmt: string, args: ..any, newline := false) -> string {
|
||||
wprintf(strings.to_writer(buf), fmt, ..args, flush=true, newline=newline)
|
||||
return strings.to_string(buf^)
|
||||
}
|
||||
// Formats and writes to a strings.Builder buffer according to the specified format string, followed by a newline.
|
||||
//
|
||||
// Inputs:
|
||||
// - buf: A pointer to a strings.Builder to store the formatted string
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
//
|
||||
// Returns: A formatted string
|
||||
//
|
||||
sbprintfln :: proc(buf: ^strings.Builder, format: string, args: ..any) -> string {
|
||||
return sbprintf(buf, format, ..args, newline=true)
|
||||
}
|
||||
// Formats and writes to an io.Writer using the default print settings
|
||||
//
|
||||
// Inputs:
|
||||
@@ -435,10 +516,11 @@ wprintln :: proc(w: io.Writer, args: ..any, sep := " ", flush := true) -> int {
|
||||
// - w: An io.Writer to write to
|
||||
// - fmt: The format string
|
||||
// - args: A variadic list of arguments to be formatted
|
||||
// - newline: Whether a trailing newline should be written. (See `wprintfln`.)
|
||||
//
|
||||
// Returns: The number of bytes written
|
||||
//
|
||||
wprintf :: proc(w: io.Writer, fmt: string, args: ..any, flush := true) -> int {
|
||||
wprintf :: proc(w: io.Writer, fmt: string, args: ..any, flush := true, newline := false) -> int {
|
||||
fi: Info
|
||||
arg_index: int = 0
|
||||
end := len(fmt)
|
||||
@@ -708,12 +790,27 @@ wprintf :: proc(w: io.Writer, fmt: string, args: ..any, flush := true) -> int {
|
||||
}
|
||||
io.write_string(fi.writer, ")", &fi.n)
|
||||
}
|
||||
|
||||
if newline {
|
||||
io.write_byte(w, '\n', &fi.n)
|
||||
}
|
||||
if flush {
|
||||
io.flush(w)
|
||||
}
|
||||
|
||||
return fi.n
|
||||
}
|
||||
// Formats and writes to an io.Writer according to the specified format string, followed by a newline.
|
||||
//
|
||||
// Inputs:
|
||||
// - w: The io.Writer to write to.
|
||||
// - args: A variadic list of arguments to be formatted.
|
||||
//
|
||||
// Returns: The number of bytes written.
|
||||
//
|
||||
wprintfln :: proc(w: io.Writer, format: string, args: ..any, flush := true) -> int {
|
||||
return wprintf(w, format, ..args, flush=flush, newline=true)
|
||||
}
|
||||
// Writes a ^runtime.Type_Info value to an io.Writer
|
||||
//
|
||||
// Inputs:
|
||||
|
||||
+16
-8
@@ -30,7 +30,7 @@ fprintln :: proc(fd: os.Handle, args: ..any, sep := " ", flush := true) -> int {
|
||||
return wprintln(w, ..args, sep=sep, flush=flush)
|
||||
}
|
||||
// fprintf formats according to the specified format string and writes to fd
|
||||
fprintf :: proc(fd: os.Handle, fmt: string, args: ..any, flush := true) -> int {
|
||||
fprintf :: proc(fd: os.Handle, fmt: string, args: ..any, flush := true, newline := false) -> int {
|
||||
buf: [1024]byte
|
||||
b: bufio.Writer
|
||||
defer bufio.writer_flush(&b)
|
||||
@@ -38,7 +38,11 @@ fprintf :: proc(fd: os.Handle, fmt: string, args: ..any, flush := true) -> int {
|
||||
bufio.writer_init_with_buf(&b, os.stream_from_handle(fd), buf[:])
|
||||
|
||||
w := bufio.writer_to_writer(&b)
|
||||
return wprintf(w, fmt, ..args, flush=flush)
|
||||
return wprintf(w, fmt, ..args, flush=flush, newline=newline)
|
||||
}
|
||||
// fprintfln formats according to the specified format string and writes to fd, followed by a newline.
|
||||
fprintfln :: proc(fd: os.Handle, fmt: string, args: ..any, flush := true) -> int {
|
||||
return fprintf(fd, fmt, ..args, flush=flush, newline=true)
|
||||
}
|
||||
fprint_type :: proc(fd: os.Handle, info: ^runtime.Type_Info, flush := true) -> (n: int, err: io.Error) {
|
||||
buf: [1024]byte
|
||||
@@ -62,15 +66,19 @@ fprint_typeid :: proc(fd: os.Handle, id: typeid, flush := true) -> (n: int, err:
|
||||
}
|
||||
|
||||
// print formats using the default print settings and writes to os.stdout
|
||||
print :: proc(args: ..any, sep := " ", flush := true) -> int { return fprint(os.stdout, ..args, sep=sep, flush=flush) }
|
||||
print :: proc(args: ..any, sep := " ", flush := true) -> int { return fprint(os.stdout, ..args, sep=sep, flush=flush) }
|
||||
// println formats using the default print settings and writes to os.stdout
|
||||
println :: proc(args: ..any, sep := " ", flush := true) -> int { return fprintln(os.stdout, ..args, sep=sep, flush=flush) }
|
||||
println :: proc(args: ..any, sep := " ", flush := true) -> int { return fprintln(os.stdout, ..args, sep=sep, flush=flush) }
|
||||
// printf formats according to the specified format string and writes to os.stdout
|
||||
printf :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stdout, fmt, ..args, flush=flush) }
|
||||
printf :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stdout, fmt, ..args, flush=flush) }
|
||||
// printfln formats according to the specified format string and writes to os.stdout, followed by a newline.
|
||||
printfln :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stdout, fmt, ..args, flush=flush, newline=true) }
|
||||
|
||||
// eprint formats using the default print settings and writes to os.stderr
|
||||
eprint :: proc(args: ..any, sep := " ", flush := true) -> int { return fprint(os.stderr, ..args, sep=sep, flush=flush) }
|
||||
eprint :: proc(args: ..any, sep := " ", flush := true) -> int { return fprint(os.stderr, ..args, sep=sep, flush=flush) }
|
||||
// eprintln formats using the default print settings and writes to os.stderr
|
||||
eprintln :: proc(args: ..any, sep := " ", flush := true) -> int { return fprintln(os.stderr, ..args, sep=sep, flush=flush) }
|
||||
eprintln :: proc(args: ..any, sep := " ", flush := true) -> int { return fprintln(os.stderr, ..args, sep=sep, flush=flush) }
|
||||
// eprintf formats according to the specified format string and writes to os.stderr
|
||||
eprintf :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stderr, fmt, ..args, flush=flush) }
|
||||
eprintf :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stderr, fmt, ..args, flush=flush) }
|
||||
// eprintfln formats according to the specified format string and writes to os.stderr, followed by a newline.
|
||||
eprintfln :: proc(fmt: string, args: ..any, flush := true) -> int { return fprintf(os.stderr, fmt, ..args, flush=flush, newline=true) }
|
||||
|
||||
@@ -369,6 +369,10 @@ datetime_to_time :: proc "contextless" (year, month, day, hour, minute, second:
|
||||
mod = year % divisor
|
||||
return
|
||||
}
|
||||
_is_leap_year :: proc "contextless" (year: int) -> bool {
|
||||
return year%4 == 0 && (year%100 != 0 || year%400 == 0)
|
||||
}
|
||||
|
||||
|
||||
ok = true
|
||||
|
||||
@@ -395,6 +399,10 @@ datetime_to_time :: proc "contextless" (year, month, day, hour, minute, second:
|
||||
|
||||
days += int(days_before[_m]) + _d
|
||||
|
||||
if _is_leap_year(year) && _m >= 2 {
|
||||
days += 1
|
||||
}
|
||||
|
||||
s += i64(days) * SECONDS_PER_DAY
|
||||
s += i64(hour) * SECONDS_PER_HOUR
|
||||
s += i64(minute) * SECONDS_PER_MINUTE
|
||||
|
||||
Reference in New Issue
Block a user