INF565/newsyntax.ml at develop · anthonyhtyao/INF565 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
open Format
module Smap = Map.Make(String)

type inst = string*int

type exp =
  | Let of inst*exp*exp  (* lex x = e in e *)
  | Letrec of inst*inst*exp*exp (* let rec f x = e in e *)
  | Fun of inst*exp (* fun x -> e *)
  | Appl of exp*exp (* e e *)
  | Oper of Syntax.oper*exp*exp (* e + e*)
  | Cond of exp*exp*exp (* if A then B else C *)
  | Inst of inst
  | Cste of Syntax.cste

let rec of_exp = function
  | Syntax.Let(i,e1,e2) -> Let((i,0),of_exp e1,of_exp e2)
  | Syntax.Letrec(i1,i2,e1,e2) -> Letrec((i1,0),(i2,0),of_exp e1,of_exp e2)
  | Syntax.Fun(i,e) -> Fun((i,0),of_exp e)
  | Syntax.Appl(e1,e2) -> Appl(of_exp e1, of_exp e2)
  | Syntax.Oper(Syntax.And,e1,e2) -> Cond(of_exp e1,of_exp e2, Cste(Syntax.False))
  | Syntax.Oper(Syntax.Or,e1,e2) -> Cond(of_exp e1,Cste(Syntax.True),of_exp e2)
  | Syntax.Oper(o,e1,e2) -> Oper(o,of_exp e1,of_exp e2)
  | Syntax.Cond(e1,e2,e3) -> Cond(of_exp e1,of_exp e2,of_exp e3)
  | Syntax.Inst i  -> Inst((i,0))
  | Syntax.Cste c  -> Cste(c)

let printCste c = Syntax.printCste c

let printOper o = Syntax.printOper o

let string_of_inst i = "("^(fst i)^","^(string_of_int (snd i))^")"

let rec printExp s = match s with
  | Cste c -> printCste c
  | Inst i -> print_string (string_of_inst i)
  | Oper (o,e1,e2) -> printExp e1;printOper o;printExp e2;print_newline ()
  | Cond (e1,e2,e3) -> print_string "if ";
                       printExp e1;
                       print_string " then ";
                       printExp e2;
                       print_string " else ";
                       printExp e3;
                       print_newline ()
  | Appl (e1,e2) -> print_string "App ";
                    printExp e1;
                    printExp e2;
                    print_newline ()
  | Fun (i,e) -> print_string ("fun "^(string_of_inst i)^" -> ");
                 printExp e;
                 print_newline ()
  | Letrec (i1,i2,e1,e2) -> print_string ("let rec "^(string_of_inst i1)^" "^(string_of_inst i2)^" = ");
                            printExp e1;
                            printf " in@.";
                            printExp e2;
                            print_newline()
  | Let (i,e1,e2)-> print_string ("let "^(string_of_inst i)^" = ");
                    printExp e1;
                    printf " in@.";
                    printExp e2;
                    print_newline ()

let cal_index exp =
  let incre m = Smap.map (fun x->x+1) m in
  let rec aux m = function
    | Cste c -> Cste c
    | Cond (e1,e2,e3) -> Cond(aux m e1,aux m e2, aux m e3)
    | Appl (e1,e2) -> Appl(aux m e1,aux m e2)
    | Oper (o,e1,e2) -> Oper(o,aux m e1,aux m e2)
    | Inst (i,_) -> let n = try Smap.find i m with _ -> -1 in Inst(i,n)
    | Fun (i,e) ->  Fun(i,aux (Smap.add (fst i) 0 (incre m)) e)
    | Let (i,e1,e2) -> let mi = incre m in
                       Let (i,aux m e1,aux (Smap.add (fst i) 0 mi) e2)
    | Letrec (i1,i2,e1,e2) -> let mi = Smap.add (fst i1) 0 (incre m) in
                    Letrec (i1,i2,aux (Smap.add (fst i2) 0 (incre mi)) e1,aux mi e2)

  in aux Smap.empty exp