2 files changed, 23 insertions, 8 deletions
diff --git a/stdlib/array.ml b/stdlib/array.ml
index aa156bef9..3089e2f83 100644
--- a/stdlib/array.ml
+++ b/stdlib/array.ml
@@ -32,13 +32,7 @@ let copy a =
     r
   end
 
-let concat_aux a1 a2 l1 l2 init =
-  let r = new (l1 + l2) init in
-  for i = 0 to l1 - 1 do unsafe_set r i (unsafe_get a1 i) done;  
-  for i = 0 to l2 - 1 do unsafe_set r (i + l1) (unsafe_get a1 i) done;  
-  r
-
-let concat a1 a2 =
+let append a1 a2 =
   let l1 = length a1 and l2 = length a2 in
   if l1 = 0 & l2 = 0 then [||] else begin
     let r = new (l1 + l2) (unsafe_get (if l1 > 0 then a1 else a2) 0) in
@@ -47,6 +41,26 @@ let concat a1 a2 =
     r
   end
 
+let concat_aux init al =
+  let size = List.fold_left (fun sz a -> sz + length a) 0 al in
+  let res = new size init in
+  let pos = ref 0 in
+  List.iter
+    (fun a ->
+      for i = 0 to length a - 1 do
+        unsafe_set res !pos (unsafe_get a i);
+        incr pos
+      done)
+    al;
+  res
+
+let concat al =
+  let rec find_init = function
+      [] -> [||]
+    | a :: rem ->
+        if length a > 0 then concat_aux (unsafe_get a 0) al else find_init rem
+  in find_init al
+
 let sub a ofs len =
   if ofs < 0 or len < 0 or ofs + len > length a then invalid_arg "Array.sub"
   else if len = 0 then [||]
diff --git a/stdlib/array.mli b/stdlib/array.mli
index dc9a4156f..28f1e586d 100644
--- a/stdlib/array.mli
+++ b/stdlib/array.mli
@@ -6,7 +6,8 @@ val get: 'a array -> int -> 'a
 val set: 'a array -> int -> 'a -> unit
 external new: int -> 'a -> 'a array = "make_vect"
 val new_matrix: int -> int -> 'a -> 'a array array
-val concat: 'a array -> 'a array -> 'a array
+val append: 'a array -> 'a array -> 'a array
+val concat: 'a array list -> 'a array
 val sub: 'a array -> int -> int -> 'a array
 val copy: 'a array -> 'a array
 val fill: 'a array -> int -> int -> 'a -> unit