ocaml/parser.ml

   1 type term =\r
   2   | Var of int\r
   3   | App of term * term\r
   4   | Lam of term\r
   5 ;;\r
   6 \r
   7 let mk_app x y = App(x, y);;\r
   8 let mk_lam x = Lam x;;\r
   9 let mk_var x = Var x;;\r
  10 \r
  11 exception ParsingError of string;;\r
  12 \r
  13 let isAlphaNum c = let n = Char.code c in\r
  14  (48 <= n && n <= 90) || (95 <= n && n <= 122) ;;\r
  15 let isSpace c = c = ' ' || c = '\n' || c = '\t' ;;\r
  16 \r
  17 let rec index_of x =\r
  18   function\r
  19   | [] -> raise (Failure "index_of: Not Found")\r
  20   | h::t -> if x = h then 0 else 1 + index_of x t\r
  21 ;;\r
  22 (* FIXME *)\r
  23 let mk_var' (bound, free) x =\r
  24   if List.mem x bound\r
  25   then free, mk_var (index_of x bound)\r
  26   else if List.mem x free\r
  27        then free, mk_var (List.length bound + index_of x free)\r
  28        else (free @ [x]), mk_var (List.length bound + List.length free)\r
  29 ;;\r
  30 \r
  31 let mk_app' = function\r
  32   | [] ->  raise (ParsingError "bug")\r
  33   | t :: ts -> List.fold_left mk_app t ts\r
  34 ;;\r
  35 \r
  36 let explode s =\r
  37   let rec aux i l =\r
  38     if i < 0 then l else aux (i - 1) (s.[i] :: l)\r
  39   in aux (String.length s - 1) []\r
  40 ;;\r
  41 \r
  42 let implode l =\r
  43   let res = String.create (List.length l) in\r
  44   let rec aux i = function\r
  45     | [] -> res\r
  46     | c :: l -> res.[i] <- c; aux (i + 1) l in\r
  47   aux 0 l\r
  48 ;;\r
  49 \r
  50 let rec strip_spaces = function\r
  51   | c::cs when isSpace c -> strip_spaces cs\r
  52   | cs -> cs\r
  53 ;;\r
  54 \r
  55 let read_var s =\r
  56   let rec aux = function\r
  57   | [] -> None, []\r
  58   | c::cs as x -> if isAlphaNum c\r
  59     then match aux cs with\r
  60          | (Some x), cs' -> Some (c :: x), cs'\r
  61          | None, cs' -> (Some [c]), cs'\r
  62     else None, x\r
  63   in match aux s with\r
  64     | None, y -> None, y\r
  65     | Some x, y -> Some (implode x), y\r
  66 ;;\r
  67 \r
  68 let read_var' (bound, free as vars) s =\r
  69   match read_var s with\r
  70   | Some varname, cs ->\r
  71     let free, v = mk_var' vars varname in\r
  72     Some [v], cs, (bound, free)\r
  73   | _, _ -> raise (ParsingError ("Can't read variable"))\r
  74 ;;\r
  75 \r
  76 let rec read_smt vars =\r
  77   let check_if_lambda cs = match read_var cs with\r
  78     | None, _ -> false\r
  79     | Some x, cs -> match strip_spaces cs with\r
  80       | [] -> false\r
  81       | c::_ -> c = '.'\r
  82   in let read_lambda (bound, free) cs = (\r
  83     match read_var (strip_spaces cs) with\r
  84       | Some varname, cs ->\r
  85       let vars' = (varname::bound, free) in\r
  86       (match strip_spaces cs with\r
  87         | [] -> raise (ParsingError "manca dopo variabile lambda")\r
  88         | c::cs -> (if c = '.' then (match read_smt vars' cs with\r
  89           | None, _, _ -> raise (ParsingError "manca corpo lambda")\r
  90           | Some [x], y, (_, free) -> Some [mk_lam x], y, (bound, free)\r
  91           | Some _, _, _ -> raise (ParsingError "???")\r
  92           ) else raise (ParsingError "manca . nel lambda")\r
  93         ))\r
  94       | _, _ -> assert false\r
  95  ) in let rec aux vars cs =\r
  96   match strip_spaces cs with\r
  97   | [] -> None, [], vars\r
  98   | c::_ as x ->\r
  99       let tms, cs, vars = (\r
 100            if c = '(' then read_pars vars x\r
 101       else if c = ')' then (None, x, vars)\r
 102       else if check_if_lambda x then read_lambda vars x\r
 103       else read_var' vars x) in\r
 104       match tms with\r
 105       | Some [tm] -> (\r
 106         match aux vars cs with\r
 107           | None, cs, vars -> Some [tm], cs, vars\r
 108           | Some ts, cs, vars -> Some (tm :: ts), cs, vars\r
 109         )\r
 110       | Some _ -> raise (ParsingError "bug")\r
 111       | None -> None, x, vars\r
 112   in fun cs -> match aux vars cs with\r
 113     | None, cs, vars -> None, cs, vars\r
 114     | Some ts, cs, vars -> Some [mk_app' ts], cs, vars\r
 115 and read_pars vars = function\r
 116   | [] -> None, [], vars\r
 117   | c::cs -> if c = '(' then (\r
 118     let tm, cs, vars = read_smt vars cs in\r
 119     let cs = strip_spaces cs in\r
 120     match cs with\r
 121       | [] -> None, [], vars\r
 122       | c::cs -> if c = ')' then tm, cs, vars else raise (ParsingError ") mancante")\r
 123     ) else raise (ParsingError "???")\r
 124 ;;\r
 125 \r
 126 let parse x =\r
 127   match read_smt ([],[]) (explode x) with\r
 128   | Some [y], [], _ -> y\r
 129   | _, _, _ -> raise (ParsingError "???")\r
 130 ;;\r
 131 \r
 132 let var_zero = "ω";; (* w is an invalid variable name *)\r
 133 \r
 134 let parse_many strs =\r
 135   let f (x, y) z = match read_smt y (explode z) with\r
 136   | Some[tm], [], vars -> (tm :: x, vars)\r
 137   | _, _, _ -> raise (ParsingError "???")\r
 138   in let aux = List.fold_left f ([], ([], [var_zero])) (* index zero is reserved *)\r
 139   in let (tms, (_, free)) = aux strs\r
 140   in (List.rev tms, free)\r
 141 ;;\r
 142 \r
 143 (**********************************************************************\r
 144 \r
 145 let rec string_of_term = function\r
 146   | Tvar n -> string_of_int n\r
 147   | Tapp(t1, t2) -> "(" ^ string_of_term t1 ^ " " ^ string_of_term t2 ^ ")"\r
 148   | Tlam(t1) -> "(\\" ^ string_of_term t1 ^ ")"\r
 149 ;;\r
 150 \r
 151 let _ = prerr_endline (">>>" ^ string_of_term (parse "(\\x. x y z z1 k) z1 z j"));;\r
 152 \r
 153 \r
 154 **********************************************************************)\r