src/Pure/Syntax/mixfix.ML
author wenzelm
Tue Apr 25 22:23:50 2006 +0200 (2006-04-25)
changeset 19467 d75570e8aa97
parent 19373 f2446ce04590
child 19482 9f11af8f7ef9
permissions -rw-r--r--
unlocalize_mixfix: fallback on NoSyn;
wenzelm@384
     1
(*  Title:      Pure/Syntax/mixfix.ML
wenzelm@384
     2
    ID:         $Id$
wenzelm@551
     3
    Author:     Tobias Nipkow and Markus Wenzel, TU Muenchen
wenzelm@384
     4
wenzelm@18719
     5
Mixfix declarations, infixes, binders.
wenzelm@384
     6
*)
wenzelm@384
     7
wenzelm@384
     8
signature MIXFIX0 =
wenzelm@2199
     9
sig
wenzelm@384
    10
  datatype mixfix =
wenzelm@384
    11
    NoSyn |
wenzelm@384
    12
    Mixfix of string * int list * int |
wenzelm@384
    13
    Delimfix of string |
wenzelm@11651
    14
    InfixName of string * int |
wenzelm@2199
    15
    InfixlName of string * int |
wenzelm@2199
    16
    InfixrName of string * int |
wenzelm@18565
    17
    Infix of int |    (*obsolete*)
wenzelm@18565
    18
    Infixl of int |   (*obsolete*)
wenzelm@18565
    19
    Infixr of int |   (*obsolete*)
wenzelm@18673
    20
    Binder of string * int * int |
wenzelm@18673
    21
    Structure
wenzelm@2199
    22
end;
wenzelm@384
    23
wenzelm@384
    24
signature MIXFIX1 =
wenzelm@2199
    25
sig
wenzelm@384
    26
  include MIXFIX0
wenzelm@12864
    27
  val literal: string -> mixfix
wenzelm@4823
    28
  val no_syn: 'a * 'b -> 'a * 'b * mixfix
wenzelm@19271
    29
  val pretty_mixfix: mixfix -> Pretty.T
wenzelm@384
    30
  val type_name: string -> mixfix -> string
wenzelm@384
    31
  val const_name: string -> mixfix -> string
wenzelm@19373
    32
  val const_mixfix: string -> mixfix -> string * mixfix
wenzelm@19373
    33
  val mixfix_const: string -> mixfix -> string * bool
wenzelm@18565
    34
  val unlocalize_mixfix: mixfix -> mixfix
wenzelm@4823
    35
  val mixfix_args: mixfix -> int
wenzelm@19020
    36
  val mixfix_content: mixfix -> string list list
wenzelm@2199
    37
end;
wenzelm@384
    38
wenzelm@384
    39
signature MIXFIX =
wenzelm@2199
    40
sig
wenzelm@384
    41
  include MIXFIX1
wenzelm@14903
    42
  val syn_ext_types: (string * int * mixfix) list -> SynExt.syn_ext
wenzelm@14903
    43
  val syn_ext_consts: (string -> bool) -> (string * typ * mixfix) list -> SynExt.syn_ext
wenzelm@2199
    44
end;
wenzelm@2199
    45
wenzelm@15751
    46
structure Mixfix: MIXFIX =
wenzelm@384
    47
struct
wenzelm@384
    48
wenzelm@2199
    49
wenzelm@384
    50
(** mixfix declarations **)
wenzelm@384
    51
wenzelm@384
    52
datatype mixfix =
wenzelm@384
    53
  NoSyn |
wenzelm@384
    54
  Mixfix of string * int list * int |
wenzelm@384
    55
  Delimfix of string |
wenzelm@11651
    56
  InfixName of string * int |
wenzelm@2199
    57
  InfixlName of string * int |
wenzelm@2199
    58
  InfixrName of string * int |
wenzelm@18565
    59
  Infix of int |      (*obsolete*)
wenzelm@18565
    60
  Infixl of int |     (*obsolete*)
wenzelm@18565
    61
  Infixr of int |     (*obsolete*)
wenzelm@18673
    62
  Binder of string * int * int |
wenzelm@18673
    63
  Structure;
wenzelm@384
    64
wenzelm@12864
    65
val literal = Delimfix o SynExt.escape_mfix;
wenzelm@12864
    66
wenzelm@4823
    67
fun no_syn (x, y) = (x, y, NoSyn);
wenzelm@4823
    68
wenzelm@384
    69
wenzelm@19271
    70
(* pretty_mixfix *)
wenzelm@19271
    71
wenzelm@19271
    72
local
wenzelm@11920
    73
wenzelm@19271
    74
val quoted = Pretty.quote o Pretty.str;
wenzelm@19271
    75
val keyword = Pretty.keyword;
wenzelm@19271
    76
val parens = Pretty.enclose "(" ")";
wenzelm@19271
    77
