ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-03 14:18:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add _internal_sort_from_indices_permuation

Browse Source
This commit is contained in:
gingerBill
2025-10-07 14:15:40 +01:00
parent af8c698b97
commit 13b8607cc3
2 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/slice/ptr.odin
+5 -5
View File
@@ -3,14 +3,14 @@ package slice
import "base:builtin" import "base:builtin"
import "base:runtime" import "base:runtime"
ptr_add :: proc(p: $P/^$T, x: int) -> ^T { ptr_add :: proc "contextless" (p: $P/^$T, x: int) -> ^T {
return ([^]T)(p)[x:] return ([^]T)(p)[x:]
} }
ptr_sub :: proc(p: $P/^$T, x: int) -> ^T { ptr_sub :: proc "contextless" (p: $P/^$T, x: int) -> ^T {
return ([^]T)(p)[-x:] return ([^]T)(p)[-x:]
} }
ptr_swap_non_overlapping :: proc(x, y: rawptr, len: int) { ptr_swap_non_overlapping :: proc "contextless" (x, y: rawptr, len: int) {
if len <= 0 { if len <= 0 {
return return
} }
@@ -44,7 +44,7 @@ ptr_swap_non_overlapping :: proc(x, y: rawptr, len: int) {
} }
} }
ptr_swap_overlapping :: proc(x, y: rawptr, len: int) { ptr_swap_overlapping :: proc "contextless" (x, y: rawptr, len: int) {
if len <= 0 { if len <= 0 {
return return
} }
@@ -68,7 +68,7 @@ ptr_swap_overlapping :: proc(x, y: rawptr, len: int) {
} }
ptr_rotate :: proc(left: int, mid: ^$T, right: int) { ptr_rotate :: proc "contextless" (left: int, mid: ^$T, right: int) {
when size_of(T) != 0 { when size_of(T) != 0 {
left, mid, right := left, mid, right left, mid, right := left, mid, right
core/slice/sort.odin
+28 -1
View File
@@ -81,6 +81,25 @@ sort_by_indices_overwrite :: proc(data: $T/[]$E, indices: []int) {
swap_with_slice(data, temp) swap_with_slice(data, temp)
} }
@(private)
_internal_sort_from_indices_permuation :: proc(data: $T/[]$E, indices: []int) {
assert(len(data) == len(indices))
if len(indices) <= 1 {
return
}
// TODO(bill): This is not O(N)
for i in 0..<len(indices) {
index_to_swap := indices[i]
for index_to_swap < i {
index_to_swap = indices[index_to_swap]
}
ptr_swap_non_overlapping(&data[i], &data[index_to_swap], size_of(E))
}
}
// sort sorts a slice and returns a slice of the original indices // sort sorts a slice and returns a slice of the original indices
// This sort is not guaranteed to be stable // This sort is not guaranteed to be stable
sort_with_indices :: proc(data: $T/[]$E, allocator := context.allocator) -> (indices: []int) where ORD(E) { sort_with_indices :: proc(data: $T/[]$E, allocator := context.allocator) -> (indices: []int) where ORD(E) {
@@ -90,11 +109,13 @@ sort_with_indices :: proc(data: $T/[]$E, allocator := context.allocator) -> (ind
for _, idx in indices { for _, idx in indices {
indices[idx] = idx indices[idx] = idx
} }
raw := ([^]byte)(raw_data(indices)) raw := ([^]byte)(raw_data(indices))
_smoothsort(raw, uint(len(indices)), size_of(int), proc(lhs, rhs: rawptr, user_data: rawptr) -> Ordering { _smoothsort(raw, uint(len(indices)), size_of(int), proc(lhs, rhs: rawptr, user_data: rawptr) -> Ordering {
data := ([^]E)(user_data) data := ([^]E)(user_data)
xi, yi := (^int)(lhs)^, (^int)(rhs)^ xi, yi := (^int)(lhs)^, (^int)(rhs)^
x, y := data[xi], data[yi] #no_bounds_check x, y := data[xi], data[yi]
if x < y { if x < y {
return .Less return .Less
} else if x > y { } else if x > y {
@@ -102,6 +123,8 @@ sort_with_indices :: proc(data: $T/[]$E, allocator := context.allocator) -> (ind
} }
return .Equal return .Equal
}, raw_data(data)) }, raw_data(data))
_internal_sort_from_indices_permuation(data, indices)
} }
return indices return indices
} }
@@ -177,6 +200,8 @@ sort_by_with_indices :: proc(data: $T/[]$E, less: proc(i, j: E) -> bool, allocat
} }
return .Equal return .Equal
}, ctx) }, ctx)
_internal_sort_from_indices_permuation(data, indices)
} }
} }
return indices return indices
@@ -208,6 +233,8 @@ sort_by_with_indices_with_data :: proc(data: $T/[]$E, less: proc(i, j: E, user_d
} }
return .Equal return .Equal
}, ctx) }, ctx)
_internal_sort_from_indices_permuation(data, indices)
} }
} }
return indices return indices