src/Pure/defs.ML
author haftmann
Thu Aug 09 15:52:54 2007 +0200 (2007-08-09)
changeset 24199 8be734b5f59f
parent 20668 00521d5e1838
child 24792 fd4655e57168
permissions -rw-r--r--
new access interface in defs.ML
wenzelm@17707
     1
(*  Title:      Pure/defs.ML
obua@16108
     2
    ID:         $Id$
wenzelm@17707
     3
    Author:     Makarius
obua@16108
     4
wenzelm@19692
     5
Global well-formedness checks for constant definitions.  Covers plain
wenzelm@19701
     6
definitions and simple sub-structural overloading.
obua@16108
     7
*)
obua@16108
     8
wenzelm@16877
     9
signature DEFS =
wenzelm@16877
    10
sig
wenzelm@19697
    11
  val pretty_const: Pretty.pp -> string * typ list -> Pretty.T
wenzelm@19701
    12
  val plain_args: typ list -> bool
wenzelm@17707
    13
  type T
haftmann@24199
    14
  val specifications_of: T -> string -> {is_def: bool, name: string,
haftmann@24199
    15
    lhs: typ list, rhs: (string * typ list) list} list
haftmann@24199
    16
  val all_specifications_of: T -> (string * {is_def: bool, name: string,
haftmann@24199
    17
    lhs: typ list, rhs: (string * typ list) list} list) list
wenzelm@19697
    18
  val dest: T ->
wenzelm@19697
    19
   {restricts: ((string * typ list) * string) list,
wenzelm@19697
    20
    reducts: ((string * typ list) * (string * typ list) list) list}
wenzelm@19590
    21
  val empty: T
wenzelm@19692
    22
  val merge: Pretty.pp -> T * T -> T
haftmann@24199
    23
  val define: Pretty.pp -> bool -> bool -> string ->
wenzelm@19727
    24
    string * typ list -> (string * typ list) list -> T -> T
obua@16108
    25
end
obua@16108
    26
wenzelm@17711
    27
structure Defs: DEFS =
wenzelm@17707
    28
struct
obua@16108
    29
wenzelm@19697
    30
(* type arguments *)
wenzelm@19613
    31
wenzelm@19697
    32
type args = typ list;
wenzelm@19697
    33
wenzelm@19697
    34
fun pretty_const pp (c, args) =
wenzelm@19613
    35
  let
wenzelm@19692
    36
    val prt_args =
wenzelm@19692
    37
      if null args then []
wenzelm@19806
    38
      else [Pretty.list "(" ")" (map (Pretty.typ pp o Logic.unvarifyT) args)];
wenzelm@19697
    39
  in Pretty.block (Pretty.str c :: prt_args) end;
wenzelm@19624
    40
wenzelm@19707
    41
fun plain_args args =
wenzelm@19707
    42
  forall Term.is_TVar args andalso not (has_duplicates (op =) args);
wenzelm@19707
    43
wenzelm@19697
    44
fun disjoint_args (Ts, Us) =
wenzelm@19697
    45
  not (Type.could_unifys (Ts, Us)) orelse
wenzelm@19697
    46
    ((Type.raw_unifys (Ts, map (Logic.incr_tvar (maxidx_of_typs Ts + 1)) Us) Vartab.empty; false)
wenzelm@19697
    47
      handle Type.TUNIFY => true);
wenzelm@19692
    48
wenzelm@19697
    49
fun match_args (Ts, Us) =
wenzelm@19697
    50
  Option.map Envir.typ_subst_TVars
wenzelm@19697
    51
    (SOME (Type.raw_matches (Ts, Us) Vartab.empty) handle Type.TYPE_MATCH => NONE);
wenzelm@19692
    52
wenzelm@19692
    53
wenzelm@19692
    54
(* datatype defs *)
wenzelm@19692
    55
haftmann@24199
    56
type spec = {is_def: bool, name: string, lhs: args, rhs: (string * args) list};
wenzelm@19697
    57
wenzelm@19692
    58
type def =
wenzelm@19712
    59
 {specs: spec Inttab.table,                 (*source specifications*)
wenzelm@19712
    60
  restricts: (args * string) list,          (*global restrictions imposed by incomplete patterns*)
wenzelm@19712
    61
  reducts: (args * (string * args) list) list};  (*specifications as reduction system*)
wenzelm@19697
    62
wenzelm@19697
    63
fun make_def (specs, restricts, reducts) =
wenzelm@19697
    64
  {specs = specs, restricts = restricts, reducts = reducts}: def;
wenzelm@19692
    65
wenzelm@19697
    66
fun map_def c f =
wenzelm@19697
    67
  Symtab.default (c, make_def (Inttab.empty, [], [])) #>
wenzelm@19697
    68
  Symtab.map_entry c (fn {specs, restricts, reducts}: def =>
wenzelm@19697
    69
    make_def (f (specs, restricts, reducts)));
wenzelm@19692
    70
wenzelm@19692
    71
wenzelm@19692
    72
datatype T = Defs of def Symtab.table;
wenzelm@19692
    73
