diff options
Diffstat (limited to 'stdlib')
| -rw-r--r-- | stdlib/array.ml | 16 | ||||
| -rw-r--r-- | stdlib/array.mli | 3 |
2 files changed, 15 insertions, 4 deletions
diff --git a/stdlib/array.ml b/stdlib/array.ml index 8aef73b8d..3c012062b 100644 --- a/stdlib/array.ml +++ b/stdlib/array.ml @@ -21,6 +21,14 @@ external unsafe_set: 'a array -> int -> 'a -> unit = "%array_unsafe_set" external make: int -> 'a -> 'a array = "make_vect" external create: int -> 'a -> 'a array = "make_vect" +let init l f = + if l = 0 then [||] else + let res = create l (f 0) in + for i = 1 to pred l do + unsafe_set res i (f i) + done; + res + let make_matrix sx sy init = let res = create sx [||] in for x = 0 to pred sx do @@ -33,11 +41,11 @@ let create_matrix = make_matrix let copy a = let l = length a in if l = 0 then [||] else begin - let r = create l (unsafe_get a 0) in - for i = 1 to l-1 do - unsafe_set r i (unsafe_get a i) + let res = create l (unsafe_get a 0) in + for i = 1 to pred l do + unsafe_set res i (unsafe_get a i) done; - r + res end let append a1 a2 = diff --git a/stdlib/array.mli b/stdlib/array.mli index 413eb15b8..218b0e740 100644 --- a/stdlib/array.mli +++ b/stdlib/array.mli @@ -37,6 +37,9 @@ external create: int -> 'a -> 'a array = "make_vect" Consequently, if [x] is mutable, it is shared among all elements of the array, and modifying [x] through one of the array entries will modify all other entries at the same time. *) +val init: int -> (int -> 'a) -> 'a array + (* [Array.init n f] returns a fresh array of length [n], + with element number [i] equal to [f i]. *) val make_matrix: int -> int -> 'a -> 'a array array val create_matrix: int -> int -> 'a -> 'a array array (* [Array.make_matrix dimx dimy e] returns a two-dimensional array |