All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.landawn.abacus.util.stream.ByteStream Maven / Gradle / Ivy

/*
 * Copyright (C) 2016, 2017, 2018, 2019 HaiYang Li
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */

package com.landawn.abacus.util.stream;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.security.SecureRandom;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Random;
import java.util.function.BiConsumer;
import java.util.function.BiFunction;
import java.util.function.BiPredicate;
import java.util.function.BinaryOperator;
import java.util.function.BooleanSupplier;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.function.Supplier;
import java.util.function.UnaryOperator;
import java.util.stream.Collector;

import com.landawn.abacus.annotation.Beta;
import com.landawn.abacus.annotation.Immutable;
import com.landawn.abacus.annotation.IntermediateOp;
import com.landawn.abacus.annotation.LazyEvaluation;
import com.landawn.abacus.annotation.ParallelSupported;
import com.landawn.abacus.annotation.SequentialOnly;
import com.landawn.abacus.annotation.TerminalOp;
import com.landawn.abacus.exception.UncheckedIOException;
import com.landawn.abacus.util.Array;
import com.landawn.abacus.util.ByteIterator;
import com.landawn.abacus.util.ByteList;
import com.landawn.abacus.util.ByteSummaryStatistics;
import com.landawn.abacus.util.Fn;
import com.landawn.abacus.util.Fn.BiConsumers;
import com.landawn.abacus.util.Fn.FB;
import com.landawn.abacus.util.IOUtil;
import com.landawn.abacus.util.IndexedByte;
import com.landawn.abacus.util.MergeResult;
import com.landawn.abacus.util.N;
import com.landawn.abacus.util.Pair;
import com.landawn.abacus.util.Percentage;
import com.landawn.abacus.util.Throwables;
import com.landawn.abacus.util.cs;
import com.landawn.abacus.util.u.Optional;
import com.landawn.abacus.util.u.OptionalByte;
import com.landawn.abacus.util.u.OptionalDouble;
import com.landawn.abacus.util.function.ByteBiFunction;
import com.landawn.abacus.util.function.ByteBiPredicate;
import com.landawn.abacus.util.function.ByteBinaryOperator;
import com.landawn.abacus.util.function.ByteConsumer;
import com.landawn.abacus.util.function.ByteFunction;
import com.landawn.abacus.util.function.ByteNFunction;
import com.landawn.abacus.util.function.BytePredicate;
import com.landawn.abacus.util.function.ByteSupplier;
import com.landawn.abacus.util.function.ByteTernaryOperator;
import com.landawn.abacus.util.function.ByteToIntFunction;
import com.landawn.abacus.util.function.ByteTriPredicate;
import com.landawn.abacus.util.function.ByteUnaryOperator;
import com.landawn.abacus.util.function.ObjByteConsumer;
import com.landawn.abacus.util.function.ToByteFunction;
import com.landawn.abacus.util.function.TriFunction;
import com.landawn.abacus.util.function.TriPredicate;

