diff options
author | Son Ho | 2021-11-23 14:02:14 +0100 |
---|---|---|
committer | Son Ho | 2021-11-23 14:02:14 +0100 |
commit | b96bd19fe1ff22c2f1e03332468d4e6abb8a947e (patch) | |
tree | 00c889edf8bc828ef3c984c2fb2945239a5594b9 /src/Print.ml | |
parent | fcc3a467886e5c95b065b1fc4c27301fd62d7c21 (diff) |
Implement formatting for eval_ctx
Diffstat (limited to 'src/Print.ml')
-rw-r--r-- | src/Print.ml | 85 |
1 files changed, 84 insertions, 1 deletions
diff --git a/src/Print.ml b/src/Print.ml index c8296754..ea368bd3 100644 --- a/src/Print.ml +++ b/src/Print.ml @@ -147,7 +147,7 @@ end open Values module Values = struct - open Types + open Types (* local module *) type value_formatter = { r_to_string : RegionVarId.id -> string; @@ -395,3 +395,86 @@ module Values = struct ^ RegionId.set_to_string abs.regions ^ "}" ^ " {\n" ^ avs ^ "\n}" end + +open Contexts +(** Pretty-printing for contexts *) + +module Contexts = struct + open Values (* local module *) + + open Contexts + + let env_value_to_string (fmt : value_formatter) (ev : env_value) : string = + match ev with + | Var (vid, tv) -> + var_id_to_string vid ^ " -> " ^ typed_value_to_string fmt tv + | Abs abs -> abs_to_string fmt abs + + let env_to_string (fmt : value_formatter) (env : env) : string = + "{\n" + ^ String.concat ";\n" + (List.map (fun ev -> " " ^ env_value_to_string fmt ev) env) + ^ "\n}" + + type ctx_formatter = value_formatter + + let eval_ctx_to_ctx_formatter (ctx : eval_ctx) : ctx_formatter = + (* We shouldn't use r_to_string *) + let r_to_string _ = failwith "Unreachable" in + let type_var_id_to_string vid = + let v = lookup_type_var ctx vid in + v.tv_name + in + let type_def_id_to_string def_id = + let def = TypeDefId.nth ctx.type_context def_id in + Types.name_to_string def.name + in + let type_def_id_variant_id_to_string def_id variant_id = + let def = TypeDefId.nth ctx.type_context def_id in + match def.kind with + | Struct _ -> failwith "Unreachable" + | Enum variants -> + let variant = VariantId.nth variants variant_id in + Types.name_to_string def.name ^ "::" ^ variant.variant_name + in + let var_id_to_string vid = + let v = VarId.Map.find vid ctx.vars in + var_to_string v + in + { + r_to_string; + type_var_id_to_string; + type_def_id_to_string; + type_def_id_variant_id_to_string; + var_id_to_string; + } + + let frame_to_string (fmt : ctx_formatter) (ctx : eval_ctx) + (frame : stack_frame) : string = + let var_binding_to_string (vid : VarId.id) : string = + let var = lookup_var ctx vid in + let v = lookup_var_value ctx vid in + let var_name = + match var.name with Some name -> "(" ^ name ^ ")" | None -> "" + in + " @" ^ VarId.to_string var.index ^ var_name ^ " --> " + ^ typed_value_to_string fmt v + in + let vars = + String.concat ",\n" (List.map var_binding_to_string frame.vars) + in + "[\n" ^ vars ^ "\n]" + + let eval_ctx_to_string (ctx : eval_ctx) : string = + let fmt = eval_ctx_to_ctx_formatter ctx in + let num_frames = List.length ctx.frames in + let frames = + List.mapi + (fun i f -> + "\n# Frame " ^ string_of_int i ^ ":\n" ^ frame_to_string fmt ctx f + ^ "\n") + ctx.frames + in + let frames = List.rev frames in + "# " ^ string_of_int num_frames ^ " frame(s)\n" ^ String.concat "" frames +end |