blob: e3a998845b37b31987393f69174c07d0ce186abd (
plain)
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
|
// Note: We are using the built-in equality to compare keys.
// Note: Our specifications are rather tied to the implementation. In
// particular, we use indices into .keys and .values. Nicer would
// be if Dafny had a built-in map type that could be used in specifications.
// Or, perhaps inductive data types could be used in specifications,
// if Dafny had those.
class Map<Key,Value> {
var keys: seq<Key>;
var values: seq<Value>;
function Valid(): bool
reads this;
{
|keys| == |values| &&
(forall i, j :: 0 <= i && i < j && j < |keys| ==> keys[i] != keys[j])
}
method Init()
modifies this;
ensures Valid() && |keys| == 0;
{
keys := [];
values := [];
}
method Find(key: Key) returns (present: bool, val: Value)
requires Valid();
ensures !present ==> key !in keys;
ensures present ==> (exists i :: 0 <= i && i < |keys| &&
keys[i] == key && values[i] == val);
{
call j := FindIndex(key);
if (j == -1) {
present := false;
} else {
present := true;
val := values[j];
}
}
method Add(key: Key, val: Value)
requires Valid();
modifies this;
ensures Valid();
ensures (forall i :: 0 <= i && i < |keys| && old(keys)[i] == key ==>
keys[i] == key && values[i] == val &&
(forall j :: 0 <= j && j < |values| && i != j ==> keys[j] == old(keys)[j] && values[j] == old(values)[j]));
ensures key !in old(keys) ==> keys == old(keys) + [key] && values == old(values) + [val];
{
call j := FindIndex(key);
if (j == -1) {
keys := keys + [key];
values := values + [val];
assert values[|keys|-1] == val; // lemma
} else {
values := values[..j] + [val] + values[j+1..];
assert values[j] == val; //lemma
}
}
method Remove(key: Key)
requires Valid();
modifies this;
ensures Valid();
// no key is introduced:
ensures (forall k :: k in keys ==> k in old(keys));
// at most one key is removed:
ensures (forall k :: k in old(keys) ==> k in keys || k == key);
// the given key is not there:
// other values don't change:
ensures key !in old(keys) ==> keys == old(keys) && values == old(values);
ensures key in old(keys) ==>
key !in keys &&
(exists h ::
0 <= h && h <= |keys| &&
keys[..h] == old(keys)[..h] &&
values[..h] == old(values)[..h] &&
keys[h..] == old(keys)[h+1..] &&
values[h..] == old(values)[h+1..]);
{
call j := FindIndex(key);
if (0 <= j) {
keys := keys[..j] + keys[j+1..];
values := values[..j] + values[j+1..];
}
}
method FindIndex(key: Key) returns (idx: int)
requires Valid();
ensures -1 <= idx && idx < |keys|;
ensures idx == -1 ==> key !in keys;
ensures 0 <= idx ==> keys[idx] == key;
{
var j := 0;
while (j < |keys|)
invariant j <= |keys|;
invariant key !in keys[..j];
decreases |keys| -j;
{
if (keys[j] == key) {
idx := j;
return;
}
j := j + 1;
}
idx := -1;
}
}
|
