src/Pure/display_goal.ML
author wenzelm
Fri Jul 24 11:50:35 2009 +0200 (2009-07-24 ago)
changeset 32167 d32817dda0e6
parent 32145 220c9e439d39
child 32168 116461b8fc01
permissions -rw-r--r--
Display_Goal.pretty_goals: always Markup.subgoal, clarified options;
wenzelm@32089
     1
(*  Title:      Pure/display_goal.ML
wenzelm@32089
     2
    Author:     Lawrence C Paulson, Cambridge University Computer Laboratory
wenzelm@32089
     3
    Author:     Makarius
wenzelm@32089
     4
wenzelm@32089
     5
Display tactical goal state.
wenzelm@32089
     6
*)
wenzelm@32089
     7
wenzelm@32089
     8
signature DISPLAY_GOAL =
wenzelm@32089
     9
sig
wenzelm@32089
    10
  val goals_limit: int ref
wenzelm@32089
    11
  val show_consts: bool ref
wenzelm@32145
    12
  val pretty_flexpair: Proof.context -> term * term -> Pretty.T
wenzelm@32167
    13
  val pretty_goals: Proof.context -> {total: bool, main: bool, maxgoals: int} ->
wenzelm@32167
    14
    thm -> Pretty.T list
wenzelm@32145
    15
  val pretty_goals_without_context: int -> thm -> Pretty.T list
wenzelm@32145
    16
  val print_goals_without_context: int -> thm -> unit
wenzelm@32089
    17
end;
wenzelm@32089
    18
wenzelm@32089
    19
structure Display_Goal: DISPLAY_GOAL =
wenzelm@32089
    20
struct
wenzelm@32089
    21
wenzelm@32089
    22
val goals_limit = ref 10;      (*max number of goals to print*)
wenzelm@32089
    23
val show_consts = ref false;   (*true: show consts with types in proof state output*)
wenzelm@32089
    24
wenzelm@32145
    25
fun pretty_flexpair ctxt (t, u) = Pretty.block
wenzelm@32145
    26
  [Syntax.pretty_term ctxt t, Pretty.str " =?=", Pretty.brk 1, Syntax.pretty_term ctxt u];
wenzelm@32089
    27
wenzelm@32089
    28
wenzelm@32089
    29
(*print thm A1,...,An/B in "goal style" -- premises as numbered subgoals*)
wenzelm@32089
    30
wenzelm@32089
    31
local
wenzelm@32089
    32
wenzelm@32089
    33
fun ins_entry (x, y) =
wenzelm@32089
    34
  AList.default (op =) (x, []) #>
wenzelm@32089
    35
  AList.map_entry (op =) x (insert (op =) y);
wenzelm@32089
    36
wenzelm@32089
    37
val add_consts = Term.fold_aterms
wenzelm@32089
    38
  (fn Const (c, T) => ins_entry (T, (c, T))
wenzelm@32089
    39
    | _ => I);
wenzelm@32089
    40
wenzelm@32089
    41
val add_vars = Term.fold_aterms
wenzelm@32089
    42
  (fn Free (x, T) => ins_entry (T, (x, ~1))
wenzelm@32089
    43
    | Var (xi, T) => ins_entry (T, xi)
wenzelm@32089
    44
    | _ => I);
wenzelm@32089
    45
wenzelm@32089
    46
val add_varsT = Term.fold_atyps
wenzelm@32089
    47
  (fn TFree (x, S) => ins_entry (S, (x, ~1))
wenzelm@32089
    48
    | TVar (xi, S) => ins_entry (S, xi)
wenzelm@32089
    49
    | _ => I);
wenzelm@32089
    50
wenzelm@32089
    51
fun sort_idxs vs = map (apsnd (sort (prod_ord string_ord int_ord))) vs;
wenzelm@32089
    52
fun sort_cnsts cs = map (apsnd (sort_wrt fst)) cs;
wenzelm@32089
    53
wenzelm@32089
    54
fun consts_of t = sort_cnsts (add_consts t []);
wenzelm@32089
    55
fun vars_of t = sort_idxs (add_vars t []);
wenzelm@32089
    56
fun varsT_of t = rev (sort_idxs (Term.fold_types add_varsT t []));
wenzelm@32089
    57
wenzelm@32089
    58
in
wenzelm@32089
    59
wenzelm@32167
    60
fun pretty_goals ctxt {total, main, maxgoals} state =
wenzelm@32089
    61
  let
wenzelm@32145
    62
    val prt_sort = Syntax.pretty_sort ctxt;
wenzelm@32145
    63
    val prt_typ = Syntax.pretty_typ ctxt;
wenzelm@32145
    64
    val prt_term = Syntax.pretty_term ctxt;
