]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | //---------------------------------------------------------------------------// |
2 | // Copyright (c) 2013 Kyle Lutz <kyle.r.lutz@gmail.com> | |
3 | // | |
4 | // Distributed under the Boost Software License, Version 1.0 | |
5 | // See accompanying file LICENSE_1_0.txt or copy at | |
6 | // http://www.boost.org/LICENSE_1_0.txt | |
7 | // | |
8 | // See http://boostorg.github.com/compute for more information. | |
9 | //---------------------------------------------------------------------------// | |
10 | ||
11 | #ifndef BOOST_COMPUTE_TYPE_TRAITS_VECTOR_SIZE_HPP | |
12 | #define BOOST_COMPUTE_TYPE_TRAITS_VECTOR_SIZE_HPP | |
13 | ||
14 | #include <boost/preprocessor/cat.hpp> | |
15 | ||
16 | #include <boost/compute/types/fundamental.hpp> | |
17 | ||
18 | namespace boost { | |
19 | namespace compute { | |
20 | ||
21 | /// Meta-function returning the size (number of components) of a vector type | |
22 | /// \p T. For scalar types this function returns \c 1. | |
23 | /// | |
24 | /// For example, | |
25 | /// \code | |
26 | /// vector_size<float>::value == 1 | |
27 | /// vector_size<float4_>::value == 4 | |
28 | /// \endcode | |
29 | template<class T> | |
30 | struct vector_size | |
31 | { | |
32 | /// \internal_ | |
33 | BOOST_STATIC_CONSTANT(size_t, value = 1); | |
34 | }; | |
35 | ||
36 | /// \internal_ | |
37 | #define BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTION(scalar, size) \ | |
38 | template<> \ | |
39 | struct vector_size<BOOST_PP_CAT(BOOST_PP_CAT(scalar, size), _)> \ | |
40 | { \ | |
41 | BOOST_STATIC_CONSTANT(size_t, value = size); \ | |
42 | }; | |
43 | ||
44 | /// \internal_ | |
45 | #define BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(scalar) \ | |
46 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTION(scalar, 2) \ | |
47 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTION(scalar, 4) \ | |
48 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTION(scalar, 8) \ | |
49 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTION(scalar, 16) | |
50 | ||
51 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(char) | |
52 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(uchar) | |
53 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(short) | |
54 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(ushort) | |
55 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(int) | |
56 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(uint) | |
57 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(long) | |
58 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(ulong) | |
59 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(float) | |
60 | BOOST_COMPUTE_DECLARE_VECTOR_SIZE_FUNCTIONS(double) | |
61 | ||
62 | } // end compute namespace | |
63 | } // end boost namespace | |
64 | ||
65 | #endif // BOOST_COMPUTE_TYPE_TRAITS_VECTOR_SIZE_HPP |