ROOT  6.07/01
Reference Guide
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
deinterleave.cpp
Go to the documentation of this file.
1 /* This file is part of the Vc library.
2 
3  Copyright (C) 2010-2011 Matthias Kretz <kretz@kde.org>
4 
5  Vc is free software: you can redistribute it and/or modify
6  it under the terms of the GNU Lesser General Public License as
7  published by the Free Software Foundation, either version 3 of
8  the License, or (at your option) any later version.
9 
10  Vc is distributed in the hope that it will be useful, but
11  WITHOUT ANY WARRANTY; without even the implied warranty of
12  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13  GNU Lesser General Public License for more details.
14 
15  You should have received a copy of the GNU Lesser General Public
16  License along with Vc. If not, see <http://www.gnu.org/licenses/>.
17 
18 */
19 
20 #include "unittest.h"
21 #include <iostream>
22 #include <limits>
23 
24 using namespace Vc;
25 
26 
27 /*
28  * V \ M | float | double | ushort | short | uint | int
29  * ---------+----------------------------------------------
30  * float_v | X | | X | X | |
31  * sfloat_v | X | | X | X | |
32  * double_v | | X | | | |
33  * int_v | | | | X | | X
34  * uint_v | | | X | | X |
35  * short_v | | | | X | |
36  * ushort_v | | | X | | |
37  */
38 template<typename A, typename B> struct TPair { typedef A V; typedef B M; };
39 
40 typedef TPair<float_v, float> float_float;
41 typedef TPair<float_v, unsigned short> float_ushort;
42 typedef TPair<float_v, short> float_short;
43 
44 typedef TPair<sfloat_v, float> sfloat_float;
45 typedef TPair<sfloat_v, unsigned short> sfloat_ushort;
46 typedef TPair<sfloat_v, short> sfloat_short;
47 
48 typedef TPair<double_v, double> double_double;
49 typedef TPair<short_v, short> short_short;
50 typedef TPair<ushort_v, unsigned short> ushort_ushort;
51 
52 typedef TPair<int_v, int> int_int;
53 typedef TPair<int_v, short> int_short;
54 
55 typedef TPair<uint_v, unsigned int> uint_uint;
56 typedef TPair<uint_v, unsigned short> uint_ushort;
57 
58 template<typename Pair> void testDeinterleave()
59 {
60  typedef typename Pair::V V;
61  typedef typename Pair::M M;
62  typedef typename V::IndexType I;
63 
64  const bool isSigned = std::numeric_limits<M>::is_signed;
65 
66  const typename V::EntryType offset = isSigned ? -512 : 0;
67  const V _0246 = static_cast<V>(I::IndexesFromZero()) * 2 + offset;
68 
69  M memory[1024];
70  for (int i = 0; i < 1024; ++i) {
71  memory[i] = static_cast<M>(i + offset);
72  }
73 
74  V a, b;
75 
76  for (int i = 0; i < 1024 - 2 * V::Size; ++i) {
77  // note that a 32 bit integer is certainly enough to decide on alignment...
78  // ... but uintptr_t is C99 but not C++ yet
79  // ... and GCC refuses to do the cast, even if I know what I'm doing
80  if (reinterpret_cast<unsigned long>(&memory[i]) & (VectorAlignment - 1)) {
81  Vc::deinterleave(&a, &b, &memory[i], Unaligned);
82  } else {
83  Vc::deinterleave(&a, &b, &memory[i]);
84  }
85  COMPARE(_0246 + i, a);
86  COMPARE(_0246 + i + 1, b);
87  }
88 }
89 
90 template<typename T, size_t N> struct SomeStruct
91 {
92  T d[N];
93 };
94 
95 template<typename V, size_t StructSize> struct Types
96 {
97  typedef typename V::EntryType T;
98  typedef typename V::IndexType I;
99  typedef typename V::AsArg VArg;
100  typedef typename I::AsArg IArg;
101  typedef SomeStruct<T, StructSize> S;
102  typedef const Vc::InterleavedMemoryWrapper<S, V> &Wrapper;
103 };
104 template<typename V, size_t StructSize, size_t N = StructSize> struct TestDeinterleaveGatherCompare;
105 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 8> {
106  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
107  {
108  V v0, v1, v2, v3, v4, v5, v6, v7;
109  (v0, v1, v2, v3, v4, v5, v6, v7) = data_v[indexes];
110  COMPARE(v0, reference + 0) << "N = 8";
111  COMPARE(v1, reference + 1) << "N = 8";
112  COMPARE(v2, reference + 2) << "N = 8";
113  COMPARE(v3, reference + 3) << "N = 8";
114  COMPARE(v4, reference + 4) << "N = 8";
115  COMPARE(v5, reference + 5) << "N = 8";
116  COMPARE(v6, reference + 6) << "N = 8";
117  COMPARE(v7, reference + 7) << "N = 8";
118  TestDeinterleaveGatherCompare<V, StructSize, 7>::test(data_v, indexes, reference);
119  }
120 };
121 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 7> {
122  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
123  {
124  V v0, v1, v2, v3, v4, v5, v6;
125  (v0, v1, v2, v3, v4, v5, v6) = data_v[indexes];
126  COMPARE(v0, reference + 0) << "N = 7";
127  COMPARE(v1, reference + 1) << "N = 7";
128  COMPARE(v2, reference + 2) << "N = 7";
129  COMPARE(v3, reference + 3) << "N = 7";
130  COMPARE(v4, reference + 4) << "N = 7";
131  COMPARE(v5, reference + 5) << "N = 7";
132  COMPARE(v6, reference + 6) << "N = 7";
133  TestDeinterleaveGatherCompare<V, StructSize, 6>::test(data_v, indexes, reference);
134  }
135 };
136 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 6> {
137  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
138  {
139  V v0, v1, v2, v3, v4, v5;
140  (v0, v1, v2, v3, v4, v5) = data_v[indexes];
141  COMPARE(v0, reference + 0) << "N = 6";
142  COMPARE(v1, reference + 1) << "N = 6";
143  COMPARE(v2, reference + 2) << "N = 6";
144  COMPARE(v3, reference + 3) << "N = 6";
145  COMPARE(v4, reference + 4) << "N = 6";
146  COMPARE(v5, reference + 5) << "N = 6";
147  TestDeinterleaveGatherCompare<V, StructSize, 5>::test(data_v, indexes, reference);
148  }
149 };
150 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 5> {
151  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
152  {
153  V v0, v1, v2, v3, v4;
154  (v0, v1, v2, v3, v4) = data_v[indexes];
155  COMPARE(v0, reference + 0) << "N = 5";
156  COMPARE(v1, reference + 1) << "N = 5";
157  COMPARE(v2, reference + 2) << "N = 5";
158  COMPARE(v3, reference + 3) << "N = 5";
159  COMPARE(v4, reference + 4) << "N = 5";
160  TestDeinterleaveGatherCompare<V, StructSize, 4>::test(data_v, indexes, reference);
161  }
162 };
163 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 4> {
164  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
165  {
166  V a, b, c, d;
167  (a, b, c, d) = data_v[indexes];
168  COMPARE(a, reference + 0) << "N = 4";
169  COMPARE(b, reference + 1) << "N = 4";
170  COMPARE(c, reference + 2) << "N = 4";
171  COMPARE(d, reference + 3) << "N = 4";
172  TestDeinterleaveGatherCompare<V, StructSize, 3>::test(data_v, indexes, reference);
173  }
174 };
175 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 3> {
176  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
177  {
178  V a, b, c;
179  (a, b, c) = data_v[indexes];
180  COMPARE(a, reference + 0) << "N = 3";
181  COMPARE(b, reference + 1) << "N = 3";
182  COMPARE(c, reference + 2) << "N = 3";
183  TestDeinterleaveGatherCompare<V, StructSize, 2>::test(data_v, indexes, reference);
184  }
185 };
186 template<typename V, size_t StructSize> struct TestDeinterleaveGatherCompare<V, StructSize, 2> {
187  static void test(typename Types<V, StructSize>::Wrapper data_v, typename Types<V, StructSize>::IArg indexes, const typename V::AsArg reference)
188  {
189  V a, b;
190  (a, b) = data_v[indexes];
191  COMPARE(a, reference + 0) << "N = 2";
192  COMPARE(b, reference + 1) << "N = 2";
193  }
194 };
195 
196 size_t createNMask(size_t N)
197 {
198  size_t NMask = (N >> 1) | (N >> 2);
199  for (size_t shift = 2; shift < sizeof(size_t) * 8; shift *= 2) {
200  NMask |= NMask >> shift;
201  }
202  return NMask;
203 }
204 
205 template<typename V, size_t StructSize> void testDeinterleaveGatherImpl()
206 {
207  typedef typename V::EntryType T;
208  typedef typename V::IndexType I;
209  typedef SomeStruct<T, StructSize> S;
210  typedef Vc::InterleavedMemoryWrapper<S, V> Wrapper;
211  const size_t N = std::min<size_t>(std::numeric_limits<typename I::EntryType>::max(), 1024 * 1024 / sizeof(S));
212  const size_t NMask = createNMask(N);
213 
214  S *data = Vc::malloc<S, Vc::AlignOnVector>(N);
215  for (size_t i = 0; i < N; ++i) {
216  for (size_t j = 0; j < StructSize; ++j) {
217  data[i].d[j] = i * StructSize + j;
218  }
219  }
220  const Wrapper data_v(data);
221 
222  for (int retest = 0; retest < 10000; ++retest) {
223  I indexes = (I::Random() >> 10) & I(NMask);
224  VERIFY(indexes >= 0);
225  VERIFY(indexes < N);
226  const V reference = static_cast<V>(indexes) * V(StructSize);
227 
228  TestDeinterleaveGatherCompare<V, StructSize>::test(data_v, indexes, reference);
229  }
230 }
231 
232 template<typename V> void testDeinterleaveGather()
233 {
234  testDeinterleaveGatherImpl<V, 2>();
235  testDeinterleaveGatherImpl<V, 3>();
236  testDeinterleaveGatherImpl<V, 4>();
237  testDeinterleaveGatherImpl<V, 5>();
238  testDeinterleaveGatherImpl<V, 6>();
239  testDeinterleaveGatherImpl<V, 7>();
240  testDeinterleaveGatherImpl<V, 8>();
241 }
242 
243 template<typename V, size_t StructSize> struct TestInterleavingScatterCompare;
244 #define _IMPL(STRUCTSIZE, _code_) \
245 template<typename V> struct TestInterleavingScatterCompare<V, STRUCTSIZE> { \
246  typedef TestInterleavingScatterCompare<V, STRUCTSIZE - 1> NextTest; \
247  template<typename Wrapper> static void test(Wrapper &data, const typename V::IndexType &i) { \
248  _code_ \
249  } \
250 }
251 _IMPL(2,
252  const V v0 = V::Random();
253  const V v1 = V::Random();
254  V t0;
255  V t1;
256  data[i] = (v0, v1);
257  (t0, t1) = data[i];
258  COMPARE(t0, v0) << 2;
259  COMPARE(t1, v1) << 2;
260  );
261 _IMPL(3,
262  const V v0 = V::Random();
263  const V v1 = V::Random();
264  const V v2 = V::Random();
265  V t0; V t1; V t2;
266  data[i] = (v0, v1, v2);
267  (t0, t1, t2) = data[i];
268  COMPARE(t0, v0) << 3;
269  COMPARE(t1, v1) << 3;
270  COMPARE(t2, v2) << 3;
271  NextTest::test(data, i);
272  );
273 _IMPL(4,
274  const V v0 = V::Random();
275  const V v1 = V::Random();
276  const V v2 = V::Random();
277  const V v3 = V::Random();
278  V t0; V t1; V t2; V t3;
279  data[i] = (v0, v1, v2, v3);
280  (t0, t1, t2, t3) = data[i];
281  COMPARE(t0, v0) << 4;
282  COMPARE(t1, v1) << 4;
283  COMPARE(t2, v2) << 4;
284  COMPARE(t3, v3) << 4;
285  NextTest::test(data, i);
286  );
287 _IMPL(5,
288  const V v0 = V::Random();
289  const V v1 = V::Random();
290  const V v2 = V::Random();
291  const V v3 = V::Random();
292  const V v4 = V::Random();
293  V t0; V t1; V t2; V t3; V t4;
294  data[i] = (v0, v1, v2, v3, v4);
295  (t0, t1, t2, t3, t4) = data[i];
296  COMPARE(t0, v0) << 5;
297  COMPARE(t1, v1) << 5;
298  COMPARE(t2, v2) << 5;
299  COMPARE(t3, v3) << 5;
300  COMPARE(t4, v4) << 5;
301  NextTest::test(data, i);
302  );
303 _IMPL(6,
304  const V v0 = V::Random();
305  const V v1 = V::Random();
306  const V v2 = V::Random();
307  const V v3 = V::Random();
308  const V v4 = V::Random();
309  const V v5 = V::Random();
310  V t0; V t1; V t2; V t3; V t4; V t5;
311  data[i] = (v0, v1, v2, v3, v4, v5);
312  (t0, t1, t2, t3, t4, t5) = data[i];
313  COMPARE(t0, v0) << 6;
314  COMPARE(t1, v1) << 6;
315  COMPARE(t2, v2) << 6;
316  COMPARE(t3, v3) << 6;
317  COMPARE(t4, v4) << 6;
318  COMPARE(t5, v5) << 6;
319  NextTest::test(data, i);
320  );
321 _IMPL(7,
322  const V v0 = V::Random();
323  const V v1 = V::Random();
324  const V v2 = V::Random();
325  const V v3 = V::Random();
326  const V v4 = V::Random();
327  const V v5 = V::Random();
328  const V v6 = V::Random();
329  V t0; V t1; V t2; V t3; V t4; V t5; V t6;
330  data[i] = (v0, v1, v2, v3, v4, v5, v6);
331  (t0, t1, t2, t3, t4, t5, t6) = data[i];
332  COMPARE(t0, v0) << 7;
333  COMPARE(t1, v1) << 7;
334  COMPARE(t2, v2) << 7;
335  COMPARE(t3, v3) << 7;
336  COMPARE(t4, v4) << 7;
337  COMPARE(t5, v5) << 7;
338  COMPARE(t6, v6) << 7;
339  NextTest::test(data, i);
340  );
341 _IMPL(8,
342  const V v0 = V::Random();
343  const V v1 = V::Random();
344  const V v2 = V::Random();
345  const V v3 = V::Random();
346  const V v4 = V::Random();
347  const V v5 = V::Random();
348  const V v6 = V::Random();
349  const V v7 = V::Random();
350  V t0; V t1; V t2; V t3; V t4; V t5; V t6; V t7;
351  data[i] = (v0, v1, v2, v3, v4, v5, v6, v7);
352  (t0, t1, t2, t3, t4, t5, t6, t7) = data[i];
353  COMPARE(t0, v0) << 8;
354  COMPARE(t1, v1) << 8;
355  COMPARE(t2, v2) << 8;
356  COMPARE(t3, v3) << 8;
357  COMPARE(t4, v4) << 8;
358  COMPARE(t5, v5) << 8;
359  COMPARE(t6, v6) << 8;
360  COMPARE(t7, v7) << 8;
361  NextTest::test(data, i);
362  );
363 
364 template<typename V, size_t StructSize> void testInterleavingScatterImpl()
365 {
366  typedef typename V::EntryType T;
367  typedef typename V::IndexType I;
368  typedef SomeStruct<T, StructSize> S;
369  typedef Vc::InterleavedMemoryWrapper<S, V> Wrapper;
370  const size_t N = std::min<size_t>(std::numeric_limits<typename I::EntryType>::max(), 1024 * 1024 / sizeof(S));
371  const size_t NMask = createNMask(N);
372 
373  S *data = Vc::malloc<S, Vc::AlignOnVector>(N);
374  std::memset(data, 0, sizeof(S) * N);
375  Wrapper data_v(data);
376 
377  for (int retest = 0; retest < 10000; ++retest) {
378  I indexes = (I::Random() >> 10) & I(NMask);
379  if (I::Size != 1) {
380  // ensure the indexes are unique
381  while(!(indexes.sorted() == indexes.sorted().rotated(1)).isEmpty()) {
382  indexes = (I::Random() >> 10) & I(NMask);
383  }
384  }
385  VERIFY(indexes >= 0);
386  VERIFY(indexes < N);
387 
388  TestInterleavingScatterCompare<V, StructSize>::test(data_v, indexes);
389  }
390 }
391 
392 template<typename V> void testInterleavingScatter()
393 {
394  testInterleavingScatterImpl<V, 2>();
395  testInterleavingScatterImpl<V, 3>();
396  testInterleavingScatterImpl<V, 4>();
397  testInterleavingScatterImpl<V, 5>();
398  testInterleavingScatterImpl<V, 6>();
399  testInterleavingScatterImpl<V, 7>();
400  testInterleavingScatterImpl<V, 8>();
401 }
402 
403 int main()
404 {
405  runTest(testDeinterleave<float_float>);
406  runTest(testDeinterleave<float_ushort>);
407  runTest(testDeinterleave<float_short>);
408  runTest(testDeinterleave<sfloat_float>);
409  runTest(testDeinterleave<sfloat_ushort>);
410  runTest(testDeinterleave<sfloat_short>);
411  runTest(testDeinterleave<double_double>);
412  runTest(testDeinterleave<int_int>);
413  runTest(testDeinterleave<int_short>);
414  runTest(testDeinterleave<uint_uint>);
415  runTest(testDeinterleave<uint_ushort>);
416  runTest(testDeinterleave<short_short>);
417  runTest(testDeinterleave<ushort_ushort>);
418 
419  testAllTypes(testDeinterleaveGather);
420  testAllTypes(testInterleavingScatter);
421 }
ROOT::Math::Cephes::B
static double B[]
Definition: SpecFuncCephes.cxx:178
ROOT::Vc::deinterleave
Vc_ALWAYS_INLINE void deinterleave(V *a, V *b, const M *memory, A align)
Loads two vectors of values from an interleaved array.
Definition: deinterleave.h:69
float_short
TPair< float_v, short > float_short
Definition: deinterleave.cpp:42
_IMPL
#define _IMPL(STRUCTSIZE, _code_)
Definition: deinterleave.cpp:244
v1
const Double_t * v1
Definition: TArcBall.cxx:33
testDeinterleave
void testDeinterleave()
Definition: deinterleave.cpp:58
c
return c
Definition: entrylist_figure1.C:47
tree.offset
tuple offset
Definition: tree.py:93
xmlio::Size
const char * Size
Definition: TXMLSetup.cxx:56
testDeinterleaveGather
void testDeinterleaveGather()
Definition: deinterleave.cpp:232
ushort_ushort
TPair< ushort_v, unsigned short > ushort_ushort
Definition: deinterleave.cpp:50
sfloat_short
TPair< sfloat_v, short > sfloat_short
Definition: deinterleave.cpp:46
t5
TText * t5
Definition: rootenv.C:53
N
#define N
t3
TText * t3
Definition: rootenv.C:37
a
TArc * a
Definition: textangle.C:12
uint_ushort
TPair< uint_v, unsigned short > uint_ushort
Definition: deinterleave.cpp:56
size_t
size_t
Definition: TBuffer.cxx:28
t1
TLatex * t1
Definition: textangle.C:20
ROOT::Math::Cephes::A
static double A[]
Definition: SpecFuncCephes.cxx:170
float_ushort
TPair< float_v, unsigned short > float_ushort
Definition: deinterleave.cpp:41
T
TTree * T
Definition: memstatExample.C:53
uint_uint
TPair< uint_v, unsigned int > uint_uint
Definition: deinterleave.cpp:55
COMPARE
#define COMPARE(a, b)
Definition: unittest.h:509
tornado.d
int d
Definition: tornado.py:11
main
int main()
Definition: deinterleave.cpp:403
testAllTypes
#define testAllTypes(name)
Definition: unittest.h:43
int_int
TPair< int_v, int > int_int
Definition: deinterleave.cpp:52
testInterleavingScatterImpl
void testInterleavingScatterImpl()
Definition: deinterleave.cpp:364
sfloat_ushort
TPair< sfloat_v, unsigned short > sfloat_ushort
Definition: deinterleave.cpp:45
t4
TText * t4
Definition: rootenv.C:45
int_short
TPair< int_v, short > int_short
Definition: deinterleave.cpp:53
testInterleavingScatter
void testInterleavingScatter()
Definition: deinterleave.cpp:392
double_double
TPair< double_v, double > double_double
Definition: deinterleave.cpp:48
VERIFY
#define VERIFY(cond)
Definition: unittest.h:515
t2
TText * t2
Definition: rootenv.C:28
testDeinterleaveGatherImpl
void testDeinterleaveGatherImpl()
Definition: deinterleave.cpp:205
TPair
Class used by TMap to store (key,value) pairs.
Definition: TMap.h:106
Random
void Random()
Definition: utils.cpp:154
S
static const float S
Definition: mandel.cpp:113
t7
TText * t7
Definition: rootenv.C:69
ROOT::Vc::AVX::max
static Vc_ALWAYS_INLINE int_v max(const int_v &x, const int_v &y)
Definition: vector.h:440
float_float
TPair< float_v, float > float_float
Definition: deinterleave.cpp:40
t6
TText * t6
Definition: rootenv.C:62
createNMask
size_t createNMask(size_t N)
Definition: deinterleave.cpp:196
I
#define I(x, y, z)
short_short
TPair< short_v, short > short_short
Definition: deinterleave.cpp:49
sfloat_float
TPair< sfloat_v, float > sfloat_float
Definition: deinterleave.cpp:44
runTest
#define runTest(name)
Definition: unittest.h:42