wenzelm@32145
    65
wenzelm@32089
    66
    fun prt_atoms prt prtT (X, xs) = Pretty.block
wenzelm@32089
    67
      [Pretty.block (Pretty.commas (map prt xs)), Pretty.str " ::",
wenzelm@32089
    68
        Pretty.brk 1, prtT X];
wenzelm@32089
    69
wenzelm@32145
    70
    fun prt_var (x, ~1) = prt_term (Syntax.free x)
wenzelm@32145
    71
      | prt_var xi = prt_term (Syntax.var xi);
wenzelm@32089
    72
wenzelm@32145
    73
    fun prt_varT (x, ~1) = prt_typ (TFree (x, []))
wenzelm@32145
    74
      | prt_varT xi = prt_typ (TVar (xi, []));
wenzelm@32089
    75
wenzelm@32145
    76
    val prt_consts = prt_atoms (prt_term o Const) prt_typ;
wenzelm@32145
    77
    val prt_vars = prt_atoms prt_var prt_typ;
wenzelm@32145
    78
    val prt_varsT = prt_atoms prt_varT prt_sort;
wenzelm@32089
    79
wenzelm@32089
    80
wenzelm@32089
    81
    fun pretty_list _ _ [] = []
wenzelm@32089
    82
      | pretty_list name prt lst = [Pretty.big_list name (map prt lst)];
wenzelm@32089
    83
wenzelm@32167
    84
    fun pretty_subgoal (n, A) = Pretty.markup Markup.subgoal
wenzelm@32145
    85
      [Pretty.str (" " ^ string_of_int n ^ ". "), prt_term A];
wenzelm@32089
    86
    fun pretty_subgoals As = map pretty_subgoal (1 upto length As ~~ As);
wenzelm@32089
    87
wenzelm@32145
    88
    val pretty_ffpairs = pretty_list "flex-flex pairs:" (pretty_flexpair ctxt);
wenzelm@32089
    89
wenzelm@32089
    90
    val pretty_consts = pretty_list "constants:" prt_consts o consts_of;
wenzelm@32089
    91
    val pretty_vars = pretty_list "variables:" prt_vars o vars_of;
wenzelm@32089
    92
    val pretty_varsT = pretty_list "type variables:" prt_varsT o varsT_of;
wenzelm@32089
    93
wenzelm@32089
    94
wenzelm@32089
    95
    val {prop, tpairs, ...} = Thm.rep_thm state;
wenzelm@32089
    96
    val (As, B) = Logic.strip_horn prop;
wenzelm@32089
    97
    val ngoals = length As;
wenzelm@32089
    98
wenzelm@32089
    99
    fun pretty_gs (types, sorts) =
wenzelm@32145
   100
      (if main then [prt_term B] else []) @
wenzelm@32089
   101
       (if ngoals = 0 then [Pretty.str "No subgoals!"]
wenzelm@32089
   102
        else if ngoals > maxgoals then
wenzelm@32089
   103
          pretty_subgoals (Library.take (maxgoals, As)) @
wenzelm@32167
   104
          (if total then [Pretty.str ("A total of " ^ string_of_int ngoals ^ " subgoals...")]
wenzelm@32089
   105
           else [])
wenzelm@32089
   106
        else pretty_subgoals As) @
wenzelm@32089
   107
      pretty_ffpairs tpairs @
wenzelm@32089
   108
      (if ! show_consts then pretty_consts prop else []) @
wenzelm@32089
   109
      (if types then pretty_vars prop else []) @
wenzelm@32089
   110
      (if sorts then pretty_varsT prop else []);
wenzelm@32089
   111
  in
wenzelm@32089
   112
    setmp show_no_free_types true
wenzelm@32089
   113
      (setmp show_types (! show_types orelse ! show_sorts orelse ! show_all_types)
wenzelm@32089
   114
        (setmp show_sorts false pretty_gs))
wenzelm@32089
   115
   (! show_types orelse ! show_sorts orelse ! show_all_types, ! show_sorts)
wenzelm@32089
   116
  end;
wenzelm@32089
   117
wenzelm@32145
   118
fun pretty_goals_without_context n th =
wenzelm@32167
   119
  pretty_goals (Syntax.init_pretty_global (Thm.theory_of_thm th))
wenzelm@32167
   120
    {total = true, main = true, maxgoals = n} th;
wenzelm@32089
   121
wenzelm@32145
   122
val print_goals_without_context =
wenzelm@32145
   123
  (Pretty.writeln o Pretty.chunks) oo pretty_goals_without_context;
wenzelm@32089
   124
wenzelm@32089
   125
end;
wenzelm@32089
   126
wenzelm@32089
   127
end;
wenzelm@32089
   128