flan/test/programs/slices.flan
Joseph Ferano 84e170b349 Slice algorithms in place, because there is nowhere to put a copy
A sequence library normally returns new sequences. There is no allocator, so
every one of these mutates the storage it was handed and a slice is the handle
that makes that useful: (slice grid 4 9) is ptr+len into grid, so sorting it
sorts those five elements and leaves the rest of grid alone. The test asserts
exactly that — it sorts a subslice and prints the whole owning array — because
it is the property that would die silently the day a slice parameter started
being copied rather than passed by value, and -O2's mem2reg would hide it.

Insertion sort rather than anything faster. Quicksort wants a stack and
mergesort wants a buffer, and neither exists; insertion sort needs a swap and
two indices. It is also the only one of the three whose inner loop is short
enough to read, which matters more than the asymptotics on the slice sizes a
frame loop actually sorts. The `and` guarding it short-circuits, and that is
load-bearing: at j = 0 the left test fails and (at s -1) is never evaluated,
so the bounds check never fires.

Over [i32] and nothing else. There are no generics, so a second element type
is a second copy of all seven functions emitted into every program that links
the prelude, and i32 is the type indices, ids and tile values already have.
An f32 set waits for a program that wants one.

min-i32 and max-i32 return (Option i32) rather than a sentinel because there
is no i32 that means "the slice was empty" and is not also a possible element.
sum-i32 accumulates in i64 and widens each element explicitly — there is no
implicit widening anywhere, and an i32 total over a screenful of i32 is how a
sum wraps without anyone noticing.
2026-09-11 18:48:11 +07:00

85 lines
3.5 KiB
Plaintext

;;;; The prelude's in-place slice algorithms.
;;;;
;;;; Every input here is chosen so that a wrong implementation passes nothing.
;;;; The sort input is unsorted, has duplicates, has negatives and has an odd
;;;; length, so a comparison with the wrong sense, an off-by-one that drops the
;;;; last element, and a swap that loses an equal key all show up. The second
;;;; sort is reverse-sorted, which is the worst case for insertion sort and the
;;;; case a no-op comparison would pass. The third sorts a *subslice* and then
;;;; prints the whole owning array: a slice is ptr+len into its owner, so the
;;;; five elements inside the range must be sorted and the three outside it
;;;; must be untouched. That last one is the property that dies silently if a
;;;; slice parameter ever starts being copied.
(defvar xs [7 i32])
(defvar ys [5 i32])
(defvar zs [8 i32])
(defn show [s [i32]]
(dotimes [i (len s)]
(when (> i 0) (print-str " "))
(print-i64 (i64 (at s i))))
(newline))
(defn load-xs []
(set (at xs 0) 5)
(set (at xs 1) -3)
(set (at xs 2) 5)
(set (at xs 3) 0)
(set (at xs 4) 12)
(set (at xs 5) -3)
(set (at xs 6) 7))
(defn main [] i32
(load-xs)
(show (slice xs 0 (len xs))) ; 5 -3 5 0 12 -3 7
;; Reading the whole slice, before anything reorders it.
(print-i64 (sum-i32 (slice xs 0 (len xs)))) (newline) ; 23
(print-i64 (i64 (match (min-i32 (slice xs 0 (len xs))) (Some v) v None 99)))
(newline) ; -3
(print-i64 (i64 (match (max-i32 (slice xs 0 (len xs))) (Some v) v None 99)))
(newline) ; 12
;; First index, not the last: 5 appears at 0 and at 2.
(print-i64 (i64 (match (index-of-i32 (slice xs 0 (len xs)) 5)
(Some v) v None -1)))
(newline) ; 0
(print-i64 (i64 (match (index-of-i32 (slice xs 0 (len xs)) 4)
(Some v) v None -1)))
(newline) ; -1
;; An empty slice has no least element, and None is the answer.
(print-i64 (i64 (match (min-i32 (slice xs 3 3)) (Some v) v None 99)))
(newline) ; 99
;; Reverse of an odd-length slice: the middle element stays put.
(reverse-i32! (slice xs 0 (len xs)))
(show (slice xs 0 (len xs))) ; 7 -3 12 0 5 -3 5
;; And of a two-element one, the smallest case that can actually move.
(reverse-i32! (slice xs 0 2))
(show (slice xs 0 (len xs))) ; -3 7 12 0 5 -3 5
(load-xs)
(sort-i32! (slice xs 0 (len xs)))
(show (slice xs 0 (len xs))) ; -3 -3 0 5 5 7 12
;; Reverse-sorted: the case a comparison that never fires would pass.
(set (at ys 0) 5) (set (at ys 1) 4) (set (at ys 2) 3)
(set (at ys 3) 2) (set (at ys 4) 1)
(sort-i32! (slice ys 0 (len ys)))
(show (slice ys 0 (len ys))) ; 1 2 3 4 5
;; A subslice, with the elements on both sides left alone.
(set (at zs 0) 100) (set (at zs 1) 9) (set (at zs 2) -1)
(set (at zs 3) 9) (set (at zs 4) 4) (set (at zs 5) 0)
(set (at zs 6) 200) (set (at zs 7) 300)
(sort-i32! (slice zs 1 6))
(show (slice zs 0 (len zs))) ; 100 -1 0 4 9 9 200 300
;; Degenerate lengths must do nothing rather than run off an end.
(sort-i32! (slice zs 0 0))
(reverse-i32! (slice zs 0 0))
(sort-i32! (slice zs 2 3))
(reverse-i32! (slice zs 2 3))
(show (slice zs 0 (len zs))) ; 100 -1 0 4 9 9 200 300
0)