author | psandoz |
Thu, 01 Aug 2013 15:28:57 +0100 | |
changeset 19188 | bbf287c5cd92 |
parent 18572 | 53b8b8c30086 |
child 19220 | d3d40ccb544e |
permissions | -rw-r--r-- |
17182 | 1 |
/* |
2 |
* Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. |
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
|
4 |
* |
|
5 |
* This code is free software; you can redistribute it and/or modify it |
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
|
7 |
* published by the Free Software Foundation. Oracle designates this |
|
8 |
* particular file as subject to the "Classpath" exception as provided |
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
|
10 |
* |
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
|
15 |
* accompanied this code). |
|
16 |
* |
|
17 |
* You should have received a copy of the GNU General Public License version |
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
20 |
* |
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
22 |
* or visit www.oracle.com if you need additional information or have any |
|
23 |
* questions. |
|
24 |
*/ |
|
25 |
package java.util.stream; |
|
26 |
||
27 |
import java.util.Comparator; |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
28 |
import java.util.Objects; |
17182 | 29 |
import java.util.Spliterator; |
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
30 |
import java.util.concurrent.atomic.AtomicLong; |
17182 | 31 |
import java.util.function.BooleanSupplier; |
32 |
import java.util.function.Consumer; |
|
33 |
import java.util.function.DoubleConsumer; |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
34 |
import java.util.function.DoubleSupplier; |
17182 | 35 |
import java.util.function.IntConsumer; |
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
36 |
import java.util.function.IntSupplier; |
17182 | 37 |
import java.util.function.LongConsumer; |
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
38 |
import java.util.function.LongSupplier; |
17182 | 39 |
import java.util.function.Supplier; |
40 |
||
41 |
/** |
|
42 |
* Spliterator implementations for wrapping and delegating spliterators, used |
|
43 |
* in the implementation of the {@link Stream#spliterator()} method. |
|
44 |
* |
|
45 |
* @since 1.8 |
|
46 |
*/ |
|
47 |
class StreamSpliterators { |
|
48 |
||
49 |
/** |
|
50 |
* Abstract wrapping spliterator that binds to the spliterator of a |
|
51 |
* pipeline helper on first operation. |
|
52 |
* |
|
53 |
* <p>This spliterator is not late-binding and will bind to the source |
|
54 |
* spliterator when first operated on. |
|
55 |
* |
|
56 |
* <p>A wrapping spliterator produced from a sequential stream |
|
57 |
* cannot be split if there are stateful operations present. |
|
58 |
*/ |
|
59 |
private static abstract class AbstractWrappingSpliterator<P_IN, P_OUT, |
|
60 |
T_BUFFER extends AbstractSpinedBuffer> |
|
61 |
implements Spliterator<P_OUT> { |
|
62 |
||
63 |
// @@@ Detect if stateful operations are present or not |
|
64 |
// If not then can split otherwise cannot |
|
65 |
||
66 |
/** |
|
67 |
* True if this spliterator supports splitting |
|
68 |
*/ |
|
69 |
final boolean isParallel; |
|
70 |
||
71 |
final PipelineHelper<P_OUT> ph; |
|
72 |
||
73 |
/** |
|
74 |
* Supplier for the source spliterator. Client provides either a |
|
75 |
* spliterator or a supplier. |
|
76 |
*/ |
|
77 |
private Supplier<Spliterator<P_IN>> spliteratorSupplier; |
|
78 |
||
79 |
/** |
|
80 |
* Source spliterator. Either provided from client or obtained from |
|
81 |
* supplier. |
|
82 |
*/ |
|
83 |
Spliterator<P_IN> spliterator; |
|
84 |
||
85 |
/** |
|
86 |
* Sink chain for the downstream stages of the pipeline, ultimately |
|
87 |
* leading to the buffer. Used during partial traversal. |
|
88 |
*/ |
|
89 |
Sink<P_IN> bufferSink; |
|
90 |
||
91 |
/** |
|
92 |
* A function that advances one element of the spliterator, pushing |
|
93 |
* it to bufferSink. Returns whether any elements were processed. |
|
94 |
* Used during partial traversal. |
|
95 |
*/ |
|
96 |
BooleanSupplier pusher; |
|
97 |
||
98 |
/** Next element to consume from the buffer, used during partial traversal */ |
|
99 |
long nextToConsume; |
|
100 |
||
101 |
/** Buffer into which elements are pushed. Used during partial traversal. */ |
|
102 |
T_BUFFER buffer; |
|
103 |
||
104 |
/** |
|
105 |
* True if full traversal has occurred (with possible cancelation). |
|
106 |
* If doing a partial traversal, there may be still elements in buffer. |
|
107 |
*/ |
|
108 |
boolean finished; |
|
109 |
||
110 |
/** |
|
111 |
* Construct an AbstractWrappingSpliterator from a |
|
112 |
* {@code Supplier<Spliterator>}. |
|
113 |
*/ |
|
114 |
AbstractWrappingSpliterator(PipelineHelper<P_OUT> ph, |
|
115 |
Supplier<Spliterator<P_IN>> spliteratorSupplier, |
|
116 |
boolean parallel) { |
|
117 |
this.ph = ph; |
|
118 |
this.spliteratorSupplier = spliteratorSupplier; |
|
119 |
this.spliterator = null; |
|
120 |
this.isParallel = parallel; |
|
121 |
} |
|
122 |
||
123 |
/** |
|
124 |
* Construct an AbstractWrappingSpliterator from a |
|
125 |
* {@code Spliterator}. |
|
126 |
*/ |
|
127 |
AbstractWrappingSpliterator(PipelineHelper<P_OUT> ph, |
|
128 |
Spliterator<P_IN> spliterator, |
|
129 |
boolean parallel) { |
|
130 |
this.ph = ph; |
|
131 |
this.spliteratorSupplier = null; |
|
132 |
this.spliterator = spliterator; |
|
133 |
this.isParallel = parallel; |
|
134 |
} |
|
135 |
||
136 |
/** |
|
137 |
* Called before advancing to set up spliterator, if needed. |
|
138 |
*/ |
|
139 |
final void init() { |
|
140 |
if (spliterator == null) { |
|
141 |
spliterator = spliteratorSupplier.get(); |
|
142 |
spliteratorSupplier = null; |
|
143 |
} |
|
144 |
} |
|
145 |
||
146 |
/** |
|
147 |
* Get an element from the source, pushing it into the sink chain, |
|
148 |
* setting up the buffer if needed |
|
149 |
* @return whether there are elements to consume from the buffer |
|
150 |
*/ |
|
151 |
final boolean doAdvance() { |
|
152 |
if (buffer == null) { |
|
153 |
if (finished) |
|
154 |
return false; |
|
155 |
||
156 |
init(); |
|
157 |
initPartialTraversalState(); |
|
158 |
nextToConsume = 0; |
|
159 |
bufferSink.begin(spliterator.getExactSizeIfKnown()); |
|
160 |
return fillBuffer(); |
|
161 |
} |
|
162 |
else { |
|
163 |
++nextToConsume; |
|
164 |
boolean hasNext = nextToConsume < buffer.count(); |
|
165 |
if (!hasNext) { |
|
166 |
nextToConsume = 0; |
|
167 |
buffer.clear(); |
|
168 |
hasNext = fillBuffer(); |
|
169 |
} |
|
170 |
return hasNext; |
|
171 |
} |
|
172 |
} |
|
173 |
||
174 |
/** |
|
175 |
* Invokes the shape-specific constructor with the provided arguments |
|
176 |
* and returns the result. |
|
177 |
*/ |
|
178 |
abstract AbstractWrappingSpliterator<P_IN, P_OUT, ?> wrap(Spliterator<P_IN> s); |
|
179 |
||
180 |
/** |
|
181 |
* Initializes buffer, sink chain, and pusher for a shape-specific |
|
182 |
* implementation. |
|
183 |
*/ |
|
184 |
abstract void initPartialTraversalState(); |
|
185 |
||
186 |
@Override |
|
187 |
public Spliterator<P_OUT> trySplit() { |
|
188 |
if (isParallel && !finished) { |
|
189 |
init(); |
|
190 |
||
191 |
Spliterator<P_IN> split = spliterator.trySplit(); |
|
192 |
return (split == null) ? null : wrap(split); |
|
193 |
} |
|
194 |
else |
|
195 |
return null; |
|
196 |
} |
|
197 |
||
198 |
/** |
|
199 |
* If the buffer is empty, push elements into the sink chain until |
|
200 |
* the source is empty or cancellation is requested. |
|
201 |
* @return whether there are elements to consume from the buffer |
|
202 |
*/ |
|
203 |
private boolean fillBuffer() { |
|
204 |
while (buffer.count() == 0) { |
|
205 |
if (bufferSink.cancellationRequested() || !pusher.getAsBoolean()) { |
|
206 |
if (finished) |
|
207 |
return false; |
|
208 |
else { |
|
209 |
bufferSink.end(); // might trigger more elements |
|
210 |
finished = true; |
|
211 |
} |
|
212 |
} |
|
213 |
} |
|
214 |
return true; |
|
215 |
} |
|
216 |
||
217 |
@Override |
|
218 |
public final long estimateSize() { |
|
219 |
init(); |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
220 |
// Use the estimate of the wrapped spliterator |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
221 |
// Note this may not be accurate if there are filter/flatMap |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
222 |
// operations filtering or adding elements to the stream |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
223 |
return spliterator.estimateSize(); |
17182 | 224 |
} |
225 |
||
226 |
@Override |
|
227 |
public final long getExactSizeIfKnown() { |
|
228 |
init(); |
|
229 |
return StreamOpFlag.SIZED.isKnown(ph.getStreamAndOpFlags()) |
|
230 |
? spliterator.getExactSizeIfKnown() |
|
231 |
: -1; |
|
232 |
} |
|
233 |
||
234 |
@Override |
|
235 |
public final int characteristics() { |
|
236 |
init(); |
|
237 |
||
238 |
// Get the characteristics from the pipeline |
|
239 |
int c = StreamOpFlag.toCharacteristics(StreamOpFlag.toStreamFlags(ph.getStreamAndOpFlags())); |
|
240 |
||
241 |
// Mask off the size and uniform characteristics and replace with |
|
242 |
// those of the spliterator |
|
243 |
// Note that a non-uniform spliterator can change from something |
|
244 |
// with an exact size to an estimate for a sub-split, for example |
|
245 |
// with HashSet where the size is known at the top level spliterator |
|
246 |
// but for sub-splits only an estimate is known |
|
247 |
if ((c & Spliterator.SIZED) != 0) { |
|
248 |
c &= ~(Spliterator.SIZED | Spliterator.SUBSIZED); |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
249 |
c |= (spliterator.characteristics() & (Spliterator.SIZED | Spliterator.SUBSIZED)); |
17182 | 250 |
} |
251 |
||
252 |
return c; |
|
253 |
} |
|
254 |
||
255 |
@Override |
|
256 |
public Comparator<? super P_OUT> getComparator() { |
|
257 |
if (!hasCharacteristics(SORTED)) |
|
258 |
throw new IllegalStateException(); |
|
259 |
return null; |
|
260 |
} |
|
261 |
||
262 |
@Override |
|
263 |
public final String toString() { |
|
264 |
return String.format("%s[%s]", getClass().getName(), spliterator); |
|
265 |
} |
|
266 |
} |
|
267 |
||
268 |
static final class WrappingSpliterator<P_IN, P_OUT> |
|
269 |
extends AbstractWrappingSpliterator<P_IN, P_OUT, SpinedBuffer<P_OUT>> { |
|
270 |
||
271 |
WrappingSpliterator(PipelineHelper<P_OUT> ph, |
|
272 |
Supplier<Spliterator<P_IN>> supplier, |
|
273 |
boolean parallel) { |
|
274 |
super(ph, supplier, parallel); |
|
275 |
} |
|
276 |
||
277 |
WrappingSpliterator(PipelineHelper<P_OUT> ph, |
|
278 |
Spliterator<P_IN> spliterator, |
|
279 |
boolean parallel) { |
|
280 |
super(ph, spliterator, parallel); |
|
281 |
} |
|
282 |
||
283 |
@Override |
|
284 |
WrappingSpliterator<P_IN, P_OUT> wrap(Spliterator<P_IN> s) { |
|
285 |
return new WrappingSpliterator<>(ph, s, isParallel); |
|
286 |
} |
|
287 |
||
288 |
@Override |
|
289 |
void initPartialTraversalState() { |
|
290 |
SpinedBuffer<P_OUT> b = new SpinedBuffer<>(); |
|
291 |
buffer = b; |
|
292 |
bufferSink = ph.wrapSink(b::accept); |
|
293 |
pusher = () -> spliterator.tryAdvance(bufferSink); |
|
294 |
} |
|
295 |
||
296 |
@Override |
|
297 |
public boolean tryAdvance(Consumer<? super P_OUT> consumer) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
298 |
Objects.requireNonNull(consumer); |
17182 | 299 |
boolean hasNext = doAdvance(); |
300 |
if (hasNext) |
|
301 |
consumer.accept(buffer.get(nextToConsume)); |
|
302 |
return hasNext; |
|
303 |
} |
|
304 |
||
305 |
@Override |
|
306 |
public void forEachRemaining(Consumer<? super P_OUT> consumer) { |
|
307 |
if (buffer == null && !finished) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
308 |
Objects.requireNonNull(consumer); |
17182 | 309 |
init(); |
310 |
||
311 |
ph.wrapAndCopyInto((Sink<P_OUT>) consumer::accept, spliterator); |
|
312 |
finished = true; |
|
313 |
} |
|
314 |
else { |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
315 |
do { } while (tryAdvance(consumer)); |
17182 | 316 |
} |
317 |
} |
|
318 |
} |
|
319 |
||
320 |
static final class IntWrappingSpliterator<P_IN> |
|
321 |
extends AbstractWrappingSpliterator<P_IN, Integer, SpinedBuffer.OfInt> |
|
322 |
implements Spliterator.OfInt { |
|
323 |
||
324 |
IntWrappingSpliterator(PipelineHelper<Integer> ph, |
|
325 |
Supplier<Spliterator<P_IN>> supplier, |
|
326 |
boolean parallel) { |
|
327 |
super(ph, supplier, parallel); |
|
328 |
} |
|
329 |
||
330 |
IntWrappingSpliterator(PipelineHelper<Integer> ph, |
|
331 |
Spliterator<P_IN> spliterator, |
|
332 |
boolean parallel) { |
|
333 |
super(ph, spliterator, parallel); |
|
334 |
} |
|
335 |
||
336 |
@Override |
|
337 |
AbstractWrappingSpliterator<P_IN, Integer, ?> wrap(Spliterator<P_IN> s) { |
|
338 |
return new IntWrappingSpliterator<>(ph, s, isParallel); |
|
339 |
} |
|
340 |
||
341 |
@Override |
|
342 |
void initPartialTraversalState() { |
|
343 |
SpinedBuffer.OfInt b = new SpinedBuffer.OfInt(); |
|
344 |
buffer = b; |
|
345 |
bufferSink = ph.wrapSink((Sink.OfInt) b::accept); |
|
346 |
pusher = () -> spliterator.tryAdvance(bufferSink); |
|
347 |
} |
|
348 |
||
349 |
@Override |
|
350 |
public Spliterator.OfInt trySplit() { |
|
351 |
return (Spliterator.OfInt) super.trySplit(); |
|
352 |
} |
|
353 |
||
354 |
@Override |
|
355 |
public boolean tryAdvance(IntConsumer consumer) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
356 |
Objects.requireNonNull(consumer); |
17182 | 357 |
boolean hasNext = doAdvance(); |
358 |
if (hasNext) |
|
359 |
consumer.accept(buffer.get(nextToConsume)); |
|
360 |
return hasNext; |
|
361 |
} |
|
362 |
||
363 |
@Override |
|
364 |
public void forEachRemaining(IntConsumer consumer) { |
|
365 |
if (buffer == null && !finished) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
366 |
Objects.requireNonNull(consumer); |
17182 | 367 |
init(); |
368 |
||
369 |
ph.wrapAndCopyInto((Sink.OfInt) consumer::accept, spliterator); |
|
370 |
finished = true; |
|
371 |
} |
|
372 |
else { |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
373 |
do { } while (tryAdvance(consumer)); |
17182 | 374 |
} |
375 |
} |
|
376 |
} |
|
377 |
||
378 |
static final class LongWrappingSpliterator<P_IN> |
|
379 |
extends AbstractWrappingSpliterator<P_IN, Long, SpinedBuffer.OfLong> |
|
380 |
implements Spliterator.OfLong { |
|
381 |
||
382 |
LongWrappingSpliterator(PipelineHelper<Long> ph, |
|
383 |
Supplier<Spliterator<P_IN>> supplier, |
|
384 |
boolean parallel) { |
|
385 |
super(ph, supplier, parallel); |
|
386 |
} |
|
387 |
||
388 |
LongWrappingSpliterator(PipelineHelper<Long> ph, |
|
389 |
Spliterator<P_IN> spliterator, |
|
390 |
boolean parallel) { |
|
391 |
super(ph, spliterator, parallel); |
|
392 |
} |
|
393 |
||
394 |
@Override |
|
395 |
AbstractWrappingSpliterator<P_IN, Long, ?> wrap(Spliterator<P_IN> s) { |
|
396 |
return new LongWrappingSpliterator<>(ph, s, isParallel); |
|
397 |
} |
|
398 |
||
399 |
@Override |
|
400 |
void initPartialTraversalState() { |
|
401 |
SpinedBuffer.OfLong b = new SpinedBuffer.OfLong(); |
|
402 |
buffer = b; |
|
403 |
bufferSink = ph.wrapSink((Sink.OfLong) b::accept); |
|
404 |
pusher = () -> spliterator.tryAdvance(bufferSink); |
|
405 |
} |
|
406 |
||
407 |
@Override |
|
408 |
public Spliterator.OfLong trySplit() { |
|
409 |
return (Spliterator.OfLong) super.trySplit(); |
|
410 |
} |
|
411 |
||
412 |
@Override |
|
413 |
public boolean tryAdvance(LongConsumer consumer) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
414 |
Objects.requireNonNull(consumer); |
17182 | 415 |
boolean hasNext = doAdvance(); |
416 |
if (hasNext) |
|
417 |
consumer.accept(buffer.get(nextToConsume)); |
|
418 |
return hasNext; |
|
419 |
} |
|
420 |
||
421 |
@Override |
|
422 |
public void forEachRemaining(LongConsumer consumer) { |
|
423 |
if (buffer == null && !finished) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
424 |
Objects.requireNonNull(consumer); |
17182 | 425 |
init(); |
426 |
||
427 |
ph.wrapAndCopyInto((Sink.OfLong) consumer::accept, spliterator); |
|
428 |
finished = true; |
|
429 |
} |
|
430 |
else { |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
431 |
do { } while (tryAdvance(consumer)); |
17182 | 432 |
} |
433 |
} |
|
434 |
} |
|
435 |
||
436 |
static final class DoubleWrappingSpliterator<P_IN> |
|
437 |
extends AbstractWrappingSpliterator<P_IN, Double, SpinedBuffer.OfDouble> |
|
438 |
implements Spliterator.OfDouble { |
|
439 |
||
440 |
DoubleWrappingSpliterator(PipelineHelper<Double> ph, |
|
441 |
Supplier<Spliterator<P_IN>> supplier, |
|
442 |
boolean parallel) { |
|
443 |
super(ph, supplier, parallel); |
|
444 |
} |
|
445 |
||
446 |
DoubleWrappingSpliterator(PipelineHelper<Double> ph, |
|
447 |
Spliterator<P_IN> spliterator, |
|
448 |
boolean parallel) { |
|
449 |
super(ph, spliterator, parallel); |
|
450 |
} |
|
451 |
||
452 |
@Override |
|
453 |
AbstractWrappingSpliterator<P_IN, Double, ?> wrap(Spliterator<P_IN> s) { |
|
454 |
return new DoubleWrappingSpliterator<>(ph, s, isParallel); |
|
455 |
} |
|
456 |
||
457 |
@Override |
|
458 |
void initPartialTraversalState() { |
|
459 |
SpinedBuffer.OfDouble b = new SpinedBuffer.OfDouble(); |
|
460 |
buffer = b; |
|
461 |
bufferSink = ph.wrapSink((Sink.OfDouble) b::accept); |
|
462 |
pusher = () -> spliterator.tryAdvance(bufferSink); |
|
463 |
} |
|
464 |
||
465 |
@Override |
|
466 |
public Spliterator.OfDouble trySplit() { |
|
467 |
return (Spliterator.OfDouble) super.trySplit(); |
|
468 |
} |
|
469 |
||
470 |
@Override |
|
471 |
public boolean tryAdvance(DoubleConsumer consumer) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
472 |
Objects.requireNonNull(consumer); |
17182 | 473 |
boolean hasNext = doAdvance(); |
474 |
if (hasNext) |
|
475 |
consumer.accept(buffer.get(nextToConsume)); |
|
476 |
return hasNext; |
|
477 |
} |
|
478 |
||
479 |
@Override |
|
480 |
public void forEachRemaining(DoubleConsumer consumer) { |
|
481 |
if (buffer == null && !finished) { |
|
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
482 |
Objects.requireNonNull(consumer); |
17182 | 483 |
init(); |
484 |
||
485 |
ph.wrapAndCopyInto((Sink.OfDouble) consumer::accept, spliterator); |
|
486 |
finished = true; |
|
487 |
} |
|
488 |
else { |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
489 |
do { } while (tryAdvance(consumer)); |
17182 | 490 |
} |
491 |
} |
|
492 |
} |
|
493 |
||
494 |
/** |
|
495 |
* Spliterator implementation that delegates to an underlying spliterator, |
|
496 |
* acquiring the spliterator from a {@code Supplier<Spliterator>} on the |
|
497 |
* first call to any spliterator method. |
|
498 |
* @param <T> |
|
499 |
*/ |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
500 |
static class DelegatingSpliterator<T, T_SPLITR extends Spliterator<T>> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
501 |
implements Spliterator<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
502 |
private final Supplier<? extends T_SPLITR> supplier; |
17182 | 503 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
504 |
private T_SPLITR s; |
17182 | 505 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
506 |
DelegatingSpliterator(Supplier<? extends T_SPLITR> supplier) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
507 |
this.supplier = supplier; |
17182 | 508 |
} |
509 |
||
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
510 |
T_SPLITR get() { |
17182 | 511 |
if (s == null) { |
512 |
s = supplier.get(); |
|
513 |
} |
|
514 |
return s; |
|
515 |
} |
|
516 |
||
517 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
518 |
public T_SPLITR trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
519 |
return (T_SPLITR) get().trySplit(); |
17182 | 520 |
} |
521 |
||
522 |
@Override |
|
523 |
public boolean tryAdvance(Consumer<? super T> consumer) { |
|
524 |
return get().tryAdvance(consumer); |
|
525 |
} |
|
526 |
||
527 |
@Override |
|
528 |
public void forEachRemaining(Consumer<? super T> consumer) { |
|
529 |
get().forEachRemaining(consumer); |
|
530 |
} |
|
531 |
||
532 |
@Override |
|
533 |
public long estimateSize() { |
|
534 |
return get().estimateSize(); |
|
535 |
} |
|
536 |
||
537 |
@Override |
|
538 |
public int characteristics() { |
|
539 |
return get().characteristics(); |
|
540 |
} |
|
541 |
||
542 |
@Override |
|
543 |
public Comparator<? super T> getComparator() { |
|
544 |
return get().getComparator(); |
|
545 |
} |
|
546 |
||
547 |
@Override |
|
548 |
public long getExactSizeIfKnown() { |
|
549 |
return get().getExactSizeIfKnown(); |
|
550 |
} |
|
551 |
||
552 |
@Override |
|
553 |
public String toString() { |
|
554 |
return getClass().getName() + "[" + get() + "]"; |
|
555 |
} |
|
556 |
||
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
557 |
static class OfPrimitive<T, T_CONS, T_SPLITR extends Spliterator.OfPrimitive<T, T_CONS, T_SPLITR>> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
558 |
extends DelegatingSpliterator<T, T_SPLITR> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
559 |
implements Spliterator.OfPrimitive<T, T_CONS, T_SPLITR> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
560 |
OfPrimitive(Supplier<? extends T_SPLITR> supplier) { |
17182 | 561 |
super(supplier); |
562 |
} |
|
563 |
||
564 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
565 |
public boolean tryAdvance(T_CONS consumer) { |
17182 | 566 |
return get().tryAdvance(consumer); |
567 |
} |
|
568 |
||
569 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
570 |
public void forEachRemaining(T_CONS consumer) { |
17182 | 571 |
get().forEachRemaining(consumer); |
572 |
} |
|
573 |
} |
|
574 |
||
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
575 |
static final class OfInt |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
576 |
extends OfPrimitive<Integer, IntConsumer, Spliterator.OfInt> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
577 |
implements Spliterator.OfInt { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
578 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
579 |
OfInt(Supplier<Spliterator.OfInt> supplier) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
580 |
super(supplier); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
581 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
582 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
583 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
584 |
static final class OfLong |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
585 |
extends OfPrimitive<Long, LongConsumer, Spliterator.OfLong> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
586 |
implements Spliterator.OfLong { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
587 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
588 |
OfLong(Supplier<Spliterator.OfLong> supplier) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
589 |
super(supplier); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
590 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
591 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
592 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
593 |
static final class OfDouble |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
594 |
extends OfPrimitive<Double, DoubleConsumer, Spliterator.OfDouble> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
595 |
implements Spliterator.OfDouble { |
17182 | 596 |
|
597 |
OfDouble(Supplier<Spliterator.OfDouble> supplier) { |
|
598 |
super(supplier); |
|
599 |
} |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
600 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
601 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
602 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
603 |
/** |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
604 |
* A slice Spliterator from a source Spliterator that reports |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
605 |
* {@code SUBSIZED}. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
606 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
607 |
*/ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
608 |
static abstract class SliceSpliterator<T, T_SPLITR extends Spliterator<T>> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
609 |
// The start index of the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
610 |
final long sliceOrigin; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
611 |
// One past the last index of the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
612 |
final long sliceFence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
613 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
614 |
// The spliterator to slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
615 |
T_SPLITR s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
616 |
// current (absolute) index, modified on advance/split |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
617 |
long index; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
618 |
// one past last (absolute) index or sliceFence, which ever is smaller |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
619 |
long fence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
620 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
621 |
SliceSpliterator(T_SPLITR s, long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
622 |
assert s.hasCharacteristics(Spliterator.SUBSIZED); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
623 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
624 |
this.sliceOrigin = sliceOrigin; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
625 |
this.sliceFence = sliceFence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
626 |
this.index = origin; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
627 |
this.fence = fence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
628 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
629 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
630 |
protected abstract T_SPLITR makeSpliterator(T_SPLITR s, long sliceOrigin, long sliceFence, long origin, long fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
631 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
632 |
public T_SPLITR trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
633 |
if (sliceOrigin >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
634 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
635 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
636 |
if (index >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
637 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
638 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
639 |
// Keep splitting until the left and right splits intersect with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
640 |
// thereby ensuring the size estimate decreases. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
641 |
// This also avoids creating empty spliterators which can result in |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
642 |
// existing and additionally created F/J tasks that perform |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
643 |
// redundant work on no elements. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
644 |
while (true) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
645 |
T_SPLITR leftSplit = (T_SPLITR) s.trySplit(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
646 |
if (leftSplit == null) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
647 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
648 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
649 |
long leftSplitFenceUnbounded = index + leftSplit.estimateSize(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
650 |
long leftSplitFence = Math.min(leftSplitFenceUnbounded, sliceFence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
651 |
if (sliceOrigin >= leftSplitFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
652 |
// The left split does not intersect with, and is to the left of, the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
653 |
// The right split does intersect |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
654 |
// Discard the left split and split further with the right split |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
655 |
index = leftSplitFence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
656 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
657 |
else if (leftSplitFence >= sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
658 |
// The right split does not intersect with, and is to the right of, the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
659 |
// The left split does intersect |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
660 |
// Discard the right split and split further with the left split |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
661 |
s = leftSplit; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
662 |
fence = leftSplitFence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
663 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
664 |
else if (index >= sliceOrigin && leftSplitFenceUnbounded <= sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
665 |
// The left split is contained within the slice, return the underlying left split |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
666 |
// Right split is contained within or intersects with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
667 |
index = leftSplitFence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
668 |
return leftSplit; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
669 |
} else { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
670 |
// The left split intersects with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
671 |
// Right split is contained within or intersects with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
672 |
return makeSpliterator(leftSplit, sliceOrigin, sliceFence, index, index = leftSplitFence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
673 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
674 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
675 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
676 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
677 |
public long estimateSize() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
678 |
return (sliceOrigin < fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
679 |
? fence - Math.max(sliceOrigin, index) : 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
680 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
681 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
682 |
public int characteristics() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
683 |
return s.characteristics(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
684 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
685 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
686 |
static final class OfRef<T> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
687 |
extends SliceSpliterator<T, Spliterator<T>> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
688 |
implements Spliterator<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
689 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
690 |
OfRef(Spliterator<T> s, long sliceOrigin, long sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
691 |
this(s, sliceOrigin, sliceFence, 0, Math.min(s.estimateSize(), sliceFence)); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
692 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
693 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
694 |
private OfRef(Spliterator<T> s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
695 |
long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
696 |
super(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
697 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
698 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
699 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
700 |
protected Spliterator<T> makeSpliterator(Spliterator<T> s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
701 |
long sliceOrigin, long sliceFence, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
702 |
long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
703 |
return new OfRef<>(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
704 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
705 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
706 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
707 |
public boolean tryAdvance(Consumer<? super T> action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
708 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
709 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
710 |
if (sliceOrigin >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
711 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
712 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
713 |
while (sliceOrigin > index) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
714 |
s.tryAdvance(e -> {}); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
715 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
716 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
717 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
718 |
if (index >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
719 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
720 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
721 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
722 |
return s.tryAdvance(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
723 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
724 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
725 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
726 |
public void forEachRemaining(Consumer<? super T> action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
727 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
728 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
729 |
if (sliceOrigin >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
730 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
731 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
732 |
if (index >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
733 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
734 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
735 |
if (index >= sliceOrigin && (index + s.estimateSize()) <= sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
736 |
// The spliterator is contained within the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
737 |
s.forEachRemaining(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
738 |
index = fence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
739 |
} else { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
740 |
// The spliterator intersects with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
741 |
while (sliceOrigin > index) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
742 |
s.tryAdvance(e -> {}); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
743 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
744 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
745 |
// Traverse elements up to the fence |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
746 |
for (;index < fence; index++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
747 |
s.tryAdvance(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
748 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
749 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
750 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
751 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
752 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
753 |
static abstract class OfPrimitive<T, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
754 |
T_SPLITR extends Spliterator.OfPrimitive<T, T_CONS, T_SPLITR>, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
755 |
T_CONS> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
756 |
extends SliceSpliterator<T, T_SPLITR> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
757 |
implements Spliterator.OfPrimitive<T, T_CONS, T_SPLITR> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
758 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
759 |
OfPrimitive(T_SPLITR s, long sliceOrigin, long sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
760 |
this(s, sliceOrigin, sliceFence, 0, Math.min(s.estimateSize(), sliceFence)); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
761 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
762 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
763 |
private OfPrimitive(T_SPLITR s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
764 |
long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
765 |
super(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
766 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
767 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
768 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
769 |
public boolean tryAdvance(T_CONS action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
770 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
771 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
772 |
if (sliceOrigin >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
773 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
774 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
775 |
while (sliceOrigin > index) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
776 |
s.tryAdvance(emptyConsumer()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
777 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
778 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
779 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
780 |
if (index >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
781 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
782 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
783 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
784 |
return s.tryAdvance(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
785 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
786 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
787 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
788 |
public void forEachRemaining(T_CONS action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
789 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
790 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
791 |
if (sliceOrigin >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
792 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
793 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
794 |
if (index >= fence) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
795 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
796 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
797 |
if (index >= sliceOrigin && (index + s.estimateSize()) <= sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
798 |
// The spliterator is contained within the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
799 |
s.forEachRemaining(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
800 |
index = fence; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
801 |
} else { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
802 |
// The spliterator intersects with the slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
803 |
while (sliceOrigin > index) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
804 |
s.tryAdvance(emptyConsumer()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
805 |
index++; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
806 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
807 |
// Traverse elements up to the fence |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
808 |
for (;index < fence; index++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
809 |
s.tryAdvance(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
810 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
811 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
812 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
813 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
814 |
protected abstract T_CONS emptyConsumer(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
815 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
816 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
817 |
static final class OfInt extends OfPrimitive<Integer, Spliterator.OfInt, IntConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
818 |
implements Spliterator.OfInt { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
819 |
OfInt(Spliterator.OfInt s, long sliceOrigin, long sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
820 |
super(s, sliceOrigin, sliceFence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
821 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
822 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
823 |
OfInt(Spliterator.OfInt s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
824 |
long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
825 |
super(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
826 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
827 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
828 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
829 |
protected Spliterator.OfInt makeSpliterator(Spliterator.OfInt s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
830 |
long sliceOrigin, long sliceFence, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
831 |
long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
832 |
return new SliceSpliterator.OfInt(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
833 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
834 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
835 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
836 |
protected IntConsumer emptyConsumer() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
837 |
return e -> {}; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
838 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
839 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
840 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
841 |
static final class OfLong extends OfPrimitive<Long, Spliterator.OfLong, LongConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
842 |
implements Spliterator.OfLong { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
843 |
OfLong(Spliterator.OfLong s, long sliceOrigin, long sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
844 |
super(s, sliceOrigin, sliceFence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
845 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
846 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
847 |
OfLong(Spliterator.OfLong s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
848 |
long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
849 |
super(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
850 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
851 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
852 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
853 |
protected Spliterator.OfLong makeSpliterator(Spliterator.OfLong s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
854 |
long sliceOrigin, long sliceFence, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
855 |
long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
856 |
return new SliceSpliterator.OfLong(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
857 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
858 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
859 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
860 |
protected LongConsumer emptyConsumer() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
861 |
return e -> {}; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
862 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
863 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
864 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
865 |
static final class OfDouble extends OfPrimitive<Double, Spliterator.OfDouble, DoubleConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
866 |
implements Spliterator.OfDouble { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
867 |
OfDouble(Spliterator.OfDouble s, long sliceOrigin, long sliceFence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
868 |
super(s, sliceOrigin, sliceFence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
869 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
870 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
871 |
OfDouble(Spliterator.OfDouble s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
872 |
long sliceOrigin, long sliceFence, long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
873 |
super(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
874 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
875 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
876 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
877 |
protected Spliterator.OfDouble makeSpliterator(Spliterator.OfDouble s, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
878 |
long sliceOrigin, long sliceFence, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
879 |
long origin, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
880 |
return new SliceSpliterator.OfDouble(s, sliceOrigin, sliceFence, origin, fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
881 |
} |
17182 | 882 |
|
883 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
884 |
protected DoubleConsumer emptyConsumer() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
885 |
return e -> {}; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
886 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
887 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
888 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
889 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
890 |
/** |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
891 |
* A slice Spliterator that does not preserve order, if any, of a source |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
892 |
* Spliterator. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
893 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
894 |
* Note: The source spliterator may report {@code ORDERED} since that |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
895 |
* spliterator be the result of a previous pipeline stage that was |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
896 |
* collected to a {@code Node}. It is the order of the pipeline stage |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
897 |
* that governs whether the this slice spliterator is to be used or not. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
898 |
*/ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
899 |
static abstract class UnorderedSliceSpliterator<T, T_SPLITR extends Spliterator<T>> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
900 |
static final int CHUNK_SIZE = 1 << 7; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
901 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
902 |
// The spliterator to slice |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
903 |
protected final T_SPLITR s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
904 |
protected final boolean unlimited; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
905 |
private final long skipThreshold; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
906 |
private final AtomicLong permits; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
907 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
908 |
UnorderedSliceSpliterator(T_SPLITR s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
909 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
910 |
this.unlimited = limit < 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
911 |
this.skipThreshold = limit >= 0 ? limit : 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
912 |
this.permits = new AtomicLong(limit >= 0 ? skip + limit : skip); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
913 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
914 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
915 |
UnorderedSliceSpliterator(T_SPLITR s, UnorderedSliceSpliterator parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
916 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
917 |
this.unlimited = parent.unlimited; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
918 |
this.permits = parent.permits; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
919 |
this.skipThreshold = parent.skipThreshold; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
920 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
921 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
922 |
/** |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
923 |
* Acquire permission to skip or process elements. The caller must |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
924 |
* first acquire the elements, then consult this method for guidance |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
925 |
* as to what to do with the data. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
926 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
927 |
* <p>We use an {@code AtomicLong} to atomically maintain a counter, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
928 |
* which is initialized as skip+limit if we are limiting, or skip only |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
929 |
* if we are not limiting. The user should consult the method |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
930 |
* {@code checkPermits()} before acquiring data elements. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
931 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
932 |
* @param numElements the number of elements the caller has in hand |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
933 |
* @return the number of elements that should be processed; any |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
934 |
* remaining elements should be discarded. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
935 |
*/ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
936 |
protected final long acquirePermits(long numElements) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
937 |
long remainingPermits; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
938 |
long grabbing; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
939 |
// permits never increase, and don't decrease below zero |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
940 |
assert numElements > 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
941 |
do { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
942 |
remainingPermits = permits.get(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
943 |
if (remainingPermits == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
944 |
return unlimited ? numElements : 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
945 |
grabbing = Math.min(remainingPermits, numElements); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
946 |
} while (grabbing > 0 && |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
947 |
!permits.compareAndSet(remainingPermits, remainingPermits - grabbing)); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
948 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
949 |
if (unlimited) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
950 |
return Math.max(numElements - grabbing, 0); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
951 |
else if (remainingPermits > skipThreshold) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
952 |
return Math.max(grabbing - (remainingPermits - skipThreshold), 0); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
953 |
else |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
954 |
return grabbing; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
955 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
956 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
957 |
enum PermitStatus { NO_MORE, MAYBE_MORE, UNLIMITED } |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
958 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
959 |
/** Call to check if permits might be available before acquiring data */ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
960 |
protected final PermitStatus permitStatus() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
961 |
if (permits.get() > 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
962 |
return PermitStatus.MAYBE_MORE; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
963 |
else |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
964 |
return unlimited ? PermitStatus.UNLIMITED : PermitStatus.NO_MORE; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
965 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
966 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
967 |
public final T_SPLITR trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
968 |
// Stop splitting when there are no more limit permits |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
969 |
if (permits.get() == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
970 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
971 |
T_SPLITR split = (T_SPLITR) s.trySplit(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
972 |
return split == null ? null : makeSpliterator(split); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
973 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
974 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
975 |
protected abstract T_SPLITR makeSpliterator(T_SPLITR s); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
976 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
977 |
public final long estimateSize() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
978 |
return s.estimateSize(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
979 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
980 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
981 |
public final int characteristics() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
982 |
return s.characteristics() & |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
983 |
~(Spliterator.SIZED | Spliterator.SUBSIZED | Spliterator.ORDERED); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
984 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
985 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
986 |
static final class OfRef<T> extends UnorderedSliceSpliterator<T, Spliterator<T>> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
987 |
implements Spliterator<T>, Consumer<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
988 |
T tmpSlot; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
989 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
990 |
OfRef(Spliterator<T> s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
991 |
super(s, skip, limit); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
992 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
993 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
994 |
OfRef(Spliterator<T> s, OfRef parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
995 |
super(s, parent); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
996 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
997 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
998 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
999 |
public final void accept(T t) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1000 |
tmpSlot = t; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1001 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1002 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1003 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1004 |
public boolean tryAdvance(Consumer<? super T> action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1005 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1006 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1007 |
while (permitStatus() != PermitStatus.NO_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1008 |
if (!s.tryAdvance(this)) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1009 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1010 |
else if (acquirePermits(1) == 1) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1011 |
action.accept(tmpSlot); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1012 |
tmpSlot = null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1013 |
return true; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1014 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1015 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1016 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1017 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1018 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1019 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1020 |
public void forEachRemaining(Consumer<? super T> action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1021 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1022 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1023 |
ArrayBuffer.OfRef<T> sb = null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1024 |
PermitStatus permitStatus; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1025 |
while ((permitStatus = permitStatus()) != PermitStatus.NO_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1026 |
if (permitStatus == PermitStatus.MAYBE_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1027 |
// Optimistically traverse elements up to a threshold of CHUNK_SIZE |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1028 |
if (sb == null) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1029 |
sb = new ArrayBuffer.OfRef<>(CHUNK_SIZE); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1030 |
else |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1031 |
sb.reset(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1032 |
long permitsRequested = 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1033 |
do { } while (s.tryAdvance(sb) && ++permitsRequested < CHUNK_SIZE); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1034 |
if (permitsRequested == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1035 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1036 |
sb.forEach(action, acquirePermits(permitsRequested)); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1037 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1038 |
else { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1039 |
// Must be UNLIMITED; let 'er rip |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1040 |
s.forEachRemaining(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1041 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1042 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1043 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1044 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1045 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1046 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1047 |
protected Spliterator<T> makeSpliterator(Spliterator<T> s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1048 |
return new UnorderedSliceSpliterator.OfRef<>(s, this); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1049 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1050 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1051 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1052 |
/** |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1053 |
* Concrete sub-types must also be an instance of type {@code T_CONS}. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1054 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1055 |
* @param <T_BUFF> the type of the spined buffer. Must also be a type of |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1056 |
* {@code T_CONS}. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1057 |
*/ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1058 |
static abstract class OfPrimitive< |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1059 |
T, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1060 |
T_CONS, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1061 |
T_BUFF extends ArrayBuffer.OfPrimitive<T_CONS>, |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1062 |
T_SPLITR extends Spliterator.OfPrimitive<T, T_CONS, T_SPLITR>> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1063 |
extends UnorderedSliceSpliterator<T, T_SPLITR> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1064 |
implements Spliterator.OfPrimitive<T, T_CONS, T_SPLITR> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1065 |
OfPrimitive(T_SPLITR s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1066 |
super(s, skip, limit); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1067 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1068 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1069 |
OfPrimitive(T_SPLITR s, UnorderedSliceSpliterator.OfPrimitive parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1070 |
super(s, parent); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1071 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1072 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1073 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1074 |
public boolean tryAdvance(T_CONS action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1075 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1076 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1077 |
while (permitStatus() != PermitStatus.NO_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1078 |
if (!s.tryAdvance((T_CONS) this)) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1079 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1080 |
else if (acquirePermits(1) == 1) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1081 |
acceptConsumed(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1082 |
return true; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1083 |
} |
17182 | 1084 |
} |
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1085 |
return false; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1086 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1087 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1088 |
protected abstract void acceptConsumed(T_CONS action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1089 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1090 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1091 |
public void forEachRemaining(T_CONS action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1092 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1093 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1094 |
T_BUFF sb = null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1095 |
PermitStatus permitStatus; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1096 |
while ((permitStatus = permitStatus()) != PermitStatus.NO_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1097 |
if (permitStatus == PermitStatus.MAYBE_MORE) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1098 |
// Optimistically traverse elements up to a threshold of CHUNK_SIZE |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1099 |
if (sb == null) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1100 |
sb = bufferCreate(CHUNK_SIZE); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1101 |
else |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1102 |
sb.reset(); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1103 |
@SuppressWarnings("unchecked") |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1104 |
T_CONS sbc = (T_CONS) sb; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1105 |
long permitsRequested = 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1106 |
do { } while (s.tryAdvance(sbc) && ++permitsRequested < CHUNK_SIZE); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1107 |
if (permitsRequested == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1108 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1109 |
sb.forEach(action, acquirePermits(permitsRequested)); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1110 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1111 |
else { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1112 |
// Must be UNLIMITED; let 'er rip |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1113 |
s.forEachRemaining(action); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1114 |
return; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1115 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1116 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1117 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1118 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1119 |
protected abstract T_BUFF bufferCreate(int initialCapacity); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1120 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1121 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1122 |
static final class OfInt |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1123 |
extends OfPrimitive<Integer, IntConsumer, ArrayBuffer.OfInt, Spliterator.OfInt> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1124 |
implements Spliterator.OfInt, IntConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1125 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1126 |
int tmpValue; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1127 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1128 |
OfInt(Spliterator.OfInt s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1129 |
super(s, skip, limit); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1130 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1131 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1132 |
OfInt(Spliterator.OfInt s, UnorderedSliceSpliterator.OfInt parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1133 |
super(s, parent); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1134 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1135 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1136 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1137 |
public void accept(int value) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1138 |
tmpValue = value; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1139 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1140 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1141 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1142 |
protected void acceptConsumed(IntConsumer action) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1143 |
action.accept(tmpValue); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1144 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1145 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1146 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1147 |
protected ArrayBuffer.OfInt bufferCreate(int initialCapacity) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1148 |
return new ArrayBuffer.OfInt(initialCapacity); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1149 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1150 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1151 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1152 |
protected Spliterator.OfInt makeSpliterator(Spliterator.OfInt s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1153 |
return new UnorderedSliceSpliterator.OfInt(s, this); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1154 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1155 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1156 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1157 |
static final class OfLong |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1158 |
extends OfPrimitive<Long, LongConsumer, ArrayBuffer.OfLong, Spliterator.OfLong> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1159 |
implements Spliterator.OfLong, LongConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1160 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1161 |
long tmpValue; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1162 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1163 |
OfLong(Spliterator.OfLong s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1164 |
super(s, skip, limit); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1165 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1166 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1167 |
OfLong(Spliterator.OfLong s, UnorderedSliceSpliterator.OfLong parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1168 |
super(s, parent); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1169 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1170 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1171 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1172 |
public void accept(long value) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1173 |
tmpValue = value; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1174 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1175 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1176 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1177 |
protected void acceptConsumed(LongConsumer action) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1178 |
action.accept(tmpValue); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1179 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1180 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1181 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1182 |
protected ArrayBuffer.OfLong bufferCreate(int initialCapacity) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1183 |
return new ArrayBuffer.OfLong(initialCapacity); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1184 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1185 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1186 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1187 |
protected Spliterator.OfLong makeSpliterator(Spliterator.OfLong s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1188 |
return new UnorderedSliceSpliterator.OfLong(s, this); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1189 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1190 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1191 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1192 |
static final class OfDouble |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1193 |
extends OfPrimitive<Double, DoubleConsumer, ArrayBuffer.OfDouble, Spliterator.OfDouble> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1194 |
implements Spliterator.OfDouble, DoubleConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1195 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1196 |
double tmpValue; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1197 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1198 |
OfDouble(Spliterator.OfDouble s, long skip, long limit) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1199 |
super(s, skip, limit); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1200 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1201 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1202 |
OfDouble(Spliterator.OfDouble s, UnorderedSliceSpliterator.OfDouble parent) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1203 |
super(s, parent); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1204 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1205 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1206 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1207 |
public void accept(double value) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1208 |
tmpValue = value; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1209 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1210 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1211 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1212 |
protected void acceptConsumed(DoubleConsumer action) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1213 |
action.accept(tmpValue); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1214 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1215 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1216 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1217 |
protected ArrayBuffer.OfDouble bufferCreate(int initialCapacity) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1218 |
return new ArrayBuffer.OfDouble(initialCapacity); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1219 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1220 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1221 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1222 |
protected Spliterator.OfDouble makeSpliterator(Spliterator.OfDouble s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1223 |
return new UnorderedSliceSpliterator.OfDouble(s, this); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1224 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1225 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1226 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1227 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1228 |
/** |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1229 |
* A Spliterator that infinitely supplies elements in no particular order. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1230 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1231 |
* <p>Splitting divides the estimated size in two and stops when the |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1232 |
* estimate size is 0. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1233 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1234 |
* <p>The {@code forEachRemaining} method if invoked will never terminate. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1235 |
* The {@coe tryAdvance} method always returns true. |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1236 |
* |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1237 |
*/ |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1238 |
static abstract class InfiniteSupplyingSpliterator<T> implements Spliterator<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1239 |
long estimate; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1240 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1241 |
protected InfiniteSupplyingSpliterator(long estimate) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1242 |
this.estimate = estimate; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1243 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1244 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1245 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1246 |
public long estimateSize() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1247 |
return estimate; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1248 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1249 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1250 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1251 |
public int characteristics() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1252 |
return IMMUTABLE; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1253 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1254 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1255 |
static final class OfRef<T> extends InfiniteSupplyingSpliterator<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1256 |
final Supplier<T> s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1257 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1258 |
OfRef(long size, Supplier<T> s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1259 |
super(size); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1260 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1261 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1262 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1263 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1264 |
public boolean tryAdvance(Consumer<? super T> action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1265 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1266 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1267 |
action.accept(s.get()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1268 |
return true; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1269 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1270 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1271 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1272 |
public Spliterator<T> trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1273 |
if (estimate == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1274 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1275 |
return new InfiniteSupplyingSpliterator.OfRef<>(estimate >>>= 1, s); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1276 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1277 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1278 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1279 |
static final class OfInt extends InfiniteSupplyingSpliterator<Integer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1280 |
implements Spliterator.OfInt { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1281 |
final IntSupplier s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1282 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1283 |
OfInt(long size, IntSupplier s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1284 |
super(size); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1285 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1286 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1287 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1288 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1289 |
public boolean tryAdvance(IntConsumer action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1290 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1291 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1292 |
action.accept(s.getAsInt()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1293 |
return true; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1294 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1295 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1296 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1297 |
public Spliterator.OfInt trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1298 |
if (estimate == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1299 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1300 |
return new InfiniteSupplyingSpliterator.OfInt(estimate = estimate >>> 1, s); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1301 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1302 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1303 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1304 |
static final class OfLong extends InfiniteSupplyingSpliterator<Long> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1305 |
implements Spliterator.OfLong { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1306 |
final LongSupplier s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1307 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1308 |
OfLong(long size, LongSupplier s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1309 |
super(size); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1310 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1311 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1312 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1313 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1314 |
public boolean tryAdvance(LongConsumer action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1315 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1316 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1317 |
action.accept(s.getAsLong()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1318 |
return true; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1319 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1320 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1321 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1322 |
public Spliterator.OfLong trySplit() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1323 |
if (estimate == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1324 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1325 |
return new InfiniteSupplyingSpliterator.OfLong(estimate = estimate >>> 1, s); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1326 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1327 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1328 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1329 |
static final class OfDouble extends InfiniteSupplyingSpliterator<Double> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1330 |
implements Spliterator.OfDouble { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1331 |
final DoubleSupplier s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1332 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1333 |
OfDouble(long size, DoubleSupplier s) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1334 |
super(size); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1335 |
this.s = s; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1336 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1337 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1338 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1339 |
public boolean tryAdvance(DoubleConsumer action) { |
19188
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1340 |
Objects.requireNonNull(action); |
bbf287c5cd92
8020016: Numerous splitereator impls do not throw NPE for null Consumers
psandoz
parents:
18572
diff
changeset
|
1341 |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1342 |
action.accept(s.getAsDouble()); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1343 |
return true; |
17182 | 1344 |
} |
1345 |
||
1346 |
@Override |
|
1347 |
public Spliterator.OfDouble trySplit() { |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1348 |
if (estimate == 0) |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1349 |
return null; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1350 |
return new InfiniteSupplyingSpliterator.OfDouble(estimate = estimate >>> 1, s); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1351 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1352 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1353 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1354 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1355 |
// @@@ Consolidate with Node.Builder |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1356 |
static abstract class ArrayBuffer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1357 |
int index; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1358 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1359 |
void reset() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1360 |
index = 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1361 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1362 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1363 |
static final class OfRef<T> extends ArrayBuffer implements Consumer<T> { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1364 |
final Object[] array; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1365 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1366 |
OfRef(int size) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1367 |
this.array = new Object[size]; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1368 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1369 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1370 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1371 |
public void accept(T t) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1372 |
array[index++] = t; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1373 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1374 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1375 |
public void forEach(Consumer<? super T> action, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1376 |
for (int i = 0; i < fence; i++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1377 |
@SuppressWarnings("unchecked") |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1378 |
T t = (T) array[i]; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1379 |
action.accept(t); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1380 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1381 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1382 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1383 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1384 |
static abstract class OfPrimitive<T_CONS> extends ArrayBuffer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1385 |
int index; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1386 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1387 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1388 |
void reset() { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1389 |
index = 0; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1390 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1391 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1392 |
abstract void forEach(T_CONS action, long fence); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1393 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1394 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1395 |
static final class OfInt extends OfPrimitive<IntConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1396 |
implements IntConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1397 |
final int[] array; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1398 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1399 |
OfInt(int size) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1400 |
this.array = new int[size]; |
17182 | 1401 |
} |
1402 |
||
1403 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1404 |
public void accept(int t) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1405 |
array[index++] = t; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1406 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1407 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1408 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1409 |
public void forEach(IntConsumer action, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1410 |
for (int i = 0; i < fence; i++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1411 |
action.accept(array[i]); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1412 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1413 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1414 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1415 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1416 |
static final class OfLong extends OfPrimitive<LongConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1417 |
implements LongConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1418 |
final long[] array; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1419 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1420 |
OfLong(int size) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1421 |
this.array = new long[size]; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1422 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1423 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1424 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1425 |
public void accept(long t) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1426 |
array[index++] = t; |
17182 | 1427 |
} |
1428 |
||
1429 |
@Override |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1430 |
public void forEach(LongConsumer action, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1431 |
for (int i = 0; i < fence; i++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1432 |
action.accept(array[i]); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1433 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1434 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1435 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1436 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1437 |
static final class OfDouble extends OfPrimitive<DoubleConsumer> |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1438 |
implements DoubleConsumer { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1439 |
final double[] array; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1440 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1441 |
OfDouble(int size) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1442 |
this.array = new double[size]; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1443 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1444 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1445 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1446 |
public void accept(double t) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1447 |
array[index++] = t; |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1448 |
} |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1449 |
|
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1450 |
@Override |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1451 |
void forEach(DoubleConsumer action, long fence) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1452 |
for (int i = 0; i < fence; i++) { |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1453 |
action.accept(array[i]); |
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1454 |
} |
17182 | 1455 |
} |
1456 |
} |
|
1457 |
} |
|
18572
53b8b8c30086
8012987: Optimizations for Stream.limit/substream
psandoz
parents:
17182
diff
changeset
|
1458 |
} |