mirror of
https://github.com/MorizzG/MLox.git
synced 2025-12-06 04:22:41 +00:00
78 lines
2.8 KiB
OCaml
78 lines
2.8 KiB
OCaml
type literal = String of string | Number of float | Bool of bool | Nil
|
|
[@@deriving show { with_path = false }]
|
|
|
|
let show_literal literal =
|
|
match literal with
|
|
| String s -> "\"" ^ s ^ "\""
|
|
| Number x -> string_of_float x
|
|
| Bool b -> string_of_bool b
|
|
| Nil -> "nil"
|
|
|
|
type binary_op =
|
|
| Plus
|
|
| Minus
|
|
| Mul
|
|
| Div
|
|
| Equal
|
|
| Less
|
|
| Greater
|
|
| LessEqual
|
|
| GreaterEqual
|
|
| And
|
|
| Or
|
|
[@@deriving show { with_path = false }]
|
|
|
|
type unary_op = Neg | Not [@@deriving show { with_path = false }]
|
|
type logical_op = And | Or [@@deriving show { with_path = false }]
|
|
|
|
type expr =
|
|
| Literal of literal
|
|
| Variable of string
|
|
| Assignment of { name : string; expr : expr_node }
|
|
| Unary of { op : unary_op; expr : expr_node }
|
|
| Binary of { op : binary_op; left : expr_node; right : expr_node }
|
|
| Logical of { op : logical_op; left : expr_node; right : expr_node }
|
|
|
|
and expr_node = { expr : expr; pos : Error.code_pos }
|
|
|
|
let rec show_expr ?(indent = 0) expr =
|
|
let show_indented = show_expr ~indent:(indent + 2) in
|
|
let indent_s = String.make indent ' ' in
|
|
match expr with
|
|
| Literal literal -> indent_s ^ show_literal literal
|
|
| Variable name -> indent_s ^ "Variable " ^ name
|
|
| Assignment { name; expr } -> indent_s ^ name ^ " = \n" ^ show_indented expr.expr
|
|
| Unary { op; expr } -> indent_s ^ show_unary_op op ^ "\n" ^ show_indented expr.expr
|
|
| Binary { op; left; right } ->
|
|
indent_s ^ show_binary_op op ^ "\n" ^ show_indented left.expr ^ "\n"
|
|
^ show_indented right.expr
|
|
| Logical { op; left; right } ->
|
|
indent_s ^ show_logical_op op ^ "\n" ^ show_indented left.expr ^ "\n"
|
|
^ show_indented right.expr
|
|
|
|
let show_expr_node expr_node = show_expr expr_node.expr
|
|
let make_expr_node (pos : Error.code_pos) (expr : expr) : expr_node = { expr; pos }
|
|
|
|
let make_string (pos : Error.code_pos) (s : string) : expr_node =
|
|
Literal (String s) |> make_expr_node pos
|
|
|
|
let make_number (pos : Error.code_pos) (x : float) : expr_node =
|
|
Literal (Number x) |> make_expr_node pos
|
|
|
|
let make_bool (pos : Error.code_pos) (b : bool) : expr_node = Literal (Bool b) |> make_expr_node pos
|
|
let make_nil (pos : Error.code_pos) = Literal Nil |> make_expr_node pos
|
|
|
|
let make_variable (pos : Error.code_pos) (name : string) : expr_node =
|
|
Variable name |> make_expr_node pos
|
|
|
|
let make_assignment (pos : Error.code_pos) (name : string) (expr : expr_node) : expr_node =
|
|
Assignment { name; expr } |> make_expr_node pos
|
|
|
|
let make_unary (pos : Error.code_pos) (op : unary_op) (expr : expr_node) =
|
|
Unary { op; expr } |> make_expr_node pos
|
|
|
|
let make_binary (pos : Error.code_pos) (op : binary_op) (left : expr_node) (right : expr_node) =
|
|
Binary { op; left; right } |> make_expr_node pos
|
|
|
|
let make_logical (pos : Error.code_pos) (op : logical_op) (left : expr_node) (right : expr_node) =
|
|
Logical { op; left; right } |> make_expr_node pos
|