summaryrefslogtreecommitdiff
path: root/compiler/TranslateCore.ml
blob: 3e4c7375761c1cc00900a0721f9d03274736e257 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
(** Some utilities for the translation *)

open Contexts
open ExtractName

(** The local logger *)
let log = Logging.translate_log

type trans_ctx = decls_ctx [@@deriving show]
type fun_and_loops = { f : Pure.fun_decl; loops : Pure.fun_decl list }
type pure_fun_translation_no_loops = Pure.fun_decl
type pure_fun_translation = fun_and_loops

let trans_ctx_to_fmt_env (ctx : trans_ctx) : Print.fmt_env =
  Print.Contexts.decls_ctx_to_fmt_env ctx

let trans_ctx_to_pure_fmt_env (ctx : trans_ctx) : PrintPure.fmt_env =
  PrintPure.decls_ctx_to_fmt_env ctx

let name_to_string (ctx : trans_ctx) =
  Print.Types.name_to_string (trans_ctx_to_fmt_env ctx)

let match_name_find_opt (ctx : trans_ctx) (name : Types.name)
    (m : 'a NameMatcherMap.t) : 'a option =
  let open Charon.NameMatcher in
  let open ExtractBuiltin in
  let mctx : ctx =
    {
      type_decls = ctx.type_ctx.type_decls;
      global_decls = ctx.global_ctx.global_decls;
      fun_decls = ctx.fun_ctx.fun_decls;
      trait_decls = ctx.trait_decls_ctx.trait_decls;
      trait_impls = ctx.trait_impls_ctx.trait_impls;
    }
  in
  NameMatcherMap.find_opt mctx name m

let match_name_with_generics_find_opt (ctx : trans_ctx) (name : Types.name)
    (generics : Types.generic_args) (m : 'a NameMatcherMap.t) : 'a option =
  let open Charon.NameMatcher in
  let open ExtractBuiltin in
  let mctx : ctx =
    {
      type_decls = ctx.type_ctx.type_decls;
      global_decls = ctx.global_ctx.global_decls;
      fun_decls = ctx.fun_ctx.fun_decls;
      trait_decls = ctx.trait_decls_ctx.trait_decls;
      trait_impls = ctx.trait_impls_ctx.trait_impls;
    }
  in
  NameMatcherMap.find_with_generics_opt mctx name generics m

let name_to_simple_name (ctx : trans_ctx) (n : Types.name) : string list =
  let mctx : Charon.NameMatcher.ctx =
    {
      type_decls = ctx.type_ctx.type_decls;
      global_decls = ctx.global_ctx.global_decls;
      fun_decls = ctx.fun_ctx.fun_decls;
      trait_decls = ctx.trait_decls_ctx.trait_decls;
      trait_impls = ctx.trait_impls_ctx.trait_impls;
    }
  in
  name_to_simple_name mctx n

let trait_name_with_generics_to_simple_name (ctx : trans_ctx)
    ?(prefix : Types.name option = None) (n : Types.name)
    (p : Types.generic_params) (g : Types.generic_args) : string list =
  let mctx : Charon.NameMatcher.ctx =
    {
      type_decls = ctx.type_ctx.type_decls;
      global_decls = ctx.global_ctx.global_decls;
      fun_decls = ctx.fun_ctx.fun_decls;
      trait_decls = ctx.trait_decls_ctx.trait_decls;
      trait_impls = ctx.trait_impls_ctx.trait_impls;
    }
  in
  name_with_generics_to_simple_name mctx ~prefix n p g

let name_to_pattern_string (ctx : trans_ctx) (n : Types.name) : string =
  let mctx : Charon.NameMatcher.ctx =
    {
      type_decls = ctx.type_ctx.type_decls;
      global_decls = ctx.global_ctx.global_decls;
      fun_decls = ctx.fun_ctx.fun_decls;
      trait_decls = ctx.trait_decls_ctx.trait_decls;
      trait_impls = ctx.trait_impls_ctx.trait_impls;
    }
  in
  let c : Charon.NameMatcher.to_pat_config =
    { tgt = TkPattern; use_trait_decl_refs = match_with_trait_decl_refs }
  in
  let pat = Charon.NameMatcher.name_to_pattern mctx c n in
  Charon.NameMatcher.pattern_to_string { tgt = TkPattern } pat