/* * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. Oracle designates this * particular file as subject to the "Classpath" exception as provided * by Oracle in the LICENSE file that accompanied this code. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have * questions. */ package jdk.incubator.vector; import java.nio.ByteBuffer; import java.nio.ByteOrder; import java.util.Objects; import java.util.function.IntUnaryOperator; import java.util.concurrent.ThreadLocalRandom; import jdk.internal.misc.Unsafe; import jdk.internal.vm.annotation.ForceInline; import static jdk.incubator.vector.VectorIntrinsics.*; /** * A specialized {@link Vector} representing an ordered immutable sequence of * {@code byte} values. */ @SuppressWarnings("cast") public abstract class ByteVector extends Vector { ByteVector() {} private static final int ARRAY_SHIFT = 31 - Integer.numberOfLeadingZeros(Unsafe.ARRAY_BYTE_INDEX_SCALE); // Unary operator interface FUnOp { byte apply(int i, byte a); } abstract ByteVector uOp(FUnOp f); abstract ByteVector uOp(Mask m, FUnOp f); // Binary operator interface FBinOp { byte apply(int i, byte a, byte b); } abstract ByteVector bOp(Vector v, FBinOp f); abstract ByteVector bOp(Vector v, Mask m, FBinOp f); // Trinary operator interface FTriOp { byte apply(int i, byte a, byte b, byte c); } abstract ByteVector tOp(Vector v1, Vector v2, FTriOp f); abstract ByteVector tOp(Vector v1, Vector v2, Mask m, FTriOp f); // Reduction operator abstract byte rOp(byte v, FBinOp f); // Binary test interface FBinTest { boolean apply(int i, byte a, byte b); } abstract Mask bTest(Vector v, FBinTest f); // Foreach interface FUnCon { void apply(int i, byte a); } abstract void forEach(FUnCon f); abstract void forEach(Mask m, FUnCon f); // Static factories /** * Returns a vector where all lane elements are set to the default * primitive value. * * @param species species of desired vector * @return a zero vector of given species */ @ForceInline @SuppressWarnings("unchecked") public static ByteVector zero(ByteSpecies species) { return species.zero(); } /** * Loads a vector from a byte array starting at an offset. *

* Bytes are composed into primitive lane elements according to the * native byte order of the underlying platform *

* This method behaves as if it returns the result of calling the * byte buffer, offset, and mask accepting * {@link #fromByteBuffer(ByteSpecies, ByteBuffer, int, Mask) method} as follows: *

{@code
     * return this.fromByteBuffer(ByteBuffer.wrap(a), i, this.maskAllTrue());
     * }
* * @param species species of desired vector * @param a the byte array * @param ix the offset into the array * @return a vector loaded from a byte array * @throws IndexOutOfBoundsException if {@code i < 0} or * {@code i > a.length - (this.length() * this.elementSize() / Byte.SIZE)} */ @ForceInline @SuppressWarnings("unchecked") public static ByteVector fromByteArray(ByteSpecies species, byte[] a, int ix) { Objects.requireNonNull(a); ix = VectorIntrinsics.checkIndex(ix, a.length, species.bitSize() / Byte.SIZE); return VectorIntrinsics.load((Class) species.boxType(), byte.class, species.length(), a, ((long) ix) + Unsafe.ARRAY_BYTE_BASE_OFFSET, a, ix, species, (c, idx, s) -> { ByteBuffer bbc = ByteBuffer.wrap(c, idx, a.length - idx).order(ByteOrder.nativeOrder()); ByteBuffer tb = bbc; return ((ByteSpecies)s).op(i -> tb.get()); }); } /** * Loads a vector from a byte array starting at an offset and using a * mask. *

* Bytes are composed into primitive lane elements according to the * native byte order of the underlying platform. *

* This method behaves as if it returns the result of calling the * byte buffer, offset, and mask accepting * {@link #fromByteBuffer(ByteSpecies, ByteBuffer, int, Mask) method} as follows: *

{@code
     * return this.fromByteBuffer(ByteBuffer.wrap(a), i, m);
     * }
* * @param species species of desired vector * @param a the byte array * @param ix the offset into the array * @param m the mask * @return a vector loaded from a byte array * @throws IndexOutOfBoundsException if {@code i < 0} or * {@code i > a.length - (this.length() * this.elementSize() / Byte.SIZE)} * @throws IndexOutOfBoundsException if the offset is {@code < 0}, * or {@code > a.length}, * for any vector lane index {@code N} where the mask at lane {@code N} * is set * {@code i >= a.length - (N * this.elementSize() / Byte.SIZE)} */ @ForceInline public static ByteVector fromByteArray(ByteSpecies species, byte[] a, int ix, Mask m) { return zero(species).blend(fromByteArray(species, a, ix), m); } /** * Loads a vector from an array starting at offset. *

* For each vector lane, where {@code N} is the vector lane index, the * array element at index {@code i + N} is placed into the * resulting vector at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param i the offset into the array * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code i < 0}, or * {@code i > a.length - this.length()} */ @ForceInline public static ByteVector fromArray(ByteSpecies species, byte[] a, int i) { Objects.requireNonNull(a); i = VectorIntrinsics.checkIndex(i, a.length, species.length()); return fromArrayWithoutCheck(species, a, i); } @ForceInline @SuppressWarnings("unchecked") static ByteVector fromArrayWithoutCheck(ByteSpecies species, byte[] a, int i) { return VectorIntrinsics.load((Class) species.boxType(), byte.class, species.length(), a, (((long) i) << ARRAY_SHIFT) + Unsafe.ARRAY_BYTE_BASE_OFFSET, a, i, species, (c, idx, s) -> ((ByteSpecies)s).op(n -> c[idx + n])); } /** * Loads a vector from an array starting at offset and using a mask. *

* For each vector lane, where {@code N} is the vector lane index, * if the mask lane at index {@code N} is set then the array element at * index {@code i + N} is placed into the resulting vector at lane index * {@code N}, otherwise the default element value is placed into the * resulting vector at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param i the offset into the array * @param m the mask * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code i < 0}, or * for any vector lane index {@code N} where the mask at lane {@code N} * is set {@code i > a.length - N} */ @ForceInline public static ByteVector fromArray(ByteSpecies species, byte[] a, int i, Mask m) { Objects.requireNonNull(a); if (i + species.length() <= a.length) { return zero(species).blend(fromArrayWithoutCheck(species, a, i), m); } else { return species.op(m, n -> a[i + n]); } } /** * Loads a vector from an array using indexes obtained from an index * map. *

* For each vector lane, where {@code N} is the vector lane index, the * array element at index {@code i + indexMap[j + N]} is placed into the * resulting vector at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param i the offset into the array, may be negative if relative * indexes in the index map compensate to produce a value within the * array bounds * @param indexMap the index map * @param j the offset into the index map * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code j < 0}, or * {@code j > indexMap.length - this.length()}, * or for any vector lane index {@code N} the result of * {@code i + indexMap[j + N]} is {@code < 0} or {@code >= a.length} */ public static ByteVector fromArray(ByteSpecies species, byte[] a, int i, int[] indexMap, int j) { return species.op(n -> a[i + indexMap[j + n]]); } /** * Loads a vector from an array using indexes obtained from an index * map and using a mask. *

* For each vector lane, where {@code N} is the vector lane index, * if the mask lane at index {@code N} is set then the array element at * index {@code i + indexMap[j + N]} is placed into the resulting vector * at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param i the offset into the array, may be negative if relative * indexes in the index map compensate to produce a value within the * array bounds * @param m the mask * @param indexMap the index map * @param j the offset into the index map * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code j < 0}, or * {@code j > indexMap.length - this.length()}, * or for any vector lane index {@code N} where the mask at lane * {@code N} is set the result of {@code i + indexMap[j + N]} is * {@code < 0} or {@code >= a.length} */ public static ByteVector fromArray(ByteSpecies species, byte[] a, int i, Mask m, int[] indexMap, int j) { return species.op(m, n -> a[i + indexMap[j + n]]); } /** * Loads a vector from a {@link ByteBuffer byte buffer} starting at an * offset into the byte buffer. *

* Bytes are composed into primitive lane elements according to the * native byte order of the underlying platform. *

* This method behaves as if it returns the result of calling the * byte buffer, offset, and mask accepting * {@link #fromByteBuffer(ByteSpecies, ByteBuffer, int, Mask)} method} as follows: *

{@code
     *   return this.fromByteBuffer(b, i, this.maskAllTrue())
     * }
* * @param species species of desired vector * @param bb the byte buffer * @param ix the offset into the byte buffer * @return a vector loaded from a byte buffer * @throws IndexOutOfBoundsException if the offset is {@code < 0}, * or {@code > b.limit()}, * or if there are fewer than * {@code this.length() * this.elementSize() / Byte.SIZE} bytes * remaining in the byte buffer from the given offset */ @ForceInline @SuppressWarnings("unchecked") public static ByteVector fromByteBuffer(ByteSpecies species, ByteBuffer bb, int ix) { if (bb.order() != ByteOrder.nativeOrder()) { throw new IllegalArgumentException(); } ix = VectorIntrinsics.checkIndex(ix, bb.limit(), species.bitSize() / Byte.SIZE); return VectorIntrinsics.load((Class) species.boxType(), byte.class, species.length(), U.getReference(bb, BYTE_BUFFER_HB), U.getLong(bb, BUFFER_ADDRESS) + ix, bb, ix, species, (c, idx, s) -> { ByteBuffer bbc = c.duplicate().position(idx).order(ByteOrder.nativeOrder()); ByteBuffer tb = bbc; return ((ByteSpecies)s).op(i -> tb.get()); }); } /** * Loads a vector from a {@link ByteBuffer byte buffer} starting at an * offset into the byte buffer and using a mask. *

* This method behaves as if the byte buffer is viewed as a primitive * {@link java.nio.Buffer buffer} for the primitive element type, * according to the native byte order of the underlying platform, and * the returned vector is loaded with a mask from a primitive array * obtained from the primitive buffer. * The following pseudocode expresses the behaviour, where * {@coce EBuffer} is the primitive buffer type, {@code e} is the * primitive element type, and {@code ESpecies} is the primitive * species for {@code e}: *

{@code
     * EBuffer eb = b.duplicate().
     *     order(ByteOrder.nativeOrder()).position(i).
     *     asEBuffer();
     * e[] es = new e[this.length()];
     * for (int n = 0; n < t.length; n++) {
     *     if (m.isSet(n))
     *         es[n] = eb.get(n);
     * }
     * Vector r = ((ESpecies)this).fromArray(es, 0, m);
     * }
* * @param species species of desired vector * @param bb the byte buffer * @param ix the offset into the byte buffer * @param m the mask * @return a vector loaded from a byte buffer * @throws IndexOutOfBoundsException if the offset is {@code < 0}, * or {@code > b.limit()}, * for any vector lane index {@code N} where the mask at lane {@code N} * is set * {@code i >= b.limit() - (N * this.elementSize() / Byte.SIZE)} */ @ForceInline public static ByteVector fromByteBuffer(ByteSpecies species, ByteBuffer bb, int ix, Mask m) { return zero(species).blend(fromByteBuffer(species, bb, ix), m); } /** * Returns a mask where each lane is set or unset according to given * {@code boolean} values *

* For each mask lane, where {@code N} is the mask lane index, * if the given {@code boolean} value at index {@code N} is {@code true} * then the mask lane at index {@code N} is set, otherwise it is unset. * * @param species mask species * @param bits the given {@code boolean} values * @return a mask where each lane is set or unset according to the given {@code boolean} value * @throws IndexOutOfBoundsException if {@code bits.length < species.length()} */ @ForceInline public static Mask maskFromValues(ByteSpecies species, boolean... bits) { if (species.boxType() == ByteMaxVector.class) return new ByteMaxVector.ByteMaxMask(bits); switch (species.bitSize()) { case 64: return new Byte64Vector.Byte64Mask(bits); case 128: return new Byte128Vector.Byte128Mask(bits); case 256: return new Byte256Vector.Byte256Mask(bits); case 512: return new Byte512Vector.Byte512Mask(bits); default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } // @@@ This is a bad implementation -- makes lambdas capturing -- fix this static Mask trueMask(ByteSpecies species) { if (species.boxType() == ByteMaxVector.class) return ByteMaxVector.ByteMaxMask.TRUE_MASK; switch (species.bitSize()) { case 64: return Byte64Vector.Byte64Mask.TRUE_MASK; case 128: return Byte128Vector.Byte128Mask.TRUE_MASK; case 256: return Byte256Vector.Byte256Mask.TRUE_MASK; case 512: return Byte512Vector.Byte512Mask.TRUE_MASK; default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } static Mask falseMask(ByteSpecies species) { if (species.boxType() == ByteMaxVector.class) return ByteMaxVector.ByteMaxMask.FALSE_MASK; switch (species.bitSize()) { case 64: return Byte64Vector.Byte64Mask.FALSE_MASK; case 128: return Byte128Vector.Byte128Mask.FALSE_MASK; case 256: return Byte256Vector.Byte256Mask.FALSE_MASK; case 512: return Byte512Vector.Byte512Mask.FALSE_MASK; default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } /** * Loads a mask from a {@code boolean} array starting at an offset. *

* For each mask lane, where {@code N} is the mask lane index, * if the array element at index {@code ix + N} is {@code true} then the * mask lane at index {@code N} is set, otherwise it is unset. * * @param species mask species * @param bits the {@code boolean} array * @param ix the offset into the array * @return the mask loaded from a {@code boolean} array * @throws IndexOutOfBoundsException if {@code ix < 0}, or * {@code ix > bits.length - species.length()} */ @ForceInline @SuppressWarnings("unchecked") public static Mask maskFromArray(ByteSpecies species, boolean[] bits, int ix) { Objects.requireNonNull(bits); ix = VectorIntrinsics.checkIndex(ix, bits.length, species.length()); return VectorIntrinsics.load((Class>) species.maskType(), byte.class, species.length(), bits, (((long) ix) << ARRAY_SHIFT) + Unsafe.ARRAY_BOOLEAN_BASE_OFFSET, bits, ix, species, (c, idx, s) -> (Mask) ((ByteSpecies)s).opm(n -> c[idx + n])); } /** * Returns a mask where all lanes are set. * * @param species mask species * @return a mask where all lanes are set */ @ForceInline @SuppressWarnings("unchecked") public static Mask maskAllTrue(ByteSpecies species) { return VectorIntrinsics.broadcastCoerced((Class>) species.maskType(), byte.class, species.length(), (byte)-1, species, ((z, s) -> trueMask((ByteSpecies)s))); } /** * Returns a mask where all lanes are unset. * * @param species mask species * @return a mask where all lanes are unset */ @ForceInline @SuppressWarnings("unchecked") public static Mask maskAllFalse(ByteSpecies species) { return VectorIntrinsics.broadcastCoerced((Class>) species.maskType(), byte.class, species.length(), 0, species, ((z, s) -> falseMask((ByteSpecies)s))); } /** * Returns a shuffle of mapped indexes where each lane element is * the result of applying a mapping function to the corresponding lane * index. *

* Care should be taken to ensure Shuffle values produced from this * method are consumed as constants to ensure optimal generation of * code. For example, values held in static final fields or values * held in loop constant local variables. *

* This method behaves as if a shuffle is created from an array of * mapped indexes as follows: *

{@code
     *   int[] a = new int[species.length()];
     *   for (int i = 0; i < a.length; i++) {
     *       a[i] = f.applyAsInt(i);
     *   }
     *   return this.shuffleFromValues(a);
     * }
* * @param species shuffle species * @param f the lane index mapping function * @return a shuffle of mapped indexes */ @ForceInline public static Shuffle shuffle(ByteSpecies species, IntUnaryOperator f) { if (species.boxType() == ByteMaxVector.class) return new ByteMaxVector.ByteMaxShuffle(f); switch (species.bitSize()) { case 64: return new Byte64Vector.Byte64Shuffle(f); case 128: return new Byte128Vector.Byte128Shuffle(f); case 256: return new Byte256Vector.Byte256Shuffle(f); case 512: return new Byte512Vector.Byte512Shuffle(f); default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } /** * Returns a shuffle where each lane element is the value of its * corresponding lane index. *

* This method behaves as if a shuffle is created from an identity * index mapping function as follows: *

{@code
     *   return this.shuffle(i -> i);
     * }
* * @param species shuffle species * @return a shuffle of lane indexes */ @ForceInline public static Shuffle shuffleIota(ByteSpecies species) { if (species.boxType() == ByteMaxVector.class) return new ByteMaxVector.ByteMaxShuffle(AbstractShuffle.IDENTITY); switch (species.bitSize()) { case 64: return new Byte64Vector.Byte64Shuffle(AbstractShuffle.IDENTITY); case 128: return new Byte128Vector.Byte128Shuffle(AbstractShuffle.IDENTITY); case 256: return new Byte256Vector.Byte256Shuffle(AbstractShuffle.IDENTITY); case 512: return new Byte512Vector.Byte512Shuffle(AbstractShuffle.IDENTITY); default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } /** * Returns a shuffle where each lane element is set to a given * {@code int} value logically AND'ed by the species length minus one. *

* For each shuffle lane, where {@code N} is the shuffle lane index, the * the {@code int} value at index {@code N} logically AND'ed by * {@code species.length() - 1} is placed into the resulting shuffle at * lane index {@code N}. * * @param species shuffle species * @param ixs the given {@code int} values * @return a shuffle where each lane element is set to a given * {@code int} value * @throws IndexOutOfBoundsException if the number of int values is * {@code < species.length()} */ @ForceInline public static Shuffle shuffleFromValues(ByteSpecies species, int... ixs) { if (species.boxType() == ByteMaxVector.class) return new ByteMaxVector.ByteMaxShuffle(ixs); switch (species.bitSize()) { case 64: return new Byte64Vector.Byte64Shuffle(ixs); case 128: return new Byte128Vector.Byte128Shuffle(ixs); case 256: return new Byte256Vector.Byte256Shuffle(ixs); case 512: return new Byte512Vector.Byte512Shuffle(ixs); default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } /** * Loads a shuffle from an {@code int} array starting at an offset. *

* For each shuffle lane, where {@code N} is the shuffle lane index, the * array element at index {@code i + N} logically AND'ed by * {@code species.length() - 1} is placed into the resulting shuffle at lane * index {@code N}. * * @param species shuffle species * @param ixs the {@code int} array * @param i the offset into the array * @return a shuffle loaded from the {@code int} array * @throws IndexOutOfBoundsException if {@code i < 0}, or * {@code i > a.length - species.length()} */ @ForceInline public static Shuffle shuffleFromArray(ByteSpecies species, int[] ixs, int i) { if (species.boxType() == ByteMaxVector.class) return new ByteMaxVector.ByteMaxShuffle(ixs, i); switch (species.bitSize()) { case 64: return new Byte64Vector.Byte64Shuffle(ixs, i); case 128: return new Byte128Vector.Byte128Shuffle(ixs, i); case 256: return new Byte256Vector.Byte256Shuffle(ixs, i); case 512: return new Byte512Vector.Byte512Shuffle(ixs, i); default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); } } // Ops @Override public abstract ByteVector add(Vector v); /** * Adds this vector to the broadcast of an input scalar. *

* This is a vector binary operation where the primitive addition operation * ({@code +}) is applied to lane elements. * * @param s the input scalar * @return the result of adding this vector to the broadcast of an input * scalar */ public abstract ByteVector add(byte s); @Override public abstract ByteVector add(Vector v, Mask m); /** * Adds this vector to broadcast of an input scalar, * selecting lane elements controlled by a mask. *

* This is a vector binary operation where the primitive addition operation * ({@code +}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the result of adding this vector to the broadcast of an input * scalar */ public abstract ByteVector add(byte s, Mask m); @Override public abstract ByteVector sub(Vector v); /** * Subtracts the broadcast of an input scalar from this vector. *

* This is a vector binary operation where the primitive subtraction * operation ({@code -}) is applied to lane elements. * * @param s the input scalar * @return the result of subtracting the broadcast of an input * scalar from this vector */ public abstract ByteVector sub(byte s); @Override public abstract ByteVector sub(Vector v, Mask m); /** * Subtracts the broadcast of an input scalar from this vector, selecting * lane elements controlled by a mask. *

* This is a vector binary operation where the primitive subtraction * operation ({@code -}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the result of subtracting the broadcast of an input * scalar from this vector */ public abstract ByteVector sub(byte s, Mask m); @Override public abstract ByteVector mul(Vector v); /** * Multiplies this vector with the broadcast of an input scalar. *

* This is a vector binary operation where the primitive multiplication * operation ({@code *}) is applied to lane elements. * * @param s the input scalar * @return the result of multiplying this vector with the broadcast of an * input scalar */ public abstract ByteVector mul(byte s); @Override public abstract ByteVector mul(Vector v, Mask m); /** * Multiplies this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a vector binary operation where the primitive multiplication * operation ({@code *}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the result of multiplying this vector with the broadcast of an * input scalar */ public abstract ByteVector mul(byte s, Mask m); @Override public abstract ByteVector neg(); @Override public abstract ByteVector neg(Mask m); @Override public abstract ByteVector abs(); @Override public abstract ByteVector abs(Mask m); @Override public abstract ByteVector min(Vector v); @Override public abstract ByteVector min(Vector v, Mask m); /** * Returns the minimum of this vector and the broadcast of an input scalar. *

* This is a vector binary operation where the operation * {@code (a, b) -> Math.min(a, b)} is applied to lane elements. * * @param s the input scalar * @return the minimum of this vector and the broadcast of an input scalar */ public abstract ByteVector min(byte s); @Override public abstract ByteVector max(Vector v); @Override public abstract ByteVector max(Vector v, Mask m); /** * Returns the maximum of this vector and the broadcast of an input scalar. *

* This is a vector binary operation where the operation * {@code (a, b) -> Math.max(a, b)} is applied to lane elements. * * @param s the input scalar * @return the maximum of this vector and the broadcast of an input scalar */ public abstract ByteVector max(byte s); @Override public abstract Mask equal(Vector v); /** * Tests if this vector is equal to the broadcast of an input scalar. *

* This is a vector binary test operation where the primitive equals * operation ({@code ==}) is applied to lane elements. * * @param s the input scalar * @return the result mask of testing if this vector is equal to the * broadcast of an input scalar */ public abstract Mask equal(byte s); @Override public abstract Mask notEqual(Vector v); /** * Tests if this vector is not equal to the broadcast of an input scalar. *

* This is a vector binary test operation where the primitive not equals * operation ({@code !=}) is applied to lane elements. * * @param s the input scalar * @return the result mask of testing if this vector is not equal to the * broadcast of an input scalar */ public abstract Mask notEqual(byte s); @Override public abstract Mask lessThan(Vector v); /** * Tests if this vector is less than the broadcast of an input scalar. *

* This is a vector binary test operation where the primitive less than * operation ({@code <}) is applied to lane elements. * * @param s the input scalar * @return the mask result of testing if this vector is less than the * broadcast of an input scalar */ public abstract Mask lessThan(byte s); @Override public abstract Mask lessThanEq(Vector v); /** * Tests if this vector is less or equal to the broadcast of an input scalar. *

* This is a vector binary test operation where the primitive less than * or equal to operation ({@code <=}) is applied to lane elements. * * @param s the input scalar * @return the mask result of testing if this vector is less than or equal * to the broadcast of an input scalar */ public abstract Mask lessThanEq(byte s); @Override public abstract Mask greaterThan(Vector v); /** * Tests if this vector is greater than the broadcast of an input scalar. *

* This is a vector binary test operation where the primitive greater than * operation ({@code >}) is applied to lane elements. * * @param s the input scalar * @return the mask result of testing if this vector is greater than the * broadcast of an input scalar */ public abstract Mask greaterThan(byte s); @Override public abstract Mask greaterThanEq(Vector v); /** * Tests if this vector is greater than or equal to the broadcast of an * input scalar. *

* This is a vector binary test operation where the primitive greater than * or equal to operation ({@code >=}) is applied to lane elements. * * @param s the input scalar * @return the mask result of testing if this vector is greater than or * equal to the broadcast of an input scalar */ public abstract Mask greaterThanEq(byte s); @Override public abstract ByteVector blend(Vector v, Mask m); /** * Blends the lane elements of this vector with those of the broadcast of an * input scalar, selecting lanes controlled by a mask. *

* For each lane of the mask, at lane index {@code N}, if the mask lane * is set then the lane element at {@code N} from the input vector is * selected and placed into the resulting vector at {@code N}, * otherwise the the lane element at {@code N} from this input vector is * selected and placed into the resulting vector at {@code N}. * * @param s the input scalar * @param m the mask controlling lane selection * @return the result of blending the lane elements of this vector with * those of the broadcast of an input scalar */ public abstract ByteVector blend(byte s, Mask m); @Override public abstract ByteVector rearrange(Vector v, Shuffle s, Mask m); @Override public abstract ByteVector rearrange(Shuffle m); @Override public abstract ByteVector reshape(Species s); @Override public abstract ByteVector rotateEL(int i); @Override public abstract ByteVector rotateER(int i); @Override public abstract ByteVector shiftEL(int i); @Override public abstract ByteVector shiftER(int i); /** * Bitwise ANDs this vector with an input vector. *

* This is a vector binary operation where the primitive bitwise AND * operation ({@code &}) is applied to lane elements. * * @param v the input vector * @return the bitwise AND of this vector with the input vector */ public abstract ByteVector and(Vector v); /** * Bitwise ANDs this vector with the broadcast of an input scalar. *

* This is a vector binary operation where the primitive bitwise AND * operation ({@code &}) is applied to lane elements. * * @param s the input scalar * @return the bitwise AND of this vector with the broadcast of an input * scalar */ public abstract ByteVector and(byte s); /** * Bitwise ANDs this vector with an input vector, selecting lane elements * controlled by a mask. *

* This is a vector binary operation where the primitive bitwise AND * operation ({@code &}) is applied to lane elements. * * @param v the input vector * @param m the mask controlling lane selection * @return the bitwise AND of this vector with the input vector */ public abstract ByteVector and(Vector v, Mask m); /** * Bitwise ANDs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a vector binary operation where the primitive bitwise AND * operation ({@code &}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the bitwise AND of this vector with the broadcast of an input * scalar */ public abstract ByteVector and(byte s, Mask m); /** * Bitwise ORs this vector with an input vector. *

* This is a vector binary operation where the primitive bitwise OR * operation ({@code |}) is applied to lane elements. * * @param v the input vector * @return the bitwise OR of this vector with the input vector */ public abstract ByteVector or(Vector v); /** * Bitwise ORs this vector with the broadcast of an input scalar. *

* This is a vector binary operation where the primitive bitwise OR * operation ({@code |}) is applied to lane elements. * * @param s the input scalar * @return the bitwise OR of this vector with the broadcast of an input * scalar */ public abstract ByteVector or(byte s); /** * Bitwise ORs this vector with an input vector, selecting lane elements * controlled by a mask. *

* This is a vector binary operation where the primitive bitwise OR * operation ({@code |}) is applied to lane elements. * * @param v the input vector * @param m the mask controlling lane selection * @return the bitwise OR of this vector with the input vector */ public abstract ByteVector or(Vector v, Mask m); /** * Bitwise ORs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a vector binary operation where the primitive bitwise OR * operation ({@code |}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the bitwise OR of this vector with the broadcast of an input * scalar */ public abstract ByteVector or(byte s, Mask m); /** * Bitwise XORs this vector with an input vector. *

* This is a vector binary operation where the primitive bitwise XOR * operation ({@code ^}) is applied to lane elements. * * @param v the input vector * @return the bitwise XOR of this vector with the input vector */ public abstract ByteVector xor(Vector v); /** * Bitwise XORs this vector with the broadcast of an input scalar. *

* This is a vector binary operation where the primitive bitwise XOR * operation ({@code ^}) is applied to lane elements. * * @param s the input scalar * @return the bitwise XOR of this vector with the broadcast of an input * scalar */ public abstract ByteVector xor(byte s); /** * Bitwise XORs this vector with an input vector, selecting lane elements * controlled by a mask. *

* This is a vector binary operation where the primitive bitwise XOR * operation ({@code ^}) is applied to lane elements. * * @param v the input vector * @param m the mask controlling lane selection * @return the bitwise XOR of this vector with the input vector */ public abstract ByteVector xor(Vector v, Mask m); /** * Bitwise XORs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a vector binary operation where the primitive bitwise XOR * operation ({@code ^}) is applied to lane elements. * * @param s the input scalar * @param m the mask controlling lane selection * @return the bitwise XOR of this vector with the broadcast of an input * scalar */ public abstract ByteVector xor(byte s, Mask m); /** * Bitwise NOTs this vector. *

* This is a vector unary operation where the primitive bitwise NOT * operation ({@code ~}) is applied to lane elements. * * @return the bitwise NOT of this vector */ public abstract ByteVector not(); /** * Bitwise NOTs this vector, selecting lane elements controlled by a mask. *

* This is a vector unary operation where the primitive bitwise NOT * operation ({@code ~}) is applied to lane elements. * * @param m the mask controlling lane selection * @return the bitwise NOT of this vector */ public abstract ByteVector not(Mask m); /** * Logically left shifts this vector by the broadcast of an input scalar. *

* This is a vector binary operation where the primitive logical left shift * operation ({@code <<}) is applied to lane elements to left shift the * element by shift value as specified by the input scalar. Only the 3 * lowest-order bits of shift value are used. It is as if the shift value * were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to left shift * @return the result of logically left shifting left this vector by the * broadcast of an input scalar */ public abstract ByteVector shiftL(int s); /** * Logically left shifts this vector by the broadcast of an input scalar, * selecting lane elements controlled by a mask. *

* This is a vector binary operation where the primitive logical left shift * operation ({@code <<}) is applied to lane elements to left shift the * element by shift value as specified by the input scalar. Only the 3 * lowest-order bits of shift value are used. It is as if the shift value * were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to left shift * @param m the mask controlling lane selection * @return the result of logically left shifting left this vector by the * broadcast of an input scalar */ public abstract ByteVector shiftL(int s, Mask m); // logical, or unsigned, shift right /** * Logically right shifts (or unsigned right shifts) this vector by the * broadcast of an input scalar. *

* This is a vector binary operation where the primitive logical right shift * operation ({@code >>>}) is applied to lane elements to logically right shift the * element by shift value as specified by the input scalar. Only the 3 * lowest-order bits of shift value are used. It is as if the shift value * were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to right shift * @return the result of logically right shifting this vector by the * broadcast of an input scalar */ public abstract ByteVector shiftR(int s); /** * Logically right shifts (or unsigned right shifts) this vector by the * broadcast of an input scalar, selecting lane elements controlled by a * mask. *

* This is a vector binary operation where the primitive logical right shift * operation ({@code >>>}) is applied to lane elements to logically right shift the * element by shift value as specified by the input scalar. Only the 3 * lowest-order bits of shift value are used. It is as if the shift value * were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to right shift * @param m the mask controlling lane selection * @return the result of logically right shifting this vector by the * broadcast of an input scalar */ public abstract ByteVector shiftR(int s, Mask m); /** * Arithmetically right shifts (or signed right shifts) this vector by the * broadcast of an input scalar. *

* This is a vector binary operation where the primitive arithmetic right * shift operation ({@code >>}) is applied to lane elements to arithmetically * right shift the element by shift value as specified by the input scalar. * Only the 3 lowest-order bits of shift value are used. It is as if the shift * value were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to right shift * @return the result of arithmetically right shifting this vector by the * broadcast of an input scalar */ public abstract ByteVector aShiftR(int s); /** * Arithmetically right shifts (or signed right shifts) this vector by the * broadcast of an input scalar, selecting lane elements controlled by a * mask. *

* This is a vector binary operation where the primitive arithmetic right * shift operation ({@code >>}) is applied to lane elements to arithmetically * right shift the element by shift value as specified by the input scalar. * Only the 3 lowest-order bits of shift value are used. It is as if the shift * value were subjected to a bitwise logical AND operator ({@code &}) with the mask value 0x7. * The shift distance actually used is therefore always in the range 0 to 7, inclusive. * * @param s the input scalar; the number of the bits to right shift * @param m the mask controlling lane selection * @return the result of arithmetically right shifting this vector by the * broadcast of an input scalar */ public abstract ByteVector aShiftR(int s, Mask m); @Override public abstract void intoByteArray(byte[] a, int ix); @Override public abstract void intoByteArray(byte[] a, int ix, Mask m); @Override public abstract void intoByteBuffer(ByteBuffer bb, int ix); @Override public abstract void intoByteBuffer(ByteBuffer bb, int ix, Mask m); // Type specific horizontal reductions /** * Adds all lane elements of this vector. *

* This is an associative vector reduction operation where the addition * operation ({@code +}) is applied to lane elements, * and the identity value is {@code 0}. * * @return the addition of all the lane elements of this vector */ public abstract byte addAll(); /** * Adds all lane elements of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the addition * operation ({@code +}) is applied to lane elements, * and the identity value is {@code 0}. * * @param m the mask controlling lane selection * @return the addition of the selected lane elements of this vector */ public abstract byte addAll(Mask m); /** * Multiplies all lane elements of this vector. *

* This is an associative vector reduction operation where the * multiplication operation ({@code *}) is applied to lane elements, * and the identity value is {@code 1}. * * @return the multiplication of all the lane elements of this vector */ public abstract byte mulAll(); /** * Multiplies all lane elements of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the * multiplication operation ({@code *}) is applied to lane elements, * and the identity value is {@code 1}. * * @param m the mask controlling lane selection * @return the multiplication of all the lane elements of this vector */ public abstract byte mulAll(Mask m); /** * Returns the minimum lane element of this vector. *

* This is an associative vector reduction operation where the operation * {@code (a, b) -> Math.min(a, b)} is applied to lane elements, * and the identity value is * {@link Byte#MAX_VALUE}. * * @return the minimum lane element of this vector */ public abstract byte minAll(); /** * Returns the minimum lane element of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the operation * {@code (a, b) -> Math.min(a, b)} is applied to lane elements, * and the identity value is * {@link Byte#MAX_VALUE}. * * @param m the mask controlling lane selection * @return the minimum lane element of this vector */ public abstract byte minAll(Mask m); /** * Returns the maximum lane element of this vector. *

* This is an associative vector reduction operation where the operation * {@code (a, b) -> Math.max(a, b)} is applied to lane elements, * and the identity value is * {@link Byte#MIN_VALUE}. * * @return the maximum lane element of this vector */ public abstract byte maxAll(); /** * Returns the maximum lane element of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the operation * {@code (a, b) -> Math.max(a, b)} is applied to lane elements, * and the identity value is * {@link Byte#MIN_VALUE}. * * @param m the mask controlling lane selection * @return the maximum lane element of this vector */ public abstract byte maxAll(Mask m); /** * Logically ORs all lane elements of this vector. *

* This is an associative vector reduction operation where the logical OR * operation ({@code |}) is applied to lane elements, * and the identity value is {@code 0}. * * @return the logical OR all the lane elements of this vector */ public abstract byte orAll(); /** * Logically ORs all lane elements of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the logical OR * operation ({@code |}) is applied to lane elements, * and the identity value is {@code 0}. * * @param m the mask controlling lane selection * @return the logical OR all the lane elements of this vector */ public abstract byte orAll(Mask m); /** * Logically ANDs all lane elements of this vector. *

* This is an associative vector reduction operation where the logical AND * operation ({@code |}) is applied to lane elements, * and the identity value is {@code -1}. * * @return the logical AND all the lane elements of this vector */ public abstract byte andAll(); /** * Logically ANDs all lane elements of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the logical AND * operation ({@code |}) is applied to lane elements, * and the identity value is {@code -1}. * * @param m the mask controlling lane selection * @return the logical AND all the lane elements of this vector */ public abstract byte andAll(Mask m); /** * Logically XORs all lane elements of this vector. *

* This is an associative vector reduction operation where the logical XOR * operation ({@code ^}) is applied to lane elements, * and the identity value is {@code 0}. * * @return the logical XOR all the lane elements of this vector */ public abstract byte xorAll(); /** * Logically XORs all lane elements of this vector, selecting lane elements * controlled by a mask. *

* This is an associative vector reduction operation where the logical XOR * operation ({@code ^}) is applied to lane elements, * and the identity value is {@code 0}. * * @param m the mask controlling lane selection * @return the logical XOR all the lane elements of this vector */ public abstract byte xorAll(Mask m); // Type specific accessors /** * Gets the lane element at lane index {@code i} * * @param i the lane index * @return the lane element at lane index {@code i} * @throws IllegalArgumentException if the index is is out of range * ({@code < 0 || >= length()}) */ public abstract byte get(int i); /** * Replaces the lane element of this vector at lane index {@code i} with * value {@code e}. *

* This is a cross-lane operation and behaves as if it returns the result * of blending this vector with an input vector that is the result of * broadcasting {@code e} and a mask that has only one lane set at lane * index {@code i}. * * @param i the lane index of the lane element to be replaced * @param e the value to be placed * @return the result of replacing the lane element of this vector at lane * index {@code i} with value {@code e}. * @throws IllegalArgumentException if the index is is out of range * ({@code < 0 || >= length()}) */ public abstract ByteVector with(int i, byte e); // Type specific extractors /** * Returns an array containing the lane elements of this vector. *

* This method behaves as if it {@link #intoArray(byte[], int)} stores} * this vector into an allocated array and returns the array as follows: *

{@code
     *   byte[] a = new byte[this.length()];
     *   this.intoArray(a, 0);
     *   return a;
     * }
* * @return an array containing the the lane elements of this vector */ @ForceInline public final byte[] toArray() { byte[] a = new byte[species().length()]; intoArray(a, 0); return a; } /** * Stores this vector into an array starting at offset. *

* For each vector lane, where {@code N} is the vector lane index, * the lane element at index {@code N} is stored into the array at index * {@code i + N}. * * @param a the array * @param i the offset into the array * @throws IndexOutOfBoundsException if {@code i < 0}, or * {@code i > a.length - this.length()} */ public abstract void intoArray(byte[] a, int i); /** * Stores this vector into an array starting at offset and using a mask. *

* For each vector lane, where {@code N} is the vector lane index, * if the mask lane at index {@code N} is set then the lane element at * index {@code N} is stored into the array index {@code i + N}. * * @param a the array * @param i the offset into the array * @param m the mask * @throws IndexOutOfBoundsException if {@code i < 0}, or * for any vector lane index {@code N} where the mask at lane {@code N} * is set {@code i >= a.length - N} */ public abstract void intoArray(byte[] a, int i, Mask m); /** * Stores this vector into an array using indexes obtained from an index * map. *

* For each vector lane, where {@code N} is the vector lane index, the * lane element at index {@code N} is stored into the array at index * {@code i + indexMap[j + N]}. * * @param a the array * @param i the offset into the array, may be negative if relative * indexes in the index map compensate to produce a value within the * array bounds * @param indexMap the index map * @param j the offset into the index map * @throws IndexOutOfBoundsException if {@code j < 0}, or * {@code j > indexMap.length - this.length()}, * or for any vector lane index {@code N} the result of * {@code i + indexMap[j + N]} is {@code < 0} or {@code >= a.length} */ public void intoArray(byte[] a, int i, int[] indexMap, int j) { forEach((n, e) -> a[i + indexMap[j + n]] = e); } /** * Stores this vector into an array using indexes obtained from an index * map and using a mask. *

* For each vector lane, where {@code N} is the vector lane index, * if the mask lane at index {@code N} is set then the lane element at * index {@code N} is stored into the array at index * {@code i + indexMap[j + N]}. * * @param a the array * @param i the offset into the array, may be negative if relative * indexes in the index map compensate to produce a value within the * array bounds * @param m the mask * @param indexMap the index map * @param j the offset into the index map * @throws IndexOutOfBoundsException if {@code j < 0}, or * {@code j > indexMap.length - this.length()}, * or for any vector lane index {@code N} where the mask at lane * {@code N} is set the result of {@code i + indexMap[j + N]} is * {@code < 0} or {@code >= a.length} */ public void intoArray(byte[] a, int i, Mask m, int[] indexMap, int j) { forEach(m, (n, e) -> a[i + indexMap[j + n]] = e); } // Species @Override public abstract ByteSpecies species(); /** * Class representing {@link ByteVector}'s of the same {@link Vector.Shape Shape}. */ public static abstract class ByteSpecies extends Vector.Species { interface FOp { byte apply(int i); } abstract ByteVector op(FOp f); abstract ByteVector op(Mask m, FOp f); interface FOpm { boolean apply(int i); } abstract Mask opm(FOpm f); // Factories @Override public abstract ByteVector zero(); /** * Returns a vector where all lane elements are set to the primitive * value {@code e}. * * @param e the value * @return a vector of vector where all lane elements are set to * the primitive value {@code e} */ public abstract ByteVector broadcast(byte e); /** * Returns a vector where the first lane element is set to the primtive * value {@code e}, all other lane elements are set to the default * value. * * @param e the value * @return a vector where the first lane element is set to the primitive * value {@code e} */ @ForceInline public final ByteVector single(byte e) { return zero().with(0, e); } /** * Returns a vector where each lane element is set to a randomly * generated primitive value. * * The semantics are equivalent to calling * {@code (byte)ThreadLocalRandom#nextInt()}. * * @return a vector where each lane elements is set to a randomly * generated primitive value */ public ByteVector random() { ThreadLocalRandom r = ThreadLocalRandom.current(); return op(i -> (byte) r.nextInt()); } /** * Returns a vector where each lane element is set to a given * primitive value. *

* For each vector lane, where {@code N} is the vector lane index, the * the primitive value at index {@code N} is placed into the resulting * vector at lane index {@code N}. * * @param es the given primitive values * @return a vector where each lane element is set to a given primitive * value * @throws IndexOutOfBoundsException if {@code es.length < this.length()} */ public abstract ByteVector scalars(byte... es); } /** * Finds the preferred species for an element type of {@code byte}. *

* A preferred species is a species chosen by the platform that has a * shape of maximal bit size. A preferred species for different element * types will have the same shape, and therefore vectors, masks, and * shuffles created from such species will be shape compatible. * * @return the preferred species for an element type of {@code byte} */ @SuppressWarnings("unchecked") public static ByteSpecies preferredSpecies() { return (ByteSpecies) Species.ofPreferred(byte.class); } /** * Finds a species for an element type of {@code byte} and shape. * * @param s the shape * @return a species for an element type of {@code byte} and shape * @throws IllegalArgumentException if no such species exists for the shape */ @SuppressWarnings("unchecked") public static ByteSpecies species(Vector.Shape s) { Objects.requireNonNull(s); switch (s) { case S_64_BIT: return Byte64Vector.SPECIES; case S_128_BIT: return Byte128Vector.SPECIES; case S_256_BIT: return Byte256Vector.SPECIES; case S_512_BIT: return Byte512Vector.SPECIES; case S_Max_BIT: return ByteMaxVector.SPECIES; default: throw new IllegalArgumentException("Bad shape: " + s); } } }