X-Git-Url: https://svn.cri.ensmp.fr/git/Faustine.git/blobdiff_plain/14e4f8cd6102d273c520c2d2e0168a5221a7daeb..d1f93175cc683d72876cb56b4a8f566578296bb9:/interpretor/beam.ml

diff --git a/interpretor/beam.ml b/interpretor/beam.ml
index cf5c003..f9ea667 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 = 
@@ -16,13 +18,17 @@ class beam : signal_type array -> beam_type =
       val signals = signals_init
 
       method get = signals
-      method length = Array.length signals
+      method width = Array.length signals
 
       method sub : int -> int -> beam_type = 
 	fun start ->
 	  fun len ->
 	    new beam (Array.sub self#get start len)
 
+      method cut : int -> beam_type * beam_type =
+	fun (cut_width : int)->
+	  ((self#sub 0 cut_width),(self#sub cut_width (self#width - cut_width))) 
+
       method append : beam_type -> beam_type =
 	fun (b : beam_type) -> 
 	  new beam (Array.append self#get b#get)
@@ -37,16 +43,16 @@ class beam : signal_type array -> beam_type =
 	  
       method matching : int -> beam_type = 
 	  fun size ->
-	    if size = self#length then new beam self#get
+	    if size = self#width then new beam self#get
 
-	    else if (size > self#length) && (size mod self#length = 0) then
+	    else if (size > self#width) && (size mod self#width = 0) then
 	      let () = self#add_memory 1 in
 	      let split_signals = 
 		Array.concat (Array.to_list 
-		     (Array.make (size/self#length) self#get)) in
+		     (Array.make (size/self#width) self#get)) in
 	      new beam split_signals
 
-	    else if (size < self#length) && (self#length mod size = 0) then
+	    else if (size < self#width) && (self#width mod size = 0) then
 	      let rec merge_rec  = 
 		fun (sa : signal_type array) ->
 		  fun (l : int) ->
@@ -59,11 +65,60 @@ 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 list) * (float array list) = 
-	  
+      method output : int -> float array array array = 
+	  fun (length_max : int) ->
+	    let () = print_string("    Faustine -> Calculating...") in
+	    let tic = Sys.time () in
+
+	    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 value2float = fun (v : value_type) -> v#to_float_array in
+            let init = [|0.|] in
+	    let container = Array.make length_max 
+		(Array.make self#width init) in
+	    let index = ref 0 in
+
+	    try
+	      while !index < length_max do
+		container.(!index) <- Array.map value2float (self#at !index);
+		incr index;
+	      done;
+	      let result = transpose container in
+	      let toc = Sys.time () in
+	      let () = print_endline (" Done. (duration: " ^ (string_of_float (toc -. tic)) ^ "s.)") in
+	      result
+
+	    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
+		| Invalid_argument s -> ""
+		| _ -> "Unknown error"
+	      in
+	      let () = print_string (error_message) in
+	      let result = transpose (Array.sub container 0 !index) in
+	      let toc = Sys.time () in
+	      let () = print_endline (" Done. (duration: " ^ (string_of_float (toc -. tic)) ^ "s.)") in
+	      result
+
+      method frequency : int array = 
+	let each_rate : signal -> int = 
+	  fun (s : signal) -> 
+	    let rate = s#frequency in
+	    if rate > 0 then rate
+	    else if rate = 0 then 44100
+	    else raise (Beam_matching "frequency error.") in	
+	Array.map each_rate self#get 
+
     end