New Faustine tested by sin.dsp and fft.dsp.
[Faustine.git] / interpretor / faustio.ml
1 (**
2 Module: Faustio
3 Description: audio input/output, csv input/output
4 @author WANG Haisheng
5 Created: 12/08/2013 Modified: 13/08/2013
6 *)
7
8 open Types;;
9 open Basic;;
10 open Value;;
11 open Signal;;
12 open Beam;;
13 open Aux;;
14
15 let default_output_path = "../output_sounds/";;
16
17 class virtual io =
18 object
19 method virtual read : string array -> beam
20 method virtual write : int array -> value_type array array -> string array
21 method private to_float : value_type array array -> float array array =
22 fun (origin : value_type array array) ->
23 let data =
24 let value2float = fun (v : value_type) -> v#to_float_array in
25 Array.map (Array.map value2float) origin in
26 Array.map Array.concat (Array.map Array.to_list data)
27 end;;
28
29 class waveio : io_type =
30 object (self)
31 inherit io
32 method read : string array -> beam =
33 fun (paths : string array) ->
34 let n = Array.length paths in
35 if n = 0 then
36 new beam [||]
37 else
38 let signals =
39 let files = Array.map Sndfile.openfile paths in
40 let frames = Array.map Int64.to_int
41 (Array.map Sndfile.frames files) in
42 let rates = Array.map Sndfile.samplerate files in
43 let create_container = fun l -> Array.create l 1. in
44 let containers = Array.map create_container frames in
45 let _ = array_map2 Sndfile.read files containers in
46 let _ = Array.map Sndfile.close files in
47 let stream2func =
48 fun stream -> fun t -> new value (R stream.(t)) in
49 array_map2 (new signal) rates (Array.map stream2func containers) in
50 new beam signals
51
52 method write : int array -> value_type array array -> string array =
53 fun (rates : int array) ->
54 fun (output : value_type array array) ->
55 let () = print_string(" Faustine -> Writing wave files...") in
56 let tic = Sys.time () in
57
58 let n = Array.length output in
59 let paths = Array.init n (fun i ->
60 default_output_path ^ "output" ^ (string_of_int (i + 1)) ^ ".wav") in
61
62 let files =
63 let channels =
64 let get_channel = fun s ->
65 let l = Array.length s in
66 match s.(l - 1)#get with
67 | Vec vec -> vec#size
68 | _ -> 1 in
69 Array.map get_channel output in
70 let file_format = Sndfile.format
71 Sndfile.MAJOR_WAV Sndfile.MINOR_PCM_16 in
72 let openwr = fun path -> fun channel -> fun rate ->
73 Sndfile.openfile ~info:(Sndfile.RDWR, file_format, channel, rate) path in
74 array_map3 openwr paths channels rates in
75 let () =
76 let data = self#to_float output in
77 let _ = array_map2 Sndfile.write files data in
78 let _ = Array.map Sndfile.close files in
79 let toc = Sys.time () in
80 let duration = toc -. tic in
81 print_endline
82 (" Done. (duration: " ^ (string_of_float duration) ^ "s)") in
83 paths
84 end;;
85
86
87 (*
88 class csvio : io_type =
89 object (self)
90 inherit io
91 method private csvread =
92 method read : string array -> beam =
93
94
95
96 method write : value_type array array -> string array
97 end;;
98 *)
99
100
101 (*
102 let csvread = fun (ic : in_channel) ->
103 let string_list = ref [] in
104 try
105 while true do
106 string_list := !string_list @ [(input_line ic)]
107 done;
108 [||]
109 with End_of_file ->
110 (*let () = print_endline(List.nth !string_list 0) in*)
111 Array.of_list (List.map float_of_string !string_list);;
112
113 let read_input_csv = fun argv ->
114 let n_input = (Array.length argv) - 4 in
115 if n_input < 0 then
116 raise (Invalid_argument "xxx")
117 else if n_input = 0 then
118 ([], [])
119 else
120 (* open csv file *)
121 let file_string_array = Array.sub argv 4 n_input in
122 let make_chemin s = io_macro_to_string Input_Route_string ^ s in
123 let file_chemin_string_array = Array.map make_chemin file_string_array in
124 let file_array = Array.map open_in file_chemin_string_array in
125 let file_list = Array.to_list file_array in
126
127 (* read sample rates and data *)
128 let rate_list = Array.to_list (Array.create n_input 0) in
129 let data_float_array_list = List.map csvread file_list in
130 let _ = List.map close_in file_list in
131 (rate_list, data_float_array_list);;
132
133
134 let write_output_csv = fun channel_int_list -> fun data_float_array_list ->
135 let () = print_string(" Faustine -> Writing csv files...") in
136 let tic = Sys.time () in
137
138 (* make output txt file names : output0, output1, ... *)
139 let n_output = List.length data_float_array_list in
140 let n_array = Array.init n_output (fun n -> n) in
141 let make_file_name i = "output" ^ (string_of_int i) ^ ".csv" in
142
143 (* make output wave file routes *)
144 let make_chemin s = io_macro_to_string Output_Route_string ^ s in
145 let file_name_string_array = Array.map make_file_name n_array in
146 let file_chemin_string_array = Array.map make_chemin file_name_string_array in
147 let file_chemin_string_list = Array.to_list file_chemin_string_array in
148
149 (* open output channels *)
150 let file_list = List.map open_out file_chemin_string_list in
151 let data_string_array_list = List.map (Array.map string_of_float) data_float_array_list in
152 let array_to_string = fun data_string_array -> fun channel_int ->
153 let data_length = Array.length data_string_array in
154 let rec to_string_rec =
155 fun data -> fun channel -> fun n -> fun i -> fun column ->
156 if i < n then
157 (
158 let element = data.(i) in
159 if column < (channel - 1) then
160 element ^ "," ^ (to_string_rec data channel n (i + 1) (column + 1))
161 else if column = (channel - 1) then
162 element ^ "\n" ^ (to_string_rec data channel n (i + 1) 0)
163 else raise (Invalid_argument "write_output_txt.")
164 )
165 else "" in
166 to_string_rec data_string_array channel_int data_length 0 0 in
167
168 let data_string_list = List.map2 array_to_string data_string_array_list channel_int_list in
169 let _ = List.map2 output_string file_list data_string_list in
170 let _ = List.map close_out file_list in
171 let toc = Sys.time () in
172 print_endline(" Done. (duration: " ^ (string_of_float (toc -. tic)) ^ "s)");;
173
174 *)