(** 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 * Pure.fun_decl list type pure_fun_translation = { keep_fwd : bool; (** Should we extract the forward function? If the forward function returns `()` and there is exactly one backward function, we may merge the forward into the backward function and thus don't extract the forward function)? *) fwd : fun_and_loops; backs : fun_and_loops list; } 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 let is_trait_impl = false in name_to_simple_name mctx is_trait_impl 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 let is_trait_impl = true in name_with_generics_to_simple_name mctx is_trait_impl ~prefix n p g