Merge remote-tracking branch 'origin/master' into pipe-video
[mesa.git] / src / gallium / auxiliary / util / u_sse.h
1 /**************************************************************************
2 *
3 * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28 /**
29 * @file
30 * SSE intrinsics portability header.
31 *
32 * Although the SSE intrinsics are support by all modern x86 and x86-64
33 * compilers, there are some intrisincs missing in some implementations
34 * (especially older MSVC versions). This header abstracts that away.
35 */
36
37 #ifndef U_SSE_H_
38 #define U_SSE_H_
39
40 #include "pipe/p_config.h"
41
42 #if defined(PIPE_ARCH_SSE)
43
44 #include <emmintrin.h>
45
46
47 /* MSVC before VC8 does not support the _mm_castxxx_yyy */
48 #if defined(_MSC_VER) && _MSC_VER < 1500
49
50 union __declspec(align(16)) m128_types {
51 __m128 m128;
52 __m128i m128i;
53 __m128d m128d;
54 };
55
56 static __inline __m128
57 _mm_castsi128_ps(__m128i a)
58 {
59 union m128_types u;
60 u.m128i = a;
61 return u.m128;
62 }
63
64 static __inline __m128i
65 _mm_castps_si128(__m128 a)
66 {
67 union m128_types u;
68 u.m128 = a;
69 return u.m128i;
70 }
71
72 #endif /* defined(_MSC_VER) && _MSC_VER < 1500 */
73
74 union m128i {
75 __m128i m;
76 ubyte ub[16];
77 ushort us[8];
78 uint ui[4];
79 };
80
81 static INLINE void u_print_epi8(const char *name, __m128i r)
82 {
83 union { __m128i m; ubyte ub[16]; } u;
84 u.m = r;
85
86 debug_printf("%s: "
87 "%02x/"
88 "%02x/"
89 "%02x/"
90 "%02x/"
91 "%02x/"
92 "%02x/"
93 "%02x/"
94 "%02x/"
95 "%02x/"
96 "%02x/"
97 "%02x/"
98 "%02x/"
99 "%02x/"
100 "%02x/"
101 "%02x/"
102 "%02x\n",
103 name,
104 u.ub[0], u.ub[1], u.ub[2], u.ub[3],
105 u.ub[4], u.ub[5], u.ub[6], u.ub[7],
106 u.ub[8], u.ub[9], u.ub[10], u.ub[11],
107 u.ub[12], u.ub[13], u.ub[14], u.ub[15]);
108 }
109
110 static INLINE void u_print_epi16(const char *name, __m128i r)
111 {
112 union { __m128i m; ushort us[8]; } u;
113 u.m = r;
114
115 debug_printf("%s: "
116 "%04x/"
117 "%04x/"
118 "%04x/"
119 "%04x/"
120 "%04x/"
121 "%04x/"
122 "%04x/"
123 "%04x\n",
124 name,
125 u.us[0], u.us[1], u.us[2], u.us[3],
126 u.us[4], u.us[5], u.us[6], u.us[7]);
127 }
128
129 static INLINE void u_print_epi32(const char *name, __m128i r)
130 {
131 union { __m128i m; uint ui[4]; } u;
132 u.m = r;
133
134 debug_printf("%s: "
135 "%08x/"
136 "%08x/"
137 "%08x/"
138 "%08x\n",
139 name,
140 u.ui[0], u.ui[1], u.ui[2], u.ui[3]);
141 }
142
143 static INLINE void u_print_ps(const char *name, __m128 r)
144 {
145 union { __m128 m; float f[4]; } u;
146 u.m = r;
147
148 debug_printf("%s: "
149 "%f/"
150 "%f/"
151 "%f/"
152 "%f\n",
153 name,
154 u.f[0], u.f[1], u.f[2], u.f[3]);
155 }
156
157
158 #define U_DUMP_EPI32(a) u_print_epi32(#a, a)
159 #define U_DUMP_EPI16(a) u_print_epi16(#a, a)
160 #define U_DUMP_EPI8(a) u_print_epi8(#a, a)
161 #define U_DUMP_PS(a) u_print_ps(#a, a)
162
163
164
165 #if defined(PIPE_ARCH_SSSE3)
166
167 #include <tmmintrin.h>
168
169 #else /* !PIPE_ARCH_SSSE3 */
170
171 /**
172 * Describe _mm_shuffle_epi8() with gcc extended inline assembly, for cases
173 * where -mssse3 is not supported/enabled.
174 *
175 * MSVC will never get in here as its intrinsics support do not rely on
176 * compiler command line options.
177 */
178 static __inline __m128i __attribute__((__gnu_inline__, __always_inline__, __artificial__))
179 _mm_shuffle_epi8(__m128i a, __m128i mask)
180 {
181 __m128i result;
182 __asm__("pshufb %1, %0"
183 : "=x" (result)
184 : "xm" (mask), "0" (a));
185 return result;
186 }
187
188 #endif /* !PIPE_ARCH_SSSE3 */
189
190
191
192
193 /* Provide an SSE2 implementation of _mm_mullo_epi32() in terms of
194 * _mm_mul_epu32().
195 *
196 * I suspect this works fine for us because one of our operands is
197 * always positive, but not sure that this can be used for general
198 * signed integer multiplication.
199 *
200 * This seems close enough to the speed of SSE4 and the real
201 * _mm_mullo_epi32() intrinsic as to not justify adding an sse4
202 * dependency at this point.
203 */
204 static INLINE __m128i mm_mullo_epi32(const __m128i a, const __m128i b)
205 {
206 __m128i a4 = _mm_srli_epi64(a, 32); /* shift by one dword */
207 __m128i b4 = _mm_srli_epi64(b, 32); /* shift by one dword */
208 __m128i ba = _mm_mul_epu32(b, a); /* multply dwords 0, 2 */
209 __m128i b4a4 = _mm_mul_epu32(b4, a4); /* multiply dwords 1, 3 */
210
211 /* Interleave the results, either with shuffles or (slightly
212 * faster) direct bit operations:
213 */
214 #if 0
215 __m128i ba8 = _mm_shuffle_epi32(ba, 8);
216 __m128i b4a48 = _mm_shuffle_epi32(b4a4, 8);
217 __m128i result = _mm_unpacklo_epi32(ba8, b4a48);
218 #else
219 __m128i mask = _mm_setr_epi32(~0,0,~0,0);
220 __m128i ba_mask = _mm_and_si128(ba, mask);
221 __m128i b4a4_mask_shift = _mm_slli_epi64(b4a4, 32);
222 __m128i result = _mm_or_si128(ba_mask, b4a4_mask_shift);
223 #endif
224
225 return result;
226 }
227
228
229 static INLINE void
230 transpose4_epi32(const __m128i * restrict a,
231 const __m128i * restrict b,
232 const __m128i * restrict c,
233 const __m128i * restrict d,
234 __m128i * restrict o,
235 __m128i * restrict p,
236 __m128i * restrict q,
237 __m128i * restrict r)
238 {
239 __m128i t0 = _mm_unpacklo_epi32(*a, *b);
240 __m128i t1 = _mm_unpacklo_epi32(*c, *d);
241 __m128i t2 = _mm_unpackhi_epi32(*a, *b);
242 __m128i t3 = _mm_unpackhi_epi32(*c, *d);
243
244 *o = _mm_unpacklo_epi64(t0, t1);
245 *p = _mm_unpackhi_epi64(t0, t1);
246 *q = _mm_unpacklo_epi64(t2, t3);
247 *r = _mm_unpackhi_epi64(t2, t3);
248 }
249
250 #define SCALAR_EPI32(m, i) _mm_shuffle_epi32((m), _MM_SHUFFLE(i,i,i,i))
251
252
253 #endif /* PIPE_ARCH_SSE */
254
255 #endif /* U_SSE_H_ */