pretty.ml 12 KB
Newer Older
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
(**************************************************************************)
(*                                                                        *)
(*  Copyright (C) 2010-                                                   *)
(*    Francois Bobot                                                      *)
(*    Jean-Christophe Filliatre                                           *)
(*    Johannes Kanig                                                      *)
(*    Andrei Paskevich                                                    *)
(*                                                                        *)
(*  This software is free software; you can redistribute it and/or        *)
(*  modify it under the terms of the GNU Library General Public           *)
(*  License version 2.1, with the special exception on linking            *)
(*  described in file LICENSE.                                            *)
(*                                                                        *)
(*  This software is distributed in the hope that it will be useful,      *)
(*  but WITHOUT ANY WARRANTY; without even the implied warranty of        *)
(*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  *)
(*                                                                        *)
(**************************************************************************)

open Format
open Pp
open Util
open Ident
open Ty
open Term
open Theory

28
let iprinter,tprinter,lprinter,pprinter =
29 30 31 32 33 34 35
  let bl = ["theory"; "type"; "logic"; "inductive";
            "axiom"; "lemma"; "goal"; "use"; "clone";
            "namespace"; "import"; "export"; "end";
            "forall"; "exists"; "and"; "or"; "not";
            "true"; "false"; "if"; "then"; "else";
            "let"; "in"; "match"; "with"; "as"; "epsilon" ]
  in
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
  let isanitize = sanitizer char_to_alpha char_to_alnumus in
  let lsanitize = sanitizer char_to_lalpha char_to_alnumus in
  let usanitize = sanitizer char_to_ualpha char_to_alnumus in
  create_ident_printer bl ~sanitizer:isanitize,
  create_ident_printer bl ~sanitizer:lsanitize,
  create_ident_printer bl ~sanitizer:lsanitize,
  create_ident_printer bl ~sanitizer:usanitize

let thash = Hid.create 63
let lhash = Hid.create 63
let phash = Hid.create 63

let forget_all () =
  forget_all iprinter;
  forget_all tprinter;
  forget_all lprinter;
  forget_all pprinter;
  Hid.clear thash;
  Hid.clear lhash;
  Hid.clear phash
56

57
let tv_set = ref Sid.empty
58

59 60
(* type variables always start with a quote *)
let print_tv fmt tv =
61
  tv_set := Sid.add (tv_name tv) !tv_set;
62
  let sanitize n = "'" ^ n in
63
  let n = id_unique iprinter ~sanitizer:sanitize (tv_name tv) in
64 65
  fprintf fmt "%s" n

66 67 68 69 70 71
let forget_tvs () =
  Sid.iter (forget_id iprinter) !tv_set;
  tv_set := Sid.empty

(* logic variables always start with a lower case letter *)
let print_vs fmt vs =
72
  let sanitize = String.uncapitalize in
73
  let n = id_unique iprinter ~sanitizer:sanitize vs.vs_name in
74 75
  fprintf fmt "%s" n

76
let forget_var vs = forget_id iprinter vs.vs_name
77

78 79 80 81
(* theory names always start with an upper case letter *)
let print_th fmt th =
  let sanitize = String.capitalize in
  let n = id_unique iprinter ~sanitizer:sanitize th.th_name in
82 83
  fprintf fmt "%s" n

84 85 86
let print_ts fmt ts =
  Hid.replace thash ts.ts_name ts;
  fprintf fmt "%s" (id_unique tprinter ts.ts_name)
87

88 89
let print_ls fmt ls =
  Hid.replace lhash ls.ls_name ls;
90 91 92 93 94
  let n = if ls.ls_constr
    then id_unique lprinter ~sanitizer:String.capitalize ls.ls_name
    else id_unique lprinter ls.ls_name
  in
  fprintf fmt "%s" n
95

96
let print_pr fmt pr =
97 98
  Hid.replace phash (pr_name pr) pr;
  fprintf fmt "%s" (id_unique pprinter (pr_name pr))
99 100 101 102 103

(** Types *)

let rec ns_comma fmt () = fprintf fmt ",@,"

