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.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23 package java.util.stream;
24
25 import java.util.Spliterator;
26 import java.util.function.IntFunction;
27
28 /**
29 * The base type for a stateful test operation.
30 */
31 interface StatefulTestOp<E> extends IntermediateTestOp<E, E> {
32
33 @SuppressWarnings({"rawtypes", "unchecked"})
34 public static<T> AbstractPipeline chain(AbstractPipeline upstream,
35 StatefulTestOp op) {
36 switch (op.outputShape()) {
37 case REFERENCE:
38 return new ReferencePipeline.StatefulOp<Object, T>(upstream, op.inputShape(), op.opGetFlags()) {
39 @Override
40 Sink opWrapSink(int flags, Sink sink) {
41 return op.opWrapSink(flags, isParallel(), sink);
42 }
43
44 @Override
45 <P_IN> Spliterator<T> opEvaluateParallelLazy(PipelineHelper<T> helper,
46 Spliterator<P_IN> spliterator) {
47 return op.opEvaluateParallelLazy(helper, spliterator);
48 }
49
50 @Override
51 <P_IN> Node<T> opEvaluateParallel(PipelineHelper<T> helper,
52 Spliterator<P_IN> spliterator,
53 IntFunction<T[]> generator) {
54 return op.opEvaluateParallel(helper, spliterator, generator);
55 }
56 };
57 case INT_VALUE:
58 return new IntPipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
59 @Override
60 Sink opWrapSink(int flags, Sink sink) {
61 return op.opWrapSink(flags, isParallel(), sink);
62 }
63
64 @Override
65 <P_IN> Spliterator<Integer> opEvaluateParallelLazy(PipelineHelper<Integer> helper,
66 Spliterator<P_IN> spliterator) {
67 return op.opEvaluateParallelLazy(helper, spliterator);
68 }
69
70 @Override
71 <P_IN> Node<Integer> opEvaluateParallel(PipelineHelper<Integer> helper,
72 Spliterator<P_IN> spliterator,
73 IntFunction<Integer[]> generator) {
74 return (Node<Integer>) op.opEvaluateParallel(helper, spliterator, generator);
75 }
76 };
77 case LONG_VALUE:
78 return new LongPipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
79 @Override
80 Sink opWrapSink(int flags, Sink sink) {
81 return op.opWrapSink(flags, isParallel(), sink);
82 }
83
84 @Override
85 <P_IN> Spliterator<Long> opEvaluateParallelLazy(PipelineHelper<Long> helper,
86 Spliterator<P_IN> spliterator) {
87 return op.opEvaluateParallelLazy(helper, spliterator);
88 }
89
90 @Override
91 <P_IN> Node<Long> opEvaluateParallel(PipelineHelper<Long> helper,
92 Spliterator<P_IN> spliterator,
93 IntFunction<Long[]> generator) {
94 return (Node<Long>) op.opEvaluateParallel(helper, spliterator, generator);
95 }
96 };
97 case DOUBLE_VALUE:
98 return new DoublePipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
99 @Override
100 Sink opWrapSink(int flags, Sink sink) {
101 return op.opWrapSink(flags, isParallel(), sink);
102 }
103
104 @Override
105 <P_IN> Spliterator<Double> opEvaluateParallelLazy(PipelineHelper<Double> helper,
106 Spliterator<P_IN> spliterator) {
107 return op.opEvaluateParallelLazy(helper, spliterator);
108 }
109
110 @Override
111 <P_IN> Node<Double> opEvaluateParallel(PipelineHelper<Double> helper,
112 Spliterator<P_IN> spliterator,
113 IntFunction<Double[]> generator) {
114 return (Node<Double>) op.opEvaluateParallel(helper, spliterator, generator);
115 }
116 };
117 default: throw new IllegalStateException(op.outputShape().toString());
118 }
119 }
120
121 default StreamShape inputShape() { return StreamShape.REFERENCE; }
122
123 default StreamShape outputShape() { return StreamShape.REFERENCE; }
124
125 default int opGetFlags() { return 0; }
126
127 Sink<E> opWrapSink(int flags, boolean parallel, Sink<E> sink);
128
129 @SuppressWarnings("unchecked")
130 default <P_IN> Spliterator<E> opEvaluateParallelLazy(PipelineHelper<E> helper,
131 Spliterator<P_IN> spliterator) {
132 return opEvaluateParallel(helper, spliterator, i -> (E[]) new Object[i]).spliterator();
133 }
134
135 <P_IN> Node<E> opEvaluateParallel(PipelineHelper<E> helper,
136 Spliterator<P_IN> spliterator,
137 IntFunction<E[]> generator);
138 }