val brackets = Pretty.enclose "[" "]";
wenzelm@19271
    78
val int = Pretty.str o string_of_int;
wenzelm@19271
    79
wenzelm@19271
    80
in
wenzelm@11920
    81
wenzelm@19271
    82
fun pretty_mixfix NoSyn = Pretty.str ""
wenzelm@19271
    83
  | pretty_mixfix (Mixfix (s, ps, p)) =
wenzelm@19271
    84
      parens (Pretty.breaks [quoted s, brackets (Pretty.commas (map int ps)), int p])
wenzelm@19271
    85
  | pretty_mixfix (Delimfix s) = parens [quoted s]
wenzelm@19271
    86
  | pretty_mixfix (InfixName (s, p)) = parens (Pretty.breaks [keyword "infix", quoted s, int p])
wenzelm@19271
    87
  | pretty_mixfix (InfixlName (s, p)) = parens (Pretty.breaks [keyword "infixl", quoted s, int p])
wenzelm@19271
    88
  | pretty_mixfix (InfixrName (s, p)) = parens (Pretty.breaks [keyword "infixl", quoted s, int p])
wenzelm@19271
    89
  | pretty_mixfix (Infix p) = parens (Pretty.breaks [keyword "infix", int p])
wenzelm@19271
    90
  | pretty_mixfix (Infixl p) = parens (Pretty.breaks [keyword "infixl", int p])
wenzelm@19271
    91
  | pretty_mixfix (Infixr p) = parens (Pretty.breaks [keyword "infixr", int p])
wenzelm@19271
    92
  | pretty_mixfix (Binder (s, p1, p2)) =
wenzelm@19271
    93
      parens (Pretty.breaks [keyword "binder", quoted s, brackets [int p1], int p2])
wenzelm@19271
    94
  | pretty_mixfix Structure = parens [keyword "structure"];
wenzelm@19271
    95
wenzelm@19271
    96
end;
wenzelm@11920
    97
wenzelm@11920
    98
wenzelm@12531
    99
(* syntax specifications *)
wenzelm@384
   100
wenzelm@384
   101
fun strip ("'" :: c :: cs) = c :: strip cs
wenzelm@384
   102
  | strip ["'"] = []
wenzelm@384
   103
  | strip (c :: cs) = c :: strip cs
wenzelm@384
   104
  | strip [] = [];
wenzelm@384
   105
wenzelm@4697
   106
val strip_esc = implode o strip o Symbol.explode;
wenzelm@384
   107
wenzelm@17284
   108
fun deprecated c = (warning ("Unnamed infix operator " ^ quote c ^ " -- deprecated"); c);
wenzelm@17284
   109
wenzelm@11651
   110
fun type_name t (InfixName _) = t
wenzelm@11651
   111
  | type_name t (InfixlName _) = t
wenzelm@2199
   112
  | type_name t (InfixrName _) = t
wenzelm@17284
   113
  | type_name t (Infix _) = deprecated (strip_esc t)
wenzelm@17284
   114
  | type_name t (Infixl _) = deprecated (strip_esc t)
wenzelm@17284
   115
  | type_name t (Infixr _) = deprecated (strip_esc t)
wenzelm@384
   116
  | type_name t _ = t;
wenzelm@384
   117
wenzelm@11651
   118
fun const_name c (InfixName _) = c
wenzelm@11651
   119
  | const_name c (InfixlName _) = c
wenzelm@2199
   120
  | const_name c (InfixrName _) = c
wenzelm@17284
   121
  | const_name c (Infix _) = "op " ^ deprecated (strip_esc c)
wenzelm@17284
   122
  | const_name c (Infixl _) = "op " ^ deprecated (strip_esc c)
wenzelm@17284
   123
  | const_name c (Infixr _) = "op " ^ deprecated (strip_esc c)
wenzelm@384
   124
  | const_name c _ = c;
wenzelm@384
   125
wenzelm@19373
   126
fun fix_mixfix c (Infix p) = InfixName (c, p)
wenzelm@19373
   127
  | fix_mixfix c (Infixl p) = InfixlName (c, p)
wenzelm@19373
   128
  | fix_mixfix c (Infixr p) = InfixrName (c, p)
wenzelm@5056
   129
  | fix_mixfix _ mx = mx;
wenzelm@5056
   130
wenzelm@19373
   131
fun const_mixfix c mx = (const_name c mx, fix_mixfix c mx);
wenzelm@19373
   132
wenzelm@19373
   133
fun mixfix_const c NoSyn = (c, true)
wenzelm@19373
   134
  | mixfix_const c _ = (Lexicon.constN ^ c, false);
wenzelm@19373
   135
wenzelm@18565
   136
fun map_mixfix _ NoSyn = NoSyn
wenzelm@18565
   137
  | map_mixfix f (Mixfix (sy, ps, p)) = Mixfix (f sy, ps, p)
