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 */
23package java.util.stream;
24
25import java.util.Spliterator;
26import java.util.function.IntFunction;
27
28/**
29 * The base type for a stateful test operation.
30 */
31interface 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                    public Sink opWrapSink(int flags, Sink sink) {
41                        return op.opWrapSink(flags, isParallel(), sink);
42                    }
43
44                    @Override
45                    public <P_IN> Spliterator<T> opEvaluateParallelLazy(PipelineHelper<T> helper,
46                                                                 Spliterator<P_IN> spliterator) {
47                        return op.opEvaluateParallelLazy(helper, spliterator);
48                    }
49
50                    @Override
51                    public <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                    public Sink opWrapSink(int flags, Sink sink) {
61                        return op.opWrapSink(flags, isParallel(), sink);
62                    }
63
64                    @Override
65                    public <P_IN> Spliterator<Integer> opEvaluateParallelLazy(PipelineHelper<Integer> helper,
66                                                                 Spliterator<P_IN> spliterator) {
67                        return op.opEvaluateParallelLazy(helper, spliterator);
68                    }
69
70                    @Override
71                    public <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                    public Sink opWrapSink(int flags, Sink sink) {
81                        return op.opWrapSink(flags, isParallel(), sink);
82                    }
83
84                    @Override
85                    public <P_IN> Spliterator<Long> opEvaluateParallelLazy(PipelineHelper<Long> helper,
86                                                                 Spliterator<P_IN> spliterator) {
87                        return op.opEvaluateParallelLazy(helper, spliterator);
88                    }
89
90                    @Override
91                    public <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                    public Sink opWrapSink(int flags, Sink sink) {
101                        return op.opWrapSink(flags, isParallel(), sink);
102                    }
103
104                    @Override
105                    public <P_IN> Spliterator<Double> opEvaluateParallelLazy(PipelineHelper<Double> helper,
106                                                                    Spliterator<P_IN> spliterator) {
107                        return op.opEvaluateParallelLazy(helper, spliterator);
108                    }
109
110                    @Override
111                    public <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}
139