/**
 * The ByteStream class is an abstract class that represents a stream of byte elements and supports different kinds of computations. 
* The Stream operations are divided into intermediate and terminal operations, and are combined to form stream pipelines.
* *
* The Stream will be automatically closed after a terminal method is called/triggered. * *
*
* Refer to {@code com.landawn.abacus.util.stream.BaseStream} and {@code com.landawn.abacus.util.stream.Stream} for more APIs docs. * * @see Stream * @see BaseStream */ @Immutable @LazyEvaluation public abstract class ByteStream extends StreamBase { static final Random RAND = new SecureRandom(); ByteStream(final boolean sorted, final Collection closeHandlers) { super(sorted, null, closeHandlers); } /** * * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream map(ByteUnaryOperator mapper); /** * * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract IntStream mapToInt(ByteToIntFunction mapper); /** * * @param * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract Stream mapToObj(ByteFunction mapper); /** * * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream flatMap(ByteFunction mapper); // @ParallelSupported // @IntermediateOp // public abstract ByteStream flatmap(ByteFunction mapper); /** * * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream flatmap(ByteFunction mapper); //NOSONAR /** * * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract IntStream flatMapToInt(ByteFunction mapper); /** * * @param * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract Stream flatMapToObj(ByteFunction> mapper); /** * * @param * @param mapper * @return */ @ParallelSupported @IntermediateOp public abstract Stream flatmapToObj(ByteFunction> mapper); //NOSONAR /** * * @param * @param mapper * @return */ @Beta @ParallelSupported @IntermediateOp public abstract Stream flattMapToObj(ByteFunction mapper); /** * Note: copied from StreamEx: StreamEx * * @param mapper * @return */ @Beta @ParallelSupported @IntermediateOp public abstract ByteStream mapPartial(ByteFunction mapper); /** * Note: copied from StreamEx: StreamEx *
* * Returns a stream consisting of results of applying the given function to * the ranges created from the source elements. * This is a quasi-intermediate * partial reduction operation. * * @param sameRange a non-interfering, stateless predicate to apply to * the leftmost and next elements which returns {@code true} for elements * which belong to the same range. * @param mapper a non-interfering, stateless function to apply to the * range borders and produce the resulting element. If value was * not merged to the interval, then mapper will receive the same * value twice, otherwise it will receive the leftmost and the * rightmost values which were merged to the range. * @return * @see #collapse(ByteBiPredicate, ByteBinaryOperator) * @see Stream#rangeMap(BiPredicate, BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream rangeMap(final ByteBiPredicate sameRange, final ByteBinaryOperator mapper); /** * Note: copied from StreamEx: StreamEx *
* * Returns a stream consisting of results of applying the given function to * the ranges created from the source elements. * This is a quasi-intermediate * partial reduction operation. * * @param * @param sameRange a non-interfering, stateless predicate to apply to * the leftmost and next elements which returns {@code true} for elements * which belong to the same range. * @param mapper a non-interfering, stateless function to apply to the * range borders and produce the resulting element. If value was * not merged to the interval, then mapper will receive the same * value twice, otherwise it will receive the leftmost and the * rightmost values which were merged to the range. * @return * @see Stream#rangeMap(BiPredicate, BiFunction) */ @SequentialOnly @IntermediateOp public abstract Stream rangeMapToObj(final ByteBiPredicate sameRange, final ByteBiFunction mapper); /** * Merges a series of adjacent elements in the stream which satisfy the given predicate into a List. * The predicate takes two parameters: the previous element and the current element in the stream. * If the predicate returns {@code true}, the current element and its previous element are considered as a series of adjacent elements. * These elements are then collapsed into a List. * * This is an intermediate operation, meaning it's always lazy. It doesn't start processing the data until a terminal operation is invoked on the stream pipeline. * It's also a stateful operation since it needs to remember the previous element when processing the current element. * * This operation is not parallelizable and requires the stream to be ordered. * * @param collapsible a BiPredicate that takes two parameters: the previous element and the current element in the stream. * @return a new Stream where each element is a List of adjacent elements which satisfy the given predicate. * @see Stream#collapse(BiPredicate) */ @SequentialOnly @IntermediateOp public abstract Stream collapse(final ByteBiPredicate collapsible); /** * Merges a series of adjacent elements in the stream which satisfy the given predicate using the merger function and returns a new stream. * The predicate takes two parameters: the previous element and the current element in the stream. * If the predicate returns {@code true}, the current element and its previous element are considered as a series of adjacent elements. * These elements are then merged using the provided BiFunction. * * This is an intermediate operation, meaning it's always lazy. It doesn't start processing the data until a terminal operation is invoked on the stream pipeline. * It's also a stateful operation since it needs to remember the previous element when processing the current element. * * This operation is not parallelizable and requires the stream to be ordered. * * @param collapsible a BiPredicate that takes two parameters: the previous element and the current element in the stream. * @param mergeFunction a BiFunction that takes two parameters: the result of the previous merge operation (or the first element if no merge has been performed yet) and the current element, and returns the result of the merge operation. * @return a new Stream where each element is the result of merging adjacent elements which satisfy the given predicate. * @see Stream#collapse(BiPredicate, BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream collapse(final ByteBiPredicate collapsible, final ByteBinaryOperator mergeFunction); /** * Merges a series of adjacent elements in the stream which satisfy the given predicate into a single element and returns a new stream. * The predicate takes three parameters: the first element of the series, the previous element and the current element in the stream. * If the predicate returns {@code true}, the current element, its previous element and the first element of the series are considered as a series of adjacent elements. * These elements are then collapsed into a single element using the provided merge function. * * This is an intermediate operation, meaning it's always lazy. It doesn't start processing the data until a terminal operation is invoked on the stream pipeline. * It's also a stateful operation since it needs to remember the first and previous elements when processing the current element. * * This operation is not parallelizable and requires the stream to be ordered. * * @param collapsible a TriPredicate that takes three parameters: the first element of the series, the previous element and the current element in the stream. * @param mergeFunction a BiFunction that takes two parameters: the current element and its previous element. It returns a single element that represents the collapsed elements. * @return a new Stream where each element is the result of collapsing adjacent elements which satisfy the given predicate. * @see Stream#collapse(TriPredicate, BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream collapse(final ByteTriPredicate collapsible, final ByteBinaryOperator mergeFunction); /** * Performs a scan (also known as prefix sum, cumulative sum, running total, or integral) operation on the elements of the stream. * The scan operation takes a binary operator (the accumulator) and applies it cumulatively on the stream elements, * successively combining each element in order from the start to produce a stream of accumulated results. * * For example, given a stream of numbers [1, 2, 3, 4], and an accumulator that performs addition, * the output would be a stream of numbers [1, 3, 6, 10]. * * This is an intermediate operation. * This operation is sequential only, even when called on a parallel stream. * * @param accumulator a {@code ByteBinaryOperator} that takes two parameters: the current accumulated value and the current stream element, and returns a new accumulated value. * @return a new {@code ByteStream} consisting of the results of the scan operation on the elements of the original stream. * @see Stream#scan(BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream scan(final ByteBinaryOperator accumulator); /** * Performs a scan (also known as prefix sum, cumulative sum, running total, or integral) operation on the elements of the stream. * The scan operation takes an initial value and a binary operator (the accumulator) and applies it cumulatively on the stream elements, * successively combining each element in order from the start to produce a stream of accumulated results. * * For example, given a stream of numbers [1, 2, 3, 4], an initial value of 10, and an accumulator that performs addition, * the output would be a stream of numbers [11, 13, 16, 20]. * This is an intermediate operation. * This operation is sequential only, even when called on a parallel stream. * * @param init the initial value. It's only used once by the accumulator to calculate the first element in the returned stream. * It will be ignored if this stream is empty and won't be the first element of the returned stream. * @param accumulator a {@code ByteBinaryOperator} that takes two parameters: the current accumulated value and the current stream element, and returns a new accumulated value. * @return a new {@code ByteStream} consisting of the results of the scan operation on the elements of the original stream. * @see Stream#scan(Object, BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream scan(final byte init, final ByteBinaryOperator accumulator); /** * Performs a scan (also known as prefix sum, cumulative sum, running total, or integral) operation on the elements of the stream. * The scan operation takes an initial value and a binary operator (the accumulator) and applies it cumulatively on the stream elements, * successively combining each element in order from the start to produce a stream of accumulated results. * * This is an intermediate operation. * This operation is sequential only, even when called on a parallel stream. * * @param init the initial value. It's only used once by the accumulator to calculate the first element in the returned stream. * @param initIncluded a boolean value that determines if the initial value should be included as the first element in the returned stream. * @param accumulator a {@code ByteBinaryOperator} that takes two parameters: the current accumulated value and the current stream element, and returns a new accumulated value. * @return a new {@code ByteStream} consisting of the results of the scan operation on the elements of the original stream. * @see Stream#scan(Object, boolean, BiFunction) */ @SequentialOnly @IntermediateOp public abstract ByteStream scan(final byte init, final boolean initIncluded, final ByteBinaryOperator accumulator); /** * * @param a * @return */ @SequentialOnly @IntermediateOp public abstract ByteStream prepend(final byte... a); /** * * @param a * @return */ @SequentialOnly @IntermediateOp public abstract ByteStream append(final byte... a); /** * * @param a * @return */ @SequentialOnly @IntermediateOp public abstract ByteStream appendIfEmpty(final byte... a); @SequentialOnly @TerminalOp public abstract ByteList toByteList(); /** * * @param * @param * @param * @param * @param keyMapper * @param valueMapper * @return * @throws E * @throws E2 * @see Collectors#toMap(Function, Function) */ @ParallelSupported @TerminalOp public abstract Map toMap(Throwables.ByteFunction keyMapper, Throwables.ByteFunction valueMapper) throws E, E2; /** * * @param * @param * @param * @param * @param * @param keyMapper * @param valueMapper * @param mapFactory * @return * @throws E * @throws E2 * @see Collectors#toMap(Function, Function, Supplier) */ @ParallelSupported @TerminalOp public abstract , E extends Exception, E2 extends Exception> M toMap(Throwables.ByteFunction keyMapper, Throwables.ByteFunction valueMapper, Supplier mapFactory) throws E, E2; /** * * @param * @param * @param * @param * @param keyMapper * @param valueMapper * @param mergeFunction * @return * @throws E * @throws E2 * @see Collectors#toMap(Function, Function, BinaryOperator) */ @ParallelSupported @TerminalOp public abstract Map toMap(Throwables.ByteFunction keyMapper, Throwables.ByteFunction valueMapper, BinaryOperator mergeFunction) throws E, E2; /** * * @param * @param * @param * @param * @param * @param keyMapper * @param valueMapper * @param mergeFunction * @param mapFactory * @return * @throws E * @throws E2 * @see Collectors#toMap(Function, Function, BinaryOperator, Supplier) */ @ParallelSupported @TerminalOp public abstract , E extends Exception, E2 extends Exception> M toMap(Throwables.ByteFunction keyMapper, Throwables.ByteFunction valueMapper, BinaryOperator mergeFunction, Supplier mapFactory) throws E, E2; /** * * @param * @param * @param * @param keyMapper * @param downstream * @return * @throws E * @see Collectors#groupingBy(Function, Collector) */ @ParallelSupported @TerminalOp public abstract Map groupTo(Throwables.ByteFunction keyMapper, final Collector downstream) throws E; /** * * @param * @param * @param * @param * @param keyMapper * @param downstream * @param mapFactory * @return * @throws E * @see Collectors#groupingBy(Function, Collector, Supplier) */ @ParallelSupported @TerminalOp public abstract , E extends Exception> M groupTo(Throwables.ByteFunction keyMapper, final Collector downstream, final Supplier mapFactory) throws E; /** * Performs a reduction on the elements of this stream, using the provided accumulator function, and returns the reduced value. * The accumulator function takes two parameters: the current reduced value (or the initial value for the first element), and the current stream element. * * @param identity the initial value of the reduction operation * @param op the function for combining the current reduced value and the current stream element * @return the result of the reduction * @see Stream#reduce(Object, BinaryOperator) */ @ParallelSupported @TerminalOp public abstract byte reduce(byte identity, ByteBinaryOperator op); /** * Performs a reduction on the elements of this stream, using the provided accumulator function, and returns the reduced value. * The accumulator function takes two parameters: the current reduced value and the current stream element. * * @param op the function for combining the current reduced value and the current stream element * @return an OptionalByte describing the result of the reduction. If the stream is empty, an empty OptionalByte is returned. */ @ParallelSupported @TerminalOp public abstract OptionalByte reduce(ByteBinaryOperator op); /** * Performs a mutable reduction operation on the elements of this stream using a Collector. * * @param The type of the result * @param supplier a function that creates a new result container. For a parallel execution, this function may be called multiple times and must return a fresh value each time. * @param accumulator an associative, non-interfering, stateless function for incorporating an additional element into a result * @param combiner an associative, non-interfering, stateless function for combining two values, which must be compatible with the accumulator function. * It's unnecessary to specify {@code combiner} if {@code R} is a {@code Map/Collection/StringBuilder/Multiset/LongMultiset/Multimap/BooleanList/IntList/.../DoubleList}. * @return the result of the reduction * @see Stream#collect(Supplier, BiConsumer, BiConsumer) * @see BiConsumers#ofAddAll() * @see BiConsumers#ofPutAll() */ @ParallelSupported @TerminalOp public abstract R collect(Supplier supplier, ObjByteConsumer accumulator, BiConsumer combiner); /** * Performs a mutable reduction operation on the elements of this stream using a Collector. * *
* Only call this method when the returned type {@code R} is one types: {@code Collection/Map/StringBuilder/Multiset/LongMultiset/Multimap/BooleanList/IntList/.../DoubleList}. * Otherwise, please call {@link #collect(Supplier, ObjByteConsumer, BiConsumer)}. * * @param The type of the result. It must be {@code Collection/Map/StringBuilder/Multiset/LongMultiset/Multimap/BooleanList/IntList/.../DoubleList}. * @param supplier A function that creates a new result container. For a parallel execution, this function may be called multiple times and must return a fresh value each time. * @param accumulator An associative, non-interfering, stateless function for incorporating an additional element into a result. * @throws IllegalArgumentException if the returned type {@code R} is not one of the types: {@code Collection/Map/StringBuilder/Multiset/LongMultiset/Multimap/BooleanList/IntList/.../DoubleList}. * @return the result of the reduction * @see #collect(Supplier, ObjByteConsumer, BiConsumer) * @see Stream#collect(Supplier, BiConsumer) * @see Stream#collect(Supplier, BiConsumer, BiConsumer) */ @ParallelSupported @TerminalOp public abstract R collect(Supplier supplier, ObjByteConsumer accumulator); /** * * @param * @param action * @throws E */ @ParallelSupported @TerminalOp public abstract void forEach(Throwables.ByteConsumer action) throws E; /** * * @param * @param action * @throws E */ @ParallelSupported @TerminalOp public abstract void forEachIndexed(Throwables.IntByteConsumer action) throws E; /** * * @param * @param predicate * @return * @throws E */ @ParallelSupported @TerminalOp public abstract boolean anyMatch(final Throwables.BytePredicate predicate) throws E; /** * * @param * @param predicate * @return * @throws E */ @ParallelSupported @TerminalOp public abstract boolean allMatch(final Throwables.BytePredicate predicate) throws E; /** * * @param * @param predicate * @return * @throws E */ @ParallelSupported @TerminalOp public abstract boolean noneMatch(final Throwables.BytePredicate predicate) throws E; /** * * @param * @param predicate * @return * @throws E */ @ParallelSupported @TerminalOp public abstract OptionalByte findFirst(final Throwables.BytePredicate predicate) throws E; /** * * @param * @param predicate * @return * @throws E */ @ParallelSupported @TerminalOp public abstract OptionalByte findAny(final Throwables.BytePredicate predicate) throws E; /** * Consider using: {@code stream.reversed().findFirst(predicate)} for better performance if possible. * * @param * @param predicate * @return * @throws E */ @Beta @ParallelSupported @TerminalOp public abstract OptionalByte findLast(final Throwables.BytePredicate predicate) throws E; // /** // * Returns the first element matched by {@code predicateForFirst} if found or the first element if this stream is not empty // * Otherwise an empty {@code OptionalByte} will be returned. // * // * @param // * @param predicateForFirst // * @return // * @throws E // */ // @ParallelSupported // @TerminalOp // public abstract OptionalByte findFirstOrElseAny(Throwables.BytePredicate predicateForFirst) throws E; // // /** // * Returns the first element matched by {@code predicateForFirst} if found or the last element if this stream is not empty // * Otherwise an empty {@code OptionalByte} will be returned. // * // * @param // * @param predicateForFirst // * @return // * @throws E // */ // @ParallelSupported // @TerminalOp // public abstract OptionalByte findFirstOrElseLast(Throwables.BytePredicate predicateForFirst) throws E; @SequentialOnly @TerminalOp public abstract OptionalByte min(); @SequentialOnly @TerminalOp public abstract OptionalByte max(); /** * Returns the k-th largest element in the stream. * If the stream is empty or the count of elements is less than k, an empty OptionalByte is returned. * * @param k the position (1-based) of the largest element to retrieve * @return an OptionalByte containing the k-th largest element, or an empty OptionalByte if the stream is empty or the count of elements is less than k */ @SequentialOnly @TerminalOp public abstract OptionalByte kthLargest(int k); /** * * @return int */ @SequentialOnly @TerminalOp public abstract int sum(); @SequentialOnly @TerminalOp public abstract OptionalDouble average(); @SequentialOnly @TerminalOp public abstract ByteSummaryStatistics summarize(); @SequentialOnly @TerminalOp public abstract Pair>> summarizeAndPercentiles(); // /** // * // * @param b // * @param nextSelector a function to determine which element should be selected as next element. // * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. // * @return // * @deprecated replaced by {@code mergeWith(ByteStream, ByteBiFunction)} // * @see #mergeWith(ByteStream, ByteBiFunction) // */ // @SequentialOnly // @IntermediateOp // @Deprecated // public ByteStream merge(final ByteStream b, final ByteBiFunction nextSelector) { // return mergeWith(b, nextSelector); // } /** * * @param b * @param nextSelector a function to determine which element should be selected as next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return */ @SequentialOnly @IntermediateOp public abstract ByteStream mergeWith(final ByteStream b, final ByteBiFunction nextSelector); /** * * @param b * @param zipFunction * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream zipWith(ByteStream b, ByteBinaryOperator zipFunction); /** * * @param b * @param c * @param zipFunction * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream zipWith(ByteStream b, ByteStream c, ByteTernaryOperator zipFunction); /** * * @param b * @param valueForNoneA * @param valueForNoneB * @param zipFunction * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream zipWith(ByteStream b, byte valueForNoneA, byte valueForNoneB, ByteBinaryOperator zipFunction); /** * * @param b * @param c * @param valueForNoneA * @param valueForNoneB * @param valueForNoneC * @param zipFunction * @return */ @ParallelSupported @IntermediateOp public abstract ByteStream zipWith(ByteStream b, ByteStream c, byte valueForNoneA, byte valueForNoneB, byte valueForNoneC, ByteTernaryOperator zipFunction); @SequentialOnly @IntermediateOp public abstract IntStream asIntStream(); @SequentialOnly @IntermediateOp public abstract Stream boxed(); abstract ByteIteratorEx iteratorEx(); // private static final ByteStream EMPTY_STREAM = new ArrayByteStream(N.EMPTY_BYTE_ARRAY, true, null); /** * Returns an empty ByteStream. * * @return an empty ByteStream */ public static ByteStream empty() { return new ArrayByteStream(N.EMPTY_BYTE_ARRAY, true, null); } /** * Creates a new ByteStream that is supplied by the given supplier. * The supplier is only invoked when the stream is actually used. * This allows for lazy evaluation of the stream. * * @implNote it's equivalent to {@code Stream.just(supplier).flatMapToByte(it -> it.get())}. * * @param supplier the supplier that provides the ByteStream * @return a new ByteStream supplied by the given supplier * @throws IllegalArgumentException if the supplier is null * @see Stream#defer(Supplier) */ public static ByteStream defer(final Supplier supplier) throws IllegalArgumentException { N.checkArgNotNull(supplier, cs.supplier); //noinspection resource return Stream.just(supplier).flatMapToByte(Supplier::get); } /** * * @param e * @return */ public static ByteStream ofNullable(final Byte e) { return e == null ? empty() : of(e); } /** * * @param a * @return */ @SafeVarargs public static ByteStream of(final byte... a) { return N.isEmpty(a) ? empty() : new ArrayByteStream(a); } /** * * @param a * @param startIndex * @param endIndex * @return */ public static ByteStream of(final byte[] a, final int startIndex, final int endIndex) { return N.isEmpty(a) && (startIndex == 0 && endIndex == 0) ? empty() : new ArrayByteStream(a, startIndex, endIndex); } /** * * @param a * @return */ public static ByteStream of(final Byte[] a) { //noinspection resource return Stream.of(a).mapToByte(FB.unbox()); } /** * * @param a * @param startIndex * @param endIndex * @return */ public static ByteStream of(final Byte[] a, final int startIndex, final int endIndex) { //noinspection resource return Stream.of(a, startIndex, endIndex).mapToByte(FB.unbox()); } /** * * @param c * @return */ public static ByteStream of(final Collection c) { //noinspection resource return Stream.of(c).mapToByte(FB.unbox()); } /** * * @param iterator * @return */ public static ByteStream of(final ByteIterator iterator) { return iterator == null ? empty() : new IteratorByteStream(iterator); } /** * * @param buf * @return */ public static ByteStream of(final ByteBuffer buf) { if (buf == null) { return empty(); } //noinspection resource return IntStream.range(buf.position(), buf.limit()).mapToByte(buf::get); } /** * * @param file * @return */ public static ByteStream of(final File file) { return of(IOUtil.newFileInputStream(file), true); } /** * * @param is * @return */ public static ByteStream of(final InputStream is) { return of(is, false); } /** * * @param is * @param closeInputStreamWhenStreamIsClosed * @return */ public static ByteStream of(final InputStream is, final boolean closeInputStreamWhenStreamIsClosed) { if (is == null) { return empty(); } final ByteIterator iter = new ByteIterator() { private final byte[] buf = new byte[8192]; private boolean isEnd = false; private int count = 0; private int idx = 0; @Override public boolean hasNext() { if (idx >= count && !isEnd) { try { count = is.read(buf); idx = 0; isEnd = count <= 0; } catch (final IOException e) { throw new UncheckedIOException(e); } } return count > idx; } @Override public byte nextByte() { if (!hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return buf[idx++]; } }; if (closeInputStreamWhenStreamIsClosed) { //noinspection resource return of(iter).onClose(Fn.close(is)); } else { return of(iter); } } private static final Function flatMapper = ByteStream::of; private static final Function flattMapper = ByteStream::flatten; /** * * @param a * @return */ public static ByteStream flatten(final byte[][] a) { //noinspection resource return N.isEmpty(a) ? empty() : Stream.of(a).flatMapToByte(flatMapper); } /** * * @param a * @param vertically * @return */ public static ByteStream flatten(final byte[][] a, final boolean vertically) { if (N.isEmpty(a)) { return empty(); } else if (a.length == 1) { return of(a[0]); } else if (!vertically) { //noinspection resource return Stream.of(a).flatMapToByte(flatMapper); } long n = 0; for (final byte[] e : a) { n += N.len(e); } if (n == 0) { return empty(); } final int rows = N.len(a); final long count = n; final ByteIterator iter = new ByteIteratorEx() { private int rowNum = 0; private int colNum = 0; private long cnt = 0; @Override public boolean hasNext() { return cnt < count; } @Override public byte nextByte() { if (cnt++ >= count) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } if (rowNum == rows) { rowNum = 0; colNum++; } while (a[rowNum] == null || colNum >= a[rowNum].length) { if (rowNum < rows - 1) { rowNum++; } else { rowNum = 0; colNum++; } } return a[rowNum++][colNum]; } }; return of(iter); } /** * * @param a * @param valueForAlignment element to append so there are same size of elements in all rows/columns * @param vertically * @return */ public static ByteStream flatten(final byte[][] a, final byte valueForAlignment, final boolean vertically) { if (N.isEmpty(a)) { return empty(); } else if (a.length == 1) { return of(a[0]); } long n = 0; int maxLen = 0; for (final byte[] e : a) { n += N.len(e); maxLen = N.max(maxLen, N.len(e)); } if (n == 0) { return empty(); } final int rows = N.len(a); final int cols = maxLen; final long count = (long) rows * cols; ByteIterator iter = null; if (vertically) { iter = new ByteIteratorEx() { private int rowNum = 0; private int colNum = 0; private long cnt = 0; @Override public boolean hasNext() { return cnt < count; } @Override public byte nextByte() { if (cnt++ >= count) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } if (rowNum == rows) { rowNum = 0; colNum++; } if (a[rowNum] == null || colNum >= a[rowNum].length) { rowNum++; return valueForAlignment; } else { return a[rowNum++][colNum]; } } }; } else { iter = new ByteIteratorEx() { private int rowNum = 0; private int colNum = 0; private long cnt = 0; @Override public boolean hasNext() { return cnt < count; } @Override public byte nextByte() { if (cnt++ >= count) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } if (colNum >= cols) { colNum = 0; rowNum++; } if (a[rowNum] == null || colNum >= a[rowNum].length) { colNum++; return valueForAlignment; } else { return a[rowNum][colNum++]; } } }; } return of(iter); } /** * * @param a * @return */ public static ByteStream flatten(final byte[][][] a) { //noinspection resource return N.isEmpty(a) ? empty() : Stream.of(a).flatMapToByte(flattMapper); } /** * * @param startInclusive * @param endExclusive * @return */ public static ByteStream range(final byte startInclusive, final byte endExclusive) { if (startInclusive >= endExclusive) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private byte next = startInclusive; private int cnt = endExclusive - startInclusive; @Override public boolean hasNext() { return cnt > 0; } @Override public byte nextByte() { if (cnt-- <= 0) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return next++; } @Override public void advance(final long n) { cnt = n >= cnt ? 0 : cnt - (int) n; next += (byte) n; } @Override public long count() { return cnt; } @Override public byte[] toArray() { final byte[] result = new byte[cnt]; for (int i = 0; i < cnt; i++) { result[i] = next++; } cnt = 0; return result; } }); } /** * * @param startInclusive * @param endExclusive * @param by * @return */ public static ByteStream range(final byte startInclusive, final byte endExclusive, final byte by) { if (by == 0) { throw new IllegalArgumentException("'by' can't be zero"); } if (endExclusive == startInclusive || endExclusive > startInclusive != by > 0) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private byte next = startInclusive; private int cnt = (endExclusive - startInclusive) / by + ((endExclusive - startInclusive) % by == 0 ? 0 : 1); @Override public boolean hasNext() { return cnt > 0; } @Override public byte nextByte() { if (cnt-- <= 0) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } final byte result = next; next += by; return result; } @Override public void advance(final long n) { cnt = n >= cnt ? 0 : cnt - (int) n; next += (byte) (n * by); } @Override public long count() { return cnt; } @Override public byte[] toArray() { final byte[] result = new byte[cnt]; for (int i = 0; i < cnt; i++, next += by) { result[i] = next; } cnt = 0; return result; } }); } /** * * @param startInclusive * @param endInclusive * @return */ public static ByteStream rangeClosed(final byte startInclusive, final byte endInclusive) { if (startInclusive > endInclusive) { //noinspection resource empty(); } else if (startInclusive == endInclusive) { return of(startInclusive); } return new IteratorByteStream(new ByteIteratorEx() { private byte next = startInclusive; private int cnt = endInclusive - startInclusive + 1; @Override public boolean hasNext() { return cnt > 0; } @Override public byte nextByte() { if (cnt-- <= 0) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return next++; } @Override public void advance(final long n) { cnt = n >= cnt ? 0 : cnt - (int) n; next += (byte) n; } @Override public long count() { return cnt; } @Override public byte[] toArray() { final byte[] result = new byte[cnt]; for (int i = 0; i < cnt; i++) { result[i] = next++; } cnt = 0; return result; } }); } /** * * @param startInclusive * @param endInclusive * @param by * @return */ public static ByteStream rangeClosed(final byte startInclusive, final byte endInclusive, final byte by) { if (by == 0) { throw new IllegalArgumentException("'by' can't be zero"); } if (endInclusive == startInclusive) { return of(startInclusive); } else if (endInclusive > startInclusive != by > 0) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private byte next = startInclusive; private int cnt = (endInclusive - startInclusive) / by + 1; @Override public boolean hasNext() { return cnt > 0; } @Override public byte nextByte() { if (cnt-- <= 0) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } final byte result = next; next += by; return result; } @Override public void advance(final long n) { cnt = n >= cnt ? 0 : cnt - (int) n; next += (byte) (n * by); } @Override public long count() { return cnt; } @Override public byte[] toArray() { final byte[] result = new byte[cnt]; for (int i = 0; i < cnt; i++, next += by) { result[i] = next; } cnt = 0; return result; } }); } /** * * @param element * @param n * @return * @throws IllegalArgumentException */ public static ByteStream repeat(final byte element, final long n) throws IllegalArgumentException { N.checkArgNotNegative(n, cs.n); if (n == 0) { return empty(); } else if (n < 10) { return of(Array.repeat(element, (int) n)); } return new IteratorByteStream(new ByteIteratorEx() { private long cnt = n; @Override public boolean hasNext() { return cnt > 0; } @Override public byte nextByte() { if (cnt-- <= 0) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return element; } @Override public void advance(final long n) { cnt = n >= cnt ? 0 : cnt - (int) n; } @Override public long count() { return cnt; } @Override public byte[] toArray() { final byte[] result = new byte[(int) cnt]; for (int i = 0; i < cnt; i++) { result[i] = element; } cnt = 0; return result; } }); } public static ByteStream random() { final int bound = Byte.MAX_VALUE - Byte.MIN_VALUE + 1; return generate(() -> (byte) (RAND.nextInt(bound) + Byte.MIN_VALUE)); } /** * Creates a stream that iterates using the given hasNext and next suppliers. * * @param hasNext a BooleanSupplier that returns {@code true} if the iteration should continue * @param next a ByteSupplier that provides the next byte in the iteration * @return a ByteStream of elements generated by the iteration * @throws IllegalArgumentException if hasNext or next is null * @see Stream#iterate(BooleanSupplier, Supplier) */ public static ByteStream iterate(final BooleanSupplier hasNext, final ByteSupplier next) throws IllegalArgumentException { N.checkArgNotNull(hasNext); N.checkArgNotNull(next); return new IteratorByteStream(new ByteIteratorEx() { private boolean hasNextVal = false; @Override public boolean hasNext() { if (!hasNextVal) { hasNextVal = hasNext.getAsBoolean(); } return hasNextVal; } @Override public byte nextByte() { if (!hasNextVal && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } hasNextVal = false; return next.getAsByte(); } }); } /** * Creates a stream that iterates from an initial value, applying a function to generate subsequent values, * and continues as long as a predicate is satisfied. * * @param init the initial value * @param hasNext a BooleanSupplier that returns {@code true} if the iteration should continue * @param f a function to apply to the previous element to generate the next element * @return a ByteStream of elements generated by the iteration * @throws IllegalArgumentException if hasNext or f is null * @see Stream#iterate(Object, BooleanSupplier, UnaryOperator) */ public static ByteStream iterate(final byte init, final BooleanSupplier hasNext, final ByteUnaryOperator f) throws IllegalArgumentException { N.checkArgNotNull(hasNext); N.checkArgNotNull(f); return new IteratorByteStream(new ByteIteratorEx() { private byte cur = 0; private boolean isFirst = true; private boolean hasNextVal = false; @Override public boolean hasNext() { if (!hasNextVal) { hasNextVal = hasNext.getAsBoolean(); } return hasNextVal; } @Override public byte nextByte() { if (!hasNextVal && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } hasNextVal = false; if (isFirst) { isFirst = false; cur = init; } else { cur = f.applyAsByte(cur); } return cur; } }); } /** * Creates a stream that iterates from an initial value, applying a function to generate subsequent values, * and continues as long as a predicate is satisfied. * * @param init the initial value * @param hasNext determinate if the returned stream has next by hasNext.test(init) for first time and hasNext.test(f.apply(previous)) for remaining. * @param f a function to apply to the previous element to generate the next element * @return a ByteStream of elements generated by the iteration * @throws IllegalArgumentException if hasNext or f is null * @see Stream#iterate(Object, Predicate, UnaryOperator) */ public static ByteStream iterate(final byte init, final BytePredicate hasNext, final ByteUnaryOperator f) throws IllegalArgumentException { N.checkArgNotNull(hasNext); N.checkArgNotNull(f); return new IteratorByteStream(new ByteIteratorEx() { private byte cur = 0; private boolean isFirst = true; private boolean hasMore = true; private boolean hasNextVal = false; @Override public boolean hasNext() { if (!hasNextVal && hasMore) { if (isFirst) { isFirst = false; hasNextVal = hasNext.test(cur = init); } else { hasNextVal = hasNext.test(cur = f.applyAsByte(cur)); } if (!hasNextVal) { hasMore = false; } } return hasNextVal; } @Override public byte nextByte() { if (!hasNextVal && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } hasNextVal = false; return cur; } }); } /** * Creates a stream that iterates from an initial value, applying a function to generate subsequent values. * * @param init the initial value * @param f a function to apply to the previous element to generate the next element * @return a ByteStream of elements generated by the iteration * @throws IllegalArgumentException if f is null * @see Stream#iterate(Object, UnaryOperator) */ public static ByteStream iterate(final byte init, final ByteUnaryOperator f) throws IllegalArgumentException { N.checkArgNotNull(f); return new IteratorByteStream(new ByteIteratorEx() { private byte cur = 0; private boolean isFirst = true; @Override public boolean hasNext() { return true; } @Override public byte nextByte() { if (isFirst) { isFirst = false; cur = init; } else { cur = f.applyAsByte(cur); } return cur; } }); } /** * Generates a ByteStream using the provided ByteSupplier. * The supplier is used to generate each element of the stream. * * @param s the ByteSupplier that provides the elements of the stream * @return a ByteStream generated by the given supplier * @throws IllegalArgumentException if the supplier is null * @see Stream#generate(Supplier) */ public static ByteStream generate(final ByteSupplier s) throws IllegalArgumentException { N.checkArgNotNull(s); return new IteratorByteStream(new ByteIteratorEx() { @Override public boolean hasNext() { return true; } @Override public byte nextByte() { return s.getAsByte(); } }); } /** * Concatenates multiple arrays of bytes into a single ByteStream. * * @param a the arrays of bytes to concatenate * @return a ByteStream containing all the bytes from the input arrays * @see Stream#concat(Object[][]) */ @SafeVarargs public static ByteStream concat(final byte[]... a) { if (N.isEmpty(a)) { return empty(); } return concat(Arrays.asList(a)); } /** * Concatenates multiple ByteIterators into a single ByteStream. * * @param a the arrays of ByteIterator to concatenate * @return a ByteStream containing all the bytes from the input ByteIterators * @see Stream#concat(Iterator[]) */ @SafeVarargs public static ByteStream concat(final ByteIterator... a) { if (N.isEmpty(a)) { return empty(); } return concatIterators(Array.asList(a)); } /** * Concatenates multiple ByteStreams into a single ByteStream. * * @param a the arrays of ByteStream to concatenate * @return a ByteStream containing all the bytes from the input ByteStreams * @see Stream#concat(Stream[]) */ @SafeVarargs public static ByteStream concat(final ByteStream... a) { if (N.isEmpty(a)) { return empty(); } return concat(Array.asList(a)); } /** * Concatenates a list of byte array into a single ByteStream. * * @param c the list of byte array to concatenate * @return a ByteStream containing all the bytes from the input list of byte array * @see Stream#concat(Object[][]) */ @Beta public static ByteStream concat(final List c) { if (N.isEmpty(c)) { return empty(); } return of(new ByteIteratorEx() { private final Iterator iter = c.iterator(); private byte[] cur; private int cursor = 0; @Override public boolean hasNext() { while ((N.isEmpty(cur) || cursor >= cur.length) && iter.hasNext()) { cur = iter.next(); cursor = 0; } return cur != null && cursor < cur.length; } @Override public byte nextByte() { if ((cur == null || cursor >= cur.length) && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return cur[cursor++]; } }); } /** * Concatenates a collection of ByteStream into a single ByteStream. * * @param streams the collection of ByteStream to concatenate * @return a ByteStream containing all the bytes from the input collection of ByteStream * @see Stream#concat(Collection) */ public static ByteStream concat(final Collection streams) { return N.isEmpty(streams) ? empty() : new IteratorByteStream(new ByteIteratorEx() { //NOSONAR private final Iterator iterators = streams.iterator(); private ByteStream cur; private ByteIterator iter; @Override public boolean hasNext() { while ((iter == null || !iter.hasNext()) && iterators.hasNext()) { if (cur != null) { cur.close(); } cur = iterators.next(); iter = cur == null ? null : cur.iteratorEx(); } return iter != null && iter.hasNext(); } @Override public byte nextByte() { if ((iter == null || !iter.hasNext()) && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return iter.nextByte(); } }).onClose(newCloseHandler(streams)); } /** * Concatenates a collection of ByteIterator into a single ByteStream. * * @param byteIterators the collection of ByteIterator to concatenate * @return a ByteStream containing all the bytes from the input collection of ByteIterator * @see Stream#concatIterators(Collection) */ @Beta public static ByteStream concatIterators(final Collection byteIterators) { if (N.isEmpty(byteIterators)) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private final Iterator iter = byteIterators.iterator(); private ByteIterator cur; @Override public boolean hasNext() { while ((cur == null || !cur.hasNext()) && iter.hasNext()) { cur = iter.next(); } return cur != null && cur.hasNext(); } @Override public byte nextByte() { if ((cur == null || !cur.hasNext()) && !hasNext()) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return cur.nextByte(); } }); } /** * Zips two byte arrays into a single ByteStream until one of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * * @param a the first byte array * @param b the second byte array * @param zipFunction the function to combine elements from both arrays * @return a ByteStream containing the results of applying the zip function to the elements of the input arrays * @see Stream#zip(Object[], Object[], BiFunction) */ public static ByteStream zip(final byte[] a, final byte[] b, final ByteBinaryOperator zipFunction) { if (N.isEmpty(a) || N.isEmpty(b)) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private final int len = N.min(N.len(a), N.len(b)); private int cursor = 0; @Override public boolean hasNext() { return cursor < len; } @Override public byte nextByte() { if (cursor >= len) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return zipFunction.applyAsByte(a[cursor], b[cursor++]); } }); } /** * Zips three byte arrays into a single ByteStream until one of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * * @param a the first byte array * @param b the second byte array * @param c the third byte array * @param zipFunction the function to combine elements from all three arrays * @return a ByteStream containing the results of applying the zip function to the elements of the input arrays * @see Stream#zip(Object[], Object[], Object[], TriFunction) */ public static ByteStream zip(final byte[] a, final byte[] b, final byte[] c, final ByteTernaryOperator zipFunction) { if (N.isEmpty(a) || N.isEmpty(b) || N.isEmpty(c)) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private final int len = N.min(N.len(a), N.len(b), N.len(c)); private int cursor = 0; @Override public boolean hasNext() { return cursor < len; } @Override public byte nextByte() { if (cursor >= len) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } return zipFunction.applyAsByte(a[cursor], b[cursor], c[cursor++]); } }); } /** * Zips two ByteIterators into a single ByteStream until one of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * * @param a the first ByteIterator * @param b the second ByteIterator * @param zipFunction the function to combine elements from both iterators * @return a ByteStream containing the results of applying the zip function to the elements of the input iterators * @see Stream#zip(Iterator, Iterator, BiFunction) */ public static ByteStream zip(final ByteIterator a, final ByteIterator b, final ByteBinaryOperator zipFunction) { return new IteratorByteStream(new ByteIteratorEx() { private final ByteIterator iterA = a == null ? ByteIterator.empty() : a; private final ByteIterator iterB = b == null ? ByteIterator.empty() : b; @Override public boolean hasNext() { return iterA.hasNext() && iterB.hasNext(); } @Override public byte nextByte() { return zipFunction.applyAsByte(iterA.nextByte(), iterB.nextByte()); } }); } /** * Zips three ByteIterators into a single ByteStream until one of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * * @param a the first ByteIterator * @param b the second ByteIterator * @param c the third ByteIterator * @param zipFunction the function to combine elements from all three iterators * @return a ByteStream containing the results of applying the zip function to the elements of the input iterators * @see Stream#zip(Iterator, Iterator, Iterator, TriFunction) */ public static ByteStream zip(final ByteIterator a, final ByteIterator b, final ByteIterator c, final ByteTernaryOperator zipFunction) { return new IteratorByteStream(new ByteIteratorEx() { private final ByteIterator iterA = a == null ? ByteIterator.empty() : a; private final ByteIterator iterB = b == null ? ByteIterator.empty() : b; private final ByteIterator iterC = c == null ? ByteIterator.empty() : c; @Override public boolean hasNext() { return iterA.hasNext() && iterB.hasNext() && iterC.hasNext(); } @Override public byte nextByte() { return zipFunction.applyAsByte(iterA.nextByte(), iterB.nextByte(), iterC.nextByte()); } }); } /** * Zips two ByteStreams into a single ByteStream until one of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * * @param a the first ByteStream * @param b the second ByteStream * @param zipFunction the function to combine elements from both streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Stream, Stream, BiFunction) */ public static ByteStream zip(final ByteStream a, final ByteStream b, final ByteBinaryOperator zipFunction) { return zip(iterate(a), iterate(b), zipFunction).onClose(newCloseHandler(a, b)); } /** * Zips three ByteStreams into a single ByteStream until one of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * * @param a the first ByteStream * @param b the second ByteStream * @param c the third ByteStream * @param zipFunction the function to combine elements from all three streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Stream, Stream, Stream, TriFunction) */ public static ByteStream zip(final ByteStream a, final ByteStream b, final ByteStream c, final ByteTernaryOperator zipFunction) { return zip(iterate(a), iterate(b), iterate(c), zipFunction).onClose(newCloseHandler(Array.asList(a, b, c))); } /** * Zips multiple ByteStreams into a single ByteStream until one of them runs out of values. * Each list of values is combined into a single value using the supplied zipFunction. * * @param streams the collection of ByteStream to zip * @param zipFunction the function to combine elements from all the streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Collection, Function) */ public static ByteStream zip(final Collection streams, final ByteNFunction zipFunction) { //noinspection resource return Stream.zip(streams, zipFunction).mapToByte(ToByteFunction.UNBOX); } /** * Zips two byte arrays into a single ByteStream until all of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * If one array runs out of values before the other, the specified valueForNoneA or valueForNoneB is used. * * @param a the first byte array * @param b the second byte array * @param valueForNoneA the default value to use if the first array is shorter * @param valueForNoneB the default value to use if the second array is shorter * @param zipFunction the function to combine elements from both arrays * @return a ByteStream containing the results of applying the zip function to the elements of the input arrays * @see Stream#zip(Object[], Object[], Object, Object, BiFunction) */ public static ByteStream zip(final byte[] a, final byte[] b, final byte valueForNoneA, final byte valueForNoneB, final ByteBinaryOperator zipFunction) { if (N.isEmpty(a) && N.isEmpty(b)) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private final int aLen = N.len(a), bLen = N.len(b), len = N.max(aLen, bLen); private int cursor = 0; private byte ret = 0; @Override public boolean hasNext() { return cursor < len; } @Override public byte nextByte() { if (cursor >= len) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } ret = zipFunction.applyAsByte(cursor < aLen ? a[cursor] : valueForNoneA, cursor < bLen ? b[cursor] : valueForNoneB); cursor++; return ret; } }); } /** * Zips three byte arrays into a single ByteStream until all of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * If one array runs out of values before the other, the specified valueForNoneA, valueForNoneB or valueForNoneC is used. * * @param a the first byte array * @param b the second byte array * @param c the third byte array * @param valueForNoneA the default value to use if the first array is shorter * @param valueForNoneB the default value to use if the second array is shorter * @param valueForNoneC the default value to use if the third array is shorter * @param zipFunction the function to combine elements from all three arrays * @return a ByteStream containing the results of applying the zip function to the elements of the input arrays * @see Stream#zip(Object[], Object[], Object[], Object, Object, Object, TriFunction) */ public static ByteStream zip(final byte[] a, final byte[] b, final byte[] c, final byte valueForNoneA, final byte valueForNoneB, final byte valueForNoneC, final ByteTernaryOperator zipFunction) { if (N.isEmpty(a) && N.isEmpty(b) && N.isEmpty(c)) { return empty(); } return new IteratorByteStream(new ByteIteratorEx() { private final int aLen = N.len(a), bLen = N.len(b), cLen = N.len(c), len = N.max(aLen, bLen, cLen); private int cursor = 0; private byte ret = 0; @Override public boolean hasNext() { return cursor < len; } @Override public byte nextByte() { if (cursor >= len) { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } ret = zipFunction.applyAsByte(cursor < aLen ? a[cursor] : valueForNoneA, cursor < bLen ? b[cursor] : valueForNoneB, cursor < cLen ? c[cursor] : valueForNoneC); cursor++; return ret; } }); } /** * Zips two ByteIterators into a single ByteStream until all of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * If one iterator runs out of values before the other, the specified valueForNoneA or valueForNoneB is used. * * @param a the first ByteIterator * @param b the second ByteIterator * @param valueForNoneA the default value to use if the first iterator is shorter * @param valueForNoneB the default value to use if the second iterator is shorter * @param zipFunction the function to combine elements from both iterators * @return a ByteStream containing the results of applying the zip function to the elements of the input iterators * @see Stream#zip(Iterator, Iterator, Object, Object, BiFunction) */ public static ByteStream zip(final ByteIterator a, final ByteIterator b, final byte valueForNoneA, final byte valueForNoneB, final ByteBinaryOperator zipFunction) { return new IteratorByteStream(new ByteIteratorEx() { private final ByteIterator iterA = a == null ? ByteIterator.empty() : a; private final ByteIterator iterB = b == null ? ByteIterator.empty() : b; @Override public boolean hasNext() { return iterA.hasNext() || iterB.hasNext(); } @Override public byte nextByte() { if (iterA.hasNext()) { return zipFunction.applyAsByte(iterA.nextByte(), iterB.hasNext() ? iterB.nextByte() : valueForNoneB); } else { return zipFunction.applyAsByte(valueForNoneA, iterB.nextByte()); } } }); } /** * Zips three ByteIterators into a single ByteStream until all of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * If one iterator runs out of values before the other, the specified valueForNoneA, valueForNoneB or valueForNoneC is used. * * @param a the first ByteIterator * @param b the second ByteIterator * @param c the third ByteIterator * @param valueForNoneA the default value to use if the first iterator is shorter * @param valueForNoneB the default value to use if the second iterator is shorter * @param valueForNoneC the default value to use if the third iterator is shorter * @param zipFunction the function to combine elements from all three iterators * @return a ByteStream containing the results of applying the zip function to the elements of the input iterators * @see Stream#zip(Iterator, Iterator, Iterator, Object, Object, Object, TriFunction) */ public static ByteStream zip(final ByteIterator a, final ByteIterator b, final ByteIterator c, final byte valueForNoneA, final byte valueForNoneB, final byte valueForNoneC, final ByteTernaryOperator zipFunction) { return new IteratorByteStream(new ByteIteratorEx() { private final ByteIterator iterA = a == null ? ByteIterator.empty() : a; private final ByteIterator iterB = b == null ? ByteIterator.empty() : b; private final ByteIterator iterC = c == null ? ByteIterator.empty() : c; @Override public boolean hasNext() { return iterA.hasNext() || iterB.hasNext() || iterC.hasNext(); } @Override public byte nextByte() { if (iterA.hasNext()) { return zipFunction.applyAsByte(iterA.nextByte(), iterB.hasNext() ? iterB.nextByte() : valueForNoneB, iterC.hasNext() ? iterC.nextByte() : valueForNoneC); } else if (iterB.hasNext()) { return zipFunction.applyAsByte(valueForNoneA, iterB.nextByte(), iterC.hasNext() ? iterC.nextByte() : valueForNoneC); } else { return zipFunction.applyAsByte(valueForNoneA, valueForNoneB, iterC.nextByte()); } } }); } /** * Zips two ByteStreams into a single ByteStream until all of them runs out of values. * Each pair of values is combined into a single value using the supplied zipFunction. * If one stream runs out of values before the other, the specified valueForNoneA or valueForNoneB is used. * * @param a the first ByteStream * @param b the second ByteStream * @param valueForNoneA the default value to use if the first stream is shorter * @param valueForNoneB the default value to use if the second stream is shorter * @param zipFunction the function to combine elements from both streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Stream, Stream, Object, Object, BiFunction) */ public static ByteStream zip(final ByteStream a, final ByteStream b, final byte valueForNoneA, final byte valueForNoneB, final ByteBinaryOperator zipFunction) { return zip(iterate(a), iterate(b), valueForNoneA, valueForNoneB, zipFunction).onClose(newCloseHandler(a, b)); } /** * Zips three ByteStreams into a single ByteStream until all of them runs out of values. * Each triple of values is combined into a single value using the supplied zipFunction. * If one stream runs out of values before the other, the specified valueForNoneA, valueForNoneB or valueForNoneC is used. * * @param a the first ByteStream * @param b the second ByteStream * @param c the third ByteStream * @param valueForNoneA the default value to use if the first stream is shorter * @param valueForNoneB the default value to use if the second stream is shorter * @param valueForNoneC the default value to use if the third stream is shorter * @param zipFunction the function to combine elements from all three streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Stream, Stream, Stream, Object, Object, Object, TriFunction) */ public static ByteStream zip(final ByteStream a, final ByteStream b, final ByteStream c, final byte valueForNoneA, final byte valueForNoneB, final byte valueForNoneC, final ByteTernaryOperator zipFunction) { return zip(iterate(a), iterate(b), iterate(c), valueForNoneA, valueForNoneB, valueForNoneC, zipFunction) .onClose(newCloseHandler(Array.asList(a, b, c))); } /** * Zips multiple ByteStreams into a single ByteStream until all of them runs out of values. * Each list of values is combined into a single value using the supplied zipFunction. * If one stream runs out of values before the other, the specified valuesForNone is used. * * @param streams the collection of ByteStream instances to zip * @param valuesForNone the default value to use if the corresponding stream is shorter * @param zipFunction the function to combine elements from all the streams * @return a ByteStream containing the results of applying the zip function to the elements of the input streams * @see Stream#zip(Collection, List, Function) */ public static ByteStream zip(final Collection streams, final byte[] valuesForNone, final ByteNFunction zipFunction) { //noinspection resource return Stream.zip(streams, valuesForNone, zipFunction).mapToByte(ToByteFunction.UNBOX); } /** * Merges two byte arrays into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the two arrays. * * @param a the first byte array * @param b the second byte array * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the two input arrays * @see Stream#merge(Object[], Object[], BiFunction) */ public static ByteStream merge(final byte[] a, final byte[] b, final ByteBiFunction nextSelector) { if (N.isEmpty(a)) { return of(b); } else if (N.isEmpty(b)) { return of(a); } return new IteratorByteStream(new ByteIteratorEx() { private final int lenA = a.length; private final int lenB = b.length; private int cursorA = 0; private int cursorB = 0; @Override public boolean hasNext() { return cursorA < lenA || cursorB < lenB; } @Override public byte nextByte() { if (cursorA < lenA) { if ((cursorB >= lenB) || (nextSelector.apply(a[cursorA], b[cursorB]) == MergeResult.TAKE_FIRST)) { return a[cursorA++]; } else { return b[cursorB++]; } } else if (cursorB < lenB) { return b[cursorB++]; } else { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } } }); } /** * Merges three byte arrays into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the three arrays. * * @param a the first byte array * @param b the second byte array * @param c the third byte array * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the three input arrays * @see Stream#merge(Object[], Object[], Object[], BiFunction) */ public static ByteStream merge(final byte[] a, final byte[] b, final byte[] c, final ByteBiFunction nextSelector) { //noinspection resource return merge(merge(a, b, nextSelector).iteratorEx(), ByteStream.of(c).iteratorEx(), nextSelector); } /** * Merges two ByteIterators into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the two iterators. * * @param a the first ByteIterator * @param b the second ByteIterator * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the two input iterators * @see Stream#merge(Iterator, Iterator, BiFunction) */ public static ByteStream merge(final ByteIterator a, final ByteIterator b, final ByteBiFunction nextSelector) { return new IteratorByteStream(new ByteIteratorEx() { private final ByteIterator iterA = a == null ? ByteIterator.empty() : a; private final ByteIterator iterB = b == null ? ByteIterator.empty() : b; private byte nextA = 0; private byte nextB = 0; private boolean hasNextA = false; private boolean hasNextB = false; @Override public boolean hasNext() { return iterA.hasNext() || iterB.hasNext() || hasNextA || hasNextB; } @Override public byte nextByte() { if (hasNextA) { if (iterB.hasNext()) { if (nextSelector.apply(nextA, (nextB = iterB.nextByte())) == MergeResult.TAKE_FIRST) { hasNextA = false; hasNextB = true; return nextA; } else { return nextB; } } else { hasNextA = false; return nextA; } } else if (hasNextB) { if (iterA.hasNext()) { if (nextSelector.apply((nextA = iterA.nextByte()), nextB) == MergeResult.TAKE_FIRST) { return nextA; } else { hasNextA = true; hasNextB = false; return nextB; } } else { hasNextB = false; return nextB; } } else if (iterA.hasNext()) { if (iterB.hasNext()) { if (nextSelector.apply((nextA = iterA.nextByte()), (nextB = iterB.nextByte())) == MergeResult.TAKE_FIRST) { hasNextB = true; return nextA; } else { hasNextA = true; return nextB; } } else { return iterA.nextByte(); } } else if (iterB.hasNext()) { return iterB.nextByte(); } else { throw new NoSuchElementException(ERROR_MSG_FOR_NO_SUCH_EX); } } }); } /** * Merges three ByteIterators into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the three iterators. * * @param a the first ByteIterator * @param b the second ByteIterator * @param c the third ByteIterator * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the three input iterators * @see Stream#merge(Iterator, Iterator, Iterator, BiFunction) */ public static ByteStream merge(final ByteIterator a, final ByteIterator b, final ByteIterator c, final ByteBiFunction nextSelector) { //noinspection resource return merge(merge(a, b, nextSelector).iteratorEx(), c, nextSelector); } /** * Merges two ByteStreams into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the two streams. * * @param a the first ByteStream * @param b the second ByteStream * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the two input streams * @see Stream#merge(Stream, Stream, BiFunction) */ public static ByteStream merge(final ByteStream a, final ByteStream b, final ByteBiFunction nextSelector) { return merge(iterate(a), iterate(b), nextSelector).onClose(newCloseHandler(a, b)); } /** * Merges three ByteStreams into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the three streams. * * @param a the first ByteStream * @param b the second ByteStream * @param c the third ByteStream * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the three input streams * @see Stream#merge(Stream, Stream, Stream, BiFunction) */ public static ByteStream merge(final ByteStream a, final ByteStream b, final ByteStream c, final ByteBiFunction nextSelector) { return merge(merge(a, b, nextSelector), c, nextSelector); } /** * Merges a collection of ByteStream into a single ByteStream based on the provided nextSelector function. * The nextSelector function determines which element to take next from the multiple streams. * * @param streams the collection of ByteStream instances to merge * @param nextSelector a function to determine which element should be selected as the next element. * The first parameter is selected if {@code MergeResult.TAKE_FIRST} is returned, otherwise the second parameter is selected. * @return a ByteStream containing the merged elements from the input ByteStreams * @see Stream#merge(Collection, BiFunction) */ public static ByteStream merge(final Collection streams, final ByteBiFunction nextSelector) { if (N.isEmpty(streams)) { return empty(); } else if (streams.size() == 1) { return streams.iterator().next(); } else if (streams.size() == 2) { final Iterator iter = streams.iterator(); return merge(iter.next(), iter.next(), nextSelector); } final Iterator iter = streams.iterator(); ByteStream result = merge(iter.next(), iter.next(), nextSelector); while (iter.hasNext()) { result = merge(result, iter.next(), nextSelector); } return result; } // /** // * Merges a collection of ByteStream into a single ByteStream in parallel. // * All the elements from each input ByteStream will be merged into two queues by multiple threads first. // * Then these two new queues will be merged into one ByteStream in current thread. // * This method is not totally lazy evaluation and may cause out of memory error if there are too many elements merged into the two new queues. // * Consider using {@code merge}, which is totally lazy evaluation. // * // * @param streams the collection of ByteStream to be merged // * @param nextSelector a function to determine which element should be selected as next element. // * @return a ByteStream containing the merged elements from the input ByteStreams // * @see Stream#parallelMerge(Collection, BiFunction) // */ // public static ByteStream parallelMerge(final Collection streams, final ByteBiFunction nextSelector) { // return parallelMerge(streams, nextSelector, DEFAULT_MAX_THREAD_NUM); // } // // /** // * Merges a collection of ByteStream into a single ByteStream in parallel. // * All the elements from each input ByteStream will be merged into two queues by multiple threads first. // * Then these two new queues will be merged into one ByteStream in current thread. // * This method is not totally lazy evaluation and may cause out of memory error if there are too many elements merged into the two new queues. // * Consider using {@code merge}, which is totally lazy evaluation. // * // * @param streams the collection of ByteStream to be merged // * @param nextSelector a function to determine which element should be selected as next element. // * @param maxThreadNum the max thread number for the parallel merge. // * @return a ByteStream containing the merged elements from the input ByteStreams // * @see Stream#parallelMerge(Collection, BiFunction, int) // */ // public static ByteStream parallelMerge(final Collection streams, final ByteBiFunction nextSelector, // final int maxThreadNum) throws IllegalArgumentException { // N.checkArgument(maxThreadNum > 0, "'maxThreadNum' must not less than 1"); // // if (maxThreadNum <= 1) { // return merge(streams, nextSelector); // } else if (N.isEmpty(streams)) { // return empty(); // } else if (streams.size() == 1) { // return streams.iterator().next(); // } else if (streams.size() == 2) { // final Iterator iter = streams.iterator(); // return merge(iter.next(), iter.next(), nextSelector); // } else if (streams.size() == 3) { // final Iterator iter = streams.iterator(); // return merge(iter.next(), iter.next(), iter.next(), nextSelector); // } // // final Supplier supplier = () -> { // final Queue queue = N.newLinkedList(); // // queue.addAll(streams); // // final Holder eHolder = new Holder<>(); // final MutableInt cnt = MutableInt.of(streams.size()); // final List> futureList = new ArrayList<>(streams.size() - 1); // // final int threadNum = N.min(maxThreadNum, streams.size() / 2 + 1); // // AsyncExecutor asyncExecutorToUse = checkAsyncExecutor(DEFAULT_ASYNC_EXECUTOR, threadNum, 0); // // for (int i = 0; i < threadNum; i++) { // asyncExecutorToUse = execute(asyncExecutorToUse, threadNum, 0, i, futureList, () -> { // ByteStream a = null; // ByteStream b = null; // ByteStream c1 = null; // // try { // while (eHolder.value() == null) { // synchronized (queue) { // if (cnt.value() > 2 && queue.size() > 1) { // a = queue.poll(); // b = queue.poll(); // // cnt.decrement(); // } else { // break; // } // } // // c1 = ByteStream.of(merge(a, b, nextSelector).toArray()); // // synchronized (queue) { // queue.offer(c1); // } // } // } catch (final Throwable e) { // NOSONAR // setError(eHolder, e); // } // }); // } // // completeAndShutdownTempExecutor(futureList, eHolder, streams, asyncExecutorToUse); // // return merge(queue.poll(), queue.poll(), nextSelector); // }; // // return Stream.just(supplier).flatMapToByte(Supplier::get); // } public abstract static class ByteStreamEx extends ByteStream { private ByteStreamEx(final boolean sorted, final Collection closeHandlers) { //NOSONAR super(sorted, closeHandlers); } } }




© 2015 - 2025 Weber Informatics LLC | Privacy Policy