X-Git-Url: https://scm.cri.ensmp.fr/git/Faustine.git/blobdiff_plain/c4ce4ce8d2dff5817b7caa758655465e34456368..4d5f39ea1ae1eff1d8eadf7875851be467e40a47:/interpretor/beam.ml

diff --git a/interpretor/beam.ml b/interpretor/beam.ml
index 0468e42..a4f22b8 100644
--- a/interpretor/beam.ml
+++ b/interpretor/beam.ml
@@ -8,6 +8,8 @@
 exception Beam_matching of string;;
 
 open Types;;
+open Basic;;
+open Value;;
 open Signal;;
 
 class beam : signal_type array -> beam_type = 
@@ -59,12 +61,39 @@ class beam : signal_type array -> beam_type =
 
 	    else raise (Beam_matching "matching size error")
 
-      method time : time -> value_type array = 
+      method at : time -> value_type array = 
 	fun t ->
 	  let signal_at = fun (t : time) -> fun (s : signal_type) -> s#at t in
 	  Array.map (signal_at t) self#get
 
-      method output : int -> (int array) * (float array array) = 
+      method output : int -> value_type array array = 
 	  fun (length_max : int) ->
-	    
+	    let transpose : 'a array array -> 'a array array = 
+	      fun matrix ->
+		let get_element = fun i -> fun array -> array.(i) in
+		let get_column = fun m -> fun i -> Array.map (get_element i) m in
+		Array.init self#width (get_column matrix) in
+            let value_init = new value Error in
+	    let container = Array.make length_max 
+		(Array.make self#width value_init) in
+	    let index = ref 0 in
+
+	    try
+	      while !index < length_max do
+		container.(!index) <- self#at !index;
+		incr index;
+	      done;
+	      transpose container
+	    with x ->
+
+	      let error_message = 
+		match x with
+		| Convert_Error s -> "Convert_Error: " ^ s
+		| Basic_operation s -> "Basic_operation: " ^ s
+		| Signal_operation s -> "Signal_operation: " ^ s
+		| Beam_matching s -> "Beam_Matching_Error: " ^ s
+		| _ -> "Compute finished."
+	      in
+	      let () = print_string error_message in
+	      transpose (Array.sub container 0 !index)
     end