/* * 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.ShortBuffer; import java.nio.ByteOrder; import java.util.Objects; import java.util.function.IntUnaryOperator; import java.util.function.Function; 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 short} values. */ @SuppressWarnings("cast") public abstract class ShortVector extends Vector { ShortVector() {} private static final int ARRAY_SHIFT = 31 - Integer.numberOfLeadingZeros(Unsafe.ARRAY_SHORT_INDEX_SCALE); // Unary operator interface FUnOp { short apply(int i, short a); } abstract ShortVector uOp(FUnOp f); abstract ShortVector uOp(VectorMask m, FUnOp f); // Binary operator interface FBinOp { short apply(int i, short a, short b); } abstract ShortVector bOp(Vector v, FBinOp f); abstract ShortVector bOp(Vector v, VectorMask m, FBinOp f); // Trinary operator interface FTriOp { short apply(int i, short a, short b, short c); } abstract ShortVector tOp(Vector v1, Vector v2, FTriOp f); abstract ShortVector tOp(Vector v1, Vector v2, VectorMask m, FTriOp f); // Reduction operator abstract short rOp(short v, FBinOp f); // Binary test interface FBinTest { boolean apply(int i, short a, short b); } abstract VectorMask bTest(Vector v, FBinTest f); // Foreach interface FUnCon { void apply(int i, short a); } abstract void forEach(FUnCon f); abstract void forEach(VectorMask 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 ShortVector zero(VectorSpecies species) { return VectorIntrinsics.broadcastCoerced((Class) species.boxType(), short.class, species.length(), 0, species, ((bits, s) -> ((ShortSpecies)s).op(i -> (short)bits))); } /** * 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(VectorSpecies, ByteBuffer, int, VectorMask) method} as follows: *

{@code
     * return fromByteBuffer(species, ByteBuffer.wrap(a), offset, VectorMask.allTrue());
     * }
* * @param species species of desired vector * @param a the byte array * @param offset the offset into the array * @return a vector loaded from a byte array * @throws IndexOutOfBoundsException if {@code i < 0} or * {@code offset > a.length - (species.length() * species.elementSize() / Byte.SIZE)} */ @ForceInline @SuppressWarnings("unchecked") public static ShortVector fromByteArray(VectorSpecies species, byte[] a, int offset) { Objects.requireNonNull(a); offset = VectorIntrinsics.checkIndex(offset, a.length, species.bitSize() / Byte.SIZE); return VectorIntrinsics.load((Class) species.boxType(), short.class, species.length(), a, ((long) offset) + Unsafe.ARRAY_BYTE_BASE_OFFSET, a, offset, species, (c, idx, s) -> { ByteBuffer bbc = ByteBuffer.wrap(c, idx, a.length - idx).order(ByteOrder.nativeOrder()); ShortBuffer tb = bbc.asShortBuffer(); return ((ShortSpecies)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(VectorSpecies, ByteBuffer, int, VectorMask) method} as follows: *

{@code
     * return fromByteBuffer(species, ByteBuffer.wrap(a), offset, m);
     * }
* * @param species species of desired vector * @param a the byte array * @param offset the offset into the array * @param m the mask * @return a vector loaded from a byte array * @throws IndexOutOfBoundsException if {@code offset < 0} or * for any vector lane index {@code N} where the mask at lane {@code N} * is set * {@code offset >= a.length - (N * species.elementSize() / Byte.SIZE)} */ @ForceInline public static ShortVector fromByteArray(VectorSpecies species, byte[] a, int offset, VectorMask m) { return zero(species).blend(fromByteArray(species, a, offset), 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 offset + N} is placed into the * resulting vector at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param offset the offset into the array * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code offset < 0}, or * {@code offset > a.length - species.length()} */ @ForceInline @SuppressWarnings("unchecked") public static ShortVector fromArray(VectorSpecies species, short[] a, int offset){ Objects.requireNonNull(a); offset = VectorIntrinsics.checkIndex(offset, a.length, species.length()); return VectorIntrinsics.load((Class) species.boxType(), short.class, species.length(), a, (((long) offset) << ARRAY_SHIFT) + Unsafe.ARRAY_SHORT_BASE_OFFSET, a, offset, species, (c, idx, s) -> ((ShortSpecies)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 offset + 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 offset the offset into the array * @param m the mask * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code offset < 0}, or * for any vector lane index {@code N} where the mask at lane {@code N} * is set {@code offset > a.length - N} */ @ForceInline public static ShortVector fromArray(VectorSpecies species, short[] a, int offset, VectorMask m) { return zero(species).blend(fromArray(species, a, offset), m); } /** * 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 a_offset + indexMap[i_offset + N]} is placed into the * resulting vector at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param a_offset 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 i_offset the offset into the index map * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code i_offset < 0}, or * {@code i_offset > indexMap.length - species.length()}, * or for any vector lane index {@code N} the result of * {@code a_offset + indexMap[i_offset + N]} is {@code < 0} or {@code >= a.length} */ public static ShortVector fromArray(VectorSpecies species, short[] a, int a_offset, int[] indexMap, int i_offset) { return ((ShortSpecies)species).op(n -> a[a_offset + indexMap[i_offset + 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 a_offset + indexMap[i_offset + N]} is placed into the resulting vector * at lane index {@code N}. * * @param species species of desired vector * @param a the array * @param a_offset 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 i_offset the offset into the index map * @return the vector loaded from an array * @throws IndexOutOfBoundsException if {@code i_offset < 0}, or * {@code i_offset > indexMap.length - species.length()}, * or for any vector lane index {@code N} where the mask at lane * {@code N} is set the result of {@code a_offset + indexMap[i_offset + N]} is * {@code < 0} or {@code >= a.length} */ public static ShortVector fromArray(VectorSpecies species, short[] a, int a_offset, VectorMask m, int[] indexMap, int i_offset) { return ((ShortSpecies)species).op(m, n -> a[a_offset + indexMap[i_offset + 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(VectorSpecies, ByteBuffer, int, VectorMask)} method} as follows: *

{@code
     *   return fromByteBuffer(b, offset, VectorMask.allTrue())
     * }
* * @param species species of desired vector * @param bb the byte buffer * @param offset 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 species.length() * species.elementSize() / Byte.SIZE} bytes * remaining in the byte buffer from the given offset */ @ForceInline @SuppressWarnings("unchecked") public static ShortVector fromByteBuffer(VectorSpecies species, ByteBuffer bb, int offset) { if (bb.order() != ByteOrder.nativeOrder()) { throw new IllegalArgumentException(); } offset = VectorIntrinsics.checkIndex(offset, bb.limit(), species.bitSize() / Byte.SIZE); return VectorIntrinsics.load((Class) species.boxType(), short.class, species.length(), U.getReference(bb, BYTE_BUFFER_HB), U.getLong(bb, BUFFER_ADDRESS) + offset, bb, offset, species, (c, idx, s) -> { ByteBuffer bbc = c.duplicate().position(idx).order(ByteOrder.nativeOrder()); ShortBuffer tb = bbc.asShortBuffer(); return ((ShortSpecies)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 * {@code 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(offset).
     *     asEBuffer();
     * e[] es = new e[species.length()];
     * for (int n = 0; n < t.length; n++) {
     *     if (m.isSet(n))
     *         es[n] = eb.get(n);
     * }
     * EVector r = EVector.fromArray(es, 0, m);
     * }
* * @param species species of desired vector * @param bb the byte buffer * @param offset 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 offset >= b.limit() - (N * species.elementSize() / Byte.SIZE)} */ @ForceInline public static ShortVector fromByteBuffer(VectorSpecies species, ByteBuffer bb, int offset, VectorMask m) { return zero(species).blend(fromByteBuffer(species, bb, offset), m); } /** * Returns a vector where all lane elements are set to the primitive * value {@code e}. * * @param species species of the desired vector * @param e the value * @return a vector of vector where all lane elements are set to * the primitive value {@code e} */ @ForceInline @SuppressWarnings("unchecked") public static ShortVector broadcast(VectorSpecies species, short e) { return VectorIntrinsics.broadcastCoerced( (Class) species.boxType(), short.class, species.length(), e, species, ((bits, sp) -> ((ShortSpecies)sp).op(i -> (short)bits))); } /** * Returns a vector where each lane element is set to given * primitive values. *

* 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 species species of the desired vector * @param es the given primitive values * @return a vector where each lane element is set to given primitive * values * @throws IndexOutOfBoundsException if {@code es.length < species.length()} */ @ForceInline @SuppressWarnings("unchecked") public static ShortVector scalars(VectorSpecies species, short... es) { Objects.requireNonNull(es); int ix = VectorIntrinsics.checkIndex(0, es.length, species.length()); return VectorIntrinsics.load((Class) species.boxType(), short.class, species.length(), es, Unsafe.ARRAY_SHORT_BASE_OFFSET, es, ix, species, (c, idx, sp) -> ((ShortSpecies)sp).op(n -> c[idx + n])); } /** * 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 species species of the desired vector * @param e the value * @return a vector where the first lane element is set to the primitive * value {@code e} */ @ForceInline public static final ShortVector single(VectorSpecies species, short e) { return zero(species).with(0, e); } /** * Returns a vector where each lane element is set to a randomly * generated primitive value. * * The semantics are equivalent to calling * (short){@link ThreadLocalRandom#nextInt()} * * @param species species of the desired vector * @return a vector where each lane elements is set to a randomly * generated primitive value */ public static ShortVector random(VectorSpecies species) { ThreadLocalRandom r = ThreadLocalRandom.current(); return ((ShortSpecies)species).op(i -> (short) r.nextInt()); } // Ops @Override public abstract ShortVector add(Vector v); /** * Adds this vector to the broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the primitive addition operation * ({@code +}) to each lane. * * @param s the input scalar * @return the result of adding this vector to the broadcast of an input * scalar */ public abstract ShortVector add(short s); @Override public abstract ShortVector add(Vector v, VectorMask m); /** * Adds this vector to broadcast of an input scalar, * selecting lane elements controlled by a mask. *

* This is a lane-wise binary operation which applies the primitive addition operation * ({@code +}) to each lane. * * @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 ShortVector add(short s, VectorMask m); @Override public abstract ShortVector sub(Vector v); /** * Subtracts the broadcast of an input scalar from this vector. *

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

* This is a lane-wise binary operation which applies the primitive subtraction * operation ({@code -}) to each lane. * * @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 ShortVector sub(short s, VectorMask m); @Override public abstract ShortVector mul(Vector v); /** * Multiplies this vector with the broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the primitive multiplication * operation ({@code *}) to each lane. * * @param s the input scalar * @return the result of multiplying this vector with the broadcast of an * input scalar */ public abstract ShortVector mul(short s); @Override public abstract ShortVector mul(Vector v, VectorMask m); /** * Multiplies this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a lane-wise binary operation which applies the primitive multiplication * operation ({@code *}) to each lane. * * @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 ShortVector mul(short s, VectorMask m); @Override public abstract ShortVector neg(); @Override public abstract ShortVector neg(VectorMask m); @Override public abstract ShortVector abs(); @Override public abstract ShortVector abs(VectorMask m); @Override public abstract ShortVector min(Vector v); @Override public abstract ShortVector min(Vector v, VectorMask m); /** * Returns the minimum of this vector and the broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the operation * {@code (a, b) -> Math.min(a, b)} to each lane. * * @param s the input scalar * @return the minimum of this vector and the broadcast of an input scalar */ public abstract ShortVector min(short s); @Override public abstract ShortVector max(Vector v); @Override public abstract ShortVector max(Vector v, VectorMask m); /** * Returns the maximum of this vector and the broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the operation * {@code (a, b) -> Math.max(a, b)} to each lane. * * @param s the input scalar * @return the maximum of this vector and the broadcast of an input scalar */ public abstract ShortVector max(short s); @Override public abstract VectorMask equal(Vector v); /** * Tests if this vector is equal to the broadcast of an input scalar. *

* This is a lane-wise binary test operation which applies the primitive equals * operation ({@code ==}) each lane. * * @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 VectorMask equal(short s); @Override public abstract VectorMask notEqual(Vector v); /** * Tests if this vector is not equal to the broadcast of an input scalar. *

* This is a lane-wise binary test operation which applies the primitive not equals * operation ({@code !=}) to each lane. * * @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 VectorMask notEqual(short s); @Override public abstract VectorMask lessThan(Vector v); /** * Tests if this vector is less than the broadcast of an input scalar. *

* This is a lane-wise binary test operation which applies the primitive less than * operation ({@code <}) to each lane. * * @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 VectorMask lessThan(short s); @Override public abstract VectorMask lessThanEq(Vector v); /** * Tests if this vector is less or equal to the broadcast of an input scalar. *

* This is a lane-wise binary test operation which applies the primitive less than * or equal to operation ({@code <=}) to each lane. * * @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 VectorMask lessThanEq(short s); @Override public abstract VectorMask greaterThan(Vector v); /** * Tests if this vector is greater than the broadcast of an input scalar. *

* This is a lane-wise binary test operation which applies the primitive greater than * operation ({@code >}) to each lane. * * @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 VectorMask greaterThan(short s); @Override public abstract VectorMask greaterThanEq(Vector v); /** * Tests if this vector is greater than or equal to the broadcast of an * input scalar. *

* This is a lane-wise binary test operation which applies the primitive greater than * or equal to operation ({@code >=}) to each lane. * * @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 VectorMask greaterThanEq(short s); @Override public abstract ShortVector blend(Vector v, VectorMask 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 ShortVector blend(short s, VectorMask m); @Override public abstract ShortVector rearrange(Vector v, VectorShuffle s, VectorMask m); @Override public abstract ShortVector rearrange(VectorShuffle m); @Override public abstract ShortVector reshape(VectorSpecies s); @Override public abstract ShortVector rotateEL(int i); @Override public abstract ShortVector rotateER(int i); @Override public abstract ShortVector shiftEL(int i); @Override public abstract ShortVector shiftER(int i); /** * Bitwise ANDs this vector with an input vector. *

* This is a lane-wise binary operation which applies the primitive bitwise AND * operation ({@code &}) to each lane. * * @param v the input vector * @return the bitwise AND of this vector with the input vector */ public abstract ShortVector and(Vector v); /** * Bitwise ANDs this vector with the broadcast of an input scalar. *

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

* This is a lane-wise binary operation which applies the primitive bitwise AND * operation ({@code &}) to each lane. * * @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 ShortVector and(Vector v, VectorMask m); /** * Bitwise ANDs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a lane-wise binary operation which applies the primitive bitwise AND * operation ({@code &}) to each lane. * * @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 ShortVector and(short s, VectorMask m); /** * Bitwise ORs this vector with an input vector. *

* This is a lane-wise binary operation which applies the primitive bitwise OR * operation ({@code |}) to each lane. * * @param v the input vector * @return the bitwise OR of this vector with the input vector */ public abstract ShortVector or(Vector v); /** * Bitwise ORs this vector with the broadcast of an input scalar. *

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

* This is a lane-wise binary operation which applies the primitive bitwise OR * operation ({@code |}) to each lane. * * @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 ShortVector or(Vector v, VectorMask m); /** * Bitwise ORs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a lane-wise binary operation which applies the primitive bitwise OR * operation ({@code |}) to each lane. * * @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 ShortVector or(short s, VectorMask m); /** * Bitwise XORs this vector with an input vector. *

* This is a lane-wise binary operation which applies the primitive bitwise XOR * operation ({@code ^}) to each lane. * * @param v the input vector * @return the bitwise XOR of this vector with the input vector */ public abstract ShortVector xor(Vector v); /** * Bitwise XORs this vector with the broadcast of an input scalar. *

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

* This is a lane-wise binary operation which applies the primitive bitwise XOR * operation ({@code ^}) to each lane. * * @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 ShortVector xor(Vector v, VectorMask m); /** * Bitwise XORs this vector with the broadcast of an input scalar, selecting * lane elements controlled by a mask. *

* This is a lane-wise binary operation which applies the primitive bitwise XOR * operation ({@code ^}) to each lane. * * @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 ShortVector xor(short s, VectorMask m); /** * Bitwise NOTs this vector. *

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

* This is a lane-wise unary operation which applies the primitive bitwise NOT * operation ({@code ~}) to each lane. * * @param m the mask controlling lane selection * @return the bitwise NOT of this vector */ public abstract ShortVector not(VectorMask m); /** * Logically left shifts this vector by the broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the primitive logical left shift * operation ({@code <<}) to each lane to left shift the * element by shift value as specified by the input scalar. Only the 4 * 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector 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 lane-wise binary operation which applies the primitive logical left shift * operation ({@code <<}) to each lane to left shift the * element by shift value as specified by the input scalar. Only the 4 * 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector shiftL(int s, VectorMask 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 lane-wise binary operation which applies the primitive logical right shift * operation ({@code >>>}) to each lane to logically right shift the * element by shift value as specified by the input scalar. Only the 4 * 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector 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 lane-wise binary operation which applies the primitive logical right shift * operation ({@code >>>}) to each lane to logically right shift the * element by shift value as specified by the input scalar. Only the 4 * 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector shiftR(int s, VectorMask m); /** * Arithmetically right shifts (or signed right shifts) this vector by the * broadcast of an input scalar. *

* This is a lane-wise binary operation which applies the primitive arithmetic right * shift operation ({@code >>}) to each lane to arithmetically * right shift the element by shift value as specified by the input scalar. * Only the 4 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector 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 lane-wise binary operation which applies the primitive arithmetic right * shift operation ({@code >>}) to each lane to arithmetically * right shift the element by shift value as specified by the input scalar. * Only the 4 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 0xF. * The shift distance actually used is therefore always in the range 0 to 15, 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 ShortVector aShiftR(int s, VectorMask m); @Override public abstract void intoByteArray(byte[] a, int ix); @Override public abstract void intoByteArray(byte[] a, int ix, VectorMask m); @Override public abstract void intoByteBuffer(ByteBuffer bb, int ix); @Override public abstract void intoByteBuffer(ByteBuffer bb, int ix, VectorMask m); // Type specific horizontal reductions /** * Adds all lane elements of this vector. *

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

* This is an associative cross-lane reduction operation which applies the addition * operation ({@code +}) 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 short addAll(VectorMask m); /** * Multiplies all lane elements of this vector. *

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

* This is an associative cross-lane reduction operation which applies the * multiplication operation ({@code *}) 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 short mulAll(VectorMask m); /** * Returns the minimum lane element of this vector. *

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

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

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

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

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

* This is an associative cross-lane reduction operation which applies the logical OR * operation ({@code |}) 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 short orAll(VectorMask m); /** * Logically ANDs all lane elements of this vector. *

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

* This is an associative cross-lane reduction operation which applies the logical AND * operation ({@code |}) 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 short andAll(VectorMask m); /** * Logically XORs all lane elements of this vector. *

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

* This is an associative cross-lane reduction operation which applies the logical XOR * operation ({@code ^}) 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 short xorAll(VectorMask 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 short lane(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 ShortVector with(int i, short e); // Type specific extractors /** * Returns an array containing the lane elements of this vector. *

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

{@code
     *   short[] a = new short[this.length()];
     *   this.intoArray(a, 0);
     *   return a;
     * }
* * @return an array containing the the lane elements of this vector */ @ForceInline public final short[] toArray() { short[] a = new short[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 offset + N}. * * @param a the array * @param offset the offset into the array * @throws IndexOutOfBoundsException if {@code offset < 0}, or * {@code offset > a.length - this.length()} */ public abstract void intoArray(short[] a, int offset); /** * 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 offset + N}. * * @param a the array * @param offset the offset into the array * @param m the mask * @throws IndexOutOfBoundsException if {@code offset < 0}, or * for any vector lane index {@code N} where the mask at lane {@code N} * is set {@code offset >= a.length - N} */ public abstract void intoArray(short[] a, int offset, VectorMask 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 a_offset + indexMap[i_offset + N]}. * * @param a the array * @param a_offset 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 i_offset the offset into the index map * @throws IndexOutOfBoundsException if {@code i_offset < 0}, or * {@code i_offset > indexMap.length - this.length()}, * or for any vector lane index {@code N} the result of * {@code a_offset + indexMap[i_offset + N]} is {@code < 0} or {@code >= a.length} */ public void intoArray(short[] a, int a_offset, int[] indexMap, int i_offset) { forEach((n, e) -> a[a_offset + indexMap[i_offset + 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 a_offset + indexMap[i_offset + N]}. * * @param a the array * @param a_offset 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 i_offset the offset into the index map * @throws IndexOutOfBoundsException if {@code j < 0}, or * {@code i_offset > 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 a_offset + indexMap[i_offset + N]} is * {@code < 0} or {@code >= a.length} */ public void intoArray(short[] a, int a_offset, VectorMask m, int[] indexMap, int i_offset) { forEach(m, (n, e) -> a[a_offset + indexMap[i_offset + n]] = e); } // Species @Override public abstract VectorSpecies species(); /** * Class representing {@link ShortVector}'s of the same {@link VectorShape VectorShape}. */ static final class ShortSpecies extends AbstractSpecies { final Function vectorFactory; private ShortSpecies(VectorShape shape, Class boxType, Class maskType, Function vectorFactory, Function> maskFactory, Function> shuffleFromArrayFactory, fShuffleFromArray shuffleFromOpFactory) { super(shape, short.class, Short.SIZE, boxType, maskType, maskFactory, shuffleFromArrayFactory, shuffleFromOpFactory); this.vectorFactory = vectorFactory; } interface FOp { short apply(int i); } ShortVector op(FOp f) { short[] res = new short[length()]; for (int i = 0; i < length(); i++) { res[i] = f.apply(i); } return vectorFactory.apply(res); } ShortVector op(VectorMask o, FOp f) { short[] res = new short[length()]; boolean[] mbits = ((AbstractMask)o).getBits(); for (int i = 0; i < length(); i++) { if (mbits[i]) { res[i] = f.apply(i); } } return vectorFactory.apply(res); } } /** * Finds the preferred species for an element type of {@code short}. *

* 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 short} */ private static ShortSpecies preferredSpecies() { return (ShortSpecies) VectorSpecies.ofPreferred(short.class); } /** * Finds a species for an element type of {@code short} and shape. * * @param s the shape * @return a species for an element type of {@code short} and shape * @throws IllegalArgumentException if no such species exists for the shape */ static ShortSpecies species(VectorShape s) { Objects.requireNonNull(s); switch (s) { case S_64_BIT: return (ShortSpecies) SPECIES_64; case S_128_BIT: return (ShortSpecies) SPECIES_128; case S_256_BIT: return (ShortSpecies) SPECIES_256; case S_512_BIT: return (ShortSpecies) SPECIES_512; case S_Max_BIT: return (ShortSpecies) SPECIES_MAX; default: throw new IllegalArgumentException("Bad shape: " + s); } } /** Species representing {@link ShortVector}s of {@link VectorShape#S_64_BIT VectorShape.S_64_BIT}. */ public static final VectorSpecies SPECIES_64 = new ShortSpecies(VectorShape.S_64_BIT, Short64Vector.class, Short64Vector.Short64Mask.class, Short64Vector::new, Short64Vector.Short64Mask::new, Short64Vector.Short64Shuffle::new, Short64Vector.Short64Shuffle::new); /** Species representing {@link ShortVector}s of {@link VectorShape#S_128_BIT VectorShape.S_128_BIT}. */ public static final VectorSpecies SPECIES_128 = new ShortSpecies(VectorShape.S_128_BIT, Short128Vector.class, Short128Vector.Short128Mask.class, Short128Vector::new, Short128Vector.Short128Mask::new, Short128Vector.Short128Shuffle::new, Short128Vector.Short128Shuffle::new); /** Species representing {@link ShortVector}s of {@link VectorShape#S_256_BIT VectorShape.S_256_BIT}. */ public static final VectorSpecies SPECIES_256 = new ShortSpecies(VectorShape.S_256_BIT, Short256Vector.class, Short256Vector.Short256Mask.class, Short256Vector::new, Short256Vector.Short256Mask::new, Short256Vector.Short256Shuffle::new, Short256Vector.Short256Shuffle::new); /** Species representing {@link ShortVector}s of {@link VectorShape#S_512_BIT VectorShape.S_512_BIT}. */ public static final VectorSpecies SPECIES_512 = new ShortSpecies(VectorShape.S_512_BIT, Short512Vector.class, Short512Vector.Short512Mask.class, Short512Vector::new, Short512Vector.Short512Mask::new, Short512Vector.Short512Shuffle::new, Short512Vector.Short512Shuffle::new); /** Species representing {@link ShortVector}s of {@link VectorShape#S_Max_BIT VectorShape.S_Max_BIT}. */ public static final VectorSpecies SPECIES_MAX = new ShortSpecies(VectorShape.S_Max_BIT, ShortMaxVector.class, ShortMaxVector.ShortMaxMask.class, ShortMaxVector::new, ShortMaxVector.ShortMaxMask::new, ShortMaxVector.ShortMaxShuffle::new, ShortMaxVector.ShortMaxShuffle::new); /** * Preferred species for {@link ShortVector}s. * A preferred species is a species of maximal bit size for the platform. */ public static final VectorSpecies SPECIES_PREFERRED = (VectorSpecies) preferredSpecies(); }