wenzelm@18565
   138
  | map_mixfix f (Delimfix sy) = Delimfix (f sy)
wenzelm@18565
   139
  | map_mixfix f (InfixName (sy, p)) = InfixName (f sy, p)
wenzelm@18565
   140
  | map_mixfix f (InfixlName (sy, p)) = InfixlName (f sy, p)
wenzelm@18565
   141
  | map_mixfix f (InfixrName (sy, p)) = InfixrName (f sy, p)
wenzelm@18565
   142
  | map_mixfix f (Binder (sy, p, q)) = Binder (f sy, p, q)
wenzelm@18673
   143
  | map_mixfix _ Structure = Structure
wenzelm@18565
   144
  | map_mixfix _ _ = raise Fail ("map_mixfix: illegal occurrence of unnamed infix");
wenzelm@18565
   145
wenzelm@19467
   146
fun unlocalize_mixfix mx =
wenzelm@19467
   147
  let val mx' = map_mixfix SynExt.unlocalize_mfix mx
wenzelm@19467
   148
  in if mx = mx' then NoSyn else mx' end;
wenzelm@18565
   149
wenzelm@4053
   150
fun mixfix_args NoSyn = 0
wenzelm@5690
   151
  | mixfix_args (Mixfix (sy, _, _)) = SynExt.mfix_args sy
wenzelm@5690
   152
  | mixfix_args (Delimfix sy) = SynExt.mfix_args sy
wenzelm@12531
   153
  | mixfix_args (InfixName (sy, _)) = 2 + SynExt.mfix_args sy
wenzelm@12531
   154
  | mixfix_args (InfixlName (sy, _)) = 2 + SynExt.mfix_args sy
wenzelm@12531
   155
  | mixfix_args (InfixrName (sy, _)) = 2 + SynExt.mfix_args sy
wenzelm@18341
   156
  | mixfix_args (Infix _) = 2
wenzelm@18341
   157
  | mixfix_args (Infixl _) = 2
wenzelm@18341
   158
  | mixfix_args (Infixr _) = 2
wenzelm@18673
   159
  | mixfix_args (Binder _) = 1
wenzelm@18673
   160
  | mixfix_args Structure = 0;
wenzelm@4053
   161
wenzelm@4053
   162
wenzelm@384
   163
(* syn_ext_types *)
wenzelm@384
   164
wenzelm@14903
   165
fun syn_ext_types type_decls =
wenzelm@384
   166
  let
wenzelm@384
   167
    fun name_of (t, _, mx) = type_name t mx;
wenzelm@384
   168
wenzelm@2199
   169
    fun mk_infix sy t p1 p2 p3 =
wenzelm@5690
   170
      SynExt.Mfix ("(_ " ^ sy ^ "/ _)",
wenzelm@12512
   171
        [SynExt.typeT, SynExt.typeT] ---> SynExt.typeT, t, [p1, p2], p3);
wenzelm@384
   172
wenzelm@2199
   173
    fun mfix_of decl =
wenzelm@2199
   174
      let val t = name_of decl in
wenzelm@2199
   175
        (case decl of
skalberg@15531
   176
          (_, _, NoSyn) => NONE
skalberg@15531
   177
        | (_, 2, InfixName (sy, p)) => SOME (mk_infix sy t (p + 1) (p + 1) p)
skalberg@15531
   178
        | (_, 2, InfixlName (sy, p)) => SOME (mk_infix sy t p (p + 1) p)
skalberg@15531
   179
        | (_, 2, InfixrName (sy, p)) => SOME (mk_infix sy t (p + 1) p p)
skalberg@15531
   180
        | (sy, 2, Infix p) => SOME (mk_infix sy t (p + 1) (p + 1) p)
skalberg@15531
   181
        | (sy, 2, Infixl p) => SOME (mk_infix sy t p (p + 1) p)
skalberg@15531
   182
        | (sy, 2, Infixr p) => SOME (mk_infix sy t (p + 1) p p)
wenzelm@14903
   183
        | _ => error ("Bad mixfix declaration for type: " ^ quote t))
wenzelm@2199
   184
      end;
wenzelm@384
   185
skalberg@15570
   186
    val mfix = List.mapPartial mfix_of type_decls;
wenzelm@384
   187
    val xconsts = map name_of type_decls;
wenzelm@14903
   188
  in SynExt.syn_ext mfix xconsts ([], [], [], []) [] ([], []) end;
wenzelm@384
   189
wenzelm@384
   190
wenzelm@384
   191
(* syn_ext_consts *)
wenzelm@384
   192
wenzelm@15751
   193
val binder_stamp = stamp ();
wenzelm@15751
   194
wenzelm@14903
   195
fun syn_ext_consts is_logtype const_decls =
wenzelm@384
   196
  let
