summaryrefslogtreecommitdiff
path: root/src/org/clojure/runtime/PersistentHashtableIdentityMap.java
blob: 8935493cf8f380d3c7b03488022c33c68ba12213 (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
/**
 *   Copyright (c) Rich Hickey. All rights reserved.
 *   The use and distribution terms for this software are covered by the
 *   Common Public License 1.0 (http://opensource.org/licenses/cpl.php)
 *   which can be found in the file CPL.TXT at the root of this distribution.
 *   By using this software in any fashion, you are agreeing to be bound by
 * 	 the terms of this license.
 *   You must not remove this notice, or any other, from this software.
 **/

package org.clojure.runtime;

import java.util.Iterator;

public class PersistentHashtableIdentityMap extends PersistentHashtableMap {

public PersistentHashtableIdentityMap(int initialCapacity) {
    super(initialCapacity);
}

public PersistentHashtableIdentityMap(Object[] init) {
    super(init);
}

PersistentHashtableIdentityMap(int count, PersistentArray array) {
    super(count, array);
}


public Iterator<IMapEntry> iterator() {
    return new Iter(array);
}


static class Iter implements Iterator{
    PersistentArray buckets;
    int b;
    PersistentListIdentityMap e;

    Iter(PersistentArray buckets){
        this.buckets = buckets;
        this.b = -1;
        nextBucket();
    }

    private void nextBucket() {
        e = null;
        for(b = b+1;b<buckets.length();b++)
            {
            PersistentListIdentityMap a = (PersistentListIdentityMap) buckets.get(b);
            if(a != null && a != PersistentListIdentityMap.EMPTY)
                {
                e = a;
                break;
                }
            }
    }

    public boolean hasNext() {
        return e != null;
    }

    public Object next() {
        PersistentListIdentityMap ret = e;
        e = e.rest();
        if(e == PersistentListIdentityMap.EMPTY)
            nextBucket();
        return ret;
    }

    public void remove() {
        throw new UnsupportedOperationException();
    }
}

IPersistentMap create(int capacity) {
    return new PersistentHashtableIdentityMap(capacity);
}

IPersistentMap create(int count, PersistentArray array) {
    return new PersistentHashtableIdentityMap(count, array);
}

IPersistentMap createListMap(Object key, Object val){
    return PersistentListIdentityMap.create(key,val);
}

}