src/Pure/General/table.ML
author haftmann
Tue, 23 Sep 2008 18:11:44 +0200
changeset 28337 93964076e7b8
parent 28334 7c693bb76366
child 29004 a5a91f387791
permissions -rw-r--r--
case default fallback for NBE
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
6118
caa439435666 fixed titles;
wenzelm
parents: 5681
diff changeset
     1
(*  Title:      Pure/General/table.ML
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
     2
    ID:         $Id$
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
     3
    Author:     Markus Wenzel and Stefan Berghofer, TU Muenchen
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
     4
16342
b146c31a2955 added structure Inttab;
wenzelm
parents: 16192
diff changeset
     5
Generic tables.  Efficient purely functional implementation using
b146c31a2955 added structure Inttab;
wenzelm
parents: 16192
diff changeset
     6
balanced 2-3 trees.
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
     7
*)
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
     8
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
     9
signature KEY =
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    10
sig
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    11
  type key
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    12
  val ord: key * key -> order
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    13
end;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    14
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    15
signature TABLE =
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    16
sig
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    17
  type key
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    18
  type 'a table
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    19
  exception DUP of key
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
    20
  exception SAME
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
    21
  exception UNDEF of key
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    22
  val empty: 'a table
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    23
  val is_empty: 'a table -> bool
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    24
  val map: ('a -> 'b) -> 'a table -> 'b table
16446
0ab34b9d1b1f added map', fold;
wenzelm
parents: 16342
diff changeset
    25
  val map': (key -> 'a -> 'b) -> 'a table -> 'b table
