X-Git-Url: https://scm.cri.mines-paristech.fr/git/Faustine.git/blobdiff_plain/d8214dc258396483f72533504515a4e1d1b3deef..fc2ec49c325b99370170789886b4be4f43e16d68:/interpretor/faustio.ml?ds=inline

diff --git a/interpretor/faustio.ml b/interpretor/faustio.ml
index 25a2868..3094ce9 100644
--- a/interpretor/faustio.ml
+++ b/interpretor/faustio.ml
@@ -12,18 +12,30 @@ open Signal;;
 open Beam;;
 open Aux;;
 
-let default_output_path = "../output_sounds/";;
+let csv_read_buffer_length = 0xFFFF;;
 
 class virtual io = 
   object
+    val mutable _basename = ""
+    val mutable _dir = ""
+    method set : string -> string -> unit = 
+      fun (dir : string) ->
+	fun (basename : string) ->
+	  _basename <- basename; _dir <- dir
+
     method virtual read : string array -> beam
-    method virtual write : int array -> value_type array array -> string array
-    method private to_float : value_type array array -> float array array = 
-      fun (origin : value_type array array) ->
-	let data = 
-	  let value2float = fun (v : value_type) -> v#to_float_array in
-	  Array.map (Array.map value2float) origin in
-	Array.map Array.concat (Array.map Array.to_list data)
+    method virtual write : int array -> float array array array -> string array
+    
+    method private concat : float array array array -> float array array = 
+      fun (origin : float array array array) ->
+	Array.map Array.concat (Array.map Array.to_list origin)
+
+    method private channels : float array array array -> int array = 
+      fun data ->
+	let get_channel = fun s -> 
+	  let l = Array.length s in
+	  Array.length s.(l - 1) in
+	Array.map get_channel data  
   end;;
 
 class waveio : io_type = 
@@ -49,126 +61,132 @@ class waveio : io_type =
 	    array_map2 (new signal) rates (Array.map stream2func containers) in
 	  new beam signals
 
-    method write : int array -> value_type array array -> string array = 
+    method write : int array -> float array array array -> string array = 
       fun (rates : int array) ->
-	fun (output : value_type array array) ->
-	  let () = print_string("    Faustine -> Writing wave files...") in
-	  let tic = Sys.time () in
-
+	fun (output : float array array array) ->
 	  let n = Array.length output in	  
 	  let paths = Array.init n (fun i -> 
-	    default_output_path ^ "output" ^ (string_of_int (i + 1)) ^ ".wav") in
+	    _dir ^ _basename ^ (string_of_int (i + 1)) ^ ".wav") in
 
 	  let files = 
-	    let channels = 
-	      let get_channel = fun s -> 
-		let l = Array.length s in
-		match s.(l - 1)#get with
-		| Vec vec -> vec#size
-		| _ -> 1 in
-	      Array.map get_channel output in	    
+	    let channels = self#channels output in 
 	    let file_format = Sndfile.format 
 		Sndfile.MAJOR_WAV Sndfile.MINOR_PCM_16 in
 	    let openwr = fun path -> fun channel -> fun rate ->
 	      Sndfile.openfile ~info:(Sndfile.RDWR, file_format, channel, rate) path in
 	    array_map3 openwr paths channels rates in 
-	  let () = 
-	    let data = self#to_float output in
+
+	    let data = self#concat output in
 	    let _ = array_map2 Sndfile.write files data in
 	    let _ = Array.map Sndfile.close files in
-	    let toc = Sys.time () in
-	    let duration = toc -. tic in
-	    print_endline 
-	      (" Done. (duration: " ^ (string_of_float duration) ^ "s)") in
 	  paths
   end;;
 
 
