mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
| 8 | #ifndef SkNx_sse_DEFINED |
| 9 | #define SkNx_sse_DEFINED |
| 10 | |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame^] | 11 | #include "SkCpu.h" |
| 12 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 13 | // This file may assume <= SSE2, but must check SK_CPU_SSE_LEVEL for anything more recent. |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 14 | // If you do, make sure this is in a static inline function... anywhere else risks violating ODR. |
mtklein | aa999cb | 2015-05-22 17:18:21 -0700 | [diff] [blame] | 15 | |
fmalita | c2e0ac4 | 2015-12-03 09:15:25 -0800 | [diff] [blame] | 16 | #define SKNX_IS_FAST |
| 17 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 18 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 19 | class SkNx<2, float> { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 20 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 21 | SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 22 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 23 | SkNx() {} |
| 24 | SkNx(float val) : fVec(_mm_set1_ps(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 25 | static SkNx Load(const void* ptr) { |
| 26 | return _mm_castsi128_ps(_mm_loadl_epi64((const __m128i*)ptr)); |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 27 | } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 28 | SkNx(float a, float b) : fVec(_mm_setr_ps(a,b,0,0)) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 29 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 30 | void store(void* ptr) const { _mm_storel_pi((__m64*)ptr, fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 31 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 32 | SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 33 | SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 34 | SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 35 | SkNx operator / (const SkNx& o) const { return _mm_div_ps(fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 36 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 37 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 38 | SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 39 | SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 40 | SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 41 | SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 42 | SkNx operator >= (const SkNx& o) const { return _mm_cmpge_ps (fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 43 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 44 | static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 45 | static SkNx Max(const SkNx& l, const SkNx& r) { return _mm_max_ps(l.fVec, r.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 46 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 47 | SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 48 | SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 49 | SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 50 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 51 | float operator[](int k) const { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 52 | SkASSERT(0 <= k && k < 2); |
| 53 | union { __m128 v; float fs[4]; } pun = {fVec}; |
mtklein | a156a8f | 2015-04-03 06:16:13 -0700 | [diff] [blame] | 54 | return pun.fs[k&1]; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 55 | } |
| 56 | |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 57 | bool allTrue() const { return 0xff == (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
| 58 | bool anyTrue() const { return 0x00 != (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
| 59 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 60 | __m128 fVec; |
| 61 | }; |
| 62 | |
| 63 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 64 | class SkNx<4, float> { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 65 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 66 | SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 67 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 68 | SkNx() {} |
| 69 | SkNx(float val) : fVec( _mm_set1_ps(val) ) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 70 | static SkNx Load(const void* ptr) { return _mm_loadu_ps((const float*)ptr); } |
mtklein | aba1dc8 | 2015-08-31 14:39:59 -0700 | [diff] [blame] | 71 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 72 | SkNx(float a, float b, float c, float d) : fVec(_mm_setr_ps(a,b,c,d)) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 73 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 74 | void store(void* ptr) const { _mm_storeu_ps((float*)ptr, fVec); } |
mtklein | 9db43ac | 2015-12-01 07:10:21 -0800 | [diff] [blame] | 75 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 76 | SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 77 | SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 78 | SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 79 | SkNx operator / (const SkNx& o) const { return _mm_div_ps(fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 80 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 81 | SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 82 | SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 83 | SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 84 | SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 85 | SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 86 | SkNx operator >= (const SkNx& o) const { return _mm_cmpge_ps (fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 87 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 88 | static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 89 | static SkNx Max(const SkNx& l, const SkNx& r) { return _mm_max_ps(l.fVec, r.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 90 | |
mtklein | c33065a | 2016-01-15 12:16:40 -0800 | [diff] [blame] | 91 | SkNx abs() const { return _mm_andnot_ps(_mm_set1_ps(-0.0f), fVec); } |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame^] | 92 | SkNx floor() const { |
| 93 | if (SkCpu::Supports(SkCpu::SSE41)) { |
| 94 | __m128 r; |
| 95 | #if defined(__GNUC__) || defined(__clang__) |
| 96 | asm("roundps $0x1, %[fVec], %[r]" : [r]"=x"(r) : [fVec]"x"(fVec)); |
| 97 | #else |
| 98 | r = _mm_floor_ps(fVec); |
| 99 | #endif |
| 100 | return r; |
| 101 | } |
| 102 | // Emulate _mm_floor_ps() with SSE2: |
| 103 | // - roundtrip through integers via truncation |
| 104 | // - subtract 1 if that's too big (possible for negative values). |
| 105 | // This restricts the domain of our inputs to a maximum somehwere around 2^31. |
| 106 | // Seems plenty big. |
| 107 | __m128 roundtrip = _mm_cvtepi32_ps(_mm_cvttps_epi32(fVec)); |
| 108 | __m128 too_big = _mm_cmpgt_ps(roundtrip, fVec); |
| 109 | return _mm_sub_ps(roundtrip, _mm_and_ps(too_big, _mm_set1_ps(1.0f))); |
| 110 | } |
mtklein | c33065a | 2016-01-15 12:16:40 -0800 | [diff] [blame] | 111 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 112 | SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 113 | SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 114 | SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 115 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 116 | float operator[](int k) const { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 117 | SkASSERT(0 <= k && k < 4); |
| 118 | union { __m128 v; float fs[4]; } pun = {fVec}; |
mtklein | a156a8f | 2015-04-03 06:16:13 -0700 | [diff] [blame] | 119 | return pun.fs[k&3]; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 120 | } |
| 121 | |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 122 | bool allTrue() const { return 0xffff == _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
| 123 | bool anyTrue() const { return 0x0000 != _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
| 124 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 125 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 2aab22a | 2015-06-26 10:46:31 -0700 | [diff] [blame] | 126 | return _mm_or_ps(_mm_and_ps (fVec, t.fVec), |
| 127 | _mm_andnot_ps(fVec, e.fVec)); |
| 128 | } |
| 129 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 130 | __m128 fVec; |
| 131 | }; |
| 132 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 133 | template <> |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 134 | class SkNx<4, int> { |
| 135 | public: |
| 136 | SkNx(const __m128i& vec) : fVec(vec) {} |
| 137 | |
| 138 | SkNx() {} |
| 139 | SkNx(int val) : fVec(_mm_set1_epi32(val)) {} |
| 140 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 141 | SkNx(int a, int b, int c, int d) : fVec(_mm_setr_epi32(a,b,c,d)) {} |
| 142 | |
| 143 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
| 144 | |
| 145 | SkNx operator + (const SkNx& o) const { return _mm_add_epi32(fVec, o.fVec); } |
| 146 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi32(fVec, o.fVec); } |
| 147 | SkNx operator * (const SkNx& o) const { |
| 148 | __m128i mul20 = _mm_mul_epu32(fVec, o.fVec), |
| 149 | mul31 = _mm_mul_epu32(_mm_srli_si128(fVec, 4), _mm_srli_si128(o.fVec, 4)); |
| 150 | return _mm_unpacklo_epi32(_mm_shuffle_epi32(mul20, _MM_SHUFFLE(0,0,2,0)), |
| 151 | _mm_shuffle_epi32(mul31, _MM_SHUFFLE(0,0,2,0))); |
| 152 | } |
| 153 | |
| 154 | SkNx operator << (int bits) const { return _mm_slli_epi32(fVec, bits); } |
| 155 | SkNx operator >> (int bits) const { return _mm_srai_epi32(fVec, bits); } |
| 156 | |
| 157 | int operator[](int k) const { |
| 158 | SkASSERT(0 <= k && k < 4); |
| 159 | union { __m128i v; int is[4]; } pun = {fVec}; |
| 160 | return pun.is[k&3]; |
| 161 | } |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 162 | |
| 163 | __m128i fVec; |
| 164 | }; |
| 165 | |
| 166 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 167 | class SkNx<4, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 168 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 169 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 170 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 171 | SkNx() {} |
| 172 | SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 173 | static SkNx Load(const void* ptr) { return _mm_loadl_epi64((const __m128i*)ptr); } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 174 | SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d) : fVec(_mm_setr_epi16(a,b,c,d,0,0,0,0)) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 175 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 176 | void store(void* ptr) const { _mm_storel_epi64((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 177 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 178 | SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 179 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 180 | SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 181 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 182 | SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 183 | SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 184 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 185 | uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 186 | SkASSERT(0 <= k && k < 4); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 187 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 188 | return pun.us[k&3]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 189 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 190 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 191 | __m128i fVec; |
| 192 | }; |
| 193 | |
| 194 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 195 | class SkNx<8, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 196 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 197 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 198 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 199 | SkNx() {} |
| 200 | SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 201 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 202 | SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d, |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 203 | uint16_t e, uint16_t f, uint16_t g, uint16_t h) : fVec(_mm_setr_epi16(a,b,c,d,e,f,g,h)) {} |
| 204 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 205 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 206 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 207 | SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 208 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 209 | SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 210 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 211 | SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 212 | SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 213 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 214 | static SkNx Min(const SkNx& a, const SkNx& b) { |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 215 | // No unsigned _mm_min_epu16, so we'll shift into a space where we can use the |
| 216 | // signed version, _mm_min_epi16, then shift back. |
| 217 | const uint16_t top = 0x8000; // Keep this separate from _mm_set1_epi16 or MSVC will whine. |
| 218 | const __m128i top_8x = _mm_set1_epi16(top); |
| 219 | return _mm_add_epi8(top_8x, _mm_min_epi16(_mm_sub_epi8(a.fVec, top_8x), |
| 220 | _mm_sub_epi8(b.fVec, top_8x))); |
| 221 | } |
| 222 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 223 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 4be181e | 2015-07-14 10:54:19 -0700 | [diff] [blame] | 224 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 225 | _mm_andnot_si128(fVec, e.fVec)); |
| 226 | } |
| 227 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 228 | uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 229 | SkASSERT(0 <= k && k < 8); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 230 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 231 | return pun.us[k&7]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 232 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 233 | |
| 234 | __m128i fVec; |
| 235 | }; |
| 236 | |
| 237 | template <> |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 238 | class SkNx<4, uint8_t> { |
| 239 | public: |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 240 | SkNx() {} |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 241 | SkNx(const __m128i& vec) : fVec(vec) {} |
| 242 | SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d) |
| 243 | : fVec(_mm_setr_epi8(a,b,c,d, 0,0,0,0, 0,0,0,0, 0,0,0,0)) {} |
| 244 | |
| 245 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 246 | static SkNx Load(const void* ptr) { return _mm_cvtsi32_si128(*(const int*)ptr); } |
| 247 | void store(void* ptr) const { *(int*)ptr = _mm_cvtsi128_si32(fVec); } |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 248 | |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 249 | uint8_t operator[](int k) const { |
| 250 | SkASSERT(0 <= k && k < 4); |
| 251 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 252 | return pun.us[k&3]; |
| 253 | } |
| 254 | |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 255 | // TODO as needed |
| 256 | |
| 257 | __m128i fVec; |
| 258 | }; |
| 259 | |
| 260 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 261 | class SkNx<16, uint8_t> { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 262 | public: |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 263 | SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 264 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 265 | SkNx() {} |
| 266 | SkNx(uint8_t val) : fVec(_mm_set1_epi8(val)) {} |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 267 | static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 268 | SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d, |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 269 | uint8_t e, uint8_t f, uint8_t g, uint8_t h, |
| 270 | uint8_t i, uint8_t j, uint8_t k, uint8_t l, |
| 271 | uint8_t m, uint8_t n, uint8_t o, uint8_t p) |
| 272 | : fVec(_mm_setr_epi8(a,b,c,d, e,f,g,h, i,j,k,l, m,n,o,p)) {} |
| 273 | |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 274 | void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 275 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 276 | SkNx saturatedAdd(const SkNx& o) const { return _mm_adds_epu8(fVec, o.fVec); } |
mtklein | 6cbf18c | 2015-05-12 15:48:09 -0700 | [diff] [blame] | 277 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 278 | SkNx operator + (const SkNx& o) const { return _mm_add_epi8(fVec, o.fVec); } |
| 279 | SkNx operator - (const SkNx& o) const { return _mm_sub_epi8(fVec, o.fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 280 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 281 | static SkNx Min(const SkNx& a, const SkNx& b) { return _mm_min_epu8(a.fVec, b.fVec); } |
| 282 | SkNx operator < (const SkNx& o) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 283 | // There's no unsigned _mm_cmplt_epu8, so we flip the sign bits then use a signed compare. |
| 284 | auto flip = _mm_set1_epi8(char(0x80)); |
| 285 | return _mm_cmplt_epi8(_mm_xor_si128(flip, fVec), _mm_xor_si128(flip, o.fVec)); |
| 286 | } |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 287 | |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 288 | uint8_t operator[](int k) const { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 289 | SkASSERT(0 <= k && k < 16); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 290 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 291 | return pun.us[k&15]; |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 292 | } |
| 293 | |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 294 | SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 295 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 296 | _mm_andnot_si128(fVec, e.fVec)); |
| 297 | } |
| 298 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 299 | __m128i fVec; |
| 300 | }; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 301 | |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 302 | template<> /*static*/ inline Sk4f SkNx_cast<float, int>(const Sk4i& src) { |
| 303 | return _mm_cvtepi32_ps(src.fVec); |
| 304 | } |
| 305 | |
| 306 | template <> /*static*/ inline Sk4i SkNx_cast<int, float>(const Sk4f& src) { |
| 307 | return _mm_cvttps_epi32(src.fVec); |
| 308 | } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 309 | |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 310 | template<> /*static*/ inline Sk4h SkNx_cast<uint16_t, float>(const Sk4f& src) { |
| 311 | auto _32 = _mm_cvttps_epi32(src.fVec); |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 312 | // Ideally we'd use _mm_packus_epi32 here. But that's SSE4.1+. |
| 313 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 314 | // With SSSE3, we can just shuffle the low 2 bytes from each lane right into place. |
| 315 | const int _ = ~0; |
| 316 | return _mm_shuffle_epi8(_32, _mm_setr_epi8(0,1, 4,5, 8,9, 12,13, _,_,_,_,_,_,_,_)); |
| 317 | #else |
| 318 | // With SSE2, we have to emulate _mm_packus_epi32 with _mm_packs_epi32: |
| 319 | _32 = _mm_sub_epi32(_32, _mm_set1_epi32((int)0x00008000)); |
| 320 | return _mm_add_epi16(_mm_packs_epi32(_32, _32), _mm_set1_epi16((short)0x8000)); |
| 321 | #endif |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 322 | } |
| 323 | |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 324 | template<> /*static*/ inline Sk4b SkNx_cast<uint8_t, float>(const Sk4f& src) { |
| 325 | auto _32 = _mm_cvttps_epi32(src.fVec); |
| 326 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 327 | const int _ = ~0; |
| 328 | return _mm_shuffle_epi8(_32, _mm_setr_epi8(0,4,8,12, _,_,_,_, _,_,_,_, _,_,_,_)); |
| 329 | #else |
| 330 | auto _16 = _mm_packus_epi16(_32, _32); |
| 331 | return _mm_packus_epi16(_16, _16); |
| 332 | #endif |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 333 | } |
| 334 | |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 335 | template<> /*static*/ inline Sk4f SkNx_cast<float, uint8_t>(const Sk4b& src) { |
| 336 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 337 | const int _ = ~0; |
| 338 | auto _32 = _mm_shuffle_epi8(src.fVec, _mm_setr_epi8(0,_,_,_, 1,_,_,_, 2,_,_,_, 3,_,_,_)); |
| 339 | #else |
| 340 | auto _16 = _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()), |
| 341 | _32 = _mm_unpacklo_epi16(_16, _mm_setzero_si128()); |
| 342 | #endif |
| 343 | return _mm_cvtepi32_ps(_32); |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 344 | } |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 345 | |
| 346 | template<> /*static*/ inline Sk4f SkNx_cast<float, uint16_t>(const Sk4h& src) { |
| 347 | auto _32 = _mm_unpacklo_epi16(src.fVec, _mm_setzero_si128()); |
| 348 | return _mm_cvtepi32_ps(_32); |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 349 | } |
| 350 | |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 351 | template<> /*static*/ inline Sk16b SkNx_cast<uint8_t, float>(const Sk16f& src) { |
| 352 | Sk8f ab, cd; |
| 353 | SkNx_split(src, &ab, &cd); |
| 354 | |
| 355 | Sk4f a,b,c,d; |
| 356 | SkNx_split(ab, &a, &b); |
| 357 | SkNx_split(cd, &c, &d); |
| 358 | |
| 359 | return _mm_packus_epi16(_mm_packus_epi16(_mm_cvttps_epi32(a.fVec), |
| 360 | _mm_cvttps_epi32(b.fVec)), |
| 361 | _mm_packus_epi16(_mm_cvttps_epi32(c.fVec), |
| 362 | _mm_cvttps_epi32(d.fVec))); |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 363 | } |
| 364 | |
| 365 | template<> /*static*/ inline Sk4h SkNx_cast<uint16_t, uint8_t>(const Sk4b& src) { |
| 366 | return _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()); |
| 367 | } |
| 368 | |
| 369 | template<> /*static*/ inline Sk4b SkNx_cast<uint8_t, uint16_t>(const Sk4h& src) { |
| 370 | return _mm_packus_epi16(src.fVec, src.fVec); |
mtklein | 550e9b0 | 2016-01-20 11:55:51 -0800 | [diff] [blame] | 371 | } |
| 372 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 373 | #endif//SkNx_sse_DEFINED |