src/HOL/Library/RBT.thy
changeset 47450 2ada2be850cb
parent 46565 ad21900e0ee9
child 48622 caaa1a02c650
equal deleted inserted replaced
47438:11a0aa6cc677 47450:2ada2be850cb
    33 
    33 
    34 
    34 
    35 subsection {* Primitive operations *}
    35 subsection {* Primitive operations *}
    36 
    36 
    37 definition lookup :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> 'a \<rightharpoonup> 'b" where
    37 definition lookup :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> 'a \<rightharpoonup> 'b" where
    38   [code]: "lookup t = RBT_Impl.lookup (impl_of t)"
    38   [code]: "lookup t = rbt_lookup (impl_of t)"
    39 
    39 
    40 definition empty :: "('a\<Colon>linorder, 'b) rbt" where
    40 definition empty :: "('a\<Colon>linorder, 'b) rbt" where
    41   "empty = RBT RBT_Impl.Empty"
    41   "empty = RBT RBT_Impl.Empty"
    42 
    42 
    43 lemma impl_of_empty [code abstract]:
    43 lemma impl_of_empty [code abstract]:
    44   "impl_of empty = RBT_Impl.Empty"
    44   "impl_of empty = RBT_Impl.Empty"
    45   by (simp add: empty_def RBT_inverse)
    45   by (simp add: empty_def RBT_inverse)
    46 
    46 
    47 definition insert :: "'a\<Colon>linorder \<Rightarrow> 'b \<Rightarrow> ('a, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    47 definition insert :: "'a\<Colon>linorder \<Rightarrow> 'b \<Rightarrow> ('a, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    48   "insert k v t = RBT (RBT_Impl.insert k v (impl_of t))"
    48   "insert k v t = RBT (rbt_insert k v (impl_of t))"
    49 
    49 
    50 lemma impl_of_insert [code abstract]:
    50 lemma impl_of_insert [code abstract]:
    51   "impl_of (insert k v t) = RBT_Impl.insert k v (impl_of t)"
    51   "impl_of (insert k v t) = rbt_insert k v (impl_of t)"
    52   by (simp add: insert_def RBT_inverse)
    52   by (simp add: insert_def RBT_inverse)
    53 
    53 
    54 definition delete :: "'a\<Colon>linorder \<Rightarrow> ('a, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    54 definition delete :: "'a\<Colon>linorder \<Rightarrow> ('a, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    55   "delete k t = RBT (RBT_Impl.delete k (impl_of t))"
    55   "delete k t = RBT (rbt_delete k (impl_of t))"
    56 
    56 
    57 lemma impl_of_delete [code abstract]:
    57 lemma impl_of_delete [code abstract]:
    58   "impl_of (delete k t) = RBT_Impl.delete k (impl_of t)"
    58   "impl_of (delete k t) = rbt_delete k (impl_of t)"
    59   by (simp add: delete_def RBT_inverse)
    59   by (simp add: delete_def RBT_inverse)
    60 
    60 
    61 definition entries :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a \<times> 'b) list" where
    61 definition entries :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a \<times> 'b) list" where
    62   [code]: "entries t = RBT_Impl.entries (impl_of t)"
    62   [code]: "entries t = RBT_Impl.entries (impl_of t)"
    63 
    63 
    64 definition keys :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> 'a list" where
    64 definition keys :: "('a\<Colon>linorder, 'b) rbt \<Rightarrow> 'a list" where
    65   [code]: "keys t = RBT_Impl.keys (impl_of t)"
    65   [code]: "keys t = RBT_Impl.keys (impl_of t)"
    66 
    66 
    67 definition bulkload :: "('a\<Colon>linorder \<times> 'b) list \<Rightarrow> ('a, 'b) rbt" where
    67 definition bulkload :: "('a\<Colon>linorder \<times> 'b) list \<Rightarrow> ('a, 'b) rbt" where
    68   "bulkload xs = RBT (RBT_Impl.bulkload xs)"
    68   "bulkload xs = RBT (rbt_bulkload xs)"
    69 
    69 
    70 lemma impl_of_bulkload [code abstract]:
    70 lemma impl_of_bulkload [code abstract]:
    71   "impl_of (bulkload xs) = RBT_Impl.bulkload xs"
    71   "impl_of (bulkload xs) = rbt_bulkload xs"
    72   by (simp add: bulkload_def RBT_inverse)
    72   by (simp add: bulkload_def RBT_inverse)
    73 
    73 
    74 definition map_entry :: "'a \<Rightarrow> ('b \<Rightarrow> 'b) \<Rightarrow> ('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    74 definition map_entry :: "'a \<Rightarrow> ('b \<Rightarrow> 'b) \<Rightarrow> ('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    75   "map_entry k f t = RBT (RBT_Impl.map_entry k f (impl_of t))"
    75   "map_entry k f t = RBT (rbt_map_entry k f (impl_of t))"
    76 
    76 
    77 lemma impl_of_map_entry [code abstract]:
    77 lemma impl_of_map_entry [code abstract]:
    78   "impl_of (map_entry k f t) = RBT_Impl.map_entry k f (impl_of t)"
    78   "impl_of (map_entry k f t) = rbt_map_entry k f (impl_of t)"
    79   by (simp add: map_entry_def RBT_inverse)
    79   by (simp add: map_entry_def RBT_inverse)
    80 
    80 
    81 definition map :: "('a \<Rightarrow> 'b \<Rightarrow> 'b) \<Rightarrow> ('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    81 definition map :: "('a \<Rightarrow> 'b \<Rightarrow> 'b) \<Rightarrow> ('a\<Colon>linorder, 'b) rbt \<Rightarrow> ('a, 'b) rbt" where
    82   "map f t = RBT (RBT_Impl.map f (impl_of t))"
    82   "map f t = RBT (RBT_Impl.map f (impl_of t))"
    83 
    83 
    96 
    96 
    97 
    97 
    98 subsection {* Abstract lookup properties *}
    98 subsection {* Abstract lookup properties *}
    99 
    99 
   100 lemma lookup_RBT:
   100 lemma lookup_RBT:
   101   "is_rbt t \<Longrightarrow> lookup (RBT t) = RBT_Impl.lookup t"
   101   "is_rbt t \<Longrightarrow> lookup (RBT t) = rbt_lookup t"
   102   by (simp add: lookup_def RBT_inverse)
   102   by (simp add: lookup_def RBT_inverse)
   103 
   103 
   104 lemma lookup_impl_of:
   104 lemma lookup_impl_of:
   105   "RBT_Impl.lookup (impl_of t) = lookup t"
   105   "rbt_lookup (impl_of t) = lookup t"
   106   by (simp add: lookup_def)
   106   by (simp add: lookup_def)
   107 
   107 
   108 lemma entries_impl_of:
   108 lemma entries_impl_of:
   109   "RBT_Impl.entries (impl_of t) = entries t"
   109   "RBT_Impl.entries (impl_of t) = entries t"
   110   by (simp add: entries_def)
   110   by (simp add: entries_def)
   117   "lookup empty = Map.empty"
   117   "lookup empty = Map.empty"
   118   by (simp add: empty_def lookup_RBT fun_eq_iff)
   118   by (simp add: empty_def lookup_RBT fun_eq_iff)
   119 
   119 
   120 lemma lookup_insert [simp]:
   120 lemma lookup_insert [simp]:
   121   "lookup (insert k v t) = (lookup t)(k \<mapsto> v)"
   121   "lookup (insert k v t) = (lookup t)(k \<mapsto> v)"
   122   by (simp add: insert_def lookup_RBT lookup_insert lookup_impl_of)
   122   by (simp add: insert_def lookup_RBT rbt_lookup_rbt_insert lookup_impl_of)
   123 
   123 
   124 lemma lookup_delete [simp]:
   124 lemma lookup_delete [simp]:
   125   "lookup (delete k t) = (lookup t)(k := None)"
   125   "lookup (delete k t) = (lookup t)(k := None)"
   126   by (simp add: delete_def lookup_RBT RBT_Impl.lookup_delete lookup_impl_of restrict_complement_singleton_eq)
   126   by (simp add: delete_def lookup_RBT rbt_lookup_rbt_delete lookup_impl_of restrict_complement_singleton_eq)
   127 
   127 
   128 lemma map_of_entries [simp]:
   128 lemma map_of_entries [simp]:
   129   "map_of (entries t) = lookup t"
   129   "map_of (entries t) = lookup t"
   130   by (simp add: entries_def map_of_entries lookup_impl_of)
   130   by (simp add: entries_def map_of_entries lookup_impl_of)
   131 
   131 
   132 lemma entries_lookup:
   132 lemma entries_lookup:
   133   "entries t1 = entries t2 \<longleftrightarrow> lookup t1 = lookup t2"
   133   "entries t1 = entries t2 \<longleftrightarrow> lookup t1 = lookup t2"
   134   by (simp add: entries_def lookup_def entries_lookup)
   134   by (simp add: entries_def lookup_def entries_rbt_lookup)
   135 
   135 
   136 lemma lookup_bulkload [simp]:
   136 lemma lookup_bulkload [simp]:
   137   "lookup (bulkload xs) = map_of xs"
   137   "lookup (bulkload xs) = map_of xs"
   138   by (simp add: bulkload_def lookup_RBT RBT_Impl.lookup_bulkload)
   138   by (simp add: bulkload_def lookup_RBT rbt_lookup_rbt_bulkload)
   139 
   139 
   140 lemma lookup_map_entry [simp]:
   140 lemma lookup_map_entry [simp]:
   141   "lookup (map_entry k f t) = (lookup t)(k := Option.map f (lookup t k))"
   141   "lookup (map_entry k f t) = (lookup t)(k := Option.map f (lookup t k))"
   142   by (simp add: map_entry_def lookup_RBT RBT_Impl.lookup_map_entry lookup_impl_of)
   142   by (simp add: map_entry_def lookup_RBT rbt_lookup_rbt_map_entry lookup_impl_of)
   143 
   143 
   144 lemma lookup_map [simp]:
   144 lemma lookup_map [simp]:
   145   "lookup (map f t) k = Option.map (f k) (lookup t k)"
   145   "lookup (map f t) k = Option.map (f k) (lookup t k)"
   146   by (simp add: map_def lookup_RBT RBT_Impl.lookup_map lookup_impl_of)
   146   by (simp add: map_def lookup_RBT rbt_lookup_map lookup_impl_of)
   147 
   147 
   148 lemma fold_fold:
   148 lemma fold_fold:
   149   "fold f t = List.fold (prod_case f) (entries t)"
   149   "fold f t = List.fold (prod_case f) (entries t)"
   150   by (simp add: fold_def fun_eq_iff RBT_Impl.fold_def entries_impl_of)
   150   by (simp add: fold_def fun_eq_iff RBT_Impl.fold_def entries_impl_of)
   151 
   151 
   152 lemma is_empty_empty [simp]:
   152 lemma is_empty_empty [simp]:
   153   "is_empty t \<longleftrightarrow> t = empty"
   153   "is_empty t \<longleftrightarrow> t = empty"
   154   by (simp add: rbt_eq_iff is_empty_def impl_of_empty split: rbt.split)
   154   by (simp add: rbt_eq_iff is_empty_def impl_of_empty split: rbt.split)
   155 
   155 
   156 lemma RBT_lookup_empty [simp]: (*FIXME*)
   156 lemma RBT_lookup_empty [simp]: (*FIXME*)
   157   "RBT_Impl.lookup t = Map.empty \<longleftrightarrow> t = RBT_Impl.Empty"
   157   "rbt_lookup t = Map.empty \<longleftrightarrow> t = RBT_Impl.Empty"
   158   by (cases t) (auto simp add: fun_eq_iff)
   158   by (cases t) (auto simp add: fun_eq_iff)
   159 
   159 
   160 lemma lookup_empty_empty [simp]:
   160 lemma lookup_empty_empty [simp]:
   161   "lookup t = Map.empty \<longleftrightarrow> t = empty"
   161   "lookup t = Map.empty \<longleftrightarrow> t = empty"
   162   by (cases t) (simp add: empty_def lookup_def RBT_inject RBT_inverse)
   162   by (cases t) (simp add: empty_def lookup_def RBT_inject RBT_inverse)
   163 
   163 
   164 lemma sorted_keys [iff]:
   164 lemma sorted_keys [iff]:
   165   "sorted (keys t)"
   165   "sorted (keys t)"
   166   by (simp add: keys_def RBT_Impl.keys_def sorted_entries)
   166   by (simp add: keys_def RBT_Impl.keys_def rbt_sorted_entries)
   167 
   167 
   168 lemma distinct_keys [iff]:
   168 lemma distinct_keys [iff]:
   169   "distinct (keys t)"
   169   "distinct (keys t)"
   170   by (simp add: keys_def RBT_Impl.keys_def distinct_entries)
   170   by (simp add: keys_def RBT_Impl.keys_def distinct_entries)
   171 
   171