simon/sidekick
1
0
Fork 0
mirror of https://github.com/c-cube/sidekick.git synced 2025-12-10 13:14:09 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(vec): add append, fix prepend

Browse source
This commit is contained in:
Simon Cruanes 2021-11-19 22:35:40 -05:00
parent 1f9c43afa8
commit 4d3b278754
No known key found for this signature in database
GPG key ID: 4AC01D0849AA62B6
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/util/Vec.ml
View file

@ -96,6 +96,16 @@ let[@inline] fast_remove t i =
Array.unsafe_set t.data i @@ Array.unsafe_get t.data (t.sz - 1);
t.sz <- t.sz - 1
let append ~into v : unit =
if v.sz = 0 then ()
else (
if v.sz + into.sz > Array.length into.data then (
resize_ into v.data.(0) (v.sz + into.sz);
);
Array.blit v.data 0 into.data into.sz v.sz;
into.sz <- into.sz + v.sz;
)
let prepend v ~into : unit =
if v.sz = 0 then ()
else (
@ -104,6 +114,7 @@ let prepend v ~into : unit =
);
Array.blit into.data 0 into.data v.sz into.sz; (* shift elements *)
Array.blit v.data 0 into.data 0 v.sz;
into.sz <- into.sz + v.sz;
)
let filter_in_place f vec =

3
src/util/Vec.mli
View file

@ -70,6 +70,9 @@ val fast_remove : 'a t -> int -> unit
(** Remove element at index [i] without preserving order
(swap with last element) *)
val append : into:'a t -> 'a t -> unit
(** [append ~into v] pushes elements of [v] in the vector [into] *)
val prepend : 'a t -> into:'a t -> unit
(** [prepend v ~into] pushes all elements of [v] into [into],
at the beginning. consumes [v]. *)