0ab34b9d1b1f added map', fold;
wenzelm
parents: 16342
diff changeset
    26
  val fold: (key * 'b -> 'a -> 'a) -> 'b table -> 'a -> 'a
28334
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
    27
  val fold_rev: (key * 'b -> 'a -> 'a) -> 'b table -> 'a -> 'a
17579
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
    28
  val fold_map: (key * 'b -> 'a -> 'c * 'a) -> 'b table -> 'a -> 'c table * 'a
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    29
  val dest: 'a table -> (key * 'a) list
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    30
  val keys: 'a table -> key list
27508
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
    31
  val exists: (key * 'a -> bool) -> 'a table -> bool
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
    32
  val forall: (key * 'a -> bool) -> 'a table -> bool
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
    33
  val get_first: (key * 'a -> 'b option) -> 'a table -> 'b option
8409
ef01ee11b14e Added function min_key.
berghofe
parents: 7061
diff changeset
    34
  val min_key: 'a table -> key option
15160
394fb9b8908b changes
obua
parents: 15014
diff changeset
    35
  val max_key: 'a table -> key option
16887
b24c067b32d6 added defined;
wenzelm
parents: 16864
diff changeset
    36
  val defined: 'a table -> key -> bool
17412
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
    37
  val lookup: 'a table -> key -> 'a option
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
    38
  val update: (key * 'a) -> 'a table -> 'a table
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
    39
  val update_new: (key * 'a) -> 'a table -> 'a table                   (*exception DUP*)
17179
28802c8a9816 canonical interface for 'default'
haftmann
parents: 17140
diff changeset
    40
  val default: key * 'a -> 'a table -> 'a table
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
    41
  val map_entry: key -> ('a -> 'a) -> 'a table -> 'a table
20141
cf8129ebcdd3 added Table.map_default
haftmann
parents: 20124
diff changeset
    42
  val map_default: (key * 'a) -> ('a -> 'a) -> 'a table -> 'a table
23655
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
    43
  val make: (key * 'a) list -> 'a table                                (*exception DUP*)
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
    44
  val extend: 'a table * (key * 'a) list -> 'a table                   (*exception DUP*)
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
    45
  val join: (key -> 'a * 'a -> 'a) (*exception DUP/SAME*) ->
23655
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
    46
    'a table * 'a table -> 'a table                                    (*exception DUP*)
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
    47
  val merge: ('a * 'a -> bool) -> 'a table * 'a table -> 'a table      (*exception DUP*)
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
    48
  val delete: key -> 'a table -> 'a table                              (*exception UNDEF*)
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
    49
  val delete_safe: key -> 'a table -> 'a table
16466
82e2fc13ce54 added member;
wenzelm
parents: 16446
diff changeset
    50
  val member: ('b * 'a -> bool) -> 'a table -> key * 'b -> bool
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
    51
  val insert: ('a * 'a -> bool) -> key * 'a -> 'a table -> 'a table    (*exception DUP*)
16139
21ce46ca61cc remove(_multi): generalized type;
wenzelm
parents: 16002
diff changeset
    52
  val remove: ('b * 'a -> bool) -> key * 'b -> 'a table -> 'a table
18946
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
    53
  val lookup_list: 'a list table -> key -> 'a list
25391
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
    54
  val cons_list: (key * 'a) -> 'a list table -> 'a list table
19506
980a2edf9e82 added insert_list;
wenzelm
parents: 19482
diff changeset
    55
  val insert_list: ('a * 'a -> bool) -> key * 'a -> 'a list table -> 'a list table
18946
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
    56
  val remove_list: ('b * 'a -> bool) -> key * 'b -> 'a list table -> 'a list table
25391
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
    57
  val update_list: ('a * 'a -> bool) -> key * 'a -> 'a list table -> 'a list table
18946
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
    58
  val make_list: (key * 'a) list -> 'a list table
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
    59
  val dest_list: 'a list table -> (key * 'a) list
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
    60
  val merge_list: ('a * 'a -> bool) ->
23655
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
    61
    'a list table * 'a list table -> 'a list table                     (*exception DUP*)
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    62
end;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    63
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    64
functor TableFun(Key: KEY): TABLE =
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    65
struct
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    66
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    67
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    68
(* datatype table *)
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    69
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    70
type key = Key.key;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    71
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    72
datatype 'a table =
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    73
  Empty |
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    74
  Branch2 of 'a table * (key * 'a) * 'a table |
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    75
  Branch3 of 'a table * (key * 'a) * 'a table * (key * 'a) * 'a table;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    76
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    77
exception DUP of key;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    78
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    79
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    80
(* empty *)
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    81
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    82
val empty = Empty;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    83
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    84
fun is_empty Empty = true
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    85
  | is_empty _ = false;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
    86
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    87
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    88
(* map and fold combinators *)
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    89
16657
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    90
fun map_table f =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    91
  let
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    92
    fun map Empty = Empty
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    93
      | map (Branch2 (left, (k, x), right)) =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    94
          Branch2 (map left, (k, f k x), map right)
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    95
      | map (Branch3 (left, (k1, x1), mid, (k2, x2), right)) =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    96
          Branch3 (map left, (k1, f k1 x1), map mid, (k2, f k2 x2), map right);
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    97
  in map end;
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
    98
16657
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
    99
fun fold_table f =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   100
  let
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   101
    fun fold Empty x = x
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   102
      | fold (Branch2 (left, p, right)) x =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   103
          fold right (f p (fold left x))
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   104
      | fold (Branch3 (left, p1, mid, p2, right)) x =
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   105
          fold right (f p2 (fold mid (f p1 (fold left x))));
a6f65f47eda1 low-level tuning of map, fold;
wenzelm
parents: 16466
diff changeset
   106
  in fold end;
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
   107
28334
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   108
fun fold_rev_table f =
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   109
  let
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   110
    fun fold Empty x = x
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   111
      | fold (Branch2 (left, p, right)) x =
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   112
          fold left (f p (fold right x))
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   113
      | fold (Branch3 (left, p1, mid, p2, right)) x =
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   114
          fold left (f p1 (fold mid (f p2 (fold right x))));
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   115
  in fold end;
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   116
17709
299eeb303f04 tuned default operation: use internal modify;
wenzelm
parents: 17579
diff changeset
   117
fun fold_map_table f =
17579
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   118
  let
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   119
    fun fold_map Empty s = (Empty, s)
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   120
      | fold_map (Branch2 (left, p as (k, x), right)) s =
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   121
          s
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   122
          |> fold_map left
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   123
          ||>> f p
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   124
          ||>> fold_map right
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   125
          |-> (fn ((l, e), r) => pair (Branch2 (l, (k, e), r)))
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   126
      | fold_map (Branch3 (left, p1 as (k1, x1), mid, p2 as (k2, x2), right)) s =
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   127
          s
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   128
          |> fold_map left
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   129
          ||>> f p1
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   130
          ||>> fold_map mid
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   131
          ||>> f p2
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   132
          ||>> fold_map right
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   133
          |-> (fn ((((l, e1), m), e2), r) => pair (Branch3 (l, (k1, e1), m, (k2, e2), r)))
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   134
  in fold_map end;
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   135
28334
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   136
fun dest tab = fold_rev_table cons tab [];
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   137
fun keys tab = fold_rev_table (cons o #1) tab [];
16192
733267a60e32 exists: made non-strict;
wenzelm
parents: 16139
diff changeset
   138
27508
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   139
fun get_first f tab =
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   140
  let
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   141
    exception FOUND of 'b option;
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   142
    fun get entry () = (case f entry of NONE => () | some => raise FOUND some);
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   143
  in (fold_table get tab (); NONE) handle FOUND res => res end;
16192
733267a60e32 exists: made non-strict;
wenzelm
parents: 16139
diff changeset
   144
27508
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   145
fun exists pred =
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   146
  is_some o get_first (fn entry => if pred entry then SOME () else NONE);
16192
733267a60e32 exists: made non-strict;
wenzelm
parents: 16139
diff changeset
   147
733267a60e32 exists: made non-strict;
wenzelm
parents: 16139
diff changeset
   148
fun forall pred = not o exists (not o pred);
733267a60e32 exists: made non-strict;
wenzelm
parents: 16139
diff changeset
   149
27508
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   150
abdab08677d8 added get_first;
wenzelm
parents: 25391
diff changeset
   151
(* min/max keys *)
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   152
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   153
fun min_key Empty = NONE
16864
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   154
  | min_key (Branch2 (Empty, (k, _), _)) = SOME k
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   155
  | min_key (Branch3 (Empty, (k, _), _, _, _)) = SOME k
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   156
  | min_key (Branch2 (left, _, _)) = min_key left
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   157
  | min_key (Branch3 (left, _, _, _, _)) = min_key left;
8409
ef01ee11b14e Added function min_key.
berghofe
parents: 7061
diff changeset
   158
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   159
fun max_key Empty = NONE
16864
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   160
  | max_key (Branch2 (_, (k, _), Empty)) = SOME k
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   161
  | max_key (Branch3 (_, _, _, (k, _), Empty)) = SOME k
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   162
  | max_key (Branch2 (_, _, right)) = max_key right
0f536ece46e3 tuned min_key, max_key;
wenzelm
parents: 16687
diff changeset
   163
  | max_key (Branch3 (_, _, _, _, right)) = max_key right;
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   164
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   165
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   166
(* lookup *)
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   167
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   168
fun lookup tab key =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   169
  let
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   170
    fun look Empty = NONE
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   171
      | look (Branch2 (left, (k, x), right)) =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   172
          (case Key.ord (key, k) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   173
            LESS => look left
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   174
          | EQUAL => SOME x
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   175
          | GREATER => look right)
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   176
      | look (Branch3 (left, (k1, x1), mid, (k2, x2), right)) =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   177
          (case Key.ord (key, k1) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   178
            LESS => look left
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   179
          | EQUAL => SOME x1
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   180
          | GREATER =>
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   181
              (case Key.ord (key, k2) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   182
                LESS => look mid
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   183
              | EQUAL => SOME x2
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   184
              | GREATER => look right));
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   185
  in look tab end;
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   186
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   187
fun defined tab key =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   188
  let
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   189
    fun def Empty = false
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   190
      | def (Branch2 (left, (k, x), right)) =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   191
          (case Key.ord (key, k) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   192
            LESS => def left
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   193
          | EQUAL => true
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   194
          | GREATER => def right)
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   195
      | def (Branch3 (left, (k1, x1), mid, (k2, x2), right)) =
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   196
          (case Key.ord (key, k1) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   197
            LESS => def left
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   198
          | EQUAL => true
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   199
          | GREATER =>
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   200
              (case Key.ord (key, k2) of
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   201
                LESS => def mid
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   202
              | EQUAL => true
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   203
              | GREATER => def right));
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   204
  in def tab end;
16887
b24c067b32d6 added defined;
wenzelm
parents: 16864
diff changeset
   205
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   206
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   207
(* modify *)
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   208
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   209
datatype 'a growth =
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   210
  Stay of 'a table |
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   211
  Sprout of 'a table * (key * 'a) * 'a table;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   212
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   213
exception SAME;
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   214
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   215
fun modify key f tab =
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   216
  let
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   217
    fun modfy Empty = Sprout (Empty, (key, f NONE), Empty)
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   218
      | modfy (Branch2 (left, p as (k, x), right)) =
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   219
          (case Key.ord (key, k) of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   220
            LESS =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   221
              (case modfy left of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   222
                Stay left' => Stay (Branch2 (left', p, right))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   223
              | Sprout (left1, q, left2) => Stay (Branch3 (left1, q, left2, p, right)))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   224
          | EQUAL => Stay (Branch2 (left, (k, f (SOME x)), right))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   225
          | GREATER =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   226
              (case modfy right of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   227
                Stay right' => Stay (Branch2 (left, p, right'))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   228
              | Sprout (right1, q, right2) =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   229
                  Stay (Branch3 (left, p, right1, q, right2))))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   230
      | modfy (Branch3 (left, p1 as (k1, x1), mid, p2 as (k2, x2), right)) =
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   231
          (case Key.ord (key, k1) of
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   232
            LESS =>
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   233
              (case modfy left of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   234
                Stay left' => Stay (Branch3 (left', p1, mid, p2, right))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   235
              | Sprout (left1, q, left2) =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   236
                  Sprout (Branch2 (left1, q, left2), p1, Branch2 (mid, p2, right)))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   237
          | EQUAL => Stay (Branch3 (left, (k1, f (SOME x1)), mid, p2, right))
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   238
          | GREATER =>
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   239
              (case Key.ord (key, k2) of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   240
                LESS =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   241
                  (case modfy mid of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   242
                    Stay mid' => Stay (Branch3 (left, p1, mid', p2, right))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   243
                  | Sprout (mid1, q, mid2) =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   244
                      Sprout (Branch2 (left, p1, mid1), q, Branch2 (mid2, p2, right)))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   245
              | EQUAL => Stay (Branch3 (left, p1, mid, (k2, f (SOME x2)), right))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   246
              | GREATER =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   247
                  (case modfy right of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   248
                    Stay right' => Stay (Branch3 (left, p1, mid, p2, right'))
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   249
                  | Sprout (right1, q, right2) =>
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   250
                      Sprout (Branch2 (left, p1, mid), p2, Branch2 (right1, q, right2)))));
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   251
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   252
  in
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   253
    (case modfy tab of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   254
      Stay tab' => tab'
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   255
    | Sprout br => Branch2 br)
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   256
    handle SAME => tab
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   257
  end;
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   258
17412
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
   259