wenzelm@384
   197
    fun name_of (c, _, mx) = const_name c mx;
wenzelm@384
   198
wenzelm@384
   199
    fun mk_infix sy ty c p1 p2 p3 =
wenzelm@5690
   200
      [SynExt.Mfix ("op " ^ sy, ty, c, [], SynExt.max_pri),
wenzelm@12512
   201
       SynExt.Mfix ("(_ " ^ sy ^ "/ _)", ty, c, [p1, p2], p3)];
wenzelm@384
   202
wenzelm@384
   203
    fun binder_typ _ (Type ("fun", [Type ("fun", [_, ty2]), ty3])) =
wenzelm@384
   204
          [Type ("idts", []), ty2] ---> ty3
wenzelm@14903
   205
      | binder_typ c _ = error ("Bad type of binder: " ^ quote c);
wenzelm@384
   206
wenzelm@2199
   207
    fun mfix_of decl =
wenzelm@2199
   208
      let val c = name_of decl in
wenzelm@2199
   209
        (case decl of
wenzelm@2199
   210
          (_, _, NoSyn) => []
wenzelm@5690
   211
        | (_, ty, Mixfix (sy, ps, p)) => [SynExt.Mfix (sy, ty, c, ps, p)]
wenzelm@5690
   212
        | (_, ty, Delimfix sy) => [SynExt.Mfix (sy, ty, c, [], SynExt.max_pri)]
wenzelm@11651
   213
        | (_, ty, InfixName (sy, p)) => mk_infix sy ty c (p + 1) (p + 1) p
wenzelm@2199
   214
        | (_, ty, InfixlName (sy, p)) => mk_infix sy ty c p (p + 1) p
wenzelm@2199
   215
        | (_, ty, InfixrName (sy, p)) => mk_infix sy ty c (p + 1) p p
wenzelm@11651
   216
        | (sy, ty, Infix p) => mk_infix sy ty c (p + 1) (p + 1) p
wenzelm@2199
   217
        | (sy, ty, Infixl p) => mk_infix sy ty c p (p + 1) p
wenzelm@2199
   218
        | (sy, ty, Infixr p) => mk_infix sy ty c (p + 1) p p
wenzelm@2199
   219
        | (_, ty, Binder (sy, p, q)) =>
wenzelm@18673
   220
            [SynExt.Mfix ("(3" ^ sy ^ "_./ _)", binder_typ c ty, sy, [0, p], q)]
wenzelm@18673
   221
        | _ => error ("Bad mixfix declaration for const: " ^ quote c))
wenzelm@2199
   222
    end;
wenzelm@384
   223
skalberg@15531
   224
    fun binder (c, _, Binder (sy, _, _)) = SOME (sy, c)
skalberg@15531
   225
      | binder _ = NONE;
wenzelm@384
   226
skalberg@15570
   227
    val mfix = List.concat (map mfix_of const_decls);
wenzelm@384
   228
    val xconsts = map name_of const_decls;
skalberg@15570
   229
    val binders = List.mapPartial binder const_decls;
wenzelm@16610
   230
    val binder_trs = binders |> map (SynExt.stamp_trfun binder_stamp o
wenzelm@16610
   231
        apsnd K o SynTrans.mk_binder_tr);
wenzelm@15751
   232
    val binder_trs' = binders |> map (SynExt.stamp_trfun binder_stamp o
wenzelm@16610
   233
        apsnd (K o SynTrans.non_typed_tr') o SynTrans.mk_binder_tr' o swap);
wenzelm@14903
   234
  in
wenzelm@14903
   235
    SynExt.syn_ext' true is_logtype
wenzelm@14903
   236
      mfix xconsts ([], binder_trs, binder_trs', []) [] ([], [])
wenzelm@14903
   237
  end;
wenzelm@384
   238
wenzelm@19020
   239
fun mixfix_content mx =
wenzelm@19020
   240
  let
wenzelm@19020
   241
    fun delims sy = SynExt.mfix_delims sy;
wenzelm@19020
   242
    fun op_delims sy = let val ds = delims sy in ["op" :: ds, ds] end;
wenzelm@19020
   243
  in
wenzelm@19020
   244
    case mx of
wenzelm@19020
   245
      Mixfix (sy, _, _) => [delims sy]
wenzelm@19020
   246
    | Delimfix sy => [delims sy]
wenzelm@19020
   247
    | InfixName (sy, _) => op_delims sy
wenzelm@19020
   248
    | InfixlName (sy, _) => op_delims sy
wenzelm@19020
   249
    | InfixrName (sy, _) => op_delims sy
wenzelm@19020
   250
    | Binder (sy, _, _) => [delims sy @ ["."]]
wenzelm@19020
   251
    | _ => []
wenzelm@19020
   252
  end;
wenzelm@19020
   253
wenzelm@384
   254
end;