simon/ocaml-containers
1
0
Fork 0
mirror of https://github.com/c-cube/ocaml-containers.git synced 2025-12-06 11:15:31 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

perf: implement iter separately from iteri

Browse source
This commit is contained in:
Simon Cruanes 2024-01-07 23:17:57 -05:00
parent a281476082
commit 12ff3802ce
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 23 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/pvec/containers_pvec.ml
View file

@ -224,6 +224,29 @@ let drop_last self =
else else
snd (pop self) snd (pop self)
let rec iter_rec_ f (self : _ tree) =
match self with
| Empty -> ()
| Leaf a ->
for i = 0 to A.length a - 1 do
f (Array.unsafe_get a i)
done
| Node a ->
for i = 0 to A.length a - 1 do
iter_rec_ f (Array.unsafe_get a i)
done
let iter f self =
iter_rec_ f self.t;
for i = 0 to A.length self.tail - 1 do
f (Array.unsafe_get self.tail i)
done
let fold_left f x m =
let acc = ref x in
iter (fun x -> acc := f !acc x) m;
!acc
let rec iteri_rec_ f idx (self : _ tree) = let rec iteri_rec_ f idx (self : _ tree) =
match self with match self with
| Empty -> () | Empty -> ()
@ -276,8 +299,6 @@ let foldi_rev f x m =
iteri_rev (fun i x -> acc := f !acc i x) m; iteri_rev (fun i x -> acc := f !acc i x) m;
!acc !acc
let iter f m = iteri (fun _ x -> f x) m
let fold_left f x m = fold_lefti (fun acc _ x -> f acc x) x m
let fold_rev f x m = foldi_rev (fun acc _ x -> f acc x) x m let fold_rev f x m = foldi_rev (fun acc _ x -> f acc x) x m
let rec map_t f (self : _ tree) : _ tree = let rec map_t f (self : _ tree) : _ tree =