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 | e18fa44 | 2016-06-09 13:40:56 -0700 | [diff] [blame] | 11 | #include <immintrin.h> |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 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 | |
Mike Klein | 1e76464 | 2016-10-14 17:09:03 -0400 | [diff] [blame] | 16 | namespace { |
| 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: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 21 | AI SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 22 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 23 | AI SkNx() {} |
| 24 | AI SkNx(float val) : fVec(_mm_set1_ps(val)) {} |
| 25 | AI static SkNx Load(const void* ptr) { |
mtklein | 507ef6d | 2016-01-31 08:02:47 -0800 | [diff] [blame] | 26 | return _mm_castsi128_ps(_mm_loadl_epi64((const __m128i*)ptr)); |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 27 | } |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 28 | AI 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 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 30 | AI void store(void* ptr) const { _mm_storel_pi((__m64*)ptr, fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 31 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 32 | AI SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 33 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 34 | AI SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 35 | AI 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 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 37 | AI SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 38 | AI SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 39 | AI SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 40 | AI SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 41 | AI SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 42 | AI 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 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 44 | AI static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 45 | AI 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 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 47 | AI SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 48 | AI SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 49 | AI SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 50 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 51 | AI 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 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 57 | AI bool allTrue() const { return 0xff == (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
| 58 | AI bool anyTrue() const { return 0x00 != (_mm_movemask_epi8(_mm_castps_si128(fVec)) & 0xff); } |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 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: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 66 | AI SkNx(const __m128& vec) : fVec(vec) {} |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 67 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 68 | AI SkNx() {} |
| 69 | AI SkNx(float val) : fVec( _mm_set1_ps(val) ) {} |
| 70 | AI 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] | 71 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 72 | AI static SkNx Load(const void* ptr) { return _mm_loadu_ps((const float*)ptr); } |
| 73 | AI void store(void* ptr) const { _mm_storeu_ps((float*)ptr, fVec); } |
mtklein | 9db43ac | 2015-12-01 07:10:21 -0800 | [diff] [blame] | 74 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 75 | AI static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 76 | __m128 v0 = _mm_loadu_ps(((float*)ptr) + 0), |
| 77 | v1 = _mm_loadu_ps(((float*)ptr) + 4), |
| 78 | v2 = _mm_loadu_ps(((float*)ptr) + 8), |
| 79 | v3 = _mm_loadu_ps(((float*)ptr) + 12); |
| 80 | _MM_TRANSPOSE4_PS(v0, v1, v2, v3); |
| 81 | *r = v0; |
| 82 | *g = v1; |
| 83 | *b = v2; |
| 84 | *a = v3; |
| 85 | } |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 86 | AI static void Store4(void* dst, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 87 | __m128 v0 = r.fVec, |
| 88 | v1 = g.fVec, |
| 89 | v2 = b.fVec, |
| 90 | v3 = a.fVec; |
| 91 | _MM_TRANSPOSE4_PS(v0, v1, v2, v3); |
| 92 | _mm_storeu_ps(((float*) dst) + 0, v0); |
| 93 | _mm_storeu_ps(((float*) dst) + 4, v1); |
| 94 | _mm_storeu_ps(((float*) dst) + 8, v2); |
| 95 | _mm_storeu_ps(((float*) dst) + 12, v3); |
| 96 | } |
| 97 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 98 | AI SkNx operator + (const SkNx& o) const { return _mm_add_ps(fVec, o.fVec); } |
| 99 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_ps(fVec, o.fVec); } |
| 100 | AI SkNx operator * (const SkNx& o) const { return _mm_mul_ps(fVec, o.fVec); } |
| 101 | AI SkNx operator / (const SkNx& o) const { return _mm_div_ps(fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 102 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 103 | AI SkNx operator == (const SkNx& o) const { return _mm_cmpeq_ps (fVec, o.fVec); } |
| 104 | AI SkNx operator != (const SkNx& o) const { return _mm_cmpneq_ps(fVec, o.fVec); } |
| 105 | AI SkNx operator < (const SkNx& o) const { return _mm_cmplt_ps (fVec, o.fVec); } |
| 106 | AI SkNx operator > (const SkNx& o) const { return _mm_cmpgt_ps (fVec, o.fVec); } |
| 107 | AI SkNx operator <= (const SkNx& o) const { return _mm_cmple_ps (fVec, o.fVec); } |
| 108 | AI SkNx operator >= (const SkNx& o) const { return _mm_cmpge_ps (fVec, o.fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 109 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 110 | AI static SkNx Min(const SkNx& l, const SkNx& r) { return _mm_min_ps(l.fVec, r.fVec); } |
| 111 | AI 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] | 112 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 113 | AI SkNx abs() const { return _mm_andnot_ps(_mm_set1_ps(-0.0f), fVec); } |
| 114 | AI SkNx floor() const { |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 115 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 116 | return _mm_floor_ps(fVec); |
| 117 | #else |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 118 | // Emulate _mm_floor_ps() with SSE2: |
| 119 | // - roundtrip through integers via truncation |
| 120 | // - subtract 1 if that's too big (possible for negative values). |
| 121 | // This restricts the domain of our inputs to a maximum somehwere around 2^31. |
| 122 | // Seems plenty big. |
| 123 | __m128 roundtrip = _mm_cvtepi32_ps(_mm_cvttps_epi32(fVec)); |
| 124 | __m128 too_big = _mm_cmpgt_ps(roundtrip, fVec); |
| 125 | return _mm_sub_ps(roundtrip, _mm_and_ps(too_big, _mm_set1_ps(1.0f))); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 126 | #endif |
mtklein | 244a653 | 2016-04-19 14:21:30 -0700 | [diff] [blame] | 127 | } |
mtklein | c33065a | 2016-01-15 12:16:40 -0800 | [diff] [blame] | 128 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 129 | AI SkNx sqrt() const { return _mm_sqrt_ps (fVec); } |
| 130 | AI SkNx rsqrt() const { return _mm_rsqrt_ps(fVec); } |
| 131 | AI SkNx invert() const { return _mm_rcp_ps(fVec); } |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 132 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 133 | AI float operator[](int k) const { |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 134 | SkASSERT(0 <= k && k < 4); |
| 135 | union { __m128 v; float fs[4]; } pun = {fVec}; |
mtklein | a156a8f | 2015-04-03 06:16:13 -0700 | [diff] [blame] | 136 | return pun.fs[k&3]; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 137 | } |
| 138 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 139 | AI bool allTrue() const { return 0xffff == _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
| 140 | AI bool anyTrue() const { return 0x0000 != _mm_movemask_epi8(_mm_castps_si128(fVec)); } |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 141 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 142 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 143 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
msarett | d280957 | 2016-06-20 06:07:45 -0700 | [diff] [blame] | 144 | return _mm_blendv_ps(e.fVec, t.fVec, fVec); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 145 | #else |
mtklein | 2aab22a | 2015-06-26 10:46:31 -0700 | [diff] [blame] | 146 | return _mm_or_ps(_mm_and_ps (fVec, t.fVec), |
| 147 | _mm_andnot_ps(fVec, e.fVec)); |
mtklein | 5608e2e | 2016-07-11 09:59:21 -0700 | [diff] [blame] | 148 | #endif |
mtklein | 2aab22a | 2015-06-26 10:46:31 -0700 | [diff] [blame] | 149 | } |
| 150 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 151 | __m128 fVec; |
| 152 | }; |
| 153 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 154 | template <> |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 155 | class SkNx<4, int32_t> { |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 156 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 157 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 158 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 159 | AI SkNx() {} |
| 160 | AI SkNx(int32_t val) : fVec(_mm_set1_epi32(val)) {} |
| 161 | AI static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 162 | AI SkNx(int32_t a, int32_t b, int32_t c, int32_t d) : fVec(_mm_setr_epi32(a,b,c,d)) {} |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 163 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 164 | AI void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 165 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 166 | AI SkNx operator + (const SkNx& o) const { return _mm_add_epi32(fVec, o.fVec); } |
| 167 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_epi32(fVec, o.fVec); } |
| 168 | AI SkNx operator * (const SkNx& o) const { |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 169 | __m128i mul20 = _mm_mul_epu32(fVec, o.fVec), |
| 170 | mul31 = _mm_mul_epu32(_mm_srli_si128(fVec, 4), _mm_srli_si128(o.fVec, 4)); |
| 171 | return _mm_unpacklo_epi32(_mm_shuffle_epi32(mul20, _MM_SHUFFLE(0,0,2,0)), |
| 172 | _mm_shuffle_epi32(mul31, _MM_SHUFFLE(0,0,2,0))); |
| 173 | } |
| 174 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 175 | AI SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
| 176 | AI SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
| 177 | AI SkNx operator ^ (const SkNx& o) const { return _mm_xor_si128(fVec, o.fVec); } |
mtklein | 64f061a | 2016-06-17 12:09:16 -0700 | [diff] [blame] | 178 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 179 | AI SkNx operator << (int bits) const { return _mm_slli_epi32(fVec, bits); } |
| 180 | AI SkNx operator >> (int bits) const { return _mm_srai_epi32(fVec, bits); } |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 181 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 182 | AI SkNx operator == (const SkNx& o) const { return _mm_cmpeq_epi32 (fVec, o.fVec); } |
| 183 | AI SkNx operator < (const SkNx& o) const { return _mm_cmplt_epi32 (fVec, o.fVec); } |
| 184 | AI SkNx operator > (const SkNx& o) const { return _mm_cmpgt_epi32 (fVec, o.fVec); } |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 185 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 186 | AI int32_t operator[](int k) const { |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 187 | SkASSERT(0 <= k && k < 4); |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 188 | union { __m128i v; int32_t is[4]; } pun = {fVec}; |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 189 | return pun.is[k&3]; |
| 190 | } |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 191 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 192 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 193 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 194 | return _mm_blendv_epi8(e.fVec, t.fVec, fVec); |
| 195 | #else |
| 196 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 197 | _mm_andnot_si128(fVec, e.fVec)); |
| 198 | #endif |
| 199 | } |
| 200 | |
mtklein | 8273ca4 | 2016-02-09 11:32:51 -0800 | [diff] [blame] | 201 | __m128i fVec; |
| 202 | }; |
| 203 | |
| 204 | template <> |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 205 | class SkNx<4, uint32_t> { |
| 206 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 207 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 208 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 209 | AI SkNx() {} |
| 210 | AI SkNx(uint32_t val) : fVec(_mm_set1_epi32(val)) {} |
| 211 | AI static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 212 | AI SkNx(uint32_t a, uint32_t b, uint32_t c, uint32_t d) : fVec(_mm_setr_epi32(a,b,c,d)) {} |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 213 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 214 | AI void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 215 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 216 | AI SkNx operator + (const SkNx& o) const { return _mm_add_epi32(fVec, o.fVec); } |
| 217 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_epi32(fVec, o.fVec); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 218 | // Not quite sure how to best do operator * in SSE2. We probably don't use it. |
| 219 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 220 | AI SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
| 221 | AI SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
| 222 | AI SkNx operator ^ (const SkNx& o) const { return _mm_xor_si128(fVec, o.fVec); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 223 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 224 | AI SkNx operator << (int bits) const { return _mm_slli_epi32(fVec, bits); } |
| 225 | AI SkNx operator >> (int bits) const { return _mm_srli_epi32(fVec, bits); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 226 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 227 | AI SkNx operator == (const SkNx& o) const { return _mm_cmpeq_epi32 (fVec, o.fVec); } |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 228 | // operator < and > take a little extra fiddling to make work for unsigned ints. |
| 229 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 230 | AI uint32_t operator[](int k) const { |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 231 | SkASSERT(0 <= k && k < 4); |
| 232 | union { __m128i v; uint32_t us[4]; } pun = {fVec}; |
| 233 | return pun.us[k&3]; |
| 234 | } |
| 235 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 236 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 237 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 238 | return _mm_blendv_epi8(e.fVec, t.fVec, fVec); |
| 239 | #else |
| 240 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 241 | _mm_andnot_si128(fVec, e.fVec)); |
| 242 | #endif |
| 243 | } |
| 244 | |
| 245 | __m128i fVec; |
| 246 | }; |
| 247 | |
| 248 | |
| 249 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 250 | class SkNx<4, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 251 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 252 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 253 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 254 | AI SkNx() {} |
| 255 | AI SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
| 256 | AI SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d) |
| 257 | : fVec(_mm_setr_epi16(a,b,c,d,0,0,0,0)) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 258 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 259 | AI static SkNx Load(const void* ptr) { return _mm_loadl_epi64((const __m128i*)ptr); } |
| 260 | AI void store(void* ptr) const { _mm_storel_epi64((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 261 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 262 | AI static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 263 | __m128i lo = _mm_loadu_si128(((__m128i*)ptr) + 0), |
| 264 | hi = _mm_loadu_si128(((__m128i*)ptr) + 1); |
| 265 | __m128i even = _mm_unpacklo_epi16(lo, hi), // r0 r2 g0 g2 b0 b2 a0 a2 |
| 266 | odd = _mm_unpackhi_epi16(lo, hi); // r1 r3 ... |
| 267 | __m128i rg = _mm_unpacklo_epi16(even, odd), // r0 r1 r2 r3 g0 g1 g2 g3 |
| 268 | ba = _mm_unpackhi_epi16(even, odd); // b0 b1 ... a0 a1 ... |
| 269 | *r = rg; |
| 270 | *g = _mm_srli_si128(rg, 8); |
| 271 | *b = ba; |
| 272 | *a = _mm_srli_si128(ba, 8); |
| 273 | } |
Matt Sarett | 5bee0b6 | 2017-01-19 12:04:32 -0500 | [diff] [blame] | 274 | AI static void Load3(const void* ptr, SkNx* r, SkNx* g, SkNx* b) { |
| 275 | // The idea here is to get 4 vectors that are R G B _ _ _ _ _. |
| 276 | // The second load is at a funny location to make sure we don't read past |
| 277 | // the bounds of memory. This is fine, we just need to shift it a little bit. |
| 278 | const uint8_t* ptr8 = (const uint8_t*) ptr; |
| 279 | __m128i rgb0 = _mm_loadu_si128((const __m128i*) (ptr8 + 0)); |
| 280 | __m128i rgb1 = _mm_srli_si128(rgb0, 3*2); |
| 281 | __m128i rgb2 = _mm_srli_si128(_mm_loadu_si128((const __m128i*) (ptr8 + 4*2)), 2*2); |
| 282 | __m128i rgb3 = _mm_srli_si128(rgb2, 3*2); |
| 283 | |
| 284 | __m128i rrggbb01 = _mm_unpacklo_epi16(rgb0, rgb1); |
| 285 | __m128i rrggbb23 = _mm_unpacklo_epi16(rgb2, rgb3); |
| 286 | *r = _mm_unpacklo_epi32(rrggbb01, rrggbb23); |
| 287 | *g = _mm_srli_si128(r->fVec, 4*2); |
| 288 | *b = _mm_unpackhi_epi32(rrggbb01, rrggbb23); |
| 289 | } |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 290 | AI static void Store4(void* dst, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
Mike Klein | 33cbfd7 | 2016-10-06 11:09:27 -0400 | [diff] [blame] | 291 | __m128i rg = _mm_unpacklo_epi16(r.fVec, g.fVec); |
| 292 | __m128i ba = _mm_unpacklo_epi16(b.fVec, a.fVec); |
| 293 | __m128i lo = _mm_unpacklo_epi32(rg, ba); |
| 294 | __m128i hi = _mm_unpackhi_epi32(rg, ba); |
| 295 | _mm_storeu_si128(((__m128i*) dst) + 0, lo); |
| 296 | _mm_storeu_si128(((__m128i*) dst) + 1, hi); |
| 297 | } |
| 298 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 299 | AI SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 300 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 301 | AI SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
Matt Sarett | 379938e | 2017-01-12 18:34:29 -0500 | [diff] [blame] | 302 | AI SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
| 303 | AI SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 304 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 305 | AI SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 306 | AI SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 307 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 308 | AI uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 309 | SkASSERT(0 <= k && k < 4); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 310 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 311 | return pun.us[k&3]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 312 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 313 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 314 | __m128i fVec; |
| 315 | }; |
| 316 | |
| 317 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 318 | class SkNx<8, uint16_t> { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 319 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 320 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 321 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 322 | AI SkNx() {} |
| 323 | AI SkNx(uint16_t val) : fVec(_mm_set1_epi16(val)) {} |
| 324 | AI SkNx(uint16_t a, uint16_t b, uint16_t c, uint16_t d, |
| 325 | uint16_t e, uint16_t f, uint16_t g, uint16_t h) |
| 326 | : fVec(_mm_setr_epi16(a,b,c,d,e,f,g,h)) {} |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 327 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 328 | AI static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 329 | AI void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 330 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 331 | AI static void Load4(const void* ptr, SkNx* r, SkNx* g, SkNx* b, SkNx* a) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 332 | // TODO: AVX2 version |
| 333 | __m128i _01 = _mm_loadu_si128(((__m128i*)ptr) + 0), |
| 334 | _23 = _mm_loadu_si128(((__m128i*)ptr) + 1), |
| 335 | _45 = _mm_loadu_si128(((__m128i*)ptr) + 2), |
| 336 | _67 = _mm_loadu_si128(((__m128i*)ptr) + 3); |
| 337 | |
| 338 | __m128i _02 = _mm_unpacklo_epi16(_01, _23), // r0 r2 g0 g2 b0 b2 a0 a2 |
| 339 | _13 = _mm_unpackhi_epi16(_01, _23), // r1 r3 g1 g3 b1 b3 a1 a3 |
| 340 | _46 = _mm_unpacklo_epi16(_45, _67), |
| 341 | _57 = _mm_unpackhi_epi16(_45, _67); |
| 342 | |
| 343 | __m128i rg0123 = _mm_unpacklo_epi16(_02, _13), // r0 r1 r2 r3 g0 g1 g2 g3 |
| 344 | ba0123 = _mm_unpackhi_epi16(_02, _13), // b0 b1 b2 b3 a0 a1 a2 a3 |
| 345 | rg4567 = _mm_unpacklo_epi16(_46, _57), |
| 346 | ba4567 = _mm_unpackhi_epi16(_46, _57); |
| 347 | |
| 348 | *r = _mm_unpacklo_epi64(rg0123, rg4567); |
| 349 | *g = _mm_unpackhi_epi64(rg0123, rg4567); |
| 350 | *b = _mm_unpacklo_epi64(ba0123, ba4567); |
| 351 | *a = _mm_unpackhi_epi64(ba0123, ba4567); |
| 352 | } |
Matt Sarett | 5bee0b6 | 2017-01-19 12:04:32 -0500 | [diff] [blame] | 353 | AI static void Load3(const void* ptr, SkNx* r, SkNx* g, SkNx* b) { |
| 354 | // TODO: AVX2 version |
| 355 | const uint8_t* ptr8 = (const uint8_t*) ptr; |
| 356 | __m128i rgb0 = _mm_loadu_si128((const __m128i*) (ptr8 + 0*2)); |
| 357 | __m128i rgb1 = _mm_srli_si128(rgb0, 3*2); |
| 358 | __m128i rgb2 = _mm_loadu_si128((const __m128i*) (ptr8 + 6*2)); |
| 359 | __m128i rgb3 = _mm_srli_si128(rgb2, 3*2); |
| 360 | __m128i rgb4 = _mm_loadu_si128((const __m128i*) (ptr8 + 12*2)); |
| 361 | __m128i rgb5 = _mm_srli_si128(rgb4, 3*2); |
| 362 | __m128i rgb6 = _mm_srli_si128(_mm_loadu_si128((const __m128i*) (ptr8 + 16*2)), 2*2); |
| 363 | __m128i rgb7 = _mm_srli_si128(rgb6, 3*2); |
| 364 | |
| 365 | __m128i rgb01 = _mm_unpacklo_epi16(rgb0, rgb1); |
| 366 | __m128i rgb23 = _mm_unpacklo_epi16(rgb2, rgb3); |
| 367 | __m128i rgb45 = _mm_unpacklo_epi16(rgb4, rgb5); |
| 368 | __m128i rgb67 = _mm_unpacklo_epi16(rgb6, rgb7); |
| 369 | |
| 370 | __m128i rg03 = _mm_unpacklo_epi32(rgb01, rgb23); |
| 371 | __m128i bx03 = _mm_unpackhi_epi32(rgb01, rgb23); |
| 372 | __m128i rg47 = _mm_unpacklo_epi32(rgb45, rgb67); |
| 373 | __m128i bx47 = _mm_unpackhi_epi32(rgb45, rgb67); |
| 374 | |
| 375 | *r = _mm_unpacklo_epi64(rg03, rg47); |
| 376 | *g = _mm_unpackhi_epi64(rg03, rg47); |
| 377 | *b = _mm_unpacklo_epi64(bx03, bx47); |
| 378 | } |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 379 | AI static void Store4(void* ptr, const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 380 | // TODO: AVX2 version |
| 381 | __m128i rg0123 = _mm_unpacklo_epi16(r.fVec, g.fVec), // r0 g0 r1 g1 r2 g2 r3 g3 |
| 382 | rg4567 = _mm_unpackhi_epi16(r.fVec, g.fVec), // r4 g4 r5 g5 r6 g6 r7 g7 |
| 383 | ba0123 = _mm_unpacklo_epi16(b.fVec, a.fVec), |
| 384 | ba4567 = _mm_unpackhi_epi16(b.fVec, a.fVec); |
| 385 | |
| 386 | _mm_storeu_si128((__m128i*)ptr + 0, _mm_unpacklo_epi32(rg0123, ba0123)); |
| 387 | _mm_storeu_si128((__m128i*)ptr + 1, _mm_unpackhi_epi32(rg0123, ba0123)); |
| 388 | _mm_storeu_si128((__m128i*)ptr + 2, _mm_unpacklo_epi32(rg4567, ba4567)); |
| 389 | _mm_storeu_si128((__m128i*)ptr + 3, _mm_unpackhi_epi32(rg4567, ba4567)); |
| 390 | } |
| 391 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 392 | AI SkNx operator + (const SkNx& o) const { return _mm_add_epi16(fVec, o.fVec); } |
| 393 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_epi16(fVec, o.fVec); } |
| 394 | AI SkNx operator * (const SkNx& o) const { return _mm_mullo_epi16(fVec, o.fVec); } |
Matt Sarett | 379938e | 2017-01-12 18:34:29 -0500 | [diff] [blame] | 395 | AI SkNx operator & (const SkNx& o) const { return _mm_and_si128(fVec, o.fVec); } |
| 396 | AI SkNx operator | (const SkNx& o) const { return _mm_or_si128(fVec, o.fVec); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 397 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 398 | AI SkNx operator << (int bits) const { return _mm_slli_epi16(fVec, bits); } |
| 399 | AI SkNx operator >> (int bits) const { return _mm_srli_epi16(fVec, bits); } |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 400 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 401 | AI static SkNx Min(const SkNx& a, const SkNx& b) { |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 402 | // No unsigned _mm_min_epu16, so we'll shift into a space where we can use the |
| 403 | // signed version, _mm_min_epi16, then shift back. |
| 404 | const uint16_t top = 0x8000; // Keep this separate from _mm_set1_epi16 or MSVC will whine. |
| 405 | const __m128i top_8x = _mm_set1_epi16(top); |
| 406 | return _mm_add_epi8(top_8x, _mm_min_epi16(_mm_sub_epi8(a.fVec, top_8x), |
| 407 | _mm_sub_epi8(b.fVec, top_8x))); |
| 408 | } |
| 409 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 410 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | 4be181e | 2015-07-14 10:54:19 -0700 | [diff] [blame] | 411 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 412 | _mm_andnot_si128(fVec, e.fVec)); |
| 413 | } |
| 414 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 415 | AI uint16_t operator[](int k) const { |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 416 | SkASSERT(0 <= k && k < 8); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 417 | union { __m128i v; uint16_t us[8]; } pun = {fVec}; |
| 418 | return pun.us[k&7]; |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 419 | } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 420 | |
| 421 | __m128i fVec; |
| 422 | }; |
| 423 | |
| 424 | template <> |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 425 | class SkNx<4, uint8_t> { |
| 426 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 427 | AI SkNx() {} |
| 428 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
| 429 | AI SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d) |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 430 | : fVec(_mm_setr_epi8(a,b,c,d, 0,0,0,0, 0,0,0,0, 0,0,0,0)) {} |
| 431 | |
| 432 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 433 | AI static SkNx Load(const void* ptr) { return _mm_cvtsi32_si128(*(const int*)ptr); } |
| 434 | AI void store(void* ptr) const { *(int*)ptr = _mm_cvtsi128_si32(fVec); } |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 435 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 436 | AI uint8_t operator[](int k) const { |
herb | fd5a260 | 2016-03-01 07:01:23 -0800 | [diff] [blame] | 437 | SkASSERT(0 <= k && k < 4); |
| 438 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 439 | return pun.us[k&3]; |
| 440 | } |
| 441 | |
mtklein | 6f37b4a | 2015-12-14 11:25:18 -0800 | [diff] [blame] | 442 | // TODO as needed |
| 443 | |
| 444 | __m128i fVec; |
| 445 | }; |
| 446 | |
| 447 | template <> |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 448 | class SkNx<16, uint8_t> { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 449 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 450 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 451 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 452 | AI SkNx() {} |
| 453 | AI SkNx(uint8_t val) : fVec(_mm_set1_epi8(val)) {} |
| 454 | AI static SkNx Load(const void* ptr) { return _mm_loadu_si128((const __m128i*)ptr); } |
| 455 | AI SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d, |
| 456 | uint8_t e, uint8_t f, uint8_t g, uint8_t h, |
| 457 | uint8_t i, uint8_t j, uint8_t k, uint8_t l, |
| 458 | uint8_t m, uint8_t n, uint8_t o, uint8_t p) |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 459 | : fVec(_mm_setr_epi8(a,b,c,d, e,f,g,h, i,j,k,l, m,n,o,p)) {} |
| 460 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 461 | AI void store(void* ptr) const { _mm_storeu_si128((__m128i*)ptr, fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 462 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 463 | AI SkNx saturatedAdd(const SkNx& o) const { return _mm_adds_epu8(fVec, o.fVec); } |
mtklein | 6cbf18c | 2015-05-12 15:48:09 -0700 | [diff] [blame] | 464 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 465 | AI SkNx operator + (const SkNx& o) const { return _mm_add_epi8(fVec, o.fVec); } |
| 466 | AI SkNx operator - (const SkNx& o) const { return _mm_sub_epi8(fVec, o.fVec); } |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 467 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 468 | AI static SkNx Min(const SkNx& a, const SkNx& b) { return _mm_min_epu8(a.fVec, b.fVec); } |
| 469 | AI SkNx operator < (const SkNx& o) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 470 | // There's no unsigned _mm_cmplt_epu8, so we flip the sign bits then use a signed compare. |
| 471 | auto flip = _mm_set1_epi8(char(0x80)); |
| 472 | return _mm_cmplt_epi8(_mm_xor_si128(flip, fVec), _mm_xor_si128(flip, o.fVec)); |
| 473 | } |
mtklein | 27e517a | 2015-05-14 17:53:04 -0700 | [diff] [blame] | 474 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 475 | AI uint8_t operator[](int k) const { |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 476 | SkASSERT(0 <= k && k < 16); |
mtklein | e4c0bee | 2016-02-09 10:35:27 -0800 | [diff] [blame] | 477 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 478 | return pun.us[k&15]; |
mtklein | d2ffd36 | 2015-05-12 06:11:21 -0700 | [diff] [blame] | 479 | } |
| 480 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 481 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
mtklein | b5e8611 | 2015-06-24 15:18:39 -0700 | [diff] [blame] | 482 | return _mm_or_si128(_mm_and_si128 (fVec, t.fVec), |
| 483 | _mm_andnot_si128(fVec, e.fVec)); |
| 484 | } |
| 485 | |
mtklein | 115acee | 2015-04-14 14:02:52 -0700 | [diff] [blame] | 486 | __m128i fVec; |
| 487 | }; |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 488 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 489 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_AVX2 |
| 490 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 491 | template <> |
| 492 | class SkNx<8, uint8_t> { |
| 493 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 494 | AI SkNx(const __m128i& vec) : fVec(vec) {} |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 495 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 496 | AI SkNx() {} |
| 497 | AI SkNx(uint8_t v) : fVec(_mm_set1_epi8(v)) {} |
| 498 | AI SkNx(uint8_t a, uint8_t b, uint8_t c, uint8_t d, |
| 499 | uint8_t e, uint8_t f, uint8_t g, uint8_t h) |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 500 | : fVec(_mm_setr_epi8(a,b,c,d, e,f,g,h, 0,0,0,0, 0,0,0,0)) {} |
| 501 | |
| 502 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 503 | AI static SkNx Load(const void* ptr) { return _mm_loadl_epi64((const __m128i*)ptr); } |
| 504 | AI void store(void* ptr) const { _mm_storel_epi64((__m128i*)ptr, fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 505 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 506 | AI uint8_t operator[](int k) const { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 507 | SkASSERT(0 <= k && k < 8); |
| 508 | union { __m128i v; uint8_t us[16]; } pun = {fVec}; |
| 509 | return pun.us[k&7]; |
| 510 | } |
| 511 | |
| 512 | __m128i fVec; |
| 513 | }; |
| 514 | |
| 515 | template <> |
| 516 | class SkNx<8, int32_t> { |
| 517 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 518 | AI SkNx(const __m256i& vec) : fVec(vec) {} |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 519 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 520 | AI SkNx() {} |
| 521 | AI SkNx(int32_t v) : fVec(_mm256_set1_epi32(v)) {} |
| 522 | AI SkNx(int32_t a, int32_t b, int32_t c, int32_t d, |
| 523 | int32_t e, int32_t f, int32_t g, int32_t h) |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 524 | : fVec(_mm256_setr_epi32(a,b,c,d, e,f,g,h)) {} |
| 525 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 526 | AI static SkNx Load(const void* ptr) { return _mm256_loadu_si256((const __m256i*)ptr); } |
| 527 | AI void store(void* ptr) const { _mm256_storeu_si256((__m256i*)ptr, fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 528 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 529 | AI SkNx operator + (const SkNx& o) const { return _mm256_add_epi32(fVec, o.fVec); } |
| 530 | AI SkNx operator - (const SkNx& o) const { return _mm256_sub_epi32(fVec, o.fVec); } |
| 531 | AI SkNx operator * (const SkNx& o) const { return _mm256_mullo_epi32(fVec, o.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 532 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 533 | AI SkNx operator & (const SkNx& o) const { return _mm256_and_si256(fVec, o.fVec); } |
| 534 | AI SkNx operator | (const SkNx& o) const { return _mm256_or_si256(fVec, o.fVec); } |
| 535 | AI SkNx operator ^ (const SkNx& o) const { return _mm256_xor_si256(fVec, o.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 536 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 537 | AI SkNx operator << (int bits) const { return _mm256_slli_epi32(fVec, bits); } |
| 538 | AI SkNx operator >> (int bits) const { return _mm256_srai_epi32(fVec, bits); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 539 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 540 | AI int32_t operator[](int k) const { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 541 | SkASSERT(0 <= k && k < 8); |
| 542 | union { __m256i v; int32_t is[8]; } pun = {fVec}; |
| 543 | return pun.is[k&7]; |
| 544 | } |
| 545 | |
| 546 | __m256i fVec; |
| 547 | }; |
| 548 | |
| 549 | template <> |
| 550 | class SkNx<8, uint32_t> { |
| 551 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 552 | AI SkNx(const __m256i& vec) : fVec(vec) {} |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 553 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 554 | AI SkNx() {} |
| 555 | AI SkNx(uint32_t v) : fVec(_mm256_set1_epi32(v)) {} |
| 556 | AI SkNx(uint32_t a, uint32_t b, uint32_t c, uint32_t d, |
| 557 | uint32_t e, uint32_t f, uint32_t g, uint32_t h) |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 558 | : fVec(_mm256_setr_epi32(a,b,c,d, e,f,g,h)) {} |
| 559 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 560 | AI static SkNx Load(const void* ptr) { return _mm256_loadu_si256((const __m256i*)ptr); } |
| 561 | AI void store(void* ptr) const { _mm256_storeu_si256((__m256i*)ptr, fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 562 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 563 | AI SkNx operator + (const SkNx& o) const { return _mm256_add_epi32(fVec, o.fVec); } |
| 564 | AI SkNx operator - (const SkNx& o) const { return _mm256_sub_epi32(fVec, o.fVec); } |
| 565 | AI SkNx operator * (const SkNx& o) const { return _mm256_mullo_epi32(fVec, o.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 566 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 567 | AI SkNx operator & (const SkNx& o) const { return _mm256_and_si256(fVec, o.fVec); } |
| 568 | AI SkNx operator | (const SkNx& o) const { return _mm256_or_si256(fVec, o.fVec); } |
| 569 | AI SkNx operator ^ (const SkNx& o) const { return _mm256_xor_si256(fVec, o.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 570 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 571 | AI SkNx operator << (int bits) const { return _mm256_slli_epi32(fVec, bits); } |
| 572 | AI SkNx operator >> (int bits) const { return _mm256_srli_epi32(fVec, bits); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 573 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 574 | AI uint32_t operator[](int k) const { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 575 | SkASSERT(0 <= k && k < 8); |
| 576 | union { __m256i v; uint32_t us[8]; } pun = {fVec}; |
| 577 | return pun.us[k&7]; |
| 578 | } |
| 579 | |
| 580 | __m256i fVec; |
| 581 | }; |
| 582 | |
mtklein | a4a4488 | 2016-11-04 13:20:07 -0700 | [diff] [blame] | 583 | // _mm256_unpack{lo,hi}_pd() auto-casting to and from __m256d. |
| 584 | AI static __m256 unpacklo_pd(__m256 x, __m256 y) { |
| 585 | return _mm256_castpd_ps(_mm256_unpacklo_pd(_mm256_castps_pd(x), _mm256_castps_pd(y))); |
| 586 | } |
| 587 | AI static __m256 unpackhi_pd(__m256 x, __m256 y) { |
| 588 | return _mm256_castpd_ps(_mm256_unpackhi_pd(_mm256_castps_pd(x), _mm256_castps_pd(y))); |
| 589 | } |
| 590 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 591 | template <> |
| 592 | class SkNx<8, float> { |
| 593 | public: |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 594 | AI SkNx(const __m256& vec) : fVec(vec) {} |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 595 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 596 | AI SkNx() {} |
| 597 | AI SkNx(float val) : fVec(_mm256_set1_ps(val)) {} |
| 598 | AI SkNx(float a, float b, float c, float d, |
| 599 | float e, float f, float g, float h) : fVec(_mm256_setr_ps(a,b,c,d,e,f,g,h)) {} |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 600 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 601 | AI static SkNx Load(const void* ptr) { return _mm256_loadu_ps((const float*)ptr); } |
| 602 | AI void store(void* ptr) const { _mm256_storeu_ps((float*)ptr, fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 603 | |
mtklein | a4a4488 | 2016-11-04 13:20:07 -0700 | [diff] [blame] | 604 | AI static void Store4(void* ptr, |
| 605 | const SkNx& r, const SkNx& g, const SkNx& b, const SkNx& a) { |
| 606 | __m256 rg0145 = _mm256_unpacklo_ps(r.fVec, g.fVec), // r0 g0 r1 g1 | r4 g4 r5 g5 |
| 607 | rg2367 = _mm256_unpackhi_ps(r.fVec, g.fVec), // r2 ... | r6 ... |
| 608 | ba0145 = _mm256_unpacklo_ps(b.fVec, a.fVec), // b0 a0 b1 a1 | b4 a4 b5 a5 |
| 609 | ba2367 = _mm256_unpackhi_ps(b.fVec, a.fVec); // b2 ... | b6 ... |
| 610 | |
| 611 | __m256 _04 = unpacklo_pd(rg0145, ba0145), // r0 g0 b0 a0 | r4 g4 b4 a4 |
| 612 | _15 = unpackhi_pd(rg0145, ba0145), // r1 ... | r5 ... |
| 613 | _26 = unpacklo_pd(rg2367, ba2367), // r2 ... | r6 ... |
| 614 | _37 = unpackhi_pd(rg2367, ba2367); // r3 ... | r7 ... |
| 615 | |
Matt Sarett | d052042 | 2017-01-06 17:30:55 -0500 | [diff] [blame] | 616 | __m256 _01 = _mm256_permute2f128_ps(_04, _15, 32), // 32 == 0010 0000 == lo, lo |
| 617 | _23 = _mm256_permute2f128_ps(_26, _37, 32), |
| 618 | _45 = _mm256_permute2f128_ps(_04, _15, 49), // 49 == 0011 0001 == hi, hi |
| 619 | _67 = _mm256_permute2f128_ps(_26, _37, 49); |
mtklein | a4a4488 | 2016-11-04 13:20:07 -0700 | [diff] [blame] | 620 | |
| 621 | _mm256_storeu_ps((float*)ptr + 0*8, _01); |
| 622 | _mm256_storeu_ps((float*)ptr + 1*8, _23); |
| 623 | _mm256_storeu_ps((float*)ptr + 2*8, _45); |
| 624 | _mm256_storeu_ps((float*)ptr + 3*8, _67); |
| 625 | } |
| 626 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 627 | AI SkNx operator+(const SkNx& o) const { return _mm256_add_ps(fVec, o.fVec); } |
| 628 | AI SkNx operator-(const SkNx& o) const { return _mm256_sub_ps(fVec, o.fVec); } |
| 629 | AI SkNx operator*(const SkNx& o) const { return _mm256_mul_ps(fVec, o.fVec); } |
| 630 | AI SkNx operator/(const SkNx& o) const { return _mm256_div_ps(fVec, o.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 631 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 632 | AI SkNx operator==(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_EQ_OQ); } |
| 633 | AI SkNx operator!=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_NEQ_OQ); } |
| 634 | AI SkNx operator <(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_LT_OQ); } |
| 635 | AI SkNx operator >(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_GT_OQ); } |
| 636 | AI SkNx operator<=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_LE_OQ); } |
| 637 | AI SkNx operator>=(const SkNx& o) const { return _mm256_cmp_ps(fVec, o.fVec, _CMP_GE_OQ); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 638 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 639 | AI static SkNx Min(const SkNx& l, const SkNx& r) { return _mm256_min_ps(l.fVec, r.fVec); } |
| 640 | AI static SkNx Max(const SkNx& l, const SkNx& r) { return _mm256_max_ps(l.fVec, r.fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 641 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 642 | AI SkNx sqrt() const { return _mm256_sqrt_ps (fVec); } |
| 643 | AI SkNx rsqrt() const { return _mm256_rsqrt_ps(fVec); } |
| 644 | AI SkNx invert() const { return _mm256_rcp_ps (fVec); } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 645 | |
Mike Klein | 2e35e8a | 2016-11-18 15:47:22 -0500 | [diff] [blame] | 646 | AI SkNx abs() const { return _mm256_andnot_ps(_mm256_set1_ps(-0.0f), fVec); } |
Mike Klein | b273fc4 | 2016-11-17 15:42:22 -0500 | [diff] [blame] | 647 | AI SkNx floor() const { return _mm256_floor_ps(fVec); } |
| 648 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 649 | AI float operator[](int k) const { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 650 | SkASSERT(0 <= k && k < 8); |
| 651 | union { __m256 v; float fs[8]; } pun = {fVec}; |
| 652 | return pun.fs[k&7]; |
| 653 | } |
| 654 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 655 | AI SkNx thenElse(const SkNx& t, const SkNx& e) const { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 656 | return _mm256_blendv_ps(e.fVec, t.fVec, fVec); |
| 657 | } |
| 658 | |
| 659 | __m256 fVec; |
| 660 | }; |
| 661 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 662 | AI static void SkNx_split(const Sk8f& v, Sk4f* lo, Sk4f* hi) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 663 | *lo = _mm256_extractf128_ps(v.fVec, 0); |
| 664 | *hi = _mm256_extractf128_ps(v.fVec, 1); |
| 665 | } |
| 666 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 667 | AI static Sk8f SkNx_join(const Sk4f& lo, const Sk4f& hi) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 668 | return _mm256_insertf128_ps(_mm256_castps128_ps256(lo.fVec), hi.fVec, 1); |
| 669 | } |
| 670 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 671 | AI static Sk8f SkNx_fma(const Sk8f& a, const Sk8f& b, const Sk8f& c) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 672 | return _mm256_fmadd_ps(a.fVec, b.fVec, c.fVec); |
| 673 | } |
| 674 | |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 675 | template<> AI /*static*/ Sk8i SkNx_cast<int>(const Sk8b& src) { |
| 676 | return _mm256_cvtepu8_epi32(src.fVec); |
| 677 | } |
| 678 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 679 | template<> AI /*static*/ Sk8f SkNx_cast<float>(const Sk8b& src) { |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 680 | return _mm256_cvtepi32_ps(SkNx_cast<int>(src).fVec); |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 681 | } |
| 682 | |
Matt Sarett | 379938e | 2017-01-12 18:34:29 -0500 | [diff] [blame] | 683 | template<> AI /*static*/ Sk8i SkNx_cast<int>(const Sk8h& src) { |
| 684 | return _mm256_cvtepu16_epi32(src.fVec); |
| 685 | } |
| 686 | |
| 687 | template<> AI /*static*/ Sk8f SkNx_cast<float>(const Sk8h& src) { |
| 688 | return _mm256_cvtepi32_ps(SkNx_cast<int>(src).fVec); |
| 689 | } |
| 690 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 691 | template<> AI /*static*/ Sk8f SkNx_cast<float>(const Sk8i& src) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 692 | return _mm256_cvtepi32_ps(src.fVec); |
| 693 | } |
| 694 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 695 | template<> AI /*static*/ Sk8i SkNx_cast<int>(const Sk8f& src) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 696 | return _mm256_cvttps_epi32(src.fVec); |
| 697 | } |
| 698 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 699 | template<> AI /*static*/ Sk8h SkNx_cast<uint16_t>(const Sk8i& src) { |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 700 | __m128i lo = _mm256_extractf128_si256(src.fVec, 0), |
| 701 | hi = _mm256_extractf128_si256(src.fVec, 1); |
| 702 | return _mm_packus_epi32(lo, hi); |
| 703 | } |
Matt Sarett | 1da27ef | 2017-01-19 17:14:07 -0500 | [diff] [blame^] | 704 | |
| 705 | template<> AI /*static*/ Sk8h SkNx_cast<uint16_t>(const Sk8f& src) { |
| 706 | return SkNx_cast<uint16_t>(SkNx_cast<int>(src)); |
| 707 | } |
| 708 | |
Mike Klein | e71b167 | 2017-01-13 07:59:23 -0500 | [diff] [blame] | 709 | template<> AI /*static*/ Sk8b SkNx_cast<uint8_t>(const Sk8i& src) { |
| 710 | auto _16 = SkNx_cast<uint16_t>(src); |
| 711 | return _mm_packus_epi16(_16.fVec, _16.fVec); |
| 712 | } |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 713 | |
Mike Klein | 04adfda | 2016-10-12 09:52:55 -0400 | [diff] [blame] | 714 | #endif |
| 715 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 716 | template<> AI /*static*/ Sk4f SkNx_cast<float, int32_t>(const Sk4i& src) { |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 717 | return _mm_cvtepi32_ps(src.fVec); |
| 718 | } |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 719 | template<> AI /*static*/ Sk4f SkNx_cast<float, uint32_t>(const Sk4u& src) { |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 720 | return SkNx_cast<float>(Sk4i::Load(&src)); |
| 721 | } |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 722 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 723 | template <> AI /*static*/ Sk4i SkNx_cast<int32_t, float>(const Sk4f& src) { |
mtklein | 0cf795f | 2016-02-17 07:23:36 -0800 | [diff] [blame] | 724 | return _mm_cvttps_epi32(src.fVec); |
| 725 | } |
mtklein | 6c221b4 | 2015-11-20 13:53:19 -0800 | [diff] [blame] | 726 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 727 | template<> AI /*static*/ Sk4h SkNx_cast<uint16_t, int32_t>(const Sk4i& src) { |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 728 | #if 0 && SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE41 |
| 729 | // TODO: This seems to be causing code generation problems. Investigate? |
| 730 | return _mm_packus_epi32(src.fVec); |
| 731 | #elif SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 732 | // With SSSE3, we can just shuffle the low 2 bytes from each lane right into place. |
| 733 | const int _ = ~0; |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 734 | return _mm_shuffle_epi8(src.fVec, _mm_setr_epi8(0,1, 4,5, 8,9, 12,13, _,_,_,_,_,_,_,_)); |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 735 | #else |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 736 | // With SSE2, we have to sign extend our input, making _mm_packs_epi32 do the pack we want. |
| 737 | __m128i x = _mm_srai_epi32(_mm_slli_epi32(src.fVec, 16), 16); |
| 738 | return _mm_packs_epi32(x,x); |
mtklein | 629f25a | 2016-02-08 05:54:38 -0800 | [diff] [blame] | 739 | #endif |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 740 | } |
| 741 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 742 | template<> AI /*static*/ Sk4h SkNx_cast<uint16_t, float>(const Sk4f& src) { |
mtklein | d05a875 | 2016-07-29 10:10:15 -0700 | [diff] [blame] | 743 | return SkNx_cast<uint16_t>(SkNx_cast<int32_t>(src)); |
mtklein | 036e183 | 2016-07-15 07:45:53 -0700 | [diff] [blame] | 744 | } |
| 745 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 746 | template<> AI /*static*/ Sk4b SkNx_cast<uint8_t, float>(const Sk4f& src) { |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 747 | auto _32 = _mm_cvttps_epi32(src.fVec); |
| 748 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 749 | const int _ = ~0; |
| 750 | return _mm_shuffle_epi8(_32, _mm_setr_epi8(0,4,8,12, _,_,_,_, _,_,_,_, _,_,_,_)); |
| 751 | #else |
| 752 | auto _16 = _mm_packus_epi16(_32, _32); |
| 753 | return _mm_packus_epi16(_16, _16); |
| 754 | #endif |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 755 | } |
| 756 | |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 757 | template<> AI /*static*/ Sk4i SkNx_cast<int32_t, uint8_t>(const Sk4b& src) { |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 758 | #if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSSE3 |
| 759 | const int _ = ~0; |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 760 | return _mm_shuffle_epi8(src.fVec, _mm_setr_epi8(0,_,_,_, 1,_,_,_, 2,_,_,_, 3,_,_,_)); |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 761 | #else |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 762 | auto _16 = _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()); |
| 763 | return _mm_unpacklo_epi16(_16, _mm_setzero_si128()); |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 764 | #endif |
Mike Klein | 06a65e2 | 2016-11-17 12:39:09 -0500 | [diff] [blame] | 765 | } |
| 766 | |
| 767 | template<> AI /*static*/ Sk4f SkNx_cast<float, uint8_t>(const Sk4b& src) { |
| 768 | return _mm_cvtepi32_ps(SkNx_cast<int32_t>(src).fVec); |
mtklein | c1eb311 | 2016-02-11 08:10:22 -0800 | [diff] [blame] | 769 | } |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 770 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 771 | template<> AI /*static*/ Sk4f SkNx_cast<float, uint16_t>(const Sk4h& src) { |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 772 | auto _32 = _mm_unpacklo_epi16(src.fVec, _mm_setzero_si128()); |
| 773 | return _mm_cvtepi32_ps(_32); |
mtklein | 2d340f2 | 2016-02-06 19:38:39 -0800 | [diff] [blame] | 774 | } |
| 775 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 776 | template<> AI /*static*/ Sk16b SkNx_cast<uint8_t, float>(const Sk16f& src) { |
mtklein | f8f90e4 | 2016-03-21 10:04:46 -0700 | [diff] [blame] | 777 | Sk8f ab, cd; |
| 778 | SkNx_split(src, &ab, &cd); |
| 779 | |
| 780 | Sk4f a,b,c,d; |
| 781 | SkNx_split(ab, &a, &b); |
| 782 | SkNx_split(cd, &c, &d); |
| 783 | |
| 784 | return _mm_packus_epi16(_mm_packus_epi16(_mm_cvttps_epi32(a.fVec), |
| 785 | _mm_cvttps_epi32(b.fVec)), |
| 786 | _mm_packus_epi16(_mm_cvttps_epi32(c.fVec), |
| 787 | _mm_cvttps_epi32(d.fVec))); |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 788 | } |
| 789 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 790 | template<> AI /*static*/ Sk4h SkNx_cast<uint16_t, uint8_t>(const Sk4b& src) { |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 791 | return _mm_unpacklo_epi8(src.fVec, _mm_setzero_si128()); |
| 792 | } |
| 793 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 794 | template<> AI /*static*/ Sk4b SkNx_cast<uint8_t, uint16_t>(const Sk4h& src) { |
mtklein | 97120a7 | 2016-02-12 14:19:06 -0800 | [diff] [blame] | 795 | return _mm_packus_epi16(src.fVec, src.fVec); |
mtklein | 550e9b0 | 2016-01-20 11:55:51 -0800 | [diff] [blame] | 796 | } |
| 797 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 798 | template<> AI /*static*/ Sk4i SkNx_cast<int32_t, uint16_t>(const Sk4h& src) { |
mtklein | 58e389b | 2016-07-15 07:00:11 -0700 | [diff] [blame] | 799 | return _mm_unpacklo_epi16(src.fVec, _mm_setzero_si128()); |
| 800 | } |
| 801 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 802 | template<> AI /*static*/ Sk4b SkNx_cast<uint8_t, int32_t>(const Sk4i& src) { |
msarett | 7d3ff71 | 2016-07-12 14:55:45 -0700 | [diff] [blame] | 803 | return _mm_packus_epi16(_mm_packus_epi16(src.fVec, src.fVec), src.fVec); |
| 804 | } |
| 805 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 806 | template<> AI /*static*/ Sk4i SkNx_cast<int32_t, uint32_t>(const Sk4u& src) { |
msarett | 15ee3de | 2016-08-02 11:30:30 -0700 | [diff] [blame] | 807 | return src.fVec; |
| 808 | } |
| 809 | |
Mike Klein | 7c78f3a | 2016-10-19 09:21:11 -0400 | [diff] [blame] | 810 | AI static Sk4i Sk4f_round(const Sk4f& x) { |
msarett | 7d3ff71 | 2016-07-12 14:55:45 -0700 | [diff] [blame] | 811 | return _mm_cvtps_epi32(x.fVec); |
| 812 | } |
| 813 | |
Mike Klein | 1e76464 | 2016-10-14 17:09:03 -0400 | [diff] [blame] | 814 | } // namespace |
| 815 | |
mtklein | c9adb05 | 2015-03-30 10:50:27 -0700 | [diff] [blame] | 816 | #endif//SkNx_sse_DEFINED |