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 : rate array -> data -> string array
+
+ method private concat : data -> matrix =
+ fun (origin : data) ->
+ Array.map Array.concat (Array.map Array.to_list origin)
+
+ method private channels : data -> 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 =
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 make_rate =
+ fun (denom : int) ->
+ fun (num : int) ->
+ new rate num denom in
+ let nums = Array.map Sndfile.samplerate files in
+ let rates = Array.map (make_rate 1) nums 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
array_map2 (new signal) rates (Array.map stream2func containers) in
new beam signals
- method write : int array -> value_type 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
-
+ method write : rate array -> data -> string array =
+ fun (rates : rate array) ->
+ fun (output : data) ->
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 get_freq = fun (r : rate) -> r#to_int in
+ let freqs = Array.map get_freq rates 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 openwr = fun path -> fun channel -> fun freq ->
+ Sndfile.openfile ~info:(Sndfile.RDWR, file_format, channel, freq) path in
+ array_map3 openwr paths channels freqs 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 (new rate 0 1) (Array.get values)
- method write : value_type array array -> string array
+ 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 : rate array -> data -> string array =
+ fun (rates : rate array) ->
+ fun (data : data) ->
+ 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 : rate array -> data -> string array =
+ fun (rates : rate array) ->
+ fun (data : data) ->
+ 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;;
projects / Faustine.git / blobdiff