104 105 106 107
let rec print_ty fmt ty = match ty.ty_node with
  | Tyvar v -> print_tv fmt v
  | Tyapp (ts, []) -> print_ts fmt ts
  | Tyapp (ts, [t]) -> fprintf fmt "%a@ %a" print_ty t print_ts ts
108
  | Tyapp (ts, l) -> fprintf fmt "(%a)@ %a"
109
      (print_list ns_comma print_ty) l print_ts ts
110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136

let print_const fmt = function
  | ConstInt s -> fprintf fmt "%s" s
  | ConstReal (RConstDecimal (i,f,None)) -> fprintf fmt "%s.%s" i f
  | ConstReal (RConstDecimal (i,f,Some e)) -> fprintf fmt "%s.%se%s" i f e
  | ConstReal (RConstHexa (i,f,e)) -> fprintf fmt "0x%s.%sp%s" i f e

(* can the type of a value be derived from the type of the arguments? *)
let unambig_fs fs =
  let rec lookup v ty = match ty.ty_node with
    | Tyvar u when u == v -> true
    | _ -> ty_any (lookup v) ty
  in
  let lookup v = List.exists (lookup v) fs.ls_args in
  let rec inspect ty = match ty.ty_node with
    | Tyvar u when not (lookup u) -> false
    | _ -> ty_all inspect ty
  in
  inspect (of_option fs.ls_value)

(** Patterns, terms, and formulas *)

let lparen_l fmt () = fprintf fmt "@ ("
let lparen_r fmt () = fprintf fmt "(@,"
let print_paren_l fmt x = print_list_delim lparen_l rparen comma fmt x
let print_paren_r fmt x = print_list_delim lparen_r rparen comma fmt x

137
let rec print_pat fmt p = match p.pat_node with
138
  | Pwild -> fprintf fmt "_"
139 140
  | Pvar v -> print_vs fmt v
  | Pas (p,v) -> fprintf fmt "%a as %a" print_pat p print_vs v
141
  | Papp (cs,pl) -> fprintf fmt "%a%a"
142
      print_ls cs (print_paren_r print_pat) pl
143

144 145
let print_vsty fmt v =
  fprintf fmt "%a:@,%a" print_vs v print_ty v.vs_ty
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160

let print_quant fmt = function
  | Fforall -> fprintf fmt "forall"
  | Fexists -> fprintf fmt "exists"

let print_binop fmt = function
  | Fand -> fprintf fmt "and"
  | For -> fprintf fmt "or"
  | Fimplies -> fprintf fmt "->"
  | Fiff -> fprintf fmt "<->"

let print_label fmt l = fprintf fmt "\"%s\"" l

let protect_on x s = if x then "(" ^^ s ^^ ")" else s

161 162 163 164 165 166
let rec print_term fmt t = print_lrterm false false fmt t
and     print_fmla fmt f = print_lrfmla false false fmt f
and print_opl_term fmt t = print_lrterm true  false fmt t
and print_opl_fmla fmt f = print_lrfmla true  false fmt f
and print_opr_term fmt t = print_lrterm false true  fmt t
and print_opr_fmla fmt f = print_lrfmla false true  fmt f
167

168 169
and print_lrterm opl opr fmt t = match t.t_label with
  | [] -> print_tnode opl opr fmt t
170
  | ll -> fprintf fmt "(%a %a)"
171
      (print_list space print_label) ll (print_tnode false false) t
172

173 174
and print_lrfmla opl opr fmt f = match f.f_label with
  | [] -> print_fnode opl opr fmt f
175
  | ll -> fprintf fmt "(%a %a)"
176
      (print_list space print_label) ll (print_fnode false false) f
177

178
and print_tnode opl opr fmt t = match t.t_node with
179 180 181
  | Tbvar _ ->
      assert false
  | Tvar v ->
182
      print_vs fmt v
183 184 185
  | Tconst c ->
      print_const fmt c
  | Tapp (fs, tl) when unambig_fs fs ->
186
      fprintf fmt "%a%a" print_ls fs (print_paren_r print_term) tl
187 188
  | Tapp (fs, tl) ->
      fprintf fmt (protect_on opl "%a%a:%a")
