ce/eval.ml

38 lines
838 B
OCaml
Raw Normal View History

2022-01-10 01:31:47 +09:00
open Ast
2022-01-18 16:52:33 +09:00
open Ast.Value
2022-01-10 01:31:47 +09:00
2022-01-18 16:52:33 +09:00
exception No_operation
2022-01-18 16:52:33 +09:00
let rec binop op l r =
let tl = typeof l and tr = typeof r in
let ty = Type.merge tl tr in
let rec promote_until t x =
if typeof x = t
then x
else promote_until t (promote x)
in
let l = promote_until ty l
and r = promote_until ty r in
match Binop.get op ty with
| None -> begin
try binop op (promote l) (promote r)
with _ -> raise No_operation
2022-01-18 15:33:56 +09:00
end
2022-01-18 16:52:33 +09:00
| Some f -> f l r
2022-01-10 23:11:13 +09:00
let rec eval = function
| Value v -> v
2022-01-10 01:31:47 +09:00
| Binop (l, op, r) ->
2022-01-18 16:52:33 +09:00
let l = eval l and r = eval r in
binop op l r
2022-01-10 01:31:47 +09:00
| Set_binop_pre (op, l) ->
2022-01-18 16:52:33 +09:00
let l =
match eval l with
| Int n -> n
| v -> raise @@ Invalid_type (typeof v)
in
2022-01-11 01:05:29 +09:00
Hashtbl.replace Parser.precedence op l;
2022-01-18 16:52:33 +09:00
Nop
2022-01-11 01:05:29 +09:00
| Get_binop_pre op ->
Int (Hashtbl.find Parser.precedence op)