ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/main/java/util/ArrayDeque.java
(Generate patch)

Comparing jsr166/src/main/java/util/ArrayDeque.java (file contents):
Revision 1.108 by jsr166, Sat Nov 5 14:41:14 2016 UTC vs.
Revision 1.121 by jsr166, Mon Nov 21 15:30:44 2016 UTC

# Line 68 | Line 68 | public class ArrayDeque<E> extends Abstr
68       *
69       * Because in a circular array, elements are in general stored in
70       * two disjoint such slices, we help the VM by writing unusual
71 <     * nested loops for all traversals over the elements.
71 >     * nested loops for all traversals over the elements.  Having only
72 >     * one hot inner loop body instead of two or three eases human
73 >     * maintenance and encourages VM loop inlining into the caller.
74       */
75  
76      /**
77       * The array in which the elements of the deque are stored.
78 <     * We guarantee that all array cells not holding deque elements
79 <     * are always null.
78 >     * All array cells not holding deque elements are always null.
79 >     * The array always has at least one null slot (at tail).
80       */
81      transient Object[] elements;
82  
# Line 88 | Line 90 | public class ArrayDeque<E> extends Abstr
90  
91      /**
92       * The index at which the next element would be added to the tail
93 <     * of the deque (via addLast(E), add(E), or push(E)).
93 >     * of the deque (via addLast(E), add(E), or push(E));
94 >     * elements[tail] is always null.
95       */
96      transient int tail;
97  
# Line 187 | Line 190 | public class ArrayDeque<E> extends Abstr
190       * @param numElements lower bound on initial capacity of the deque
191       */
192      public ArrayDeque(int numElements) {
193 <        elements = new Object[Math.max(1, numElements + 1)];
193 >        elements =
194 >            new Object[(numElements < 1) ? 1 :
195 >                       (numElements == Integer.MAX_VALUE) ? Integer.MAX_VALUE :
196 >                       numElements + 1];
197      }
198  
199      /**
# Line 201 | Line 207 | public class ArrayDeque<E> extends Abstr
207       * @throws NullPointerException if the specified collection is null
208       */
209      public ArrayDeque(Collection<? extends E> c) {
210 <        elements = new Object[c.size() + 1];
210 >        this(c.size());
211          addAll(c);
212      }
213  
# Line 224 | Line 230 | public class ArrayDeque<E> extends Abstr
230      }
231  
232      /**
233 <     * Adds i and j, mod modulus.
234 <     * Precondition and postcondition: 0 <= i < modulus, 0 <= j <= modulus.
233 >     * Circularly adds the given distance to index i, mod modulus.
234 >     * Precondition: 0 <= i < modulus, 0 <= distance <= modulus.
235 >     * @return index 0 <= i < modulus
236       */
237 <    static final int add(int i, int j, int modulus) {
238 <        if ((i += j) - modulus >= 0) i -= modulus;
237 >    static final int add(int i, int distance, int modulus) {
238 >        if ((i += distance) - modulus >= 0) distance -= modulus;
239          return i;
240      }
241  
242      /**
243       * Subtracts j from i, mod modulus.
244 <     * Index i must be logically ahead of j.
245 <     * Returns the "circular distance" from j to i.
246 <     * Precondition and postcondition: 0 <= i < modulus, 0 <= j < modulus.
244 >     * Index i must be logically ahead of index j.
245 >     * Precondition: 0 <= i < modulus, 0 <= j < modulus.
246 >     * @return the "circular distance" from j to i; corner case i == j
247 >     * is diambiguated to "empty", returning 0.
248       */
249      static final int sub(int i, int j, int modulus) {
250          if ((i -= j) < 0) i += modulus;
# Line 244 | Line 252 | public class ArrayDeque<E> extends Abstr
252      }
253  
254      /**
247     * Returns the array index of the last element.
248     * May return invalid index -1 if there are no elements.
249     */
250    final int last() {
251        return dec(tail, elements.length);
252    }
253
254    /**
255       * Returns element at array index i.
256       * This is a slight abuse of generics, accepted by javac.
257       */
# Line 322 | Line 322 | public class ArrayDeque<E> extends Abstr
322       *         of its elements are null
323       */
324      public boolean addAll(Collection<? extends E> c) {
325 <        final int s = size(), needed;
326 <        if ((needed = s + c.size() - elements.length + 1) > 0)
325 >        final int s, needed;
326 >        if ((needed = (s = size()) + c.size() + 1 - elements.length) > 0)
327              grow(needed);
328 <        c.forEach((e) -> addLast(e));
328 >        c.forEach(this::addLast);
329          // checkInvariants();
330          return size() > s;
331      }
# Line 477 | Line 477 | public class ArrayDeque<E> extends Abstr
477              final Object[] es = elements;
478              for (int i = tail, end = head, to = (i >= end) ? end : 0;
479                   ; i = es.length, to = end) {
480 <                while (--i >= to)
480 >                for (i--; i > to - 1; i--)
481                      if (o.equals(es[i])) {
482                          delete(i);
483                          return true;
# Line 616 | Line 616 | public class ArrayDeque<E> extends Abstr
616          // checkInvariants();
617          final Object[] es = elements;
618          final int capacity = es.length;
619 <        final int h = head;
619 >        final int h, t;
620          // number of elements before to-be-deleted elt
621 <        final int front = sub(i, h, capacity);
622 <        final int back = size() - front - 1; // number of elements after
621 >        final int front = sub(i, h = head, capacity);
622 >        // number of elements after to-be-deleted elt
623 >        final int back = sub(t = tail, i, capacity) - 1;
624          if (front < back) {
625              // move front elements forwards
626              if (h <= i) {
# Line 635 | Line 636 | public class ArrayDeque<E> extends Abstr
636              return false;
637          } else {
638              // move back elements backwards
639 <            tail = dec(tail, capacity);
639 >            tail = dec(t, capacity);
640              if (i <= tail) {
641                  System.arraycopy(es, i + 1, es, i, back);
642              } else { // Wrap around
643 <                int firstLeg = capacity - (i + 1);
643 <                System.arraycopy(es, i + 1, es, i, firstLeg);
643 >                System.arraycopy(es, i + 1, es, i, capacity - (i + 1));
644                  es[capacity - 1] = es[0];
645 <                System.arraycopy(es, 1, es, 0, back - firstLeg - 1);
645 >                System.arraycopy(es, 1, es, 0, t - 1);
646              }
647              es[tail] = null;
648              // checkInvariants();
# Line 752 | Line 752 | public class ArrayDeque<E> extends Abstr
752      }
753  
754      private class DescendingIterator extends DeqIterator {
755 <        DescendingIterator() { cursor = last(); }
755 >        DescendingIterator() { cursor = dec(tail, elements.length); }
756  
757          public final E next() {
758              if (remaining <= 0)
# Line 781 | Line 781 | public class ArrayDeque<E> extends Abstr
781                  throw new ConcurrentModificationException();
782              for (int i = cursor, end = head, to = (i >= end) ? end : 0;
783                   ; i = es.length - 1, to = end) {
784 <                for (; i >= to; i--)
784 >                // hotspot generates faster code than for: i >= to !
785 >                for (; i > to - 1; i--)
786                      action.accept(elementAt(es, i));
787                  if (to == end) {
788                      if (end != head)
789                          throw new ConcurrentModificationException();
790 <                    lastRet = head;
790 >                    lastRet = end;
791                      break;
792                  }
793              }
# Line 863 | Line 864 | public class ArrayDeque<E> extends Abstr
864          }
865  
866          public boolean tryAdvance(Consumer<? super E> action) {
867 <            if (action == null)
868 <                throw new NullPointerException();
869 <            int t, i;
870 <            if ((t = fence) < 0) t = getFence();
871 <            if (t == (i = cursor))
867 >            Objects.requireNonNull(action);
868 >            final Object[] es = elements;
869 >            if (fence < 0) { fence = tail; cursor = head; } // late-binding
870 >            final int i;
871 >            if ((i = cursor) == fence)
872                  return false;
873 <            final Object[] es;
873 <            action.accept(nonNullElementAt(es = elements, i));
873 >            E e = nonNullElementAt(es, i);
874              cursor = inc(i, es.length);
875 +            action.accept(e);
876              return true;
877          }
878  
# Line 957 | Line 958 | public class ArrayDeque<E> extends Abstr
958               ; i = 0, to = end) {
959              for (; i < to; i++)
960                  if (filter.test(elementAt(es, i)))
961 <                    return bulkRemoveModified(filter, i, to);
961 >                    return bulkRemoveModified(filter, i);
962              if (to == end) {
963                  if (end != tail) throw new ConcurrentModificationException();
964                  break;
# Line 966 | Line 967 | public class ArrayDeque<E> extends Abstr
967          return false;
968      }
969  
970 +    // A tiny bit set implementation
971 +
972 +    private static long[] nBits(int n) {
973 +        return new long[((n - 1) >> 6) + 1];
974 +    }
975 +    private static void setBit(long[] bits, int i) {
976 +        bits[i >> 6] |= 1L << i;
977 +    }
978 +    private static boolean isClear(long[] bits, int i) {
979 +        return (bits[i >> 6] & (1L << i)) == 0;
980 +    }
981 +
982      /**
983       * Helper for bulkRemove, in case of at least one deletion.
984 <     * @param i valid index of first element to be deleted
984 >     * Tolerate predicates that reentrantly access the collection for
985 >     * read (but writers still get CME), so traverse once to find
986 >     * elements to delete, a second pass to physically expunge.
987 >     *
988 >     * @param beg valid index of first element to be deleted
989       */
990      private boolean bulkRemoveModified(
991 <        Predicate<? super E> filter, int i, int to) {
991 >        Predicate<? super E> filter, final int beg) {
992          final Object[] es = elements;
993          final int capacity = es.length;
977        // a two-finger algorithm, with hare i reading, tortoise j writing
978        int j = i++;
994          final int end = tail;
995 <        try {
996 <            for (;; j = 0) {    // j rejoins i on second leg
997 <                E e;
998 <                // In this loop, i and j are on the same leg, with i > j
999 <                for (; i < to; i++)
1000 <                    if (!filter.test(e = elementAt(es, i)))
1001 <                        es[j++] = e;
1002 <                if (to == end) break;
1003 <                // In this loop, j is on the first leg, i on the second
1004 <                for (i = 0, to = end; i < to && j < capacity; i++)
1005 <                    if (!filter.test(e = elementAt(es, i)))
1006 <                        es[j++] = e;
1007 <                if (i >= to) {
1008 <                    if (j == capacity) j = 0; // "corner" case
1009 <                    break;
1010 <                }
995 >        final long[] deathRow = nBits(sub(end, beg, capacity));
996 >        deathRow[0] = 1L;   // set bit 0
997 >        for (int i = beg + 1, to = (i <= end) ? end : es.length, k = beg;
998 >             ; i = 0, to = end, k -= capacity) {
999 >            for (; i < to; i++)
1000 >                if (filter.test(elementAt(es, i)))
1001 >                    setBit(deathRow, i - k);
1002 >            if (to == end) break;
1003 >        }
1004 >        // a two-finger traversal, with hare i reading, tortoise w writing
1005 >        int w = beg;
1006 >        for (int i = beg + 1, to = (i <= end) ? end : es.length, k = beg;
1007 >             ; w = 0) { // w rejoins i on second leg
1008 >            // In this loop, i and w are on the same leg, with i > w
1009 >            for (; i < to; i++)
1010 >                if (isClear(deathRow, i - k))
1011 >                    es[w++] = es[i];
1012 >            if (to == end) break;
1013 >            // In this loop, w is on the first leg, i on the second
1014 >            for (i = 0, to = end, k -= capacity; i < to && w < capacity; i++)
1015 >                if (isClear(deathRow, i - k))
1016 >                    es[w++] = es[i];
1017 >            if (i >= to) {
1018 >                if (w == capacity) w = 0; // "corner" case
1019 >                break;
1020              }
997            return true;
998        } catch (Throwable ex) {
999            // copy remaining elements
1000            for (; i != end; i = inc(i, capacity), j = inc(j, capacity))
1001                es[j] = es[i];
1002            throw ex;
1003        } finally {
1004            if (end != tail) throw new ConcurrentModificationException();
1005            circularClear(es, tail = j, end);
1006            // checkInvariants();
1021          }
1022 +        if (end != tail) throw new ConcurrentModificationException();
1023 +        circularClear(es, tail = w, end);
1024 +        // checkInvariants();
1025 +        return true;
1026      }
1027  
1028      /**
# Line 1087 | Line 1105 | public class ArrayDeque<E> extends Abstr
1105      private <T> T[] toArray(Class<T[]> klazz) {
1106          final Object[] es = elements;
1107          final T[] a;
1108 <        final int size = size(), head = this.head, end;
1109 <        final int len = Math.min(size, es.length - head);
1110 <        if ((end = head + size) >= 0) {
1108 >        final int head = this.head, tail = this.tail, end;
1109 >        if ((end = tail + ((head <= tail) ? 0 : es.length)) >= 0) {
1110 >            // Uses null extension feature of copyOfRange
1111              a = Arrays.copyOfRange(es, head, end, klazz);
1112          } else {
1113              // integer overflow!
1114 <            a = Arrays.copyOfRange(es, 0, size, klazz);
1115 <            System.arraycopy(es, head, a, 0, len);
1114 >            a = Arrays.copyOfRange(es, 0, end - head, klazz);
1115 >            System.arraycopy(es, head, a, 0, es.length - head);
1116          }
1117 <        if (tail < head)
1118 <            System.arraycopy(es, 0, a, len, tail);
1117 >        if (end != tail)
1118 >            System.arraycopy(es, 0, a, es.length - head, tail);
1119          return a;
1120      }
1121  
# Line 1222 | Line 1240 | public class ArrayDeque<E> extends Abstr
1240  
1241      /** debugging */
1242      void checkInvariants() {
1243 +        // Use head and tail fields with empty slot at tail strategy.
1244 +        // head == tail disambiguates to "empty".
1245          try {
1246              int capacity = elements.length;
1247              // assert head >= 0 && head < capacity;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines