src/Pure/Thy/thy_output.ML
author wenzelm
Sun, 28 Nov 2010 21:07:28 +0100
changeset 41048 330eb65c9469
parent 39937 78b185bf7660
child 41051 6cfacec435e6
permissions -rw-r--r--
Parse.liberal_name for document antiquotations and attributes;
wenzelm@22124
     1
(*  Title:      Pure/Thy/thy_output.ML
wenzelm@22124
     2
    Author:     Markus Wenzel, TU Muenchen
wenzelm@22124
     3
wenzelm@30390
     4
Theory document output with antiquotations.
wenzelm@22124
     5
*)
wenzelm@22124
     6
wenzelm@22124
     7
signature THY_OUTPUT =
wenzelm@22124
     8
sig
wenzelm@39042
     9
  val display_default: bool Unsynchronized.ref
wenzelm@39042
    10
  val quotes_default: bool Unsynchronized.ref
wenzelm@39042
    11
  val indent_default: int Unsynchronized.ref
wenzelm@39042
    12
  val source_default: bool Unsynchronized.ref
wenzelm@39042
    13
  val break_default: bool Unsynchronized.ref
wenzelm@39042
    14
  val display: bool Config.T
wenzelm@39042
    15
  val quotes: bool Config.T
wenzelm@39042
    16
  val indent: int Config.T
wenzelm@39042
    17
  val source: bool Config.T
wenzelm@39042
    18
  val break: bool Config.T
wenzelm@39041
    19
  val add_wrapper: ((unit -> string) -> unit -> string) -> Proof.context -> Proof.context
wenzelm@39041
    20
  val add_option: string -> (string -> Proof.context -> Proof.context) -> unit
wenzelm@26893
    21
  val defined_command: string -> bool
wenzelm@26893
    22
  val defined_option: string -> bool
wenzelm@22124
    23
  val print_antiquotations: unit -> unit
wenzelm@22124
    24
  val boolean: string -> bool
wenzelm@22124
    25
  val integer: string -> int
wenzelm@30518
    26
  val antiquotation: string -> 'a context_parser ->