189
        print_ls fs (print_paren_r print_term) tl print_ty t.t_ty
190 191 192
  | Tlet (t1,tb) ->
      let v,t2 = t_open_bound tb in
      fprintf fmt (protect_on opr "let %a =@ %a in@ %a")
193 194
        print_vs v print_opl_term t1 print_opl_term t2;
      forget_var v
195 196
  | Tcase (t1,bl) ->
      fprintf fmt "match %a with@\n@[<hov>%a@]@\nend"
197
        print_term t1 (print_list newline print_tbranch) bl
198 199
  | Teps fb ->
      let v,f = f_open_bound fb in
Andrei Paskevich's avatar
Andrei Paskevich committed
200
      fprintf fmt (protect_on opr "epsilon %a.@ %a")
201 202
        print_vsty v print_opl_fmla f;
      forget_var v
203

204
and print_fnode opl opr fmt f = match f.f_node with
205 206
  | Fapp (ps,[t1;t2]) when ps = ps_equ ->
      fprintf fmt (protect_on (opl || opr) "%a =@ %a")
207
        print_opr_term t1 print_opl_term t2
208
  | Fapp (ps,tl) ->
209 210
      fprintf fmt "%a%a" print_ls ps
        (print_paren_r print_term) tl
211 212 213
  | Fquant (q,fq) ->
      let vl,tl,f = f_open_quant fq in
      fprintf fmt (protect_on opr "%a %a%a.@ %a") print_quant q
214 215
        (print_list comma print_vsty) vl print_tl tl print_fmla f;
      List.iter forget_var vl
216 217 218 219 220 221
  | Ftrue ->
      fprintf fmt "true"
  | Ffalse ->
      fprintf fmt "false"
  | Fbinop (b,f1,f2) ->
      fprintf fmt (protect_on (opl || opr) "%a %a@ %a")
222
        print_opr_fmla f1 print_binop b print_opl_fmla f2
223
  | Fnot f ->
224
      fprintf fmt (protect_on opr "not %a") print_opl_fmla f
225 226 227
  | Flet (t,f) ->
      let v,f = f_open_bound f in
      fprintf fmt (protect_on opr "let %a =@ %a in@ %a")
228 229
        print_vs v print_opl_term t print_opl_fmla f;
      forget_var v
230
  | Fcase (t,bl) ->
231 232
      fprintf fmt "match %a with@\n@[<hov>%a@]@\nend" print_term t
        (print_list newline print_fbranch) bl
233 234
  | Fif (f1,f2,f3) ->
      fprintf fmt (protect_on opr "if %a@ then %a@ else %a")
235
        print_fmla f1 print_fmla f2 print_opl_fmla f3
236

237
and print_tbranch fmt br =
238
  let pat,svs,t = t_open_branch br in
239 240
  fprintf fmt "@[<hov 4>| %a ->@ %a@]" print_pat pat print_term t;
  Svs.iter forget_var svs
241

242
and print_fbranch fmt br =
243
  let pat,svs,f = f_open_branch br in
244 245
  fprintf fmt "@[<hov 4>| %a ->@ %a@]" print_pat pat print_fmla f;
  Svs.iter forget_var svs
246

247
and print_tl fmt tl =
248
  if tl = [] then () else fprintf fmt "@ [%a]"
249
    (print_list alt (print_list comma print_tr)) tl
250

251
and print_tr fmt = function
252 253
  | Term t -> print_term fmt t
  | Fmla f -> print_fmla fmt f
254 255 256

(** Declarations *)

257 258 259
let print_constr fmt cs =
  fprintf fmt "@[<hov 4>| %a%a@]" print_ls cs
    (print_paren_l print_ty) cs.ls_args
260

261
let print_ty_args fmt = function
262
  | [] -> ()
263 264
  | [tv] -> fprintf fmt " %a" print_tv tv
  | l -> fprintf fmt " (%a)" (print_list ns_comma print_tv) l
265

266
let print_type_decl fmt (ts,def) = match def with
267 268 269
  | Tabstract -> begin match ts.ts_def with
      | None ->
          fprintf fmt "@[<hov 2>type%a %a@]"
