1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
/* __ *\
** ________ ___ / / ___ Scala API **
** / __/ __// _ | / / / _ | (c) 2006-2008, LAMP/EPFL **
** __\ \/ /__/ __ |/ /__/ __ | http://www.scala-lang.org/ **
** /____/\___/_/ |_/____/_/ | | **
** |/ **
\* */
// $Id$
package scala.collection.jcl
/** A wrapper around a Java map.
*
* @author Sean McDirmid
*/
trait MapWrapper[K, E] extends jcl.Map[K, E] {
def underlying: java.util.Map[K, E]
override def size = underlying.size
override def isEmpty = underlying.isEmpty
override def clear() = underlying.clear
override def put(key: K, elem: E) = {
//if (elem == null) throw new IllegalArgumentException;
val ret = underlying.put(key, elem)
if (ret == null) None else Some(ret.asInstanceOf[E])
}
override def get(key : K) : Option[E] = {
val ret = underlying.get(key);
if (ret == null) None else Some(ret.asInstanceOf[E]);
}
override def ++=(that : Iterable[Tuple2[K,E]]) : Unit = that match {
case that : MapWrapper[_,_] => underlying.putAll(that.underlying);
case _ => super.++=(that)
}
override def removeKey(key: K) = {
val ret = underlying.remove(key)
if (ret == null) None else Some(ret.asInstanceOf[E])
}
override def contains(key: K) = underlying.containsKey(key)
override def keySet: Set.Projection[K] = new KeySet
override def valueSet: MutableIterable.Projection[E] = new ValueSet
override def elements: MutableIterator[Tuple2[K,E]] = new IteratorWrapper
class IteratorWrapper extends MutableIterator[Tuple2[K,E]] {
val underlying = MapWrapper.this.underlying.entrySet.iterator
def hasNext = underlying.hasNext
def remove = underlying.remove
def next = {
val next = underlying.next.asInstanceOf[java.util.Map.Entry[K,E]]
Tuple2(next.getKey.asInstanceOf[K],next.getValue.asInstanceOf[E])
}
}
class KeySet extends super.KeySet with SetWrapper[K] with Set.Projection[K] {
val underlying = MapWrapper.this.underlying.keySet
}
class ValueSet extends IterableWrapper[E] with MutableIterable.Projection[E] {
override def size = MapWrapper.this.size
val underlying = MapWrapper.this.underlying.values
override def has(e : E) = MapWrapper.this.underlying.containsValue(e)
}
override def toString = underlying.toString
override def hashCode = underlying.hashCode
override def equals(that : Any) = that match {
case that: MapWrapper[_,_] => underlying == that.underlying
case _ => super.equals(that)
}
}
|
