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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/* __ *\
** ________ ___ / / ___ Scala API **
** / __/ __// _ | / / / _ | (c) 2003-2009, LAMP/EPFL **
** __\ \/ /__/ __ |/ /__/ __ | http://scala-lang.org/ **
** /____/\___/_/ |_/____/_/ | | **
** |/ **
\* */
// $Id$
package scala.collection.mutable
import Predef._
import generic._
/** This class implements mutable maps using a hashtable.
*
* @author Matthias Zenger
* @author Martin Odersky
* @version 2.8
*/
object LinkedHashMap extends MutableMapFactory[LinkedHashMap] {
implicit def builderFactory[A, B]: BuilderFactory[(A, B), LinkedHashMap[A, B], Coll] = new MapBuilderFactory[A, B]
def empty[A, B] = new LinkedHashMap[A, B]
}
@serializable
class LinkedHashMap[A, B] extends Map[A, B]
with MutableMapTemplate[A, B, LinkedHashMap[A, B]]
with HashTable[A] {
override def empty = LinkedHashMap.empty[A, B]
override def size = super[HashTable].size
type Entry = LinkedEntry[A, B]
protected var firstEntry: Entry = null
protected var lastEntry: Entry = null
def get(key: A): Option[B] = {
val e = findEntry(key)
if (e == null) None
else Some(e.value)
}
override def put(key: A, value: B): Option[B] = {
val e = findEntry(key)
if (e == null) {
val e = new Entry(key, value)
addEntry(e)
if (firstEntry == null) firstEntry = e
else { lastEntry.later = e; e.earlier = lastEntry }
lastEntry = e
None
} else {
val v = e.value
e.value = value
Some(v)
}
}
override def remove(key: A): Option[B] = {
val e = removeEntry(key)
if (e eq null) None
else {
if (e.earlier eq null) firstEntry = e.later
else e.earlier.later = e.later
if (e.later eq null) lastEntry = e.earlier
else e.later.earlier = e.earlier
Some(e.value)
}
}
def += (kv: (A, B)): this.type = { put(kv._1, kv._2); this }
def -=(key: A): this.type = { remove(key); this }
def iterator: Iterator[(A, B)] = new Iterator[(A, B)] {
private var cur = firstEntry
def hasNext = cur ne null
def next =
if (hasNext) { val res = (cur.key, cur.value); cur = cur.later; res }
else Iterator.empty.next
}
override def keysIterator: Iterator[A] = new Iterator[A] {
private var cur = firstEntry
def hasNext = cur ne null
def next =
if (hasNext) { val res = cur.key; cur = cur.later; res }
else Iterator.empty.next
}
override def valuesIterator: Iterator[B] = new Iterator[B] {
private var cur = firstEntry
def hasNext = cur ne null
def next = { val res = cur.value; cur = cur.later; res }
if (hasNext) { val res = cur.value; cur = cur.later; res }
else Iterator.empty.next
}
override def foreach[U](f: ((A, B)) => U) = {
var cur = firstEntry
while (cur ne null) {
f((cur.key, cur.value))
cur = cur.later
}
}
override def clear() {
super[HashTable].clear()
firstEntry = null
}
}
|