mlox/lib/value.ml

46 lines
1.5 KiB
OCaml

type lox_function = {
name : string;
arity : int;
arg_names : string list;
body : Stmt.stmt_node; [@printer fun fmt _ -> fprintf fmt "<body>"]
}
[@@deriving show { with_path = false }]
type native_function = {
name : string;
arity : int;
fn : lox_value list -> (lox_value, string) result;
}
[@@deriving show { with_path = false }]
and function_ = NativeFunction of native_function | LoxFunction of lox_function
[@@deriving show { with_path = false }]
and lox_value = Function of function_ | String of string | Number of float | Bool of bool | Nil
[@@deriving show { with_path = false }]
let string_of_lox_value lox_value =
match lox_value with
| Function (NativeFunction { name; arity; _ }) -> Printf.sprintf "<native fn %s/%d>" name arity
| Function (LoxFunction { name; arity; _ }) -> Printf.sprintf "<fn %s/%d>" name arity
| String s -> s
| Number x -> if Float.is_integer x then string_of_int (Int.of_float x) else string_of_float x
| Bool b -> string_of_bool b
| Nil -> "nil"
let type_string_of_lox_value lox_value =
match lox_value with
| Function (NativeFunction _) -> "NativeFunction"
| Function (LoxFunction _) -> "Function"
| String _ -> "String"
| Number _ -> "Number"
| Bool _ -> "Bool"
| Nil -> "Nil"
let lox_value_to_bool lox_value = match lox_value with Bool b -> b | Nil -> false | _ -> true
let make_lox_function (name : string) (arg_names : string list) (body : Stmt.stmt_node) : lox_value
=
let arity = List.length arg_names in
let fn = LoxFunction { name; arity; arg_names; body } in
Function fn