Home | History | Annotate | Download | only in renderscript
      1 /*
      2  * Copyright (C) 2013 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package android.renderscript;
     18 
     19 /**
     20  * Vector version of the basic short type.
     21  * Provides four short fields packed.
     22  */
     23 public class Short4 {
     24     public short x;
     25     public short y;
     26     public short z;
     27     public short w;
     28 
     29     public Short4() {
     30     }
     31 
     32     /** @hide */
     33     public Short4(short i) {
     34         this.x = this.y = this.z = this.w = i;
     35     }
     36 
     37     public Short4(short x, short y, short z, short w) {
     38         this.x = x;
     39         this.y = y;
     40         this.z = z;
     41         this.w = w;
     42     }
     43 
     44     /** @hide */
     45     public Short4(Short4 source) {
     46         this.x = source.x;
     47         this.y = source.y;
     48         this.z = source.z;
     49         this.w = source.w;
     50     }
     51 
     52     /** @hide
     53      * Vector add
     54      *
     55      * @param a
     56      */
     57     public void add(Short4 a) {
     58         this.x += a.x;
     59         this.y += a.y;
     60         this.z += a.z;
     61         this.w += a.w;
     62     }
     63 
     64     /** @hide
     65      * Vector add
     66      *
     67      * @param a
     68      * @param b
     69      * @return
     70      */
     71     public static Short4 add(Short4 a, Short4 b) {
     72         Short4 result = new Short4();
     73         result.x = (short)(a.x + b.x);
     74         result.y = (short)(a.y + b.y);
     75         result.z = (short)(a.z + b.z);
     76         result.w = (short)(a.w + b.w);
     77 
     78         return result;
     79     }
     80 
     81     /** @hide
     82      * Vector add
     83      *
     84      * @param value
     85      */
     86     public void add(short value) {
     87         x += value;
     88         y += value;
     89         z += value;
     90         w += value;
     91     }
     92 
     93     /** @hide
     94      * Vector add
     95      *
     96      * @param a
     97      * @param b
     98      * @return
     99      */
    100     public static Short4 add(Short4 a, short b) {
    101         Short4 result = new Short4();
    102         result.x = (short)(a.x + b);
    103         result.y = (short)(a.y + b);
    104         result.z = (short)(a.z + b);
    105         result.w = (short)(a.w + b);
    106 
    107         return result;
    108     }
    109 
    110     /** @hide
    111      * Vector subtraction
    112      *
    113      * @param a
    114      */
    115     public void sub(Short4 a) {
    116         this.x -= a.x;
    117         this.y -= a.y;
    118         this.z -= a.z;
    119         this.w -= a.w;
    120     }
    121 
    122     /** @hide
    123      * Vector subtraction
    124      *
    125      * @param a
    126      * @param b
    127      * @return
    128      */
    129     public static Short4 sub(Short4 a, Short4 b) {
    130         Short4 result = new Short4();
    131         result.x = (short)(a.x - b.x);
    132         result.y = (short)(a.y - b.y);
    133         result.z = (short)(a.z - b.z);
    134         result.w = (short)(a.w - b.w);
    135 
    136         return result;
    137     }
    138 
    139     /** @hide
    140      * Vector subtraction
    141      *
    142      * @param value
    143      */
    144     public void sub(short value) {
    145         x -= value;
    146         y -= value;
    147         z -= value;
    148         w -= value;
    149     }
    150 
    151     /** @hide
    152      * Vector subtraction
    153      *
    154      * @param a
    155      * @param b
    156      * @return
    157      */
    158     public static Short4 sub(Short4 a, short b) {
    159         Short4 result = new Short4();
    160         result.x = (short)(a.x - b);
    161         result.y = (short)(a.y - b);
    162         result.z = (short)(a.z - b);
    163         result.w = (short)(a.w - b);
    164 
    165         return result;
    166     }
    167 
    168     /** @hide
    169      * Vector multiplication
    170      *
    171      * @param a
    172      */
    173     public void mul(Short4 a) {
    174         this.x *= a.x;
    175         this.y *= a.y;
    176         this.z *= a.z;
    177         this.w *= a.w;
    178     }
    179 
    180     /** @hide
    181      * Vector multiplication
    182      *
    183      * @param a
    184      * @param b
    185      * @return
    186      */
    187     public static Short4 mul(Short4 a, Short4 b) {
    188         Short4 result = new Short4();
    189         result.x = (short)(a.x * b.x);
    190         result.y = (short)(a.y * b.y);
    191         result.z = (short)(a.z * b.z);
    192         result.w = (short)(a.w * b.w);
    193 
    194         return result;
    195     }
    196 
    197     /** @hide
    198      * Vector multiplication
    199      *
    200      * @param value
    201      */
    202     public void mul(short value) {
    203         x *= value;
    204         y *= value;
    205         z *= value;
    206         w *= value;
    207     }
    208 
    209     /** @hide
    210      * Vector multiplication
    211      *
    212      * @param a
    213      * @param b
    214      * @return
    215      */
    216     public static Short4 mul(Short4 a, short b) {
    217         Short4 result = new Short4();
    218         result.x = (short)(a.x * b);
    219         result.y = (short)(a.y * b);
    220         result.z = (short)(a.z * b);
    221         result.w = (short)(a.w * b);
    222 
    223         return result;
    224     }
    225 
    226     /** @hide
    227      * Vector division
    228      *
    229      * @param a
    230      */
    231     public void div(Short4 a) {
    232         this.x /= a.x;
    233         this.y /= a.y;
    234         this.z /= a.z;
    235         this.w /= a.w;
    236     }
    237 
    238     /** @hide
    239      * Vector division
    240      *
    241      * @param a
    242      * @param b
    243      * @return
    244      */
    245     public static Short4 div(Short4 a, Short4 b) {
    246         Short4 result = new Short4();
    247         result.x = (short)(a.x / b.x);
    248         result.y = (short)(a.y / b.y);
    249         result.z = (short)(a.z / b.z);
    250         result.w = (short)(a.w / b.w);
    251 
    252         return result;
    253     }
    254 
    255     /** @hide
    256      * Vector division
    257      *
    258      * @param value
    259      */
    260     public void div(short value) {
    261         x /= value;
    262         y /= value;
    263         z /= value;
    264         w /= value;
    265     }
    266 
    267     /** @hide
    268      * Vector division
    269      *
    270      * @param a
    271      * @param b
    272      * @return
    273      */
    274     public static Short4 div(Short4 a, short b) {
    275         Short4 result = new Short4();
    276         result.x = (short)(a.x / b);
    277         result.y = (short)(a.y / b);
    278         result.z = (short)(a.z / b);
    279         result.w = (short)(a.w / b);
    280 
    281         return result;
    282     }
    283 
    284     /** @hide
    285      * Vector Modulo
    286      *
    287      * @param a
    288      */
    289     public void mod(Short4 a) {
    290         this.x %= a.x;
    291         this.y %= a.y;
    292         this.z %= a.z;
    293         this.w %= a.w;
    294     }
    295 
    296     /** @hide
    297      * Vector Modulo
    298      *
    299      * @param a
    300      * @param b
    301      * @return
    302      */
    303     public static Short4 mod(Short4 a, Short4 b) {
    304         Short4 result = new Short4();
    305         result.x = (short)(a.x % b.x);
    306         result.y = (short)(a.y % b.y);
    307         result.z = (short)(a.z % b.z);
    308         result.w = (short)(a.w % b.w);
    309 
    310         return result;
    311     }
    312 
    313     /** @hide
    314      * Vector Modulo
    315      *
    316      * @param value
    317      */
    318     public void mod(short value) {
    319         x %= value;
    320         y %= value;
    321         z %= value;
    322         w %= value;
    323     }
    324 
    325     /** @hide
    326      * Vector Modulo
    327      *
    328      * @param a
    329      * @param b
    330      * @return
    331      */
    332     public static Short4 mod(Short4 a, short b) {
    333         Short4 result = new Short4();
    334         result.x = (short)(a.x % b);
    335         result.y = (short)(a.y % b);
    336         result.z = (short)(a.z % b);
    337         result.w = (short)(a.w % b);
    338 
    339         return result;
    340     }
    341 
    342     /** @hide
    343      * get vector length
    344      *
    345      * @return
    346      */
    347     public short length() {
    348         return 4;
    349     }
    350 
    351     /** @hide
    352      * set vector negate
    353      */
    354     public void negate() {
    355         this.x = (short)(-x);
    356         this.y = (short)(-y);
    357         this.z = (short)(-z);
    358         this.w = (short)(-w);
    359     }
    360 
    361     /** @hide
    362      * Vector dot Product
    363      *
    364      * @param a
    365      * @return
    366      */
    367     public short dotProduct(Short4 a) {
    368         return (short)((x * a.x) + (y * a.y) + (z * a.z) + (w * a.w));
    369     }
    370 
    371     /** @hide
    372      * Vector dot Product
    373      *
    374      * @param a
    375      * @param b
    376      * @return
    377      */
    378     public static short dotProduct(Short4 a, Short4 b) {
    379         return (short)((b.x * a.x) + (b.y * a.y) + (b.z * a.z) + (b.w * a.w));
    380     }
    381 
    382     /** @hide
    383      * Vector add Multiple
    384      *
    385      * @param a
    386      * @param factor
    387      */
    388     public void addMultiple(Short4 a, short factor) {
    389         x += a.x * factor;
    390         y += a.y * factor;
    391         z += a.z * factor;
    392         w += a.w * factor;
    393     }
    394 
    395     /** @hide
    396      * set vector value by Short4
    397      *
    398      * @param a
    399      */
    400     public void set(Short4 a) {
    401         this.x = a.x;
    402         this.y = a.y;
    403         this.z = a.z;
    404         this.w = a.w;
    405     }
    406 
    407     /** @hide
    408      * set the vector field value by Short
    409      *
    410      * @param a
    411      * @param b
    412      * @param c
    413      * @param d
    414      */
    415     public void setValues(short a, short b, short c, short d) {
    416         this.x = a;
    417         this.y = b;
    418         this.z = c;
    419         this.w = d;
    420     }
    421 
    422     /** @hide
    423      * return the element sum of vector
    424      *
    425      * @return
    426      */
    427     public short elementSum() {
    428         return (short)(x + y + z + w);
    429     }
    430 
    431     /** @hide
    432      * get the vector field value by index
    433      *
    434      * @param i
    435      * @return
    436      */
    437     public short get(int i) {
    438         switch (i) {
    439         case 0:
    440             return (short)(x);
    441         case 1:
    442             return (short)(y);
    443         case 2:
    444             return (short)(z);
    445         case 3:
    446             return (short)(w);
    447         default:
    448             throw new IndexOutOfBoundsException("Index: i");
    449         }
    450     }
    451 
    452     /** @hide
    453      * set the vector field value by index
    454      *
    455      * @param i
    456      * @param value
    457      */
    458     public void setAt(int i, short value) {
    459         switch (i) {
    460         case 0:
    461             x = value;
    462             return;
    463         case 1:
    464             y = value;
    465             return;
    466         case 2:
    467             z = value;
    468             return;
    469         case 3:
    470             w = value;
    471             return;
    472         default:
    473             throw new IndexOutOfBoundsException("Index: i");
    474         }
    475     }
    476 
    477     /** @hide
    478      * add the vector field value by index
    479      *
    480      * @param i
    481      * @param value
    482      */
    483     public void addAt(int i, short value) {
    484         switch (i) {
    485         case 0:
    486             x += value;
    487             return;
    488         case 1:
    489             y += value;
    490             return;
    491         case 2:
    492             z += value;
    493             return;
    494         case 3:
    495             w += value;
    496             return;
    497         default:
    498             throw new IndexOutOfBoundsException("Index: i");
    499         }
    500     }
    501 
    502     /** @hide
    503      * copy the vector to short array
    504      *
    505      * @param data
    506      * @param offset
    507      */
    508     public void copyTo(short[] data, int offset) {
    509         data[offset] = (short)(x);
    510         data[offset + 1] = (short)(y);
    511         data[offset + 2] = (short)(z);
    512         data[offset + 3] = (short)(w);
    513     }
    514 }
    515