Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1750)

Unified Diff: sdk/lib/collection/splay_tree.dart

Issue 23872008: Reapply "Make LinkedHashMap also have a factory constructor and be customizable"" (Closed) Base URL: https://dart.googlecode.com/svn/branches/bleeding_edge/dart
Patch Set: Created 7 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « sdk/lib/collection/linked_hash_map.dart ('k') | tests/corelib/map_test.dart » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: sdk/lib/collection/splay_tree.dart
diff --git a/sdk/lib/collection/splay_tree.dart b/sdk/lib/collection/splay_tree.dart
index a0c3624d265b0ffb517b597118cba7c4d0deef6b..72251d940b518ffe7bcc9e6a873854317a28e7d2 100644
--- a/sdk/lib/collection/splay_tree.dart
+++ b/sdk/lib/collection/splay_tree.dart
@@ -4,6 +4,8 @@
part of dart.collection;
+typedef bool _Predicate<T>(T value);
+
/**
* A node in a splay tree. It holds the sorting key and the left
* and right children in the tree.
@@ -229,6 +231,10 @@ abstract class _SplayTree<K> {
}
}
+class _TypeTest<T> {
+ bool test(v) => v is T;
+}
+
/*
* A [Map] of objects that can be ordered relative to each other.
*
@@ -238,19 +244,31 @@ abstract class _SplayTree<K> {
* Keys of the map are compared using the `compare` function passed in
* the constructor. If that is omitted, the objects are assumed to be
* [Comparable], and are compared using their [Comparable.compareTo]
- * method. This also means that `null` is *not* allowed as a key.
+ * method. Non-comparable objects (including `null`) will not work as keys
+ * in that case.
+ *
+ * To allow calling [operator[]], [remove] or [containsKey] with objects
+ * that are not supported by the `compare` function, an extra `isValidKey`
+ * predicate function can be supplied. This function is tested before
+ * using the `compare` function on an argument value that may not be a [K]
+ * value. If omitted, the `isValidKey` function defaults to testing if the
+ * value is a [K].
*/
class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
Comparator<K> _comparator;
+ _Predicate _validKey;
- SplayTreeMap([int compare(K key1, K key2)])
- : _comparator = (compare == null) ? Comparable.compare : compare;
+ SplayTreeMap([int compare(K key1, K key2), bool isValidKey(potentialKey)])
+ : _comparator = (compare == null) ? Comparable.compare : compare,
+ _validKey = (isValidKey != null) ? isValidKey : ((v) => v is K);
/**
* Creates a [SplayTreeMap] that contains all key value pairs of [other].
*/
- factory SplayTreeMap.from(Map<K, V> other, [int compare(K key1, K key2)]) =>
- new SplayTreeMap(compare)..addAll(other);
+ factory SplayTreeMap.from(Map<K, V> other,
+ [ int compare(K key1, K key2),
+ bool isValidKey(potentialKey)]) =>
+ new SplayTreeMap(compare, isValidKey)..addAll(other);
/**
* Creates a [SplayTreeMap] where the keys and values are computed from the
@@ -266,8 +284,9 @@ class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
* identity function.
*/
factory SplayTreeMap.fromIterable(Iterable<K> iterable,
- {K key(element), V value(element), int compare(K key1, K key2)}) {
- SplayTreeMap<K, V> map = new SplayTreeMap<K, V>(compare);
+ {K key(element), V value(element), int compare(K key1, K key2),
+ bool isValidKey(potentialKey) }) {
+ SplayTreeMap<K, V> map = new SplayTreeMap<K, V>(compare, isValidKey);
Maps._fillMapWithMappedIterable(map, iterable, key, value);
return map;
}
@@ -284,8 +303,8 @@ class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
* It is an error if the two [Iterable]s don't have the same length.
*/
factory SplayTreeMap.fromIterables(Iterable<K> keys, Iterable<V> values,
- [int compare(K key1, K key2)]) {
- SplayTreeMap<K, V> map = new SplayTreeMap<K, V>(compare);
+ [int compare(K key1, K key2), bool isValidKey(potentialKey)]) {
+ SplayTreeMap<K, V> map = new SplayTreeMap<K, V>(compare, isValidKey);
Maps._fillMapWithIterables(map, keys, values);
return map;
}
@@ -296,7 +315,7 @@ class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
V operator [](Object key) {
if (key == null) throw new ArgumentError(key);
- if (key is! K) return null;
+ if (!_validKey(key)) return null;
if (_root != null) {
int comp = _splay(key);
if (comp == 0) {
@@ -308,7 +327,7 @@ class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
}
V remove(Object key) {
- if (key is! K) return null;
+ if (!_validKey(key)) return null;
_SplayTreeMapNode mapRoot = _remove(key);
if (mapRoot != null) return mapRoot.value;
return null;
@@ -378,7 +397,7 @@ class SplayTreeMap<K, V> extends _SplayTree<K> implements Map<K, V> {
}
bool containsKey(Object key) {
- return key is K && _splay(key) == 0;
+ return _validKey(key) && _splay(key) == 0;
}
bool containsValue(Object value) {
« no previous file with comments | « sdk/lib/collection/linked_hash_map.dart ('k') | tests/corelib/map_test.dart » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698