1 /* 2 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have 23 * questions. 24 */ 25 package jdk.incubator.vector; 26 27 /** 28 * Operations on vectors that are not intrinsic candidates 29 */ 30 @SuppressWarnings("cast") 31 public final class ShortVectorHelper { 32 33 private ShortVectorHelper() {} 34 35 public interface BinaryOp { 36 short apply(int i, short a, short b); 37 } 38 39 public interface UnaryOp { 40 short apply(int i, short a); 41 } 42 43 public static 44 ShortVector map(ShortVector va, ShortVector vb, BinaryOp op) { 45 return va.bOp(vb, (i, a, b) -> op.apply(i, a, b)); 46 } 47 48 public static 49 ShortVector map(ShortVector va, ShortVector vb, Vector.Mask<Short> m, BinaryOp op) { 50 return va.bOp(vb, m, (i, a, b) -> op.apply(i, a, b)); 51 } 52 53 public static 54 ShortVector map(ShortVector va, UnaryOp op) { 55 return va.uOp((i, a) -> op.apply(i, a)); 56 } 57 58 public static 59 ShortVector map(ShortVector va, Vector.Mask<Short> m, UnaryOp op) { 60 return va.uOp(m, (i, a) -> op.apply(i, a)); 61 } 62 63 public static 64 ShortVector div(ShortVector va, ShortVector vb) { 65 return va.bOp(vb, (i, a, b) -> (short) (a / b)); 66 } 67 68 public static 69 ShortVector div(ShortVector va, ShortVector vb, Vector.Mask<Short> m) { 70 return va.bOp(vb, m, (i, a, b) -> (short) (a / b)); 71 } 72 73 public static 74 ShortVector mod(ShortVector va, ShortVector vb) { 75 return va.bOp(vb, (i, a, b) -> (short) (a % b)); 76 } 77 78 public static 79 ShortVector mod(ShortVector va, ShortVector vb, Vector.Mask<Short> m) { 80 return va.bOp(vb, m, (i, a, b) -> (short) (a % b)); 81 } 82 83 84 85 }