wenzelm@39041
    27
    ({source: Args.src, state: Toplevel.state, context: Proof.context} -> 'a -> string) -> unit
wenzelm@22124
    28
  datatype markup = Markup | MarkupEnv | Verbatim
wenzelm@32738
    29
  val modes: string list Unsynchronized.ref
wenzelm@30576
    30
  val eval_antiquote: Scan.lexicon -> Toplevel.state -> Symbol_Pos.text * Position.T -> string
wenzelm@28427
    31
  val present_thy: Scan.lexicon -> (string -> string list) -> (markup -> string -> bool) ->
wenzelm@36969
    32
    (Toplevel.transition * Toplevel.state) list -> (Token.T, 'a) Source.source -> Buffer.T
wenzelm@39042
    33
  val pretty_text: Proof.context -> string -> Pretty.T
wenzelm@28644
    34
  val pretty_term: Proof.context -> term -> Pretty.T
wenzelm@28644
    35
  val pretty_thm: Proof.context -> thm -> Pretty.T
wenzelm@30390
    36
  val str_of_source: Args.src -> string
wenzelm@39042
    37
  val maybe_pretty_source: (Proof.context -> 'a -> Pretty.T) -> Proof.context ->
wenzelm@39042
    38
    Args.src -> 'a list -> Pretty.T list
wenzelm@39042
    39
  val output: Proof.context -> Pretty.T list -> string
wenzelm@22124
    40
end;
wenzelm@22124
    41
wenzelm@37216
    42
structure Thy_Output: THY_OUTPUT =
wenzelm@22124
    43
struct
wenzelm@22124
    44
wenzelm@22124
    45
(** global options **)
wenzelm@22124
    46
wenzelm@39042
    47
val display_default = Unsynchronized.ref false;
wenzelm@39042
    48
val quotes_default = Unsynchronized.ref false;
wenzelm@39042
    49
val indent_default = Unsynchronized.ref 0;
wenzelm@39042
    50
val source_default = Unsynchronized.ref false;
wenzelm@39042
    51
val break_default = Unsynchronized.ref false;
wenzelm@39042
    52
wenzelm@39042
    53
val (display, setup_display) = Attrib.config_bool "thy_output_display" (fn _ => ! display_default);
wenzelm@39042
    54
val (quotes, setup_quotes) = Attrib.config_bool "thy_output_quotes" (fn _ => ! quotes_default);
wenzelm@39042
    55
val (indent, setup_indent) = Attrib.config_int "thy_output_indent" (fn _ => ! indent_default);
wenzelm@39042
    56
val (source, setup_source) = Attrib.config_bool "thy_output_source" (fn _ => ! source_default);
wenzelm@39042
    57
val (break, setup_break) = Attrib.config_bool "thy_output_break" (fn _ => ! break_default);
wenzelm@39042
    58
wenzelm@39042
    59
val _ = Context.>> (Context.map_theory
wenzelm@39042
    60
 (setup_display #> setup_quotes #> setup_indent #> setup_source #> setup_break));
wenzelm@22124
    61
wenzelm@22124
    62
wenzelm@39041
    63
structure Wrappers = Proof_Data
wenzelm@39041
    64
(
wenzelm@39041
    65
  type T = ((unit -> string) -> unit -> string) list;
wenzelm@39041
    66
  fun init _ = [];
wenzelm@39041
    67
);
wenzelm@39041
    68
wenzelm@39041
    69
fun add_wrapper wrapper = Wrappers.map (cons wrapper);
wenzelm@39041
    70
wenzelm@39041
    71
val wrap = Wrappers.get #> fold (fn wrapper => fn f => wrapper f);
wenzelm@39041
    72
wenzelm@39041
    73
wenzelm@22124
    74
wenzelm@30390
    75
(** maintain global antiquotations **)
wenzelm@22124
    76
wenzelm@22124
    77
local
wenzelm@22124
    78
wenzelm@22124
    79
val global_commands =
wenzelm@39041
    80
  Unsynchronized.ref
wenzelm@39041
    81
    (Symtab.empty: (Args.src -> Toplevel.state -> Proof.context -> string) Symtab.table);
wenzelm@22124
    82
wenzelm@22124
    83
val global_options =
wenzelm@39041
    84
  Unsynchronized.ref (Symtab.empty: (string -> Proof.context -> Proof.context) Symtab.table);
wenzelm@22124
    85
wenzelm@39041
    86
fun add_item kind name item tab =
wenzelm@22124
    87
 (if not (Symtab.defined tab name) then ()
wenzelm@27897
    88
  else warning ("Redefined document antiquotation " ^ kind ^ ": " ^ quote name);
wenzelm@39041
    89
  Symtab.update (name, item) tab);
wenzelm@22124
    90
wenzelm@22124
    91
in
wenzelm@22124
    92
wenzelm@39041
    93
fun add_command name cmd =
wenzelm@39041
    94
  CRITICAL (fn () => Unsynchronized.change global_commands (add_item "command" name cmd));
wenzelm@39041
    95
fun add_option name opt =
wenzelm@39041
    96
  CRITICAL (fn () => Unsynchronized.change global_options (add_item "option" name opt));
wenzelm@22124
    97
wenzelm@26893
    98
fun defined_command name = Symtab.defined (! global_commands) name;
wenzelm@26893
    99
fun defined_option name = Symtab.defined (! global_options) name;
wenzelm@26893
   100
wenzelm@22124
   101
fun command src =
wenzelm@22124
   102
  let val ((name, _), pos) = Args.dest_src src in
wenzelm@22124
   103
    (case Symtab.lookup (! global_commands) name of
wenzelm@30317
   104
      NONE => error ("Unknown document antiquotation: " ^ quote name ^ Position.str_of pos)
wenzelm@30317
   105
    | SOME f =>
wenzelm@39776
   106
       (Position.report pos (Markup.doc_antiq name);
wenzelm@39041
   107
        (fn state => fn ctxt => f src state ctxt handle ERROR msg =>
wenzelm@30317
   108
          cat_error msg ("The error(s) above occurred in document antiquotation: " ^
wenzelm@30317
   109
            quote name ^ Position.str_of pos))))
wenzelm@22124
   110
  end;
wenzelm@22124
   111
wenzelm@39041
   112
fun option (name, s) ctxt =
wenzelm@22124
   113
  (case Symtab.lookup (! global_options) name of
wenzelm@27897
   114
    NONE => error ("Unknown document antiquotation option: " ^ quote name)
wenzelm@39041
   115
  | SOME opt => opt s ctxt);
wenzelm@22124
   116
wenzelm@22124
   117
wenzelm@22124
   118
fun print_antiquotations () =
wenzelm@27897
   119
 [Pretty.big_list "document antiquotation commands:"
wenzelm@23272
   120
    (map Pretty.str (sort_strings (Symtab.keys (! global_commands)))),
wenzelm@27897
   121
  Pretty.big_list "document antiquotation options:"
wenzelm@23272
   122
    (map Pretty.str (sort_strings (Symtab.keys (! global_options))))]
wenzelm@22124
   123
 |> Pretty.chunks |> Pretty.writeln;
wenzelm@22124
   124
wenzelm@22124
   125
end;
wenzelm@22124
   126
wenzelm@39041
   127
fun antiquotation name scan out =
wenzelm@39041
   128
  add_command name
wenzelm@39041
   129
    (fn src => fn state => fn ctxt =>
wenzelm@39041
   130
      let val (x, ctxt') = Args.context_syntax "document antiquotation" scan src ctxt
wenzelm@39041
   131
      in out {source = src, state = state, context = ctxt'} x end);
wenzelm@30390
   132
wenzelm@22124
   133
wenzelm@22124
   134
wenzelm@22124
   135
(** syntax of antiquotations **)
wenzelm@22124
   136
wenzelm@22124
   137
(* option values *)
wenzelm@22124
   138
wenzelm@22124
   139
fun boolean "" = true
wenzelm@22124
   140
  | boolean "true" = true
wenzelm@22124
   141
  | boolean "false" = false
wenzelm@22124
   142
  | boolean s = error ("Bad boolean value: " ^ quote s);
wenzelm@22124
   143
wenzelm@22124
   144
fun integer s =
wenzelm@22124
   145
  let
wenzelm@22124
   146
    fun int ss =
wenzelm@22124
   147
      (case Library.read_int ss of (i, []) => i
wenzelm@22124
   148
      | _ => error ("Bad integer value: " ^ quote s));
wenzelm@22124
   149
  in (case Symbol.explode s of "-" :: ss => ~ (int ss) | ss => int ss) end;
wenzelm@22124
   150
wenzelm@22124
   151
wenzelm@22124
   152
(* outer syntax *)
wenzelm@22124
   153
wenzelm@22124
   154
local
wenzelm@22124
   155
wenzelm@36950
   156
val property =
wenzelm@36950
   157
  Parse.xname -- Scan.optional (Parse.$$$ "=" |-- Parse.!!! Parse.xname) "";
wenzelm@36950
   158
wenzelm@36950
   159
val properties =
wenzelm@36950
   160
  Scan.optional (Parse.$$$ "[" |-- Parse.!!! (Parse.enum "," property --| Parse.$$$ "]")) [];
wenzelm@22124
   161
wenzelm@22124
   162
in
wenzelm@22124
   163
wenzelm@36950
   164
val antiq =
wenzelm@41048
   165
  Parse.!!!
wenzelm@41048
   166
    (Parse.position Parse.liberal_name -- properties -- Args.parse --| Scan.ahead Parse.eof)
wenzelm@22124
   167
  >> (fn (((x, pos), y), z) => (y, Args.src ((x, z), pos)));
wenzelm@22124
   168
wenzelm@22124
   169
end;
wenzelm@22124
   170
wenzelm@22124
   171
wenzelm@22124
   172
(* eval_antiquote *)
wenzelm@22124
   173
wenzelm@32738
   174
val modes = Unsynchronized.ref ([]: string list);
wenzelm@22124
   175
wenzelm@30367
   176
fun eval_antiquote lex state (txt, pos) =
wenzelm@22124
   177
  let
wenzelm@30592
   178
    fun expand (Antiquote.Text ss) = Symbol_Pos.content ss
wenzelm@30698
   179
      | expand (Antiquote.Antiq (ss, (pos, _))) =
wenzelm@39041
   180
          let
wenzelm@39041
   181
            val (opts, src) = Token.read_antiq lex antiq (ss, pos);
wenzelm@39078
   182
            fun cmd ctxt = wrap ctxt (fn () => command src state ctxt) ();
wenzelm@39078
   183
            val preview_ctxt = fold option opts (Toplevel.presentation_context_of state);
wenzelm@39078
   184
            val print_ctxt = Context_Position.set_visible false preview_ctxt;
wenzelm@39078
   185
            val _ = cmd preview_ctxt;
wenzelm@39078
   186
          in Print_Mode.with_modes (! modes @ Latex.modes) (fn () => cmd print_ctxt) () end
wenzelm@27344
   187
      | expand (Antiquote.Open _) = ""
wenzelm@27344
   188
      | expand (Antiquote.Close _) = "";
wenzelm@30646
   189
    val ants = Antiquote.read (Symbol_Pos.explode (txt, pos), pos);
wenzelm@22124
   190
  in
wenzelm@30639
   191
    if Toplevel.is_toplevel state andalso not (forall Antiquote.is_text ants) then
wenzelm@27897
   192
      error ("Unknown context -- cannot expand document antiquotations" ^ Position.str_of pos)
wenzelm@22124
   193
    else implode (map expand ants)
wenzelm@22124
   194
  end;
wenzelm@22124
   195
wenzelm@22124
   196
wenzelm@22124
   197
wenzelm@22124
   198
(** present theory source **)
wenzelm@22124
   199
wenzelm@22124
   200
(*NB: arranging white space around command spans is a black art.*)
wenzelm@22124
   201
wenzelm@22124
   202
(* presentation tokens *)
wenzelm@22124
   203
wenzelm@22124
   204
datatype token =
wenzelm@22124
   205
    NoToken
wenzelm@36969
   206
  | BasicToken of Token.T
wenzelm@22124
   207
  | MarkupToken of string * (string * Position.T)
wenzelm@22124
   208
  | MarkupEnvToken of string * (string * Position.T)
wenzelm@22124
   209
  | VerbatimToken of string * Position.T;
wenzelm@22124
   210
wenzelm@22124
   211
fun output_token lex state =
wenzelm@30367
   212
  let val eval = eval_antiquote lex state in
wenzelm@22124
   213
    fn NoToken => ""
wenzelm@22124
   214
     | BasicToken tok => Latex.output_basic tok
wenzelm@22124
   215
     | MarkupToken (cmd, txt) => Latex.output_markup cmd (eval txt)
wenzelm@22124
   216
     | MarkupEnvToken (cmd, txt) => Latex.output_markup_env cmd (eval txt)
wenzelm@22124
   217
     | VerbatimToken txt => Latex.output_verbatim (eval txt)
wenzelm@22124
   218
  end;
wenzelm@22124
   219
wenzelm@22124
   220
fun basic_token pred (BasicToken tok) = pred tok
wenzelm@22124
   221
  | basic_token _ _ = false;
wenzelm@22124
   222
wenzelm@36969
   223
val improper_token = basic_token (not o Token.is_proper);
wenzelm@36969
   224
val comment_token = basic_token Token.is_comment;
wenzelm@36969
   225
val blank_token = basic_token Token.is_blank;
wenzelm@36969
   226
val newline_token = basic_token Token.is_newline;
wenzelm@22124
   227
wenzelm@22124
   228
wenzelm@22124
   229
(* command spans *)
wenzelm@22124
   230
wenzelm@22124
   231
type command = string * Position.T * string list;   (*name, position, tags*)
wenzelm@22124
   232
type source = (token * (string * int)) list;        (*token, markup flag, meta-comment depth*)
wenzelm@22124
   233
wenzelm@22124
   234
datatype span = Span of command * (source * source * source * source) * bool;
wenzelm@22124
   235
wenzelm@22124
   236
fun make_span cmd src =
wenzelm@22124
   237
  let
wenzelm@22124
   238
    fun take_newline (tok :: toks) =
wenzelm@22124
   239
          if newline_token (fst tok) then ([tok], toks, true)
wenzelm@22124
   240
          else ([], tok :: toks, false)
wenzelm@22124
   241
      | take_newline [] = ([], [], false);
wenzelm@22124
   242
    val (((src_prefix, src_main), src_suffix1), (src_suffix2, src_appendix, newline)) =
wenzelm@22124
   243
      src
wenzelm@22124
   244
      |> take_prefix (improper_token o fst)
wenzelm@22124
   245
      ||>> take_suffix (improper_token o fst)
wenzelm@22124
   246
      ||>> take_prefix (comment_token o fst)
wenzelm@22124
   247
      ||> take_newline;
wenzelm@22124
   248
  in Span (cmd, (src_prefix, src_main, src_suffix1 @ src_suffix2, src_appendix), newline) end;
wenzelm@22124
   249
wenzelm@22124
   250
wenzelm@22124
   251
(* present spans *)
wenzelm@22124
   252
wenzelm@22124
   253
local
wenzelm@22124
   254
wenzelm@22124
   255
fun err_bad_nesting pos =
wenzelm@22124
   256
  error ("Bad nesting of commands in presentation" ^ pos);
wenzelm@22124
   257
wenzelm@22124
   258
fun edge which f (x: string option, y) =
wenzelm@22124
   259
  if x = y then I
wenzelm@22124
   260
  else (case which (x, y) of NONE => I | SOME txt => Buffer.add (f txt));
wenzelm@22124
   261
wenzelm@22124
   262
val begin_tag = edge #2 Latex.begin_tag;
wenzelm@22124
   263
val end_tag = edge #1 Latex.end_tag;
wenzelm@22124
   264
fun open_delim delim e = edge #2 Latex.begin_delim e #> delim #> edge #2 Latex.end_delim e;
wenzelm@22124
   265
fun close_delim delim e = edge #1 Latex.begin_delim e #> delim #> edge #1 Latex.end_delim e;
wenzelm@22124
   266
wenzelm@22124
   267
in
wenzelm@22124
   268
wenzelm@22124
   269
fun present_span lex default_tags span state state'
wenzelm@22124
   270
    (tag_stack, active_tag, newline, buffer, present_cont) =
wenzelm@22124
   271
  let
wenzelm@22124
   272
    val present = fold (fn (tok, (flag, 0)) =>
wenzelm@22124
   273
        Buffer.add (output_token lex state' tok)
wenzelm@22124
   274
        #> Buffer.add flag
wenzelm@22124
   275
      | _ => I);
wenzelm@22124
   276
wenzelm@22124
   277
    val Span ((cmd_name, cmd_pos, cmd_tags), srcs, span_newline) = span;
wenzelm@22124
   278
wenzelm@22124
   279
    val (tag, tags) = tag_stack;
wenzelm@36950
   280
    val tag' = try hd (fold Keyword.update_tags cmd_tags (the_list tag));
wenzelm@22124
   281
wenzelm@22124
   282
    val active_tag' =
wenzelm@22124
   283
      if is_some tag' then tag'
wenzelm@22124
   284
      else if cmd_name = "end" andalso not (Toplevel.is_toplevel state') then NONE
wenzelm@22124
   285
      else try hd (default_tags cmd_name);
wenzelm@22124
   286
    val edge = (active_tag, active_tag');
wenzelm@22124
   287
wenzelm@22124
   288
    val newline' =
wenzelm@22124
   289
      if is_none active_tag' then span_newline else newline;
wenzelm@22124
   290
wenzelm@22124
   291
    val nesting = Toplevel.level state' - Toplevel.level state;
wenzelm@22124
   292
    val tag_stack' =
wenzelm@22124
   293
      if nesting = 0 andalso not (Toplevel.is_proof state) then tag_stack
wenzelm@22124
   294
      else if nesting >= 0 then (tag', replicate nesting tag @ tags)
wenzelm@22124
   295
      else
haftmann@33956
   296
        (case drop (~ nesting - 1) tags of
wenzelm@22124
   297
          tgs :: tgss => (tgs, tgss)
wenzelm@22124
   298
        | [] => err_bad_nesting (Position.str_of cmd_pos));
wenzelm@22124
   299
wenzelm@22124
   300
    val buffer' =
wenzelm@22124
   301
      buffer
wenzelm@22124
   302
      |> end_tag edge
wenzelm@22124
   303
      |> close_delim (fst present_cont) edge
wenzelm@22124
   304
      |> snd present_cont
wenzelm@22124
   305
      |> open_delim (present (#1 srcs)) edge
wenzelm@22124
   306
      |> begin_tag edge
wenzelm@22124
   307
      |> present (#2 srcs);
wenzelm@22124
   308
    val present_cont' =
wenzelm@22124
   309
      if newline then (present (#3 srcs), present (#4 srcs))
wenzelm@22124
   310
      else (I, present (#3 srcs) #> present (#4 srcs));
wenzelm@22124
   311
  in (tag_stack', active_tag', newline', buffer', present_cont') end;
wenzelm@22124
   312
wenzelm@22124
   313
fun present_trailer ((_, tags), active_tag, _, buffer, present_cont) =
wenzelm@22124
   314
  if not (null tags) then err_bad_nesting " at end of theory"
wenzelm@22124
   315
  else
wenzelm@22124
   316
    buffer
wenzelm@22124
   317
    |> end_tag (active_tag, NONE)
wenzelm@22124
   318
    |> close_delim (fst present_cont) (active_tag, NONE)
wenzelm@22124
   319
    |> snd present_cont;
wenzelm@22124
   320
wenzelm@22124
   321
end;
wenzelm@22124
   322
wenzelm@22124
   323
wenzelm@28427
   324
(* present_thy *)
wenzelm@22124
   325
wenzelm@22124
   326
datatype markup = Markup | MarkupEnv | Verbatim;
wenzelm@22124
   327
wenzelm@22124
   328
local
wenzelm@22124
   329
wenzelm@22124
   330
val space_proper =
wenzelm@36969
   331
  Scan.one Token.is_blank -- Scan.many Token.is_comment -- Scan.one Token.is_proper;
wenzelm@22124
   332
wenzelm@36969
   333
val is_improper = not o (Token.is_proper orf Token.is_begin_ignore orf Token.is_end_ignore);
wenzelm@22124
   334
val improper = Scan.many is_improper;
wenzelm@22124
   335
val improper_end = Scan.repeat (Scan.unless space_proper (Scan.one is_improper));
wenzelm@36969
   336
val blank_end = Scan.repeat (Scan.unless space_proper (Scan.one Token.is_blank));
wenzelm@22124
   337
wenzelm@36969
   338
val opt_newline = Scan.option (Scan.one Token.is_newline);
wenzelm@22124
   339
wenzelm@22124
   340
val ignore =
wenzelm@36969
   341
  Scan.depend (fn d => opt_newline |-- Scan.one Token.is_begin_ignore
wenzelm@22124
   342
    >> pair (d + 1)) ||
wenzelm@36969
   343
  Scan.depend (fn d => Scan.one Token.is_end_ignore --|
wenzelm@22124
   344
    (if d = 0 then Scan.fail_with (K "Bad nesting of meta-comments") else opt_newline)
wenzelm@22124
   345
    >> pair (d - 1));
wenzelm@22124
   346
wenzelm@36950
   347
val tag = (improper -- Parse.$$$ "%" -- improper) |-- Parse.!!! (Parse.tag_name --| blank_end);
wenzelm@22124
   348
wenzelm@22124
   349
val locale =
wenzelm@36950
   350
  Scan.option ((Parse.$$$ "(" -- improper -- Parse.$$$ "in") |--
wenzelm@36950
   351
    Parse.!!! (improper |-- Parse.xname --| (improper -- Parse.$$$ ")")));
wenzelm@22124
   352
wenzelm@22124
   353
in
wenzelm@22124
   354
wenzelm@28427
   355
fun present_thy lex default_tags is_markup command_results src =
wenzelm@22124
   356
  let
wenzelm@22124
   357
    (* tokens *)
wenzelm@22124
   358
wenzelm@22124
   359
    val ignored = Scan.state --| ignore
wenzelm@22124
   360
      >> (fn d => (NONE, (NoToken, ("", d))));
wenzelm@22124
   361
wenzelm@22124
   362
    fun markup mark mk flag = Scan.peek (fn d =>
wenzelm@36950
   363
      improper |--
wenzelm@36969
   364
        Parse.position
wenzelm@36969
   365
          (Scan.one (Token.is_kind Token.Command andf is_markup mark o Token.content_of)) --
wenzelm@22124
   366
      Scan.repeat tag --
wenzelm@36950
   367
      Parse.!!!! ((improper -- locale -- improper) |-- Parse.doc_source --| improper_end)
wenzelm@22124
   368
      >> (fn (((tok, pos), tags), txt) =>
wenzelm@36969
   369
        let val name = Token.content_of tok
wenzelm@22124
   370
        in (SOME (name, pos, tags), (mk (name, txt), (flag, d))) end));
wenzelm@22124
   371
wenzelm@22124
   372
    val command = Scan.peek (fn d =>
wenzelm@36969
   373
      Parse.position (Scan.one (Token.is_kind Token.Command)) --
wenzelm@22124
   374
      Scan.repeat tag
wenzelm@22124
   375
      >> (fn ((tok, pos), tags) =>
wenzelm@36969
   376
        let val name = Token.content_of tok
wenzelm@22124
   377
        in (SOME (name, pos, tags), (BasicToken tok, (Latex.markup_false, d))) end));
wenzelm@22124
   378
wenzelm@22124
   379
    val cmt = Scan.peek (fn d =>
wenzelm@36950
   380
      Parse.$$$ "--" |-- Parse.!!!! (improper |-- Parse.doc_source)
wenzelm@22124
   381
      >> (fn txt => (NONE, (MarkupToken ("cmt", txt), ("", d)))));
wenzelm@22124
   382
wenzelm@22124
   383
    val other = Scan.peek (fn d =>
wenzelm@36950
   384
       Parse.not_eof >> (fn tok => (NONE, (BasicToken tok, ("", d)))));
wenzelm@22124
   385
wenzelm@22124
   386
    val token =
wenzelm@22124
   387
      ignored ||
wenzelm@22124
   388
      markup Markup MarkupToken Latex.markup_true ||
wenzelm@22124
   389
      markup MarkupEnv MarkupEnvToken Latex.markup_true ||
wenzelm@22124
   390
      markup Verbatim (VerbatimToken o #2) "" ||
wenzelm@22124
   391
      command || cmt || other;
wenzelm@22124
   392
wenzelm@22124
   393
wenzelm@22124
   394
    (* spans *)
wenzelm@22124
   395
wenzelm@36969
   396
    val is_eof = fn (_, (BasicToken x, _)) => Token.is_eof x | _ => false;
wenzelm@36969
   397
    val stopper = Scan.stopper (K (NONE, (BasicToken Token.eof, ("", 0)))) is_eof;
wenzelm@22124
   398
wenzelm@22124
   399
    val cmd = Scan.one (is_some o fst);
wenzelm@27732
   400
    val non_cmd = Scan.one (is_none o fst andf not o is_eof) >> #2;
wenzelm@22124
   401
wenzelm@22124
   402
    val comments = Scan.many (comment_token o fst o snd);
wenzelm@22124
   403
    val blank = Scan.one (blank_token o fst o snd);
wenzelm@22124
   404
    val newline = Scan.one (newline_token o fst o snd);
wenzelm@22124
   405
    val before_cmd =
wenzelm@22124
   406
      Scan.option (newline -- comments) --
wenzelm@22124
   407
      Scan.option (newline -- comments) --
wenzelm@22124
   408
      Scan.option (blank -- comments) -- cmd;
wenzelm@22124
   409
wenzelm@22124
   410
    val span =
wenzelm@22124
   411
      Scan.repeat non_cmd -- cmd --
wenzelm@22124
   412
        Scan.repeat (Scan.unless before_cmd non_cmd) --
wenzelm@22124
   413
        Scan.option (newline >> (single o snd))
wenzelm@22124
   414
      >> (fn (((toks1, (cmd, tok2)), toks3), tok4) =>
wenzelm@22124
   415
          make_span (the cmd) (toks1 @ (tok2 :: (toks3 @ the_default [] tok4))));
wenzelm@22124
   416
wenzelm@22124
   417
    val spans =
wenzelm@22124
   418
      src
wenzelm@36969
   419
      |> Source.filter (not o Token.is_semicolon)
wenzelm@36969
   420
      |> Source.source' 0 Token.stopper (Scan.error (Scan.bulk token)) NONE
wenzelm@22124
   421
      |> Source.source stopper (Scan.error (Scan.bulk span)) NONE
wenzelm@22124
   422
      |> Source.exhaust;
wenzelm@28427
   423
wenzelm@28427
   424
wenzelm@28427
   425
    (* present commands *)
wenzelm@28427
   426
wenzelm@28427
   427
    fun present_command tr span st st' =
wenzelm@28427
   428
      Toplevel.setmp_thread_position tr (present_span lex default_tags span st st');
wenzelm@28427
   429
wenzelm@28427
   430
    fun present _ [] = I
wenzelm@28427
   431
      | present st (((tr, st'), span) :: rest) = present_command tr span st st' #> present st' rest;
wenzelm@22124
   432
  in
wenzelm@28427
   433
    if length command_results = length spans then
wenzelm@22124
   434
      ((NONE, []), NONE, true, Buffer.empty, (I, I))
wenzelm@28427
   435
      |> present Toplevel.toplevel (command_results ~~ spans)
wenzelm@22124
   436
      |> present_trailer
wenzelm@22124
   437
    else error "Messed-up outer syntax for presentation"
wenzelm@22124
   438
  end;
wenzelm@22124
   439
wenzelm@22124
   440
end;
wenzelm@22124
   441
wenzelm@22124
   442
wenzelm@22124
   443
wenzelm@22124
   444
(** setup default output **)
wenzelm@22124
   445
wenzelm@22124
   446
(* options *)
wenzelm@22124
   447
wenzelm@39393
   448
val _ = add_option "show_types" (Config.put show_types o boolean);
wenzelm@39393
   449
val _ = add_option "show_sorts" (Config.put show_sorts o boolean);
wenzelm@39373
   450
val _ = add_option "show_structs" (Config.put show_structs o boolean);
wenzelm@39285
   451
val _ = add_option "show_question_marks" (Config.put show_question_marks o boolean);
wenzelm@39041
   452
val _ = add_option "long_names" (add_wrapper o setmp_CRITICAL Name_Space.long_names o boolean);
wenzelm@39041
   453
val _ = add_option "short_names" (add_wrapper o setmp_CRITICAL Name_Space.short_names o boolean);
wenzelm@39041
   454
val _ = add_option "unique_names" (add_wrapper o setmp_CRITICAL Name_Space.unique_names o boolean);
wenzelm@39374
   455
val _ = add_option "eta_contract" (Config.put eta_contract o boolean);
wenzelm@39042
   456
val _ = add_option "display" (Config.put display o boolean);
wenzelm@39042
   457
val _ = add_option "break" (Config.put break o boolean);
wenzelm@39042
   458
val _ = add_option "quotes" (Config.put quotes o boolean);
wenzelm@39041
   459
val _ = add_option "mode" (add_wrapper o Print_Mode.with_modes o single);
wenzelm@39041
   460
val _ = add_option "margin" (add_wrapper o setmp_CRITICAL Pretty.margin_default o integer);
wenzelm@39042
   461
val _ = add_option "indent" (Config.put indent o integer);
wenzelm@39042
   462
val _ = add_option "source" (Config.put source o boolean);
wenzelm@39371
   463
val _ = add_option "goals_limit" (Config.put Goal_Display.goals_limit o integer);
wenzelm@22124
   464
wenzelm@22124
   465
wenzelm@22124
   466
(* basic pretty printing *)
wenzelm@22124
   467
wenzelm@39042
   468
fun tweak_line ctxt s =
wenzelm@39042
   469
  if Config.get ctxt display then s else Symbol.strip_blanks s;
wenzelm@22124
   470
wenzelm@39042
   471
fun pretty_text ctxt =
wenzelm@39042
   472
  Pretty.chunks o map Pretty.str o map (tweak_line ctxt) o Library.split_lines;
wenzelm@22124
   473
wenzelm@26710
   474
fun pretty_term ctxt t = Syntax.pretty_term (Variable.auto_fixes t ctxt) t;
wenzelm@24920
   475
haftmann@32898
   476
fun pretty_thm ctxt = pretty_term ctxt o Thm.full_prop_of;
wenzelm@22124
   477
haftmann@32898
   478
fun pretty_term_style ctxt (style, t) =
haftmann@32898
   479
  pretty_term ctxt (style t);
wenzelm@22124
   480
haftmann@32898
   481
fun pretty_thm_style ctxt (style, th) =
haftmann@32898
   482
  pretty_term ctxt (style (Thm.full_prop_of th));
haftmann@32898
   483
haftmann@32898
   484
fun pretty_term_typ ctxt (style, t) =
haftmann@32898
   485
  let val t' = style t
wenzelm@39541
   486
  in pretty_term ctxt (Type.constraint (Term.fastype_of t') t') end;
haftmann@32898
   487
haftmann@32898
   488
fun pretty_term_typeof ctxt (style, t) =
haftmann@32898
   489
  Syntax.pretty_typ ctxt (Term.fastype_of (style t));
wenzelm@22124
   490
wenzelm@25373
   491
fun pretty_const ctxt c =
wenzelm@25373
   492
  let
wenzelm@25373
   493
    val t = Const (c, Consts.type_scheme (ProofContext.consts_of ctxt) c)
wenzelm@25373
   494
      handle TYPE (msg, _, _) => error msg;
wenzelm@25373
   495
    val ([t'], _) = Variable.import_terms true [t] ctxt;
wenzelm@25373
   496
  in pretty_term ctxt t' end;
wenzelm@22124
   497
wenzelm@25407
   498
fun pretty_abbrev ctxt s =
wenzelm@22124
   499
  let
wenzelm@25407
   500
    val t = Syntax.parse_term ctxt s |> singleton (ProofContext.standard_infer_types ctxt);
wenzelm@24920
   501
    fun err () = error ("Abbreviated constant expected: " ^ Syntax.string_of_term ctxt t);
wenzelm@22124
   502
    val (head, args) = Term.strip_comb t;
wenzelm@22124
   503
    val (c, T) = Term.dest_Const head handle TERM _ => err ();
wenzelm@25054
   504
    val (U, u) = Consts.the_abbreviation (ProofContext.consts_of ctxt) c
wenzelm@22124
   505
      handle TYPE _ => err ();
wenzelm@22124
   506
    val t' = Term.betapplys (Envir.expand_atom T (U, u), args);
haftmann@32898
   507
    val eq = Logic.mk_equals (t, t');
haftmann@32898
   508
    val ctxt' = Variable.auto_fixes eq ctxt;
haftmann@32898
   509
  in ProofContext.pretty_term_abbrev ctxt' eq end;
wenzelm@22124
   510
haftmann@39551
   511
fun pretty_class ctxt =
haftmann@39551
   512
  Pretty.str o Type.extern_class (ProofContext.tsig_of ctxt) o ProofContext.read_class ctxt;
haftmann@39551
   513
haftmann@39555
   514
fun pretty_type ctxt s =
wenzelm@39937
   515
  let val Type (name, _) = ProofContext.read_type_name_proper ctxt false s
wenzelm@39937
   516
  in Pretty.str (Type.extern_type (ProofContext.tsig_of ctxt) name) end;
haftmann@39551
   517
wenzelm@33388
   518
fun pretty_prf full ctxt = Proof_Syntax.pretty_proof_of ctxt full;
wenzelm@22124
   519
wenzelm@22124
   520
fun pretty_theory ctxt name =
wenzelm@22124
   521
  (Theory.requires (ProofContext.theory_of ctxt) name "presentation"; Pretty.str name);
wenzelm@22124
   522
wenzelm@22124
   523
wenzelm@30390
   524
(* default output *)
wenzelm@22124
   525
wenzelm@36969
   526
val str_of_source = space_implode " " o map Token.unparse o #2 o #1 o Args.dest_src;
wenzelm@30390
   527
wenzelm@39042
   528
fun maybe_pretty_source pretty ctxt src xs =
wenzelm@39042
   529
  map (pretty ctxt) xs  (*always pretty in order to exhibit errors!*)
wenzelm@39042
   530
  |> (if Config.get ctxt source then K [pretty_text ctxt (str_of_source src)] else I);
wenzelm@30390
   531
wenzelm@39042
   532
fun output ctxt prts =
wenzelm@30390
   533
  prts
wenzelm@39042
   534
  |> (if Config.get ctxt quotes then map Pretty.quote else I)
wenzelm@39042
   535
  |> (if Config.get ctxt display then
wenzelm@39042
   536
    map (Output.output o Pretty.string_of o Pretty.indent (Config.get ctxt indent))
wenzelm@22124
   537
    #> space_implode "\\isasep\\isanewline%\n"
wenzelm@22124
   538
    #> enclose "\\begin{isabelle}%\n" "%\n\\end{isabelle}"
wenzelm@22124
   539
  else
wenzelm@39042
   540
    map (Output.output o (if Config.get ctxt break then Pretty.string_of else Pretty.str_of))
wenzelm@22124
   541
    #> space_implode "\\isasep\\isanewline%\n"
wenzelm@22124
   542
    #> enclose "\\isa{" "}");
wenzelm@22124
   543
wenzelm@30390
   544
wenzelm@30390
   545
wenzelm@30390
   546
(** concrete antiquotations **)
wenzelm@30390
   547
wenzelm@30390
   548
(* basic entities *)
wenzelm@30390
   549
wenzelm@30390
   550
local
wenzelm@30390
   551
wenzelm@30390
   552
fun basic_entities name scan pretty = antiquotation name scan
wenzelm@39042
   553
  (fn {source, context, ...} => output context o maybe_pretty_source pretty context source);
wenzelm@30390
   554
haftmann@32890
   555
fun basic_entities_style name scan pretty = antiquotation name scan
haftmann@32890
   556
  (fn {source, context, ...} => fn (style, xs) =>
wenzelm@39042
   557
    output context
wenzelm@39042
   558
      (maybe_pretty_source (fn ctxt => fn x => pretty ctxt (style, x)) context source xs));
haftmann@32890
   559
wenzelm@30390
   560
fun basic_entity name scan = basic_entities name (scan >> single);
wenzelm@30390
   561
wenzelm@30390
   562
in
wenzelm@30390
   563
haftmann@32890
   564
val _ = basic_entities_style "thm" (Term_Style.parse -- Attrib.thms) pretty_thm_style;
haftmann@32890
   565
val _ = basic_entity "prop" (Term_Style.parse -- Args.prop) pretty_term_style;
haftmann@32890
   566
val _ = basic_entity "term" (Term_Style.parse -- Args.term) pretty_term_style;
haftmann@32898
   567
val _ = basic_entity "term_type" (Term_Style.parse -- Args.term) pretty_term_typ;
haftmann@32898
   568
val _ = basic_entity "typeof" (Term_Style.parse -- Args.term) pretty_term_typeof;
wenzelm@35399
   569
val _ = basic_entity "const" (Args.const_proper false) pretty_const;
wenzelm@30390
   570
val _ = basic_entity "abbrev" (Scan.lift Args.name_source) pretty_abbrev;
wenzelm@30390
   571
val _ = basic_entity "typ" Args.typ_abbrev Syntax.pretty_typ;
haftmann@39551
   572
val _ = basic_entity "class" (Scan.lift Args.name) pretty_class;
haftmann@39551
   573
val _ = basic_entity "type" (Scan.lift Args.name) pretty_type;
wenzelm@39042
   574
val _ = basic_entity "text" (Scan.lift Args.name) pretty_text;
wenzelm@30390
   575
val _ = basic_entities "prf" Attrib.thms (pretty_prf false);
wenzelm@30390
   576
val _ = basic_entities "full_prf" Attrib.thms (pretty_prf true);
wenzelm@30390
   577
val _ = basic_entity "theory" (Scan.lift Args.name) pretty_theory;
wenzelm@30390
   578
haftmann@32898
   579
val _ = basic_entities_style "thm_style" (Term_Style.parse_bare -- Attrib.thms) pretty_thm_style;
haftmann@32898
   580
val _ = basic_entity "term_style" (Term_Style.parse_bare -- Args.term) pretty_term_style;
haftmann@32898
   581
wenzelm@30390
   582
end;
wenzelm@30390
   583
wenzelm@30390
   584
wenzelm@30390
   585
(* goal state *)
wenzelm@30390
   586
wenzelm@30390
   587
local
wenzelm@22124
   588
wenzelm@30367
   589
fun proof_state state =
wenzelm@30367
   590
  (case try Toplevel.proof_of state of
wenzelm@30367
   591
    SOME prf => prf
wenzelm@22124
   592
  | _ => error "No proof state");
wenzelm@22124
   593
wenzelm@30390
   594
fun goal_state name main_goal = antiquotation name (Scan.succeed ())
wenzelm@39371
   595
  (fn {state, context = ctxt, ...} => fn () => output ctxt
wenzelm@39375
   596
    [Pretty.chunks
wenzelm@39375
   597
      (Proof.pretty_goals main_goal (Proof.map_context (K ctxt) (proof_state state)))]);
wenzelm@22124
   598
wenzelm@30390
   599
in
wenzelm@22124
   600
wenzelm@30390
   601
val _ = goal_state "goals" true;
wenzelm@30390
   602
val _ = goal_state "subgoals" false;
wenzelm@22124
   603
wenzelm@30390
   604
end;
wenzelm@22124
   605
wenzelm@27522
   606
wenzelm@30390
   607
(* embedded lemma *)
wenzelm@27522
   608
wenzelm@36950
   609
val _ = Keyword.keyword "by";
wenzelm@27381
   610
wenzelm@30390
   611
val _ = antiquotation "lemma"
wenzelm@30390
   612
  (Args.prop -- Scan.lift (Args.$$$ "by" |-- Method.parse -- Scan.option Method.parse))
wenzelm@30390
   613
  (fn {source, context, ...} => fn (prop, methods) =>
wenzelm@30390
   614
    let
wenzelm@30390
   615
      val prop_src =
wenzelm@30390
   616
        (case Args.dest_src source of ((a, arg :: _), pos) => Args.src ((a, [arg]), pos));
wenzelm@30390
   617
      val _ = context
wenzelm@36334
   618
        |> Proof.theorem NONE (K I) [[(prop, [])]]
wenzelm@30390
   619
        |> Proof.global_terminal_proof methods;
wenzelm@39042
   620
    in output context (maybe_pretty_source pretty_term context prop_src [prop]) end);
wenzelm@30390
   621
wenzelm@30390
   622
wenzelm@30390
   623
(* ML text *)
wenzelm@30390
   624
wenzelm@30390
   625
local
wenzelm@30390
   626
wenzelm@30390
   627
fun ml_text name ml = antiquotation name (Scan.lift Args.name_source_position)
wenzelm@30390
   628
  (fn {context, ...} => fn (txt, pos) =>
wenzelm@37199
   629
   (ML_Context.eval_in (SOME context) false pos (ml pos txt);
wenzelm@30576
   630
    Symbol_Pos.content (Symbol_Pos.explode (txt, pos))
wenzelm@39042
   631
    |> (if Config.get context quotes then quote else I)
wenzelm@39042
   632
    |> (if Config.get context display then enclose "\\begin{verbatim}\n" "\n\\end{verbatim}"
wenzelm@30390
   633
    else
wenzelm@30390
   634
      split_lines
wenzelm@30390
   635
      #> map (space_implode "\\verb,|," o map (enclose "\\verb|" "|") o space_explode "|")
wenzelm@30390
   636
      #> space_implode "\\isasep\\isanewline%\n")));
wenzelm@30390
   637
wenzelm@37199
   638
fun ml_enclose bg en pos txt =
wenzelm@37199
   639
  ML_Lex.read Position.none bg @ ML_Lex.read pos txt @ ML_Lex.read Position.none en;
wenzelm@37199
   640
wenzelm@30390
   641
in
wenzelm@30390
   642
wenzelm@37199
   643
val _ = ml_text "ML" (ml_enclose "fn _ => (" ");");
wenzelm@37199
   644
val _ = ml_text "ML_type" (ml_enclose "val _ = NONE : (" ") option;");
wenzelm@37199
   645
val _ = ml_text "ML_struct" (ml_enclose "functor XXX() = struct structure XX = " " end;");
wenzelm@37199
   646
wenzelm@37199
   647
val _ = ml_text "ML_functor"   (* FIXME formal treatment of functor name (!?) *)
wenzelm@37199
   648
  (fn pos => fn txt =>
wenzelm@37199
   649
    ML_Lex.read Position.none ("ML_Env.check_functor " ^
wenzelm@37199
   650
      ML_Syntax.print_string (Symbol_Pos.content (Symbol_Pos.explode (txt, pos)))));
wenzelm@37199
   651
wenzelm@37199
   652
val _ = ml_text "ML_text" (K (K []));
wenzelm@22124
   653
wenzelm@22124
   654
end;
wenzelm@30390
   655
wenzelm@30390
   656
end;