doc-src/more_antiquote.ML
author haftmann
Fri Jan 22 13:38:28 2010 +0100 (2010-01-22)
changeset 34944 970e1466028d
parent 34896 a22b09addd78
child 35246 bcbb5ba7dbbc
permissions -rw-r--r--
code literals: distinguish numeral classes by different entries
     1 (*  Title:      doc-src/more_antiquote.ML
     2     Author:     Florian Haftmann, TU Muenchen
     3 
     4 More antiquotations.
     5 *)
     6 
     7 signature MORE_ANTIQUOTE =
     8 sig
     9   val typewriter: string -> string
    10 end;
    11 
    12 structure More_Antiquote : MORE_ANTIQUOTE =
    13 struct
    14 
    15 (* printing typewriter lines *)
    16 
    17 fun typewriter s =
    18   let
    19     val parse = Scan.repeat
    20       (Scan.this_string "\n" |-- Scan.succeed "\\\\\n\\hspace*{0pt}"
    21         || (Scan.this_string " "
    22           || Scan.this_string "."
    23           || Scan.this_string ","
    24           || Scan.this_string ":"
    25           || Scan.this_string ";"
    26           || Scan.this_string "\"" |-- Scan.succeed "{\\char34}"
    27           || Scan.this_string "#" |-- Scan.succeed "{\\char35}"
    28           || Scan.this_string "$" |-- Scan.succeed "{\\char36}"
    29           || Scan.this_string "%" |-- Scan.succeed "{\\char37}"
    30           || Scan.this_string "&" |-- Scan.succeed "{\\char38}"
    31           || Scan.this_string "\\" |-- Scan.succeed "{\\char92}"
    32           || Scan.this_string "^" |-- Scan.succeed "{\\char94}"
    33           || Scan.this_string "_" |-- Scan.succeed "{\\char95}"
    34           || Scan.this_string "{" |-- Scan.succeed "{\\char123}"
    35           || Scan.this_string "}" |-- Scan.succeed "{\\char125}"
    36           || Scan.this_string "~" |-- Scan.succeed "{\\char126}")
    37           -- Scan.repeat (Scan.this_string " ")
    38           >> (fn (x, xs) => x ^ replicate_string (length xs) "~")
    39         || Scan.one Symbol.not_eof);
    40     fun is_newline s = (s = "\n");
    41     val cs = explode s |> take_prefix is_newline |> snd
    42       |> take_suffix is_newline |> fst;
    43     val (texts, []) =  Scan.finite Symbol.stopper parse cs
    44   in if null texts
    45     then ""
    46     else implode ("\\isatypewriter%\n\\noindent%\n\\hspace*{0pt}" :: texts)
    47   end
    48 
    49 
    50 (* class and type constructor antiquotation *)
    51 
    52 local
    53 
    54 val pr_text = enclose "\\isa{" "}" o Pretty.output o Pretty.str;
    55 
    56 fun pr_class ctxt = Sign.read_class (ProofContext.theory_of ctxt)
    57   #> Sign.extern_class (ProofContext.theory_of ctxt)
    58   #> pr_text;
    59 
    60 fun pr_type ctxt = Sign.intern_type (ProofContext.theory_of ctxt)
    61   #> tap (Sign.arity_number (ProofContext.theory_of ctxt))
    62   #> Sign.extern_type (ProofContext.theory_of ctxt)
    63   #> pr_text;
    64 
    65 in
    66 
    67 val _ = ThyOutput.antiquotation "class" (Scan.lift Args.name) (pr_class o #context);
    68 val _ = ThyOutput.antiquotation "type" (Scan.lift Args.name) (pr_type o #context);
    69 
    70 end;
    71 
    72 
    73 (* code theorem antiquotation *)
    74 
    75 local
    76 
    77 fun pretty_term ctxt t = Syntax.pretty_term (Variable.auto_fixes t ctxt) t;
    78 
    79 fun pretty_thm ctxt = pretty_term ctxt o Thm.full_prop_of;
    80 
    81 fun no_vars ctxt thm =
    82   let
    83     val ctxt' = Variable.set_body false ctxt;
    84     val ((_, [thm]), _) = Variable.import true [thm] ctxt';
    85   in thm end;
    86 
    87 fun pretty_code_thm src ctxt raw_const =
    88   let
    89     val thy = ProofContext.theory_of ctxt;
    90     val const = Code.check_const thy raw_const;
    91     val (_, eqngr) = Code_Preproc.obtain thy [const] [];
    92     fun holize thm = @{thm meta_eq_to_obj_eq} OF [thm];
    93     val thms = Code_Preproc.cert eqngr const
    94       |> Code.equations_thms_cert thy
    95       |> snd
    96       |> map_filter (fn (_, (thm, proper)) => if proper then SOME thm else NONE)
    97       |> map (holize o no_vars ctxt o AxClass.overload thy);
    98   in ThyOutput.output (ThyOutput.maybe_pretty_source (pretty_thm ctxt) src thms) end;
    99 
   100 in
   101 
   102 val _ = ThyOutput.antiquotation "code_thms" Args.term
   103   (fn {source, context, ...} => pretty_code_thm source context);
   104 
   105 end;
   106 
   107 
   108 (* code antiquotation *)
   109 
   110 local
   111 
   112   val parse_const_terms = Scan.repeat1 Args.term
   113     >> (fn ts => fn thy => map (Code.check_const thy) ts);
   114   val parse_consts = Scan.lift (Args.parens (Args.$$$ "consts")) |-- parse_const_terms
   115     >> (fn mk_cs => fn thy => fn naming => map_filter (Code_Thingol.lookup_const naming) (mk_cs thy));
   116   val parse_types = Scan.lift (Args.parens (Args.$$$ "types") |-- Scan.repeat1 Args.name)
   117     >> (fn tycos => fn thy => fn naming => map_filter (Code_Thingol.lookup_tyco naming o Sign.intern_type thy) tycos);
   118   val parse_classes = Scan.lift (Args.parens (Args.$$$ "classes") |-- Scan.repeat1 Args.name)
   119     >> (fn classes => fn thy => fn naming => map_filter (Code_Thingol.lookup_class naming o Sign.intern_class thy) classes);
   120   val parse_instances = Scan.lift (Args.parens (Args.$$$ "instances") |-- Scan.repeat1 (Args.name --| Args.$$$ "::" -- Args.name))
   121     >> (fn insts => fn thy => fn naming => map_filter (Code_Thingol.lookup_instance naming o apsnd (Sign.intern_type thy) o apfst (Sign.intern_class thy) o swap) insts);
   122   val parse_names = parse_consts || parse_types || parse_classes || parse_instances;
   123 
   124 in
   125 
   126 val _ = ThyOutput.antiquotation "code_stmts"
   127   (parse_const_terms -- Scan.repeat parse_names -- Scan.lift (Args.parens Args.name))
   128   (fn {context = ctxt, ...} => fn ((mk_cs, mk_stmtss), target) =>
   129     let val thy = ProofContext.theory_of ctxt in
   130       Code_Target.code_of thy
   131         target NONE "Example" (mk_cs thy)
   132         (fn naming => maps (fn f => f thy naming) mk_stmtss)
   133       |> typewriter
   134     end);
   135 
   136 end;
   137 
   138 end;