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