wenzelm@19712
    74
fun lookup_list which defs c =
wenzelm@19692
    75
  (case Symtab.lookup defs c of
wenzelm@19713
    76
    SOME (def: def) => which def
wenzelm@19692
    77
  | NONE => []);
wenzelm@19692
    78
haftmann@24199
    79
fun specifications_of (Defs defs) = lookup_list (map snd o Inttab.dest o #specs) defs;
haftmann@24199
    80
fun all_specifications_of (Defs defs) =
haftmann@24199
    81
  ((map o apsnd) (map snd o Inttab.dest o #specs) o Symtab.dest) defs;
wenzelm@19692
    82
val restricts_of = lookup_list #restricts;
wenzelm@19692
    83
val reducts_of = lookup_list #reducts;
wenzelm@19692
    84
wenzelm@19697
    85
fun dest (Defs defs) =
wenzelm@19697
    86
  let
wenzelm@19697
    87
    val restricts = Symtab.fold (fn (c, {restricts, ...}) =>
wenzelm@19697
    88
      fold (fn (args, name) => cons ((c, args), name)) restricts) defs [];
wenzelm@19697
    89
    val reducts = Symtab.fold (fn (c, {reducts, ...}) =>
wenzelm@19697
    90
      fold (fn (args, deps) => cons ((c, args), deps)) reducts) defs [];
wenzelm@19697
    91
  in {restricts = restricts, reducts = reducts} end;
wenzelm@19692
    92
wenzelm@19697
    93
val empty = Defs Symtab.empty;
wenzelm@19697
    94
wenzelm@19697
    95
wenzelm@19697
    96
(* specifications *)
wenzelm@19692
    97
wenzelm@19697
    98
fun disjoint_specs c (i, {lhs = Ts, name = a, ...}: spec) =
wenzelm@19697
    99
  Inttab.forall (fn (j, {lhs = Us, name = b, ...}: spec) =>
wenzelm@19697
   100
    i = j orelse disjoint_args (Ts, Us) orelse
wenzelm@19697
   101
      error ("Type clash in specifications " ^ quote a ^ " and " ^ quote b ^
wenzelm@19697
   102
        " for constant " ^ quote c));
wenzelm@19692
   103
wenzelm@19697
   104
fun join_specs c ({specs = specs1, restricts, reducts}, {specs = specs2, ...}: def) =
wenzelm@19697
   105
  let
wenzelm@19697
   106
    val specs' =
wenzelm@19697
   107
      Inttab.fold (fn spec2 => (disjoint_specs c spec2 specs1; Inttab.update spec2)) specs2 specs1;
wenzelm@19697
   108
  in make_def (specs', restricts, reducts) end;
wenzelm@19697
   109
wenzelm@19697
   110
fun update_specs c spec = map_def c (fn (specs, restricts, reducts) =>
wenzelm@19697
   111
  (disjoint_specs c spec specs; (Inttab.update spec specs, restricts, reducts)));
wenzelm@19697
   112
wenzelm@19697
   113
wenzelm@19701
   114
(* normalized dependencies: reduction with well-formedness check *)
wenzelm@19697
   115
wenzelm@19697
   116
local
wenzelm@19697
   117
wenzelm@19729
   118
val prt = Pretty.string_of oo pretty_const;
wenzelm@19729
   119
fun err pp (c, args) (d, Us) s1 s2 =
wenzelm@19729
   120
  error (s1 ^ " dependency of constant " ^ prt pp (c, args) ^ " -> " ^ prt pp (d, Us) ^ s2);
wenzelm@19729
   121
wenzelm@19712
   122
fun contained (U as TVar _) (Type (_, Ts)) = exists (fn T => T = U orelse contained U T) Ts
wenzelm@19697
   123
  | contained _ _ = false;
wenzelm@19697
   124
wenzelm@19729
   125
fun acyclic pp defs (c, args) (d, Us) =
wenzelm@19729
   126
  c <> d orelse
wenzelm@19729
   127
  exists (fn U => exists (contained U) args) Us orelse
wenzelm@19729
   128
  is_none (match_args (args, Us)) orelse
wenzelm@19729
   129
  err pp (c, args) (d, Us) "Circular" "";
wenzelm@19729
   130
wenzelm@19701
   131
fun wellformed pp defs (c, args) (d, Us) =
wenzelm@19729
   132
  forall is_TVar Us orelse
wenzelm@19729
   133
  (case find_first (fn (Ts, _) => not (disjoint_args (Ts, Us))) (restricts_of defs d) of
wenzelm@19729
   134
    SOME (Ts, name) =>
wenzelm@19729
   135
      err pp (c, args) (d, Us) "Malformed"
wenzelm@19729
   136
        ("\n(restriction " ^ prt pp (d, Ts) ^ " from " ^ quote name ^ ")")
wenzelm@19729
   137
  | NONE => true);
wenzelm@19692
   138
wenzelm@19701
   139
fun reduction pp defs const deps =
wenzelm@19692
   140
  let
wenzelm@19701
   141
    fun reduct Us (Ts, rhs) =
wenzelm@19701
   142
      (case match_args (Ts, Us) of
wenzelm@19701
   143
        NONE => NONE
wenzelm@19701
   144
      | SOME subst => SOME (map (apsnd (map subst)) rhs));
wenzelm@19701
   145
    fun reducts (d, Us) = get_first (reduct Us) (reducts_of defs d);
wenzelm@19701
   146
wenzelm@19701
   147
    val reds = map (`reducts) deps;
wenzelm@19701
   148
    val deps' =
wenzelm@19701
   149
      if forall (is_none o #1) reds then NONE
wenzelm@20668
   150
      else SOME (fold_rev
wenzelm@20668
   151
        (fn (NONE, dp) => insert (op =) dp | (SOME dps, _) => fold (insert (op =)) dps) reds []);
wenzelm@19729
   152
    val _ = forall (acyclic pp defs const) (the_default deps deps');
wenzelm@19697
   153
  in deps' end;
wenzelm@19692
   154
wenzelm@19760
   155
in
wenzelm@19760
   156
wenzelm@19712
   157
fun normalize pp =
wenzelm@19692
   158
  let
wenzelm@19701
   159
    fun norm_update (c, {reducts, ...}: def) (changed, defs) =
wenzelm@19701
   160
      let
wenzelm@19701
   161
        val reducts' = reducts |> map (fn (args, deps) =>
wenzelm@19712
   162
          (args, perhaps (reduction pp defs (c, args)) deps));
wenzelm@19697
   163
      in
wenzelm@19701
   164
        if reducts = reducts' then (changed, defs)
wenzelm@19701
   165
        else (true, defs |> map_def c (fn (specs, restricts, reducts) =>
wenzelm@19701
   166
          (specs, restricts, reducts')))
wenzelm@19697
   167
      end;
wenzelm@19701
   168
    fun norm_all defs =
wenzelm@19701
   169
      (case Symtab.fold norm_update defs (false, defs) of
wenzelm@19701
   170
        (true, defs') => norm_all defs'
wenzelm@19701
   171
      | (false, _) => defs);
wenzelm@19729
   172
    fun check defs (c, {reducts, ...}: def) =
wenzelm@19729
   173
      reducts |> forall (fn (args, deps) => forall (wellformed pp defs (c, args)) deps);
wenzelm@19729
   174
  in norm_all #> (fn defs => tap (Symtab.forall (check defs)) defs) end;
wenzelm@19701
   175
wenzelm@19712
   176
fun dependencies pp (c, args) restr deps =
wenzelm@19712
   177
  map_def c (fn (specs, restricts, reducts) =>
wenzelm@19712
   178
    let
wenzelm@19712
   179
      val restricts' = Library.merge (op =) (restricts, restr);
wenzelm@19712
   180
      val reducts' = insert (op =) (args, deps) reducts;
wenzelm@19712
   181
    in (specs, restricts', reducts') end)
wenzelm@19712
   182
  #> normalize pp;
wenzelm@19697
   183
wenzelm@19697
   184
end;
wenzelm@19692
   185
wenzelm@19692
   186
wenzelm@19624
   187
(* merge *)
wenzelm@19624
   188
wenzelm@19692
   189
fun merge pp (Defs defs1, Defs defs2) =
wenzelm@19613
   190
  let
wenzelm@19697
   191
    fun add_deps (c, args) restr deps defs =
wenzelm@19692
   192
      if AList.defined (op =) (reducts_of defs c) args then defs
wenzelm@19697
   193
      else dependencies pp (c, args) restr deps defs;
wenzelm@19697
   194
    fun add_def (c, {restricts, reducts, ...}: def) =
wenzelm@19697
   195
      fold (fn (args, deps) => add_deps (c, args) restricts deps) reducts;
wenzelm@19760
   196
  in
wenzelm@19760
   197
    Defs (Symtab.join join_specs (defs1, defs2)
wenzelm@19760
   198
      |> normalize pp |> Symtab.fold add_def defs2)
wenzelm@19760
   199
  end;
wenzelm@19613
   200
wenzelm@19613
   201
wenzelm@19613
   202
(* define *)
wenzelm@19590
   203
haftmann@24199
   204
fun define pp unchecked is_def name (c, args) deps (Defs defs) =
wenzelm@17707
   205
  let
wenzelm@19697
   206
    val restr =
wenzelm@19697
   207
      if plain_args args orelse
wenzelm@19697
   208
        (case args of [Type (a, rec_args)] => plain_args rec_args | _ => false)
wenzelm@19697
   209
      then [] else [(args, name)];
wenzelm@19692
   210
    val spec =
haftmann@24199
   211
      (serial (), {is_def = is_def, name = name, lhs = args, rhs = deps});
wenzelm@19697
   212
    val defs' = defs |> update_specs c spec;
wenzelm@19712
   213
  in Defs (defs' |> (if unchecked then I else dependencies pp (c, args) restr deps)) end;
wenzelm@19697
   214
wenzelm@19697
   215
end;