src/Pure/Thy/term_style.ML
author krauss
Wed Feb 02 08:47:45 2011 +0100 (2011-02-02)
changeset 41686 d8efc2490b8e
parent 41685 e29ea98a76ce
child 42016 3b6826b3ed37
permissions -rw-r--r--
made SML/NJ happy
     1 (*  Title:      Pure/Thy/term_style.ML
     2     Author:     Florian Haftmann, TU Muenchen
     3 
     4 Styles for term printing.
     5 *)
     6 
     7 signature TERM_STYLE =
     8 sig
     9   val setup: string -> (Proof.context -> term -> term) parser -> theory -> theory
    10   val parse: (term -> term) context_parser
    11   val parse_bare: (term -> term) context_parser
    12 end;
    13 
    14 structure Term_Style: TERM_STYLE =
    15 struct
    16 
    17 (* style data *)
    18 
    19 fun err_dup_style name =
    20   error ("Duplicate declaration of antiquote style: " ^ quote name);
    21 
    22 structure StyleData = Theory_Data
    23 (
    24   type T = ((Proof.context -> term -> term) parser * stamp) Symtab.table;
    25   val empty = Symtab.empty;
    26   val extend = I;
    27   fun merge data : T = Symtab.merge (eq_snd (op =)) data
    28     handle Symtab.DUP dup => err_dup_style dup;
    29 );
    30 
    31 
    32 (* accessors *)
    33 
    34 fun the_style thy name =
    35   (case Symtab.lookup (StyleData.get thy) name of
    36     NONE => error ("Unknown antiquote style: " ^ quote name)
    37   | SOME (style, _) => style);
    38 
    39 fun setup name style thy =
    40   StyleData.map (Symtab.update_new (name, (style, stamp ()))) thy
    41     handle Symtab.DUP _ => err_dup_style name;
    42 
    43 
    44 (* style parsing *)
    45 
    46 fun parse_single ctxt = Parse.position (Parse.xname -- Args.parse)
    47   >> (fn x as ((name, _), _) => fst (Args.context_syntax "style"
    48        (Scan.lift (the_style (ProofContext.theory_of ctxt) name))
    49          (Args.src x) ctxt |>> (fn f => f ctxt)));
    50 
    51 val parse = Args.context :|-- (fn ctxt => Scan.lift 
    52   (Args.parens (parse_single ctxt ::: Scan.repeat (Args.$$$ "," |-- parse_single ctxt))
    53       >> fold I
    54   || Scan.succeed I));
    55 
    56 val parse_bare = Args.context :|-- (fn ctxt => (Output.legacy_feature "Old-style antiquotation style.";
    57   Scan.lift Args.liberal_name
    58   >> (fn name => fst (Args.context_syntax "style"
    59        (Scan.lift (the_style (ProofContext.theory_of ctxt) name))
    60           (Args.src (("style", []), Position.none)) ctxt |>> (fn f => f ctxt)))));
    61 
    62 
    63 (* predefined styles *)
    64 
    65 fun style_lhs_rhs proj = Scan.succeed (fn ctxt => fn t =>
    66   let
    67     val concl =
    68       Object_Logic.drop_judgment (ProofContext.theory_of ctxt) (Logic.strip_imp_concl t)
    69   in
    70     case concl of (_ $ l $ r) => proj (l, r)
    71     | _ => error ("Binary operator expected in term: " ^ Syntax.string_of_term ctxt concl)
    72   end);
    73 
    74 val style_prem = Args.name >> (fn raw_i => fn ctxt => fn t =>
    75   let
    76     val i = (the o Int.fromString) raw_i;
    77     val prems = Logic.strip_imp_prems t;
    78   in
    79     if i <= length prems then nth prems (i - 1)
    80     else error ("Not enough premises for prem " ^ string_of_int i ^
    81       " in propositon: " ^ Syntax.string_of_term ctxt t)
    82   end);
    83 
    84 fun style_parm_premise i = Scan.succeed (fn ctxt => fn t =>
    85   let
    86     val i_str = string_of_int i;
    87     val _ = Output.legacy_feature (quote ("prem" ^ i_str)
    88       ^ " term style encountered; use explicit argument syntax "
    89       ^ quote ("prem " ^ i_str) ^ " instead.");
    90     val prems = Logic.strip_imp_prems t;
    91   in
    92     if i <= length prems then nth prems (i - 1)
    93     else error ("Not enough premises for prem" ^ i_str ^
    94       " in propositon: " ^ Syntax.string_of_term ctxt t)
    95   end);
    96 
    97 fun isub_symbols (d :: s :: ss) =
    98       if Symbol.is_ascii_digit d andalso not (String.isPrefix ("\\<^") s)
    99       then d :: "\\<^isub>" :: isub_symbols (s :: ss)
   100       else d :: s :: ss
   101   | isub_symbols cs = cs;
   102 
   103 val isub_name = implode o rev o isub_symbols o rev o Symbol.explode;
   104 
   105 fun isub_term (Free (n, T)) = Free (isub_name n, T)
   106   | isub_term (Var ((n, idx), T)) =
   107       if idx <> 0 then Var ((isub_name (n ^ string_of_int idx), 0), T)
   108       else Var ((isub_name n, 0), T)
   109   | isub_term (t $ u) = isub_term t $ isub_term u
   110   | isub_term (Abs (n, T, b)) = Abs (isub_name n, T, isub_term b)
   111   | isub_term t = t;
   112 
   113 val _ = Context.>> (Context.map_theory
   114  (setup "lhs" (style_lhs_rhs fst) #>
   115   setup "rhs" (style_lhs_rhs snd) #>
   116   setup "prem" style_prem #>
   117   setup "concl" (Scan.succeed (K Logic.strip_imp_concl)) #>
   118   setup "isub" (Scan.succeed (K isub_term)) #>
   119   setup "prem1" (style_parm_premise 1) #>
   120   setup "prem2" (style_parm_premise 2) #>
   121   setup "prem3" (style_parm_premise 3) #>
   122   setup "prem4" (style_parm_premise 4) #>
   123   setup "prem5" (style_parm_premise 5) #>
   124   setup "prem6" (style_parm_premise 6) #>
   125   setup "prem7" (style_parm_premise 7) #>
   126   setup "prem8" (style_parm_premise 8) #>
   127   setup "prem9" (style_parm_premise 9) #>
   128   setup "prem10" (style_parm_premise 10) #>
   129   setup "prem11" (style_parm_premise 11) #>
   130   setup "prem12" (style_parm_premise 12) #>
   131   setup "prem13" (style_parm_premise 13) #>
   132   setup "prem14" (style_parm_premise 14) #>
   133   setup "prem15" (style_parm_premise 15) #>
   134   setup "prem16" (style_parm_premise 16) #>
   135   setup "prem17" (style_parm_premise 17) #>
   136   setup "prem18" (style_parm_premise 18) #>
   137   setup "prem19" (style_parm_premise 19)));
   138 
   139 end;