ba139614a3fb1152d4968beeb6ed13e21e000128
[Faustine.git] / interpretor / process.ml
1 (**
2 Module: Process
3 Description: Faust process classes
4 @author WANG Haisheng
5 Created: 03/06/2013 Modified: 14/08/2013
6 *)
7
8 open Types;;
9 open Aux;;
10 open Basic;;
11 open Symbol;;
12 open Value;;
13 open Signal;;
14 open Beam;;
15
16 exception NotYetDone;;
17 exception Dimension_error of string;;
18 exception Process_error of string;;
19
20
21 (* PARSER *)
22
23 let exp_of_string s = (Parser.main Lexer.token (Lexing.from_string s));;
24
25
26 class dimension : int * int -> dimension_type =
27 fun (init : int * int) ->
28 object (self)
29 val _input = fst init
30 val _output = snd init
31
32 method input = _input
33 method output = _output
34
35 method par : dimension_type -> dimension_type =
36 fun dim ->
37 new dimension
38 ((self#input + dim#input), (self#output + dim#output))
39
40 method seq : dimension_type -> dimension_type =
41 fun dim ->
42 if self#output = dim#input then
43 new dimension (self#input, dim#output)
44 else raise (Dimension_error "seq dimension not matched.")
45
46 method split : dimension_type -> dimension_type =
47 fun dim ->
48 if dim#input mod self#output = 0 then
49 new dimension (self#input, dim#output)
50 else raise (Dimension_error "split dimension not matched.")
51
52 method merge : dimension_type -> dimension_type =
53 fun dim ->
54 if self#output mod dim#input = 0 then
55 new dimension (self#input, dim#output)
56 else raise (Dimension_error "merge dimension not matched.")
57
58 method _rec : dimension_type -> dimension_type =
59 fun dim ->
60 if self#output >= dim#input && self#input >= dim#output then
61 new dimension (self#input - dim#output, self#output)
62 else raise (Dimension_error "rec dimension not matched.")
63 end;;
64
65 class virtual process =
66 fun (exp_init : faust_exp) ->
67 object
68 val _exp = exp_init
69 val virtual _dim : dimension_type
70 val virtual _delay : int
71 method exp = _exp
72 method dim = _dim
73 method delay = _delay
74 method virtual eval : beam_type -> beam_type
75 end
76
77 class proc_const : faust_exp -> process_type =
78 fun (exp_init : faust_exp) ->
79 let _const =
80 match exp_init with
81 | Const b -> b
82 | _ -> raise (Process_error "const process constructor.") in
83
84 object (self)
85 inherit process exp_init
86 val _dim = new dimension (0,1)
87 val _delay = 0
88 method private const = _const
89 method eval : beam_type -> beam_type =
90 fun (input : beam_type) ->
91 if input#get = [||] then
92 new beam [| new signal (new rate 0 1)
93 (fun t -> new value self#const)|]
94 else
95 raise (Process_error "proc_const accepts no input.")
96 end;;
97
98
99 class proc_ident : faust_exp -> process_type =
100 fun (exp_init : faust_exp) ->
101 let _symbol =
102 match exp_init with
103 | Ident s -> s
104 | _ -> raise (Process_error "ident process constructor.") in
105
106 object (self)
107 inherit process exp_init
108 val _dim = new dimension (dimension_of_symbol _symbol)
109 val _delay = delay_of_symbol _symbol
110 method private symb = _symbol
111
112 method private beam_of_ident : int -> signal_type -> beam_type =
113 fun (n : int) ->
114 fun (s : signal_type) ->
115 if n = (self#dim)#input then
116 new beam [|s|]
117 else raise (Process_error ("Ident " ^ string_of_symbol self#symb))
118
119 method eval : beam_type -> beam_type =
120 fun (input : beam_type) ->
121 let n = Array.length input#get in
122 match self#symb with
123 | Pass -> self#beam_of_ident n input#get.(0)
124 | Stop -> if n = 1 then new beam [||]
125 else raise (Process_error "Ident !")
126 | Add -> self#beam_of_ident n
127 ((input#get.(0))#add input#get.(1))
128 | Sub -> self#beam_of_ident n
129 ((input#get.(0))#sub input#get.(1))
130 | Mul -> self#beam_of_ident n
131 ((input#get.(0))#mul input#get.(1))
132 | Div -> self#beam_of_ident n
133 ((input#get.(0))#div input#get.(1))
134 | Power -> self#beam_of_ident n
135 ((input#get.(0))#power input#get.(1))
136 | And -> self#beam_of_ident n
137 ((input#get.(0))#_and input#get.(1))
138 | Or -> self#beam_of_ident n
139 ((input#get.(0))#_or input#get.(1))
140 | Xor -> self#beam_of_ident n
141 ((input#get.(0))#_xor input#get.(1))
142 | Mem -> self#beam_of_ident n
143 ((input#get.(0))#mem)
144 | Delay -> self#beam_of_ident n
145 ((input#get.(0))#delay input#get.(1))
146 | Floor -> self#beam_of_ident n
147 ((input#get.(0))#floor)
148 | Ceil -> self#beam_of_ident n
149 ((input#get.(0))#ceil)
150 | Rint -> self#beam_of_ident n
151 ((input#get.(0))#rint)
152 | Int -> self#beam_of_ident n
153 ((input#get.(0))#int)
154 | Float -> self#beam_of_ident n
155 ((input#get.(0))#float)
156 | Sin -> self#beam_of_ident n
157 ((input#get.(0))#sin)
158 | Asin -> self#beam_of_ident n
159 ((input#get.(0))#asin)
160 | Cos -> self#beam_of_ident n
161 ((input#get.(0))#cos)
162 | Acos -> self#beam_of_ident n
163 ((input#get.(0))#acos)
164 | Tan -> self#beam_of_ident n
165 ((input#get.(0))#tan)
166 | Atan -> self#beam_of_ident n
167 ((input#get.(0))#atan)
168 | Atan2 -> self#beam_of_ident n
169 ((input#get.(0))#atan2 input#get.(1))
170 | Exp -> self#beam_of_ident n
171 ((input#get.(0))#exp)
172 | Sqrt -> self#beam_of_ident n
173 ((input#get.(0))#sqrt)
174 | Ln -> self#beam_of_ident n
175 ((input#get.(0))#ln)
176 | Lg -> self#beam_of_ident n
177 ((input#get.(0))#lg)
178 | Abs -> self#beam_of_ident n
179 ((input#get.(0))#abs)
180 | Mod -> self#beam_of_ident n
181 ((input#get.(0))#_mod input#get.(1))
182 | Fmod -> self#beam_of_ident n
183 ((input#get.(0))#fmod input#get.(1))
184 | Remainder -> self#beam_of_ident n
185 ((input#get.(0))#remainder input#get.(1))
186 | Vectorize -> self#beam_of_ident n
187 ((input#get.(0))#vectorize input#get.(1))
188 | Vconcat -> self#beam_of_ident n
189 ((input#get.(0))#vconcat input#get.(1))
190 | Vpick -> self#beam_of_ident n
191 ((input#get.(0))#vpick input#get.(1))
192 | Serialize -> self#beam_of_ident n
193 (input#get.(0))#serialize
194 | Gt -> self#beam_of_ident n
195 ((input#get.(0))#gt input#get.(1))
196 | Lt -> self#beam_of_ident n
197 ((input#get.(0))#lt input#get.(1))
198 | Geq -> self#beam_of_ident n
199 ((input#get.(0))#geq input#get.(1))
200 | Leq -> self#beam_of_ident n
201 ((input#get.(0))#leq input#get.(1))
202 | Eq -> self#beam_of_ident n
203 ((input#get.(0))#eq input#get.(1))
204 | Neq -> self#beam_of_ident n
205 ((input#get.(0))#neq input#get.(1))
206 | Max -> self#beam_of_ident n
207 ((input#get.(0))#max input#get.(1))
208 | Min -> self#beam_of_ident n
209 ((input#get.(0))#min input#get.(1))
210 | Shl -> self#beam_of_ident n
211 ((input#get.(0))#shl input#get.(1))
212 | Shr -> self#beam_of_ident n
213 ((input#get.(0))#shr input#get.(1))
214 | Prefix -> self#beam_of_ident n
215 ((input#get.(1))#prefix input#get.(0))
216 | Select2 -> self#beam_of_ident n
217 ((input#get.(0))#select2 input#get.(1) input#get.(2))
218 | Select3 -> self#beam_of_ident n
219 ((input#get.(0))#select3 input#get.(1)
220 input#get.(2) input#get.(3))
221 | Rdtable -> self#beam_of_ident n
222 ((input#get.(1))#rdtable input#get.(0) input#get.(2))
223 | Rwtable -> self#beam_of_ident n
224 ((input#get.(0))#rwtable input#get.(1)
225 input#get.(2) input#get.(3) input#get.(4))
226 end;;
227
228 class virtual process_binary =
229 fun (exp_init : faust_exp) ->
230 let (exp_left, exp_right) =
231 match exp_init with
232 | Par (e1, e2) -> (e1, e2)
233 | Seq (e1, e2) -> (e1, e2)
234 | Split (e1, e2) -> (e1, e2)
235 | Merge (e1, e2) -> (e1, e2)
236 | Rec (e1, e2) -> (e1, e2)
237 | _ -> raise (Process_error "binary process constructor.") in
238 let proc_left = (new proc_factory)#make exp_left in
239 let proc_right = (new proc_factory)#make exp_right in
240
241 object
242 inherit process exp_init
243 method private proc_left = proc_left
244 method private proc_right = proc_right
245
246 val _dim =
247 match exp_init with
248 | Par (e1, e2) -> (proc_left#dim)#par proc_right#dim
249 | Seq (e1, e2) -> (proc_left#dim)#seq proc_right#dim
250 | Split (e1, e2) -> (proc_left#dim)#split proc_right#dim
251 | Merge (e1, e2) -> (proc_left#dim)#merge proc_right#dim
252 | Rec (e1, e2) -> (proc_left#dim)#_rec proc_right#dim
253 | _ -> raise (Process_error "binary process constructor.")
254
255 val _delay =
256 match exp_init with
257 | Par (e1, e2) -> max proc_left#delay proc_right#delay
258 | Seq (e1, e2) -> proc_left#delay + proc_right#delay
259 | Split (e1, e2) -> proc_left#delay + proc_right#delay
260 | Merge (e1, e2) -> proc_left#delay + proc_right#delay
261 | Rec (e1, e2) -> 1 + proc_left#delay + proc_right#delay
262 | _ -> raise (Process_error "binary process constructor.")
263 end
264
265 and proc_par : faust_exp -> process_type =
266 fun (exp_init : faust_exp) ->
267 object (self)
268 inherit process_binary exp_init
269 method eval : beam_type -> beam_type =
270 fun (input : beam_type) ->
271 let (sub_input1, sub_input2) = input#cut self#proc_left#dim#input in
272 let sub_output1 = self#proc_left#eval sub_input1 in
273 let sub_output2 = self#proc_right#eval sub_input2 in
274 sub_output1#append sub_output2
275 end
276
277 and proc_split : faust_exp -> process_type =
278 fun (exp_init : faust_exp) ->
279 object (self)
280 inherit process_binary exp_init
281 method eval : beam_type -> beam_type =
282 fun (input : beam_type) ->
283 let mid_output = self#proc_left#eval input in
284 let mid_input = mid_output#matching self#proc_right#dim#input in
285 self#proc_right#eval mid_input
286 end
287
288 and proc_merge : faust_exp -> process_type =
289 fun (exp_init : faust_exp) ->
290 object (self)
291 inherit process_binary exp_init
292 method eval : beam_type -> beam_type =
293 fun (input : beam_type) ->
294 let mid_output = self#proc_left#eval input in
295 let mid_input = mid_output#matching self#proc_right#dim#input in
296 self#proc_right#eval mid_input
297 end
298
299 and proc_seq : faust_exp -> process_type =
300 fun (exp_init : faust_exp) ->
301 object (self)
302 inherit process_binary exp_init
303 method eval : beam_type -> beam_type =
304 fun (input : beam_type) ->
305 let mid_output = self#proc_left#eval input in
306 self#proc_right#eval mid_output
307 end
308
309 and proc_rec : faust_exp -> process_type =
310 fun (exp_init : faust_exp) ->
311 object (self)
312 inherit process_binary exp_init
313 method eval : beam_type -> beam_type =
314 fun (input : beam_type) ->
315 let memory = Hashtbl.create self#delay in
316 let rates = ref (Array.make self#dim#output (new rate 0 1)) in
317
318 let split : (time -> value_type array) -> (time -> value_type) array =
319 fun beam_at ->
320 let get_signal =
321 fun beam_func -> fun i -> fun t ->
322 (beam_func t).(i) in
323 Array.init self#dim#output (get_signal beam_at) in
324
325 let feedback : (time -> value_type array) -> beam =
326 fun beam_at ->
327 let signals_at = split beam_at in
328 let delay_by_one = fun s -> fun t -> s (t - 1) in
329 let delay_signal_funcs = Array.map delay_by_one
330 (Array.sub signals_at 0 self#proc_right#dim#input) in
331 new beam (array_map2 (new signal)
332 (Array.sub !rates 0 self#proc_right#dim#input)
333 delay_signal_funcs) in
334
335 let rec beam_at : time -> value_type array =
336 fun (t : time) ->
337 if t < 0 then
338 Array.make self#dim#output (new value Zero)
339 else if Hashtbl.mem memory t then
340 Hashtbl.find memory t
341 else
342 let beam_fb_in = feedback beam_at in
343 let beam_fb_out = self#proc_right#eval beam_fb_in in
344 let beam_in = beam_fb_out#append input in
345 let beam_out = self#proc_left#eval beam_in in
346 let values = beam_out#at t in
347 let () = (rates := beam_out#frequency) in
348 let () = Hashtbl.add memory t values in
349 let () = if t - self#delay >= 0 then
350 Hashtbl.remove memory (t - self#delay) else () in
351 values in
352 new beam (array_map2 (new signal) !rates (split beam_at))
353 end
354
355 and proc_factory =
356 object
357 method make : faust_exp -> process_type =
358 fun (exp : faust_exp) ->
359 match exp with
360 | Const b -> new proc_const exp
361 | Ident s -> new proc_ident exp
362 | Par (e1, e2) -> new proc_par exp
363 | Seq (e1, e2) -> new proc_seq exp
364 | Split (e1, e2) -> new proc_split exp
365 | Merge (e1, e2) -> new proc_merge exp
366 | Rec (e1, e2) -> new proc_rec exp
367 end;;