fun update (key, x) tab = modify key (fn _ => x) tab;
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
   260
fun update_new (key, x) tab = modify key (fn NONE => x | SOME _ => raise DUP key) tab;
17709
299eeb303f04 tuned default operation: use internal modify;
wenzelm
parents: 17579
diff changeset
   261
fun default (key, x) tab = modify key (fn NONE => x | SOME _ => raise SAME) tab;
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   262
fun map_entry key f = modify key (fn NONE => raise SAME | SOME x => f x);
27783
cd5ae3dbd30e map_default: more explicit scope;
wenzelm
parents: 27508
diff changeset
   263
fun map_default (key, x) f = modify key (fn NONE => f x | SOME y => f y);
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   264
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   265
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   266
(* delete *)
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   267
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   268
exception UNDEF of key;
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   269
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   270
local
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   271
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   272
fun compare NONE (k2, _) = LESS
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   273
  | compare (SOME k1) (k2, _) = Key.ord (k1, k2);
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   274
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   275
fun if_eq EQUAL x y = x
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   276
  | if_eq _ x y = y;
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   277
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   278
fun del (SOME k) Empty = raise UNDEF k
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   279
  | del NONE (Branch2 (Empty, p, Empty)) = (p, (true, Empty))
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   280
  | del NONE (Branch3 (Empty, p, Empty, q, Empty)) =
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   281
      (p, (false, Branch2 (Empty, q, Empty)))
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   282
  | del k (Branch2 (Empty, p, Empty)) = (case compare k p of
16002
wenzelm
parents: 15761
diff changeset
   283
      EQUAL => (p, (true, Empty)) | _ => raise UNDEF (the k))
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   284
  | del k (Branch3 (Empty, p, Empty, q, Empty)) = (case compare k p of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   285
      EQUAL => (p, (false, Branch2 (Empty, q, Empty)))
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   286
    | _ => (case compare k q of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   287
        EQUAL => (q, (false, Branch2 (Empty, p, Empty)))
16002
wenzelm
parents: 15761
diff changeset
   288
      | _ => raise UNDEF (the k)))
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   289
  | del k (Branch2 (l, p, r)) = (case compare k p of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   290
      LESS => (case del k l of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   291
        (p', (false, l')) => (p', (false, Branch2 (l', p, r)))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   292
      | (p', (true, l')) => (p', case r of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   293
          Branch2 (rl, rp, rr) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   294
            (true, Branch3 (l', p, rl, rp, rr))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   295
        | Branch3 (rl, rp, rm, rq, rr) => (false, Branch2
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   296
            (Branch2 (l', p, rl), rp, Branch2 (rm, rq, rr)))))
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   297
    | ord => (case del (if_eq ord NONE k) r of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   298
        (p', (false, r')) => (p', (false, Branch2 (l, if_eq ord p' p, r')))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   299
      | (p', (true, r')) => (p', case l of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   300
          Branch2 (ll, lp, lr) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   301
            (true, Branch3 (ll, lp, lr, if_eq ord p' p, r'))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   302
        | Branch3 (ll, lp, lm, lq, lr) => (false, Branch2
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   303
            (Branch2 (ll, lp, lm), lq, Branch2 (lr, if_eq ord p' p, r'))))))
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   304
  | del k (Branch3 (l, p, m, q, r)) = (case compare k q of
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   305
      LESS => (case compare k p of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   306
        LESS => (case del k l of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   307
          (p', (false, l')) => (p', (false, Branch3 (l', p, m, q, r)))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   308
        | (p', (true, l')) => (p', (false, case (m, r) of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   309
            (Branch2 (ml, mp, mr), Branch2 _) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   310
              Branch2 (Branch3 (l', p, ml, mp, mr), q, r)
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   311
          | (Branch3 (ml, mp, mm, mq, mr), _) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   312
              Branch3 (Branch2 (l', p, ml), mp, Branch2 (mm, mq, mr), q, r)
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   313
          | (Branch2 (ml, mp, mr), Branch3 (rl, rp, rm, rq, rr)) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   314
              Branch3 (Branch2 (l', p, ml), mp, Branch2 (mr, q, rl), rp,
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   315
                Branch2 (rm, rq, rr)))))
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   316
      | ord => (case del (if_eq ord NONE k) m of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   317
          (p', (false, m')) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   318
            (p', (false, Branch3 (l, if_eq ord p' p, m', q, r)))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   319
        | (p', (true, m')) => (p', (false, case (l, r) of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   320
            (Branch2 (ll, lp, lr), Branch2 _) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   321
              Branch2 (Branch3 (ll, lp, lr, if_eq ord p' p, m'), q, r)
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   322
          | (Branch3 (ll, lp, lm, lq, lr), _) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   323
              Branch3 (Branch2 (ll, lp, lm), lq,
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   324
                Branch2 (lr, if_eq ord p' p, m'), q, r)
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   325
          | (_, Branch3 (rl, rp, rm, rq, rr)) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   326
              Branch3 (l, if_eq ord p' p, Branch2 (m', q, rl), rp,
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   327
                Branch2 (rm, rq, rr))))))
15531
08c8dad8e399 Deleted Library.option type.
skalberg
parents: 15160
diff changeset
   328
    | ord => (case del (if_eq ord NONE k) r of
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   329
        (q', (false, r')) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   330
          (q', (false, Branch3 (l, p, m, if_eq ord q' q, r')))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   331
      | (q', (true, r')) => (q', (false, case (l, m) of
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   332
          (Branch2 _, Branch2 (ml, mp, mr)) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   333
            Branch2 (l, p, Branch3 (ml, mp, mr, if_eq ord q' q, r'))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   334
        | (_, Branch3 (ml, mp, mm, mq, mr)) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   335
            Branch3 (l, p, Branch2 (ml, mp, mm), mq,
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   336
              Branch2 (mr, if_eq ord q' q, r'))
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   337
        | (Branch3 (ll, lp, lm, lq, lr), Branch2 (ml, mp, mr)) =>
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   338
            Branch3 (Branch2 (ll, lp, lm), lq, Branch2 (lr, p, ml), mp,
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   339
              Branch2 (mr, if_eq ord q' q, r'))))));
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   340
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   341
in
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   342
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   343
fun delete key tab = snd (snd (del (SOME key) tab));
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   344
fun delete_safe key tab = delete key tab handle UNDEF _ => tab;
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   345
15665
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   346
end;
7e7412fffc0c tuned updates, added map_entry;
wenzelm
parents: 15574
diff changeset
   347
