| Jan Vesely | ef513d3 | 2014-09-02 17:55:02 +0000 | [diff] [blame] | 1 | #include <clc/clc.h> |
| 2 | #include "../clcmacro.h" |
| 3 | |
| 4 | _CLC_OVERLOAD _CLC_DEF char mad_sat(char x, char y, char z) { |
| 5 | return clamp((short)mad24((short)x, (short)y, (short)z), (short)CHAR_MIN, (short) CHAR_MAX); |
| 6 | } |
| 7 | |
| 8 | _CLC_OVERLOAD _CLC_DEF uchar mad_sat(uchar x, uchar y, uchar z) { |
| 9 | return clamp((ushort)mad24((ushort)x, (ushort)y, (ushort)z), (ushort)0, (ushort) UCHAR_MAX); |
| 10 | } |
| 11 | |
| 12 | _CLC_OVERLOAD _CLC_DEF short mad_sat(short x, short y, short z) { |
| 13 | return clamp((int)mad24((int)x, (int)y, (int)z), (int)SHRT_MIN, (int) SHRT_MAX); |
| 14 | } |
| 15 | |
| 16 | _CLC_OVERLOAD _CLC_DEF ushort mad_sat(ushort x, ushort y, ushort z) { |
| 17 | return clamp((uint)mad24((uint)x, (uint)y, (uint)z), (uint)0, (uint) USHRT_MAX); |
| 18 | } |
| 19 | |
| 20 | _CLC_OVERLOAD _CLC_DEF int mad_sat(int x, int y, int z) { |
| 21 | int mhi = mul_hi(x, y); |
| 22 | uint mlo = x * y; |
| 23 | long m = upsample(mhi, mlo); |
| 24 | m += z; |
| 25 | if (m > INT_MAX) |
| 26 | return INT_MAX; |
| 27 | if (m < INT_MIN) |
| 28 | return INT_MIN; |
| 29 | return m; |
| 30 | } |
| 31 | |
| 32 | _CLC_OVERLOAD _CLC_DEF uint mad_sat(uint x, uint y, uint z) { |
| 33 | if (mul_hi(x, y) != 0) |
| 34 | return UINT_MAX; |
| 35 | return add_sat(x * y, z); |
| 36 | } |
| 37 | |
| 38 | _CLC_OVERLOAD _CLC_DEF long mad_sat(long x, long y, long z) { |
| 39 | long hi = mul_hi(x, y); |
| 40 | ulong ulo = x * y; |
| 41 | long slo = x * y; |
| 42 | /* Big overflow of more than 2 bits, add can't fix this */ |
| 43 | if (((x < 0) == (y < 0)) && hi != 0) |
| 44 | return LONG_MAX; |
| 45 | /* Low overflow in mul and z not neg enough to correct it */ |
| 46 | if (hi == 0 && ulo >= LONG_MAX && (z > 0 || (ulo + z) > LONG_MAX)) |
| 47 | return LONG_MAX; |
| 48 | /* Big overflow of more than 2 bits, add can't fix this */ |
| 49 | if (((x < 0) != (y < 0)) && hi != -1) |
| 50 | return LONG_MIN; |
| 51 | /* Low overflow in mul and z not pos enough to correct it */ |
| 52 | if (hi == -1 && ulo <= ((ulong)LONG_MAX + 1UL) && (z < 0 || z < (LONG_MAX - ulo))) |
| 53 | return LONG_MIN; |
| 54 | /* We have checked all conditions, any overflow in addition returns |
| 55 | * the correct value */ |
| 56 | return ulo + z; |
| 57 | } |
| 58 | |
| 59 | _CLC_OVERLOAD _CLC_DEF ulong mad_sat(ulong x, ulong y, ulong z) { |
| 60 | if (mul_hi(x, y) != 0) |
| 61 | return ULONG_MAX; |
| 62 | return add_sat(x * y, z); |
| 63 | } |
| 64 | |
| 65 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, char, mad_sat, char, char, char) |
| 66 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, uchar, mad_sat, uchar, uchar, uchar) |
| 67 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, short, mad_sat, short, short, short) |
| 68 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, ushort, mad_sat, ushort, ushort, ushort) |
| 69 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, int, mad_sat, int, int, int) |
| 70 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, uint, mad_sat, uint, uint, uint) |
| 71 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, long, mad_sat, long, long, long) |
| 72 | _CLC_TERNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, ulong, mad_sat, ulong, ulong, ulong) |