12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576 |
- #ifndef BOOST_COMPUTE_UTILITY_DIM_HPP
- #define BOOST_COMPUTE_UTILITY_DIM_HPP
- #include <boost/compute/config.hpp>
- #include <boost/compute/utility/extents.hpp>
- namespace boost {
- namespace compute {
- #ifndef BOOST_COMPUTE_NO_VARIADIC_TEMPLATES
- template<class... Args>
- inline extents<sizeof...(Args)> dim(Args... args)
- {
- return extents<sizeof...(Args)>({ static_cast<size_t>(args)... });
- }
- #if BOOST_WORKAROUND(BOOST_MSVC, <= 1800)
- template<class T>
- inline extents<1> dim(T arg)
- {
- return extents<1>(static_cast<size_t>(arg));
- }
- #endif
- #else
- #define BOOST_COMPUTE_DETAIL_ASSIGN_DIM(z, n, var) \
- var[n] = BOOST_PP_CAT(e, n);
- #define BOOST_COMPUTE_DETAIL_DEFINE_DIM(z, n, var) \
- inline extents<n> dim(BOOST_PP_ENUM_PARAMS(n, size_t e)) \
- { \
- extents<n> exts; \
- BOOST_PP_REPEAT(n, BOOST_COMPUTE_DETAIL_ASSIGN_DIM, exts) \
- return exts; \
- }
- BOOST_PP_REPEAT(BOOST_COMPUTE_MAX_ARITY, BOOST_COMPUTE_DETAIL_DEFINE_DIM, ~)
- #undef BOOST_COMPUTE_DETAIL_ASSIGN_DIM
- #undef BOOST_COMPUTE_DETAIL_DEFINE_DIM
- #endif
- template<size_t N>
- inline extents<N> dim()
- {
- return extents<N>();
- }
- }
- }
- #endif
|