source: Deliverables/D4.1/BitVectors.ml @ 138

Last change on this file since 138 was 138, checked in by sacerdot, 9 years ago

Several bug fixes and code clean-up.
New main file: test.ml to parse and execute an HEX file.
Usage: ./test.native foo.hex

File size: 3.4 KB
Line 
1type bit = bool
2type 'a vect = bit list
3type nibble = [`Four] vect
4type byte7 = [`Seven] vect
5type byte = [`Eight] vect
6type word = [`Sixteen] vect
7type word11 = [`Eleven] vect
8
9type sizes = [ `Four | `Seven | `Eight | `Eleven | `Sixteen ]
10
11let mk_nibble b1 b2 b3 b4 = [b1; b2; b3; b4]
12let from_nibble =
13 function
14    [b1; b2; b3; b4] -> b1,b2,b3,b4
15  | _ -> assert false
16let mk_byte n1 n2 = n1 @ n2
17let mk_byte_from_bits ((b1,b2,b3,b4),(b5,b6,b7,b8)) = ([b1;b2;b3;b4;b5;b6;b7;b8] : [`Eight] vect)
18let from_byte =
19 function
20    b1::b2::b3::b4::r -> [b1;b2;b3;b4],r
21  | _ -> assert false
22let bits_of_byte =
23 function
24    [b1;b2;b3;b4;b5;b6;b7;b8] -> (b1,b2,b3,b4),(b5,b6,b7,b8)
25  | _ -> assert false
26let mk_byte7 b1 b2 b3 n1 = b1::b2::b3::n1
27let from_byte7 =
28 function
29    b1::b2::b3::r -> b1,b2,b3,r
30  | _ -> assert false
31let mk_word = mk_byte
32let from_word =
33 function
34    b1::b2::b3::b4::b5::b6::b7::b8::r -> [b1;b2;b3;b4;b5;b6;b7;b8],r
35  | _ -> assert false
36let mk_word11 = mk_byte7
37let from_word11 = from_byte7
38
39let get_bit index l =
40  try
41    Some (List.nth index l)
42  with (Failure _ | Invalid_argument _) -> None
43
44let set_bit l index new_val =
45  try
46    let rec aux index l = 
47      match index, l with
48        _, [] -> raise (Invalid_argument "")
49      | 0,_::tl -> new_val::tl
50      | n,hd::tl -> hd::(aux (n-1) tl) in
51     Some (List.rev (aux index (List.rev l)))
52  with Invalid_argument "" -> None
53
54let (-&-) l1 l2 = List.map2 (fun b1 b2 -> b1 & b2) l1 l2
55let (-|-) l1 l2 = List.map2 (fun b1 b2 -> b1 || b2) l1 l2
56let xor b1 b2 = b1 <> b2
57let (-^-) l1 l2 = List.map2 xor l1 l2
58let complement l1 = List.map (not) l1
59
60let iter_bits f v = String.concat "" (List.map f v)
61let map_bits = List.map
62let map2_bits = List.map2
63
64let string_of_bit = function false -> "0" | true -> "1"
65let string_of_vect l = String.concat "" (List.map string_of_bit l)
66
67let full_add l r c = List.fold_right2 (fun b1 b2 (c,r) -> b1 & b2 || c & (b1 || b2),xor (xor b1 b2) c::r) l r (c,[])
68let half_add l r = full_add l r false
69
70let rec split_last =
71  function
72    [] -> assert false
73  | [he] -> he,[]
74  | he::tl ->
75      let l,res = split_last tl in
76        l,he::res
77
78let shift_left =
79  function
80    [] -> assert false
81  | _::tl -> tl @ [false]
82let shift_right l = false :: snd (split_last l)
83let rotate_left =
84  function
85    [] -> assert false
86  | he::tl -> tl @ [he]
87let rotate_right l =
88  let he,tl = split_last l in
89    he::tl
90
91let int_of_vect v =
92  let rec aux pow v =
93    match v with
94      [] -> 0
95    | hd::tl ->
96        if hd = true then
97          pow + (aux (pow * 2) tl)
98        else
99          aux (pow * 2) tl
100  in
101    aux 1 (List.rev v)
102
103let size_lookup =
104  function
105    `Four -> 4
106  | `Seven -> 7
107  | `Eight -> 8
108  | `Eleven -> 11
109  | `Sixteen -> 16
110
111let rec pow v p =
112  if p = 0 then
113    1
114  else
115    v * (pow v (p - 1))
116
117let divide_with_remainder x y = (x / y, x mod y)
118
119let rec aux i =
120  if i < 0 then
121    raise (Invalid_argument "Negative index")
122  else
123    let (d, r) = divide_with_remainder i 2 in
124      if (d, r) = (0, 0) then
125        []
126      else if r = 0 then
127        false :: aux d
128      else
129        true :: aux d
130
131let rec pad i l = if i = 0 then l else false :: (pad (i - 1) l)
132
133let vect_of_int i size =
134  let big_list = List.rev (aux i) in
135    if List.length big_list > size_lookup size then
136      raise (Invalid_argument "Size not big enough")
137    else
138      let diff = size_lookup size - List.length big_list in
139        pad diff big_list
140   
141let zero size = pad (size_lookup size) []
Note: See TracBrowser for help on using the repository browser.