15014
97ab70c3d955 Added delete operation.
berghofe
parents: 14981
diff changeset
   348
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   349
(* membership operations *)
16466
82e2fc13ce54 added member;
wenzelm
parents: 16446
diff changeset
   350
82e2fc13ce54 added member;
wenzelm
parents: 16446
diff changeset
   351
fun member eq tab (key, x) =
17412
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
   352
  (case lookup tab key of
16466
82e2fc13ce54 added member;
wenzelm
parents: 16446
diff changeset
   353
    NONE => false
82e2fc13ce54 added member;
wenzelm
parents: 16446
diff changeset
   354
  | SOME y => eq (x, y));
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   355
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   356
fun insert eq (key, x) =
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   357
  modify key (fn NONE => x | SOME y => if eq (x, y) then raise SAME else raise DUP key);
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   358
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   359
fun remove eq (key, x) tab =
17412
e26cb20ef0cc TableFun/Symtab: curried lookup and update;
wenzelm
parents: 17222
diff changeset
   360
  (case lookup tab key of
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   361
    NONE => tab
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   362
  | SOME y => if eq (x, y) then delete key tab else tab);
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   363
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   364
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   365
(* simultaneous modifications *)
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   366
23655
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
   367
fun extend (table, args) = fold update_new args table;
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   368
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   369
fun make entries = extend (empty, entries);
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   370
12287
7017cee2f3ac added join, merge_multi(');
wenzelm
parents: 8806
diff changeset
   371
fun join f (table1, table2) =
23655
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
   372
  let fun add (key, y) tab = modify key (fn NONE => y | SOME x => f key (x, y)) tab;
d2d1138e0ddc replaced exception TableFun/GraphFun.DUPS by TableFun/GraphFun.DUP;
wenzelm
parents: 21065
diff changeset
   373
  in fold_table add table2 table1 end;
12287
7017cee2f3ac added join, merge_multi(');
wenzelm
parents: 8806
diff changeset
   374
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   375
fun merge eq = join (fn key => fn xy => if eq xy then raise SAME else raise DUP key);
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   376
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   377
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   378
(* list tables *)
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   379
18946
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
   380
fun lookup_list tab key = these (lookup tab key);
25391
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   381
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   382
fun cons_list (key, x) tab = modify key (fn NONE => [x] | SOME xs => x :: xs) tab;
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   383
20124
caf3a129b90d tuned insert_list;
wenzelm
parents: 19506
diff changeset
   384
fun insert_list eq (key, x) =
caf3a129b90d tuned insert_list;
wenzelm
parents: 19506
diff changeset
   385
  modify key (fn NONE => [x] | SOME xs => if Library.member eq xs x then raise SAME else x :: xs);
25391
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   386
18946
ce65d2d2e0c2 renamed xxx_multi to xxx_list;
wenzelm
parents: 18007
diff changeset
   387
fun remove_list eq (key, x) tab =
15761
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   388
  map_entry key (fn xs => (case Library.remove eq x xs of [] => raise UNDEF key | ys => ys)) tab
c9561302c74a added delete_safe, insert, remove, remove_multi;
wenzelm
parents: 15665
diff changeset
   389
  handle UNDEF _ => delete key tab;
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   390
25391
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   391
fun update_list eq (key, x) =
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   392
  modify key (fn NONE => [x] | SOME [] => [x] | SOME (xs as y :: _) =>
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   393
    if eq (x, y) then raise SAME else Library.update eq x xs);
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   394
783e5de2a6c7 renamed update_list to cons_list;
wenzelm
parents: 23655
diff changeset
   395
fun make_list args = fold_rev cons_list args empty;
19482
9f11af8f7ef9 tuned basic list operators (flat, maps, map_filter);
wenzelm
parents: 19073
diff changeset
   396
fun dest_list tab = maps (fn (key, xs) => map (pair key) xs) (dest tab);
19031
0059b5b195a2 export exception SAME (for join);
wenzelm
parents: 18946
diff changeset
   397
fun merge_list eq = join (fn _ => Library.merge eq);
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   398
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   399
5681
464913c6086a added foldl, keys;
wenzelm
parents: 5015
diff changeset
   400
(*final declarations of this structure!*)
16446
0ab34b9d1b1f added map', fold;
wenzelm
parents: 16342
diff changeset
   401
fun map f = map_table (K f);
0ab34b9d1b1f added map', fold;
wenzelm
parents: 16342
diff changeset
   402
val map' = map_table;
0ab34b9d1b1f added map', fold;
wenzelm
parents: 16342
diff changeset
   403
val fold = fold_table;
28334
7c693bb76366 added fold_rev;
wenzelm
parents: 27783
diff changeset
   404
val fold_rev = fold_rev_table;
17579
830207835ab5 added fold_map_table
haftmann
parents: 17412
diff changeset
   405
val fold_map = fold_map_table;
5015
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   406
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   407
end;
44fd9e09c637 moved table.ML to General/table.ML;
wenzelm
parents:
diff changeset
   408
16342
b146c31a2955 added structure Inttab;
wenzelm
parents: 16192
diff changeset
   409
structure Inttab = TableFun(type key = int val ord = int_ord);
16687
51fa05ce0f32 Symtab: use fast_string_ord instead of string_ord -- affects order of Symtab.dest etc.;
wenzelm
parents: 16657
diff changeset
   410
structure Symtab = TableFun(type key = string val ord = fast_string_ord);