File: test03.ml

package info (click to toggle)
ocaml-visitors 20200210-3
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 1,896 kB
  • sloc: ml: 4,077; makefile: 44; sh: 18
file content (63 lines) | stat: -rw-r--r-- 1,545 bytes parent folder | download | duplicates (5)
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
type ('var, 'binder) term =
  | TVar of 'var
  | TAbs of 'binder * ('var, 'binder) term
  | TApp of ('var, 'binder) term * ('var, 'binder) term
[@@deriving
     visitors { variety = "iter" },
     visitors { variety = "map" },
     visitors { variety = "reduce" },
     visitors { variety = "endo" },
     visitors { variety = "iter2" },
     visitors { variety = "map2" },
     visitors { variety = "reduce2" }
]

(* Nominal. *)

module StringSet = Set.Make(String)

let iter = object(self)
  inherit [_] iter
  (* Descending methods for local types. *)
  method! visit_TAbs env x t =
    let env = StringSet.add x env in
    self#visit_term env t
  (* Descending methods for nonlocal types. *)
  method visit_'binder _env _x = ()
  method visit_'var env x =
    if StringSet.mem x env then
      Printf.printf "%s is a bound variable.\n%!" x
    else
      Printf.printf "%s is a free variable.\n%!" x

end

let t : (_, _) term =
  TAbs ("x", TApp (TVar "x", TVar "y"))

let () =
  iter#visit_term StringSet.empty t

(* De Bruijn. *)

let iter = object(self)
  inherit [_] iter
  (* Descending methods for local types. *)
  method! visit_TAbs env _x t =
    let env = 1 + env in
    self#visit_term env t
  (* Descending methods for nonlocal types. *)
  method visit_'binder _env _x = ()
  method visit_'var env x =
    if x < env then
      Printf.printf "%d is a bound variable.\n%!" x
    else
      Printf.printf "%d is a free variable.\n%!" x

end

let t : (_, _) term =
  TAbs ((), TApp (TVar 0, TVar 1))

let () =
  iter#visit_term 0 t