-(*
+
 class csvio : io_type = 
   object (self)
     inherit io
-    method private csvread = 
-    method read : string array -> beam =
-      
+    method private csvread : in_channel -> signal = 
+      fun (ic : in_channel) ->
+	let buffer = Buffer.create csv_read_buffer_length in
+	let () = 
+	  try
+	    while true do
+	      Buffer.add_string buffer (input_line ic);
+	      Buffer.add_char buffer '\t';
+	    done; 
+	  with End_of_file -> () in
+	let content = Buffer.contents buffer in
+	let lines = Str.split (Str.regexp "\t") content in
+	let elements = List.map (Str.split (Str.regexp ",")) lines in
+	let data = 
+	  let data_in_list = List.map (List.map float_of_string) elements in
+	  Array.of_list (List.map Array.of_list data_in_list) in      
+	let values = 
+	  let convertor = new value Zero in
+	  Array.map (convertor#of_float_array) data in
+	new signal 0 (Array.get values)
 
+    method read : string array -> beam =
+      fun (paths : string array) ->
+	let files = Array.map open_in paths in
+	let signals = Array.map self#csvread files in
+	new beam signals
 
-    method write : value_type array array -> string array
+    method write : int array -> float array array array -> string array = 
+      fun (rates : int array) ->
+	fun (data : float array array array) ->
+	  let paths = 
+	    let n = Array.length data in
+	    let path_pattern = fun i -> 
+	      _dir ^ _basename ^ (string_of_int (i + 1)) ^ ".csv" in
+	    Array.init n path_pattern in	  
+
+	  let files = Array.map open_out paths in
+	  let strings = 
+	    let value2string : float array -> string =
+	      fun (v : float array) ->
+		let strings = Array.map string_of_float v in
+		String.concat "," (Array.to_list strings) in
+	    let signal2string : float array array -> string = 
+	      fun (s : float array array) -> 
+		let lines = Array.map value2string s in
+		String.concat "\n" (Array.to_list lines) in
+	    Array.map signal2string data in
+	  let _ = array_map2 output_string files strings in
+	  let _ = Array.map close_out files in
+	  paths
   end;;
-*)
-
 
-(*
-let csvread = fun (ic : in_channel) ->
-  let string_list = ref [] in
-  try
-    while true do
-      string_list := !string_list @ [(input_line ic)]
-    done;
-    [||]
-  with End_of_file ->
-    (*let () = print_endline(List.nth !string_list 0) in*)
-    Array.of_list (List.map float_of_string !string_list);;
-
-let read_input_csv = fun argv ->
-        let n_input = (Array.length argv) - 4 in
-	if n_input < 0 then 
-	        raise (Invalid_argument "xxx")
-	else if n_input = 0 then 
-	        ([], [])
-	else
-	        (* open csv file *)
-		let file_string_array = Array.sub argv 4 n_input in
-		let make_chemin s = io_macro_to_string Input_Route_string ^ s in
-		let file_chemin_string_array = Array.map make_chemin file_string_array in
-		let file_array = Array.map open_in file_chemin_string_array in
-		let file_list = Array.to_list file_array in
-
-		(* read sample rates and data *)
-		let rate_list = Array.to_list (Array.create n_input 0) in
-		let data_float_array_list = List.map csvread file_list in
-		let _ = List.map close_in file_list in
-		(rate_list, data_float_array_list);;
-
-
-let write_output_csv = fun channel_int_list -> fun data_float_array_list ->
-        let () = print_string("    Faustine -> Writing csv files...") in
-	let tic = Sys.time () in
-
-	(* make output txt file names : output0, output1, ... *)
-	let n_output = List.length data_float_array_list in
-	let n_array = Array.init n_output (fun n -> n) in
-	let make_file_name i = "output" ^ (string_of_int i) ^ ".csv" in
-
-	(* make output wave file routes *)
-	let make_chemin s = io_macro_to_string Output_Route_string ^ s in
-	let file_name_string_array = Array.map make_file_name n_array in
-	let file_chemin_string_array = Array.map make_chemin file_name_string_array in
-	let file_chemin_string_list = Array.to_list file_chemin_string_array in
-
-	(* open output channels *)
-        let file_list = List.map open_out file_chemin_string_list in
-	let data_string_array_list = List.map (Array.map string_of_float) data_float_array_list in
-	let array_to_string = fun data_string_array -> fun channel_int ->
-	  let data_length = Array.length data_string_array in
-	  let rec to_string_rec = 
-	    fun data -> fun channel -> fun n -> fun i -> fun column ->
-	      if i < n then 
-		(
-		let element = data.(i) in
-		if column < (channel - 1) then 
-		  element ^ "," ^ (to_string_rec data channel n (i + 1) (column + 1))
-		else if column = (channel - 1) then
-		  element ^ "\n" ^ (to_string_rec data channel n (i + 1) 0)
-		else raise (Invalid_argument "write_output_txt.")
-		    )
-	      else "" in
-	  to_string_rec data_string_array channel_int data_length 0 0 in
-
-	let data_string_list = List.map2 array_to_string data_string_array_list channel_int_list in
-	let _ = List.map2 output_string file_list data_string_list in
-	let _ = List.map close_out file_list in
-	let toc = Sys.time () in
-	print_endline(" Done. (duration: " ^ (string_of_float (toc -. tic)) ^ "s)");;
 
-*)
+class iomanager = 
+  object (self)
+    val wave = new waveio
+    val csv = new csvio
+    val mutable _dir = ""
+    val mutable _format = ""
+    val mutable _basename = ""
+
+    method private grab_format : string -> string = 
+      fun (path : string) ->
+	let fragments = Str.split (Str.regexp "\.") path in
+	let n = List.length fragments in
+	List.nth fragments (n - 1)	
+
+    method read : string list -> beam_type = 
+      fun (paths : string list) ->
+	let formats = List.map self#grab_format paths in
+	let read_one : string -> string -> beam_type = 
+	  fun (format : string) ->
+	    fun (path : string) ->
+	      if format = "wav" then wave#read [|path|]
+	      else if format = "csv" then csv#read [|path|]
+	      else raise (Invalid_argument "Unknown format.") in
+	let beams = List.map2 read_one formats paths in
+	let concat : beam_type -> beam_type -> beam_type = 
+	  fun b1 -> fun b2 -> b1#append b2 in
+	List.fold_left concat (new beam [||]) beams
+	      
+    method set : string -> string -> string -> unit = 
+      fun (dir : string) ->
+	fun (format : string) ->
+	  fun (basename : string) ->
+	    _dir <- dir; 
+	    _format <- format; 
+	    _basename <- basename;
+	    wave#set _dir _basename;
+	    csv#set _dir _basename
+
+    method write : int array -> float array array array -> string array = 
+      fun (rates : int array) ->
+	fun (data : float array array array) ->
+	  if _format = "" then
+	    raise (Invalid_argument "output format unset.")
+	  else if _format = "wav" then 
+	    wave#write rates data
+	  else if _format = "csv" then
+	    csv#write rates data 
+	  else raise (Invalid_argument "unknown format.")
+	
+  end;;