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 org.openjdk.testlib.java.util.stream;
24
25import java.util.Spliterator;
26import java.util.function.IntFunction;
27import java.util.stream.*;
28
29/**
30 * The base type for a stateful test operation.
31 */
32interface StatefulTestOp<E> extends IntermediateTestOp<E, E> {
33
34    @SuppressWarnings({"rawtypes", "unchecked"})
35    public static<T> AbstractPipeline chain(AbstractPipeline upstream,
36                                            StatefulTestOp op) {
37        switch (op.outputShape()) {
38            case REFERENCE:
39                return new ReferencePipeline.StatefulOp<Object, T>(upstream, op.inputShape(), op.opGetFlags()) {
40                    @Override
41                    public Sink opWrapSink(int flags, Sink sink) {
42                        return op.opWrapSink(flags, isParallel(), sink);
43                    }
44
45                    @Override
46                    public <P_IN> Spliterator<T> opEvaluateParallelLazy(PipelineHelper<T> helper,
47                                                                 Spliterator<P_IN> spliterator) {
48                        return op.opEvaluateParallelLazy(helper, spliterator);
49                    }
50
51                    @Override
52                    public <P_IN> Node<T> opEvaluateParallel(PipelineHelper<T> helper,
53                                                      Spliterator<P_IN> spliterator,
54                                                      IntFunction<T[]> generator) {
55                        return op.opEvaluateParallel(helper, spliterator, generator);
56                    }
57                };
58            case INT_VALUE:
59                return new IntPipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
60                    @Override
61                    public Sink opWrapSink(int flags, Sink sink) {
62                        return op.opWrapSink(flags, isParallel(), sink);
63                    }
64
65                    @Override
66                    public <P_IN> Spliterator<Integer> opEvaluateParallelLazy(PipelineHelper<Integer> helper,
67                                                                 Spliterator<P_IN> spliterator) {
68                        return op.opEvaluateParallelLazy(helper, spliterator);
69                    }
70
71                    @Override
72                    public <P_IN> Node<Integer> opEvaluateParallel(PipelineHelper<Integer> helper,
73                                                            Spliterator<P_IN> spliterator,
74                                                            IntFunction<Integer[]> generator) {
75                        return (Node<Integer>) op.opEvaluateParallel(helper, spliterator, generator);
76                    }
77                };
78            case LONG_VALUE:
79                return new LongPipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
80                    @Override
81                    public Sink opWrapSink(int flags, Sink sink) {
82                        return op.opWrapSink(flags, isParallel(), sink);
83                    }
84
85                    public @Override
86                    <P_IN> Spliterator<Long> opEvaluateParallelLazy(PipelineHelper<Long> helper,
87                                                                 Spliterator<P_IN> spliterator) {
88                        return op.opEvaluateParallelLazy(helper, spliterator);
89                    }
90
91                    public @Override
92                    <P_IN> Node<Long> opEvaluateParallel(PipelineHelper<Long> helper,
93                                                         Spliterator<P_IN> spliterator,
94                                                         IntFunction<Long[]> generator) {
95                        return (Node<Long>) op.opEvaluateParallel(helper, spliterator, generator);
96                    }
97                };
98            case DOUBLE_VALUE:
99                return new DoublePipeline.StatefulOp<Object>(upstream, op.inputShape(), op.opGetFlags()) {
100                    @Override
101                    public Sink opWrapSink(int flags, Sink sink) {
102                        return op.opWrapSink(flags, isParallel(), sink);
103                    }
104
105                    @Override
106                    public <P_IN> Spliterator<Double> opEvaluateParallelLazy(PipelineHelper<Double> helper,
107                                                                    Spliterator<P_IN> spliterator) {
108                        return op.opEvaluateParallelLazy(helper, spliterator);
109                    }
110
111                    @Override
112                    public <P_IN> Node<Double> opEvaluateParallel(PipelineHelper<Double> helper,
113                                                           Spliterator<P_IN> spliterator,
114                                                           IntFunction<Double[]> generator) {
115                        return (Node<Double>) op.opEvaluateParallel(helper, spliterator, generator);
116                    }
117                };
118            default: throw new IllegalStateException(op.outputShape().toString());
119        }
120    }
121
122    default StreamShape inputShape() { return StreamShape.REFERENCE; }
123
124    default StreamShape outputShape() { return StreamShape.REFERENCE; }
125
126    default int opGetFlags() { return 0; }
127
128    Sink<E> opWrapSink(int flags, boolean parallel, Sink<E> sink);
129
130    @SuppressWarnings("unchecked")
131    default <P_IN> Spliterator<E> opEvaluateParallelLazy(PipelineHelper<E> helper,
132                                                         Spliterator<P_IN> spliterator) {
133        return opEvaluateParallel(helper, spliterator, i -> (E[]) new Object[i]).spliterator();
134    }
135
136    <P_IN> Node<E> opEvaluateParallel(PipelineHelper<E> helper,
137                                      Spliterator<P_IN> spliterator,
138                                      IntFunction<E[]> generator);
139}
140