(** Module: Faustio Description: audio input/output, csv input/output @author WANG Haisheng Created: 12/08/2013 Modified: 13/08/2013 *) open Types;; open Basic;; open Value;; open Signal;; open Beam;; open Aux;; 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 -> 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 = object (self) inherit io method read : string array -> beam = fun (paths : string array) -> let n = Array.length paths in if n = 0 then new beam [||] else let signals = let files = Array.map Sndfile.openfile paths in let frames = Array.map Int64.to_int (Array.map Sndfile.frames files) in let rates = Array.map Sndfile.samplerate files in let create_container = fun l -> Array.create l 1. in let containers = Array.map create_container frames in let _ = array_map2 Sndfile.read files containers in let _ = Array.map Sndfile.close files in let stream2func = fun stream -> fun t -> new value (R stream.(t)) in array_map2 (new signal) rates (Array.map stream2func containers) in new beam signals method write : int array -> float array array array -> string array = fun (rates : int array) -> fun (output : float array array array) -> let n = Array.length output in let paths = Array.init n (fun i -> _dir ^ _basename ^ (string_of_int (i + 1)) ^ ".wav") in let files = 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 data = self#concat output in let _ = array_map2 Sndfile.write files data in let _ = Array.map Sndfile.close files in paths end;; class csvio : io_type = object (self) inherit io 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 : 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;; 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;;