270
            print_ty_args ts.ts_args print_ts ts
271 272
      | Some ty ->
          fprintf fmt "@[<hov 2>type%a %a =@ %a@]"
273
            print_ty_args ts.ts_args print_ts ts print_ty ty
274 275 276
      end
  | Talgebraic csl ->
      fprintf fmt "@[<hov 2>type%a %a =@\n@[<hov>%a@]@]"
277 278
        print_ty_args ts.ts_args print_ts ts
        (print_list newline print_constr) csl
279

280
let print_type_decl fmt d = print_type_decl fmt d; forget_tvs ()
281

282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
let print_ld fmt ld =
  let _,vl,e = open_ls_defn ld in
  begin match e with
    | Term t -> print_term fmt t
    | Fmla f -> print_fmla fmt f
  end;
  List.iter forget_var vl

let print_ls_defn fmt = option_iter (fprintf fmt " =@ %a" print_ld)
let print_ls_type fmt = option_iter (fprintf fmt " :@ %a" print_ty)

let print_logic_decl fmt (ls,ld) =
  fprintf fmt "@[<hov 2>logic %a%a%a%a@]"
    print_ls ls (print_paren_l print_ty) ls.ls_args
    print_ls_type ls.ls_value print_ls_defn ld;
  forget_tvs ()
298

299 300
let print_ind fmt (pr,f) =
  fprintf fmt "@[<hov 4>| %a : %a@]" print_pr pr print_fmla f
301

302
let print_ind_decl fmt (ps,bl) =
303
  fprintf fmt "@[<hov 2>inductive %a%a =@ @[<hov>%a@]@]"
304 305
    print_ls ps (print_paren_l print_ty) ps.ls_args
    (print_list newline print_ind) bl;
306
  forget_tvs ()
307 308 309 310 311 312

let print_pkind fmt = function
  | Paxiom -> fprintf fmt "axiom"
  | Plemma -> fprintf fmt "lemma"
  | Pgoal  -> fprintf fmt "goal"

313 314 315 316 317
let print_prop_decl fmt (k,pr,f) =
  fprintf fmt "@[<hov 2>%a %a : %a@]" print_pkind k
    print_pr pr print_fmla f;
  forget_tvs ()

318 319 320
let print_inst fmt (id1,id2) =
  if Hid.mem thash id2 then
    let n = id_unique tprinter id1 in
321
    fprintf fmt "type %s = %a" n print_ts (Hid.find thash id2)
322 323
  else if Hid.mem lhash id2 then
    let n = id_unique lprinter id1 in
324
    fprintf fmt "logic %s = %a" n print_ls (Hid.find lhash id2)
325 326
  else if Hid.mem phash id2 then
    let n = id_unique pprinter id1 in
327
    fprintf fmt "prop %s = %a" n print_pr (Hid.find phash id2)
328 329 330
  else assert false

let print_decl fmt d = match d.d_node with
331 332 333
  | Dtype tl  -> print_list newline print_type_decl fmt tl
  | Dlogic ll -> print_list newline print_logic_decl fmt ll
  | Dind il   -> print_list newline print_ind_decl fmt il
334
  | Dprop p   -> print_prop_decl fmt p
335 336 337 338
  | Duse th ->
      fprintf fmt "@[<hov 2>(* use %a *)@]" print_th th
  | Dclone (th,inst) ->
      fprintf fmt "@[<hov 2>(* clone %a with %a *)@]"
339
        print_th th (print_list comma print_inst) inst
340

341
let print_decls fmt dl =
342
  fprintf fmt "@[<hov>%a@\n@]" (print_list newline2 print_decl) dl
343

344
let print_context fmt ctxt = print_decls fmt (Context.get_decls ctxt)
345

346
let print_theory fmt th =
347
  fprintf fmt "@[<hov 2>theory %a@\n%a@]@\nend@\n@."
348
    print_th th print_context th.th_ctxt
349

350
let print_named_context fmt name ctxt =
351
  fprintf fmt "@[<hov 2>context %s@\n%a@]@\nend@\n@."
352
    name print_context ctxt
353