X-Git-Url: https://scm.cri.ensmp.fr/git/Faustine.git/blobdiff_plain/d8214dc258396483f72533504515a4e1d1b3deef..a6be79ccb7456a5181561f094c1f2a4f90aadb02:/interpretor/beam.ml

diff --git a/interpretor/beam.ml b/interpretor/beam.ml
index c4c3b9e..273ae1c 100644
--- a/interpretor/beam.ml
+++ b/interpretor/beam.ml
@@ -70,33 +70,35 @@ class beam : signal_type array -> beam_type =
 	  let signal_at = fun (t : time) -> fun (s : signal_type) -> s#at t in
 	  Array.map (signal_at t) self#get
 
-      method output : int -> value_type array array = 
+      method output : int -> float array 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 value2float = fun (v : value_type) -> v#to_float_array in
+            let init = [|0.|] in
 	    let container = Array.make length_max 
-		(Array.make self#width value_init) in
+		(Array.make self#width init) in
 	    let index = ref 0 in
 
 	    try
 	      while !index < length_max do
-		container.(!index) <- self#at !index;
+		container.(!index) <- Array.map value2float (self#at !index);
 		incr index;
 	      done;
 	      transpose container
-	    with x ->
 
+	    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."
+		| Invalid_argument s -> "Compute finished."
+		| _ -> "Unknown error"
 	      in
 	      let () = print_string error_message in
 	      transpose (Array.sub container 0 !index)