0468e42bc69de39f2ae0c440695c75508d1b1a4a
[Faustine.git] / interpretor / beam.ml
1 (**
2 Module: Beam
3 Description: beam definition and operations
4 @author WANG Haisheng
5 Created: 21/07/2013 Modified: 21/07/2013
6 *)
7
8 exception Beam_matching of string;;
9
10 open Types;;
11 open Signal;;
12
13 class beam : signal_type array -> beam_type =
14 fun (signals_init : signal_type array) ->
15 object (self)
16 val signals = signals_init
17
18 method get = signals
19 method width = Array.length signals
20
21 method sub : int -> int -> beam_type =
22 fun start ->
23 fun len ->
24 new beam (Array.sub self#get start len)
25
26 method append : beam_type -> beam_type =
27 fun (b : beam_type) ->
28 new beam (Array.append self#get b#get)
29
30 method private add_memory : int -> unit =
31 fun (window : int) ->
32 let signal_add_memory : int -> signal_type -> unit =
33 fun (l : int) ->
34 fun (s : signal) ->
35 s#add_memory l in
36 let _ = Array.map (signal_add_memory window) self#get in ()
37
38 method matching : int -> beam_type =
39 fun size ->
40 if size = self#width then new beam self#get
41
42 else if (size > self#width) && (size mod self#width = 0) then
43 let () = self#add_memory 1 in
44 let split_signals =
45 Array.concat (Array.to_list
46 (Array.make (size/self#width) self#get)) in
47 new beam split_signals
48
49 else if (size < self#width) && (self#width mod size = 0) then
50 let rec merge_rec =
51 fun (sa : signal_type array) ->
52 fun (l : int) ->
53 fun (i : int) ->
54 if i + l < Array.length sa then
55 (sa.(i))#add (merge_rec sa l (i + l))
56 else sa.(i) in
57 let merge_signals = Array.init size (merge_rec self#get size) in
58 new beam merge_signals
59
60 else raise (Beam_matching "matching size error")
61
62 method time : time -> value_type array =
63 fun t ->
64 let signal_at = fun (t : time) -> fun (s : signal_type) -> s#at t in
65 Array.map (signal_at t) self#get
66
67 method output : int -> (int array) * (float array array) =
68 fun (length_max : int) ->
69
70 end