Merge.
1 (* Title: Pure/Thy/thy_output.ML
2 Author: Markus Wenzel, TU Muenchen
4 Theory document output.
14 val add_commands: (string * (Args.src -> Toplevel.node option -> string)) list -> unit
15 val add_options: (string * (string -> (unit -> string) -> unit -> string)) list -> unit
16 val defined_command: string -> bool
17 val defined_option: string -> bool
18 val print_antiquotations: unit -> unit
19 val boolean: string -> bool
20 val integer: string -> int
21 val args: (Context.generic * Args.T list -> 'a * (Context.generic * Args.T list)) ->
22 (Args.src -> Proof.context -> 'a -> string) -> Args.src -> Toplevel.node option -> string
23 datatype markup = Markup | MarkupEnv | Verbatim
24 val modes: string list ref
25 val eval_antiquote: Scan.lexicon -> Toplevel.node option -> SymbolPos.text * Position.T -> string
26 val present_thy: Scan.lexicon -> (string -> string list) -> (markup -> string -> bool) ->
27 (Toplevel.transition * Toplevel.state) list -> (OuterLex.token, 'a) Source.source -> Buffer.T
28 val str_of_source: Args.src -> string
29 val pretty_text: string -> Pretty.T
30 val pretty_term: Proof.context -> term -> Pretty.T
31 val pretty_thm: Proof.context -> thm -> Pretty.T
32 val output_list: (Proof.context -> 'a -> Pretty.T) -> Args.src ->
33 Proof.context -> 'a list -> string
34 val output: (Proof.context -> 'a -> Pretty.T) -> Args.src -> Proof.context -> 'a -> string
37 structure ThyOutput: THY_OUTPUT =
40 structure T = OuterLex;
41 structure P = OuterParse;
44 (** global options **)
47 val display = ref false;
48 val quotes = ref false;
50 val source = ref false;
51 val break = ref false;
55 (** maintain global commands **)
60 ref (Symtab.empty: (Args.src -> Toplevel.node option -> string) Symtab.table);
63 ref (Symtab.empty: (string -> (unit -> string) -> unit -> string) Symtab.table);
65 fun add_item kind (name, x) tab =
66 (if not (Symtab.defined tab name) then ()
67 else warning ("Redefined document antiquotation " ^ kind ^ ": " ^ quote name);
68 Symtab.update (name, x) tab);
72 fun add_commands xs = CRITICAL (fn () => change global_commands (fold (add_item "command") xs));
73 fun add_options xs = CRITICAL (fn () => change global_options (fold (add_item "option") xs));
75 fun defined_command name = Symtab.defined (! global_commands) name;
76 fun defined_option name = Symtab.defined (! global_options) name;
79 let val ((name, _), pos) = Args.dest_src src in
80 (case Symtab.lookup (! global_commands) name of
81 NONE => error ("Unknown document antiquotation command: " ^ quote name ^ Position.str_of pos)
82 | SOME f => (Position.report (Markup.doc_antiq name) pos; f src))
85 fun option (name, s) f () =
86 (case Symtab.lookup (! global_options) name of
87 NONE => error ("Unknown document antiquotation option: " ^ quote name)
88 | SOME opt => opt s f ());
91 | options (opt :: opts) f = option opt (options opts f);
94 fun print_antiquotations () =
95 [Pretty.big_list "document antiquotation commands:"
96 (map Pretty.str (sort_strings (Symtab.keys (! global_commands)))),
97 Pretty.big_list "document antiquotation options:"
98 (map Pretty.str (sort_strings (Symtab.keys (! global_options))))]
99 |> Pretty.chunks |> Pretty.writeln;
105 (** syntax of antiquotations **)
109 fun boolean "" = true
110 | boolean "true" = true
111 | boolean "false" = false
112 | boolean s = error ("Bad boolean value: " ^ quote s);
117 (case Library.read_int ss of (i, []) => i
118 | _ => error ("Bad integer value: " ^ quote s));
119 in (case Symbol.explode s of "-" :: ss => ~ (int ss) | ss => int ss) end;
124 fun syntax scan = Args.context_syntax "document antiquotation" scan;
126 fun args scan f src node : string =
128 val loc = if ! locale = "" then NONE else SOME (! locale);
129 val (x, ctxt) = syntax scan src (Toplevel.presentation_context node loc);
137 val property = P.xname -- Scan.optional (P.$$$ "=" |-- P.!!! P.xname) "";
138 val properties = Scan.optional (P.$$$ "[" |-- P.!!! (P.enum "," property --| P.$$$ "]")) [];
142 val antiq = P.!!! (P.position P.xname -- properties -- Args.parse --| Scan.ahead P.eof)
143 >> (fn (((x, pos), y), z) => (y, Args.src ((x, z), pos)));
150 val modes = ref ([]: string list);
152 fun eval_antiquote lex node (txt, pos) =
154 fun expand (Antiquote.Text s) = s
155 | expand (Antiquote.Antiq x) =
156 let val (opts, src) = Antiquote.read_antiq lex antiq x in
157 options opts (fn () => command src node) (); (*preview errors!*)
158 PrintMode.with_modes (! modes @ Latex.modes)
159 (Output.no_warnings (options opts (fn () => command src node))) ()
161 | expand (Antiquote.Open _) = ""
162 | expand (Antiquote.Close _) = "";
163 val ants = Antiquote.read (SymbolPos.explode (txt, pos), pos);
165 if is_none node andalso exists Antiquote.is_antiq ants then
166 error ("Unknown context -- cannot expand document antiquotations" ^ Position.str_of pos)
167 else implode (map expand ants)
172 (** present theory source **)
174 (*NB: arranging white space around command spans is a black art.*)
176 (* presentation tokens *)
180 | BasicToken of T.token
181 | MarkupToken of string * (string * Position.T)
182 | MarkupEnvToken of string * (string * Position.T)
183 | VerbatimToken of string * Position.T;
185 fun output_token lex state =
187 val eval = eval_antiquote lex (try Toplevel.node_of state)
190 | BasicToken tok => Latex.output_basic tok
191 | MarkupToken (cmd, txt) => Latex.output_markup cmd (eval txt)
192 | MarkupEnvToken (cmd, txt) => Latex.output_markup_env cmd (eval txt)
193 | VerbatimToken txt => Latex.output_verbatim (eval txt)
196 fun basic_token pred (BasicToken tok) = pred tok
197 | basic_token _ _ = false;
199 val improper_token = basic_token (not o T.is_proper);
200 val comment_token = basic_token T.is_comment;
201 val blank_token = basic_token T.is_blank;
202 val newline_token = basic_token T.is_newline;
207 type command = string * Position.T * string list; (*name, position, tags*)
208 type source = (token * (string * int)) list; (*token, markup flag, meta-comment depth*)
210 datatype span = Span of command * (source * source * source * source) * bool;
212 fun make_span cmd src =
214 fun take_newline (tok :: toks) =
215 if newline_token (fst tok) then ([tok], toks, true)
216 else ([], tok :: toks, false)
217 | take_newline [] = ([], [], false);
218 val (((src_prefix, src_main), src_suffix1), (src_suffix2, src_appendix, newline)) =
220 |> take_prefix (improper_token o fst)
221 ||>> take_suffix (improper_token o fst)
222 ||>> take_prefix (comment_token o fst)
224 in Span (cmd, (src_prefix, src_main, src_suffix1 @ src_suffix2, src_appendix), newline) end;
231 fun err_bad_nesting pos =
232 error ("Bad nesting of commands in presentation" ^ pos);
234 fun edge which f (x: string option, y) =
236 else (case which (x, y) of NONE => I | SOME txt => Buffer.add (f txt));
238 val begin_tag = edge #2 Latex.begin_tag;
239 val end_tag = edge #1 Latex.end_tag;
240 fun open_delim delim e = edge #2 Latex.begin_delim e #> delim #> edge #2 Latex.end_delim e;
241 fun close_delim delim e = edge #1 Latex.begin_delim e #> delim #> edge #1 Latex.end_delim e;
245 fun present_span lex default_tags span state state'
246 (tag_stack, active_tag, newline, buffer, present_cont) =
248 val present = fold (fn (tok, (flag, 0)) =>
249 Buffer.add (output_token lex state' tok)
253 val Span ((cmd_name, cmd_pos, cmd_tags), srcs, span_newline) = span;
255 val (tag, tags) = tag_stack;
256 val tag' = try hd (fold OuterKeyword.update_tags cmd_tags (the_list tag));
259 if is_some tag' then tag'
260 else if cmd_name = "end" andalso not (Toplevel.is_toplevel state') then NONE
261 else try hd (default_tags cmd_name);
262 val edge = (active_tag, active_tag');
265 if is_none active_tag' then span_newline else newline;
267 val nesting = Toplevel.level state' - Toplevel.level state;
269 if nesting = 0 andalso not (Toplevel.is_proof state) then tag_stack
270 else if nesting >= 0 then (tag', replicate nesting tag @ tags)
272 (case Library.drop (~ nesting - 1, tags) of
273 tgs :: tgss => (tgs, tgss)
274 | [] => err_bad_nesting (Position.str_of cmd_pos));
279 |> close_delim (fst present_cont) edge
281 |> open_delim (present (#1 srcs)) edge
283 |> present (#2 srcs);
285 if newline then (present (#3 srcs), present (#4 srcs))
286 else (I, present (#3 srcs) #> present (#4 srcs));
287 in (tag_stack', active_tag', newline', buffer', present_cont') end;
289 fun present_trailer ((_, tags), active_tag, _, buffer, present_cont) =
290 if not (null tags) then err_bad_nesting " at end of theory"
293 |> end_tag (active_tag, NONE)
294 |> close_delim (fst present_cont) (active_tag, NONE)
302 datatype markup = Markup | MarkupEnv | Verbatim;
307 Scan.one T.is_blank -- Scan.many T.is_comment -- Scan.one T.is_proper;
309 val is_improper = not o (T.is_proper orf T.is_begin_ignore orf T.is_end_ignore);
310 val improper = Scan.many is_improper;
311 val improper_end = Scan.repeat (Scan.unless space_proper (Scan.one is_improper));
312 val blank_end = Scan.repeat (Scan.unless space_proper (Scan.one T.is_blank));
314 val opt_newline = Scan.option (Scan.one T.is_newline);
317 Scan.depend (fn d => opt_newline |-- Scan.one T.is_begin_ignore
319 Scan.depend (fn d => Scan.one T.is_end_ignore --|
320 (if d = 0 then Scan.fail_with (K "Bad nesting of meta-comments") else opt_newline)
323 val tag = (improper -- P.$$$ "%" -- improper) |-- P.!!! (P.tag_name --| blank_end);
326 Scan.option ((P.$$$ "(" -- improper -- P.$$$ "in") |--
327 P.!!! (improper |-- P.xname --| (improper -- P.$$$ ")")));
331 fun present_thy lex default_tags is_markup command_results src =
335 val ignored = Scan.state --| ignore
336 >> (fn d => (NONE, (NoToken, ("", d))));
338 fun markup mark mk flag = Scan.peek (fn d =>
339 improper |-- P.position (Scan.one (T.is_kind T.Command andf is_markup mark o T.content_of)) --
341 P.!!!! ((improper -- locale -- improper) |-- P.doc_source --| improper_end)
342 >> (fn (((tok, pos), tags), txt) =>
343 let val name = T.content_of tok
344 in (SOME (name, pos, tags), (mk (name, txt), (flag, d))) end));
346 val command = Scan.peek (fn d =>
347 P.position (Scan.one (T.is_kind T.Command)) --
349 >> (fn ((tok, pos), tags) =>
350 let val name = T.content_of tok
351 in (SOME (name, pos, tags), (BasicToken tok, (Latex.markup_false, d))) end));
353 val cmt = Scan.peek (fn d =>
354 P.$$$ "--" |-- P.!!!! (improper |-- P.doc_source)
355 >> (fn txt => (NONE, (MarkupToken ("cmt", txt), ("", d)))));
357 val other = Scan.peek (fn d =>
358 P.not_eof >> (fn tok => (NONE, (BasicToken tok, ("", d)))));
362 markup Markup MarkupToken Latex.markup_true ||
363 markup MarkupEnv MarkupEnvToken Latex.markup_true ||
364 markup Verbatim (VerbatimToken o #2) "" ||
365 command || cmt || other;
370 val is_eof = fn (_, (BasicToken x, _)) => T.is_eof x | _ => false;
371 val stopper = Scan.stopper (K (NONE, (BasicToken T.eof, ("", 0)))) is_eof;
373 val cmd = Scan.one (is_some o fst);
374 val non_cmd = Scan.one (is_none o fst andf not o is_eof) >> #2;
376 val comments = Scan.many (comment_token o fst o snd);
377 val blank = Scan.one (blank_token o fst o snd);
378 val newline = Scan.one (newline_token o fst o snd);
380 Scan.option (newline -- comments) --
381 Scan.option (newline -- comments) --
382 Scan.option (blank -- comments) -- cmd;
385 Scan.repeat non_cmd -- cmd --
386 Scan.repeat (Scan.unless before_cmd non_cmd) --
387 Scan.option (newline >> (single o snd))
388 >> (fn (((toks1, (cmd, tok2)), toks3), tok4) =>
389 make_span (the cmd) (toks1 @ (tok2 :: (toks3 @ the_default [] tok4))));
393 |> Source.filter (not o T.is_semicolon)
394 |> Source.source' 0 T.stopper (Scan.error (Scan.bulk token)) NONE
395 |> Source.source stopper (Scan.error (Scan.bulk span)) NONE
399 (* present commands *)
401 fun present_command tr span st st' =
402 Toplevel.setmp_thread_position tr (present_span lex default_tags span st st');
405 | present st (((tr, st'), span) :: rest) = present_command tr span st st' #> present st' rest;
407 if length command_results = length spans then
408 ((NONE, []), NONE, true, Buffer.empty, (I, I))
409 |> present Toplevel.toplevel (command_results ~~ spans)
411 else error "Messed-up outer syntax for presentation"
418 (** setup default output **)
423 [("show_types", Library.setmp Syntax.show_types o boolean),
424 ("show_sorts", Library.setmp Syntax.show_sorts o boolean),
425 ("show_structs", Library.setmp show_structs o boolean),
426 ("show_question_marks", Library.setmp show_question_marks o boolean),
427 ("long_names", Library.setmp NameSpace.long_names o boolean),
428 ("short_names", Library.setmp NameSpace.short_names o boolean),
429 ("unique_names", Library.setmp NameSpace.unique_names o boolean),
430 ("eta_contract", Library.setmp Syntax.eta_contract o boolean),
431 ("locale", Library.setmp locale),
432 ("display", Library.setmp display o boolean),
433 ("break", Library.setmp break o boolean),
434 ("quotes", Library.setmp quotes o boolean),
435 ("mode", fn s => fn f => PrintMode.with_modes [s] f),
436 ("margin", Pretty.setmp_margin o integer),
437 ("indent", Library.setmp indent o integer),
438 ("source", Library.setmp source o boolean),
439 ("goals_limit", Library.setmp goals_limit o integer)];
442 (* basic pretty printing *)
444 val str_of_source = space_implode " " o map T.unparse o #2 o #1 o Args.dest_src;
447 if ! display then s else Symbol.strip_blanks s;
449 val pretty_text = Pretty.chunks o map Pretty.str o map tweak_line o Library.split_lines;
451 fun pretty_term ctxt t = Syntax.pretty_term (Variable.auto_fixes t ctxt) t;
453 fun pretty_term_abbrev ctxt t = ProofContext.pretty_term_abbrev (Variable.auto_fixes t ctxt) t;
455 fun pretty_term_typ ctxt t =
456 Syntax.pretty_term ctxt (TypeInfer.constrain (Term.fastype_of t) t);
458 fun pretty_term_typeof ctxt = Syntax.pretty_typ ctxt o Term.fastype_of;
460 fun pretty_const ctxt c =
462 val t = Const (c, Consts.type_scheme (ProofContext.consts_of ctxt) c)
463 handle TYPE (msg, _, _) => error msg;
464 val ([t'], _) = Variable.import_terms true [t] ctxt;
465 in pretty_term ctxt t' end;
467 fun pretty_abbrev ctxt s =
469 val t = Syntax.parse_term ctxt s |> singleton (ProofContext.standard_infer_types ctxt);
470 fun err () = error ("Abbreviated constant expected: " ^ Syntax.string_of_term ctxt t);
471 val (head, args) = Term.strip_comb t;
472 val (c, T) = Term.dest_Const head handle TERM _ => err ();
473 val (U, u) = Consts.the_abbreviation (ProofContext.consts_of ctxt) c
474 handle TYPE _ => err ();
475 val t' = Term.betapplys (Envir.expand_atom T (U, u), args);
476 in pretty_term_abbrev ctxt (Logic.mk_equals (t, t')) end;
478 fun pretty_thm ctxt = pretty_term ctxt o Thm.full_prop_of;
480 fun pretty_term_style ctxt (name, t) =
481 pretty_term ctxt (TermStyle.the_style (ProofContext.theory_of ctxt) name ctxt t);
483 fun pretty_thm_style ctxt (name, th) =
484 pretty_term_style ctxt (name, Thm.full_prop_of th);
486 fun pretty_prf full ctxt thms =
487 Pretty.chunks (map (ProofSyntax.pretty_proof_of ctxt full) thms);
489 fun pretty_theory ctxt name =
490 (Theory.requires (ProofContext.theory_of ctxt) name "presentation"; Pretty.str name);
495 fun output_list pretty src ctxt xs =
496 map (pretty ctxt) xs (*always pretty in order to exhibit errors!*)
497 |> (if ! source then K [pretty_text (str_of_source src)] else I)
498 |> (if ! quotes then map Pretty.quote else I)
499 |> (if ! display then
500 map (Output.output o Pretty.string_of o Pretty.indent (! indent))
501 #> space_implode "\\isasep\\isanewline%\n"
502 #> enclose "\\begin{isabelle}%\n" "%\n\\end{isabelle}"
504 map (Output.output o (if ! break then Pretty.string_of else Pretty.str_of))
505 #> space_implode "\\isasep\\isanewline%\n"
506 #> enclose "\\isa{" "}");
508 fun output pretty src ctxt = output_list pretty src ctxt o single;
510 fun proof_state node =
511 (case Option.map Toplevel.proof_node node of
512 SOME (SOME prf) => ProofNode.current prf
513 | _ => error "No proof state");
515 fun output_goals main_goal src node = args (Scan.succeed ()) (output (fn _ => fn _ =>
516 Pretty.chunks (Proof.pretty_goals main_goal (proof_state node)))) src node;
518 fun ml_val txt = "fn _ => (" ^ txt ^ ");";
519 fun ml_type txt = "val _ = NONE : (" ^ txt ^ ") option;";
520 fun ml_struct txt = "functor DUMMY_FUNCTOR() = struct structure DUMMY = " ^ txt ^ " end;"
522 fun output_ml ml _ ctxt (txt, pos) =
523 (ML_Context.eval_in (SOME ctxt) false pos (ml txt);
524 SymbolPos.content (SymbolPos.explode (txt, pos))
525 |> (if ! quotes then quote else I)
526 |> (if ! display then enclose "\\begin{verbatim}\n" "\n\\end{verbatim}"
529 #> map (space_implode "\\verb,|," o map (enclose "\\verb|" "|") o space_explode "|")
530 #> space_implode "\\isasep\\isanewline%\n"));
533 (* embedded lemmas *)
535 fun pretty_lemma ctxt (prop, methods) =
538 |> Proof.theorem_i NONE (K I) [[(prop, [])]]
539 |> Proof.global_terminal_proof methods;
540 in pretty_term ctxt prop end;
543 args (Args.prop -- Scan.lift (Args.$$$ "by" |-- Method.parse -- Scan.option Method.parse))
544 (output pretty_lemma o (fn ((a, arg :: _), p) => (Args.src ((a, [arg]), p))) o Args.dest_src);
549 val _ = OuterKeyword.keyword "by";
552 [("thm", args Attrib.thms (output_list pretty_thm)),
553 ("thm_style", args (Scan.lift Args.liberal_name -- Attrib.thm) (output pretty_thm_style)),
554 ("prop", args Args.prop (output pretty_term)),
555 ("lemma", embedded_lemma),
556 ("term", args Args.term (output pretty_term)),
557 ("term_style", args (Scan.lift Args.liberal_name -- Args.term) (output pretty_term_style)),
558 ("term_type", args Args.term (output pretty_term_typ)),
559 ("typeof", args Args.term (output pretty_term_typeof)),
560 ("const", args Args.const_proper (output pretty_const)),
561 ("abbrev", args (Scan.lift Args.name_source) (output pretty_abbrev)),
562 ("typ", args Args.typ_abbrev (output Syntax.pretty_typ)),
563 ("text", args (Scan.lift Args.name) (output (K pretty_text))),
564 ("goals", output_goals true),
565 ("subgoals", output_goals false),
566 ("prf", args Attrib.thms (output (pretty_prf false))),
567 ("full_prf", args Attrib.thms (output (pretty_prf true))),
568 ("theory", args (Scan.lift Args.name) (output pretty_theory)),
569 ("ML", args (Scan.lift Args.name_source_position) (output_ml ml_val)),
570 ("ML_type", args (Scan.lift Args.name_source_position) (output_ml ml_type)),
571 ("ML_struct", args (Scan.lift Args.name_source_position) (output_ml ml_struct))];