summaryrefslogtreecommitdiff
path: root/chapter20/src/main/java/com/seaofnodes/simple/Ary.java
blob: f43179ad593c94206d0df42ae51c2b459a1d787a (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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
package com.seaofnodes.simple;

import java.lang.StringBuilder;
import java.lang.reflect.Array;
import java.util.*;

// ArrayList with saner syntax
public class Ary<E> extends AbstractList<E> implements List<E> {
    public E[] _es;
    public int _len;

    public Ary(E[] es) { this(es,es.length); }
    public Ary(E[] es, int len) { if( es.length==0 ) es=Arrays.copyOf(es,1); _es=es; _len=len; }
    @SuppressWarnings("unchecked")
    public Ary(Class<E> clazz) { this((E[]) Array.newInstance(clazz, 1),0); }

    /** @return list is empty */
    public boolean isEmpty() { return _len==0; }
    /** @return active list length */
    public int len() { return _len; }
    public int size() { return len(); }

    /** @param i element index
     *  @return element being returned; throws if OOB
     *  @exception AIOOBE if !(0 <= i < _len)
     */
    public E get( int i ) { return at(i); }
    public E at( int i ) {
        range_check(i);
        return _es[i];
    }
    public E atX( int i ) {
        if( i >= _len ) return null;
        return _es[i];
    }

    /** @return last element */
    public E last( ) { return at(_len-1); }

    /** Add element in amortized constant time
     *  @param e Element to add at end of list
     **/
    public E push( E e ) {
        if( _len >= _es.length ) _es = Arrays.copyOf(_es,Math.max(1,_es.length<<1));
        return (_es[_len++] = e);
    }
    /** @return remove and return last element */
    public E pop( ) {
        return _len==0 ? null : _es[--_len];
    }
    @Override public E removeLast() { return pop(); }


    /** Add element in amortized constant time
     *  @param e Element to add at end of list
     *  @return true */
    public boolean add( E e ) {
        if( _len >= _es.length ) _es = Arrays.copyOf(_es,Math.max(1,_es.length<<1));
        _es[_len++] = e;
        return true;
    }

    /** Fast, constant-time, element removal.  Does not preserve order
     *  @param i element to be removed
     *  @return element removed */
    public E del( int i ) {
        range_check(i);
        E tmp = _es[i];
        _es[i]=_es[--_len];
        return tmp;
    }

    /** Slow, linear-time, element removal.  Preserves order.
     *  @param i element to be removed
     *  @return element removed */
    @Override public E remove( int i ) {
        range_check(i);
        E e = _es[i];
        System.arraycopy(_es,i+1,_es,i,(--_len)-i);
        return e;
    }

    /** Slow, linear-time, element insertion.  Preserves order.
     *  @param i where to insert
     *  @return element inserted */
    public E insert( E e, int i ) {
        setLen(_len+1);
        System.arraycopy(_es,i,_es,i+1,_len-i);
        _es[i] = e;
        return e;
    }


    /** Remove all elements */
    public void clear( ) { _len=0; }

    /** Set existing element
     *  @param i element to set
     *  @param e value to set
     *  @return old value
     *  @exception AIOOBE if !(0 <= i < _len)
     */
    public E set( int i, E e ) {
        range_check(i);
        E old = _es[i];
        _es[i] = e;
        return old;
    }

    /** Set existing element
     *  @param i element to set
     *  @param e value to set
     *  @return old value
     *  @exception AIOOBE if !(0 <= i < _len)
     */
    public E setX( int i, E e ) {
        if( i >= _len ) {
            while( i >= _es.length ) _es = Arrays.copyOf( _es, _es.length << 1 );
            _len = i+1;
        }
        E old = _es[i];
        _es[i] = e;
        return old;
    }

    public Ary<E> setLen( int len ) {
        if( len > _len )
            while( len>= _es.length ) _es = Arrays.copyOf(_es,_es.length<<1);
        _len = len;
        while( _es.length > (len<<1) ) // Shrink if hugely too large
            _es = Arrays.copyOf(_es,_es.length>>1);
        Arrays.fill(_es,len,_es.length,null);
        return this;
    }

    public E swap( int i, int j ) {
        range_check(i);
        range_check(j);
        E tmp = _es[i];
        if( i==j ) return tmp;
        _es[i] = _es[j];
        _es[j] = tmp;
        return tmp;
    }


    /** @param c Collection to be added */
    public boolean addAll( Collection<? extends E> c ) {
        if( c==null ) return false;
        for( E e : c ) add(e);
        return true;
    }
    public Ary<E> addAll( Iterable<? extends E> it ) { for( E e : it ) add(e); return this; }

    /** @param es Array to be added */
    public <F extends E> Ary<E> addAll( F[] es ) {
        if( es==null || es.length==0 ) return this;
        while( _len+es.length > _es.length ) _es = Arrays.copyOf(_es,_es.length<<1);
        System.arraycopy(es,0,_es,_len,es.length);
        _len += es.length;
        return this;
    }

    /** @param c Collection to be added */
    public Ary<E> addAll( Ary<? extends E> c ) {
        if( c==null || c._len==0 ) return this;
        while( _len+c._len > _es.length ) _es = Arrays.copyOf(_es,_es.length<<1);
        System.arraycopy(c._es,0,_es,_len,c._len);
        _len += c._len;
        return this;
    }

    /** @return compact array version */
    public E[] asAry() { return Arrays.copyOf(_es,_len); }

    /** Find the first matching element using ==, or -1 if none.  Note that
     *  most del calls shuffle the list, so the first element might be random.
     *  @param e Element to find
     *  @return index of first matching element, or -1 if none */
    public int find( E e ) {
        for( int i=0; i<_len; i++ )  if( _es[i]==e )  return i;
        return -1;
    }


    /** @return an iterator */
    @Override public Iterator<E> iterator() { return new Iter(); }
    private class Iter implements Iterator<E> {
        int _i=0;
        @Override public boolean hasNext() { return _i<_len; }
        @Override public E next() { return _es[_i++]; }
    }

    @Override public String toString() {
        StringBuilder sb = new StringBuilder().append('{');
        for( int i=0; i<_len; i++ ) {
            if( i>0 ) sb.append(',');
            if( _es[i] != null ) sb.append(_es[i].toString());
        }
        return sb.append('}').toString();
    }

    private void range_check( int i ) {
        if( i < 0 || i>=_len )
            throw new ArrayIndexOutOfBoundsException(""+i+" >= "+_len);
    }

    // Note that the hashCode() and equals() are not invariant to changes in the
    // underlying array.  If the hashCode() is used (e.g., inserting into a
    // HashMap) and the then the array changes, the hashCode() will change also.
    @Override public boolean equals( Object o ) {
        if( this==o ) return true;
        if( !(o instanceof Ary ary) ) return false;
        if( _len != ary._len ) return false;
        if( _es == ary._es ) return true;
        for( int i=0; i<_len; i++ )
            if( !(_es[i]==null ? (ary._es[i] == null) : _es[i].equals(ary._es[i])) )
                return false;
        return true;
    }
    @Override public int hashCode( ) {
        int sum=_len;
        for( int i=0; i<_len; i++ )
            sum += _es[i]==null ? 0 : _es[i].hashCode();
        return sum;
    }

}