create_vmpl_sequence.hpp

/*
 * mpl_sequence.hpp
 *
 * Set of classes to create an mpl::vector with a filled sequence
 *
 *  Created on: Aug 27, 2014
 *      Author: Pietro Incardona
 */
 
#ifndef MPL_SEQUENCE_HPP
#define MPL_SEQUENCE_HPP
 
#include <boost/mpl/int.hpp>
#include <boost/mpl/reverse.hpp>
 
template<int ...> struct index_tuple_sq{};
 
template<int c,int end>
struct exit_impl_sq : boost::mpl::equal_to<boost::mpl::int_<c>,boost::mpl::int_<end>>
{};
 
template<int c , int end, bool exit,int ... vars>
struct to_variadic_impl
{
   typedef typename exit_impl_sq<c,end>::type exit_;
   typedef typename to_variadic_impl<c+1,end,exit_::value,vars ..., c>::type type;
};
 
 
template<int c, int end, int ... vars>
struct to_variadic_impl<c,end,true, vars ...>
{
   typedef index_tuple_sq<vars ...> type;
};
 
template<unsigned int N, unsigned int M>
struct to_int_sequence
{
    typedef typename exit_impl_sq<N,M>::type exit_;
 
    typedef typename to_variadic_impl<N+1,M,exit_::value,N>::type type;
};
 
 
 
template<int c , int end, int ele, bool exit,int ... vars>
struct to_variadic_const_impl
{
   typedef typename exit_impl_sq<c,end-1>::type exit_;
   typedef typename to_variadic_const_impl<c+1,end,ele,exit_::value,vars ..., ele>::type type;
};
 
 
template<int c, int end, int ele, int ... vars>
struct to_variadic_const_impl<c,end,ele,true, vars ...>
{
   typedef boost::mpl::vector<boost::mpl::int_<vars> ...> type;
};
 
template<unsigned int N, unsigned int M>
struct vmpl_create_constant
{
    typedef typename exit_impl_sq<N,M>::type exit_;
 
    typedef typename to_variadic_const_impl<1,N,M,exit_::value,M>::type type;
};
 
 
template<typename bint, unsigned int ele>
struct sum_ele
{
};
 
template<int op1, unsigned int ele>
struct sum_ele<boost::mpl::int_<op1>,ele>
{
    typedef boost::mpl::int_<op1+ele> type;
};
 
template<unsigned int ele>
struct sum_ele<boost::mpl::na,ele>
{
    typedef boost::mpl::na type;
};
 
 
template <unsigned int ele, typename ... vmpl>
struct vmpl_sum_constant_impl
{
    typedef boost::mpl::vector<typename sum_ele<vmpl,ele>::type ...> type;
};
 
template <unsigned int ele, typename vmpl>
struct vmpl_sum_constant
{
    typedef int type;
};
template <unsigned int ele, typename ... vars>
struct vmpl_sum_constant<ele, boost::mpl::vector<vars ... > >
{
    typedef boost::mpl::vector<typename sum_ele<vars,ele>::type ...> type;
};
 
 
 
 
template<int c, int accu, int stop, typename vmpl, bool exit>
struct vmpl_reduce_prod_impl
{
   typedef typename exit_impl_sq<c,stop>::type exit_;
   typedef typename boost::mpl::at<vmpl,boost::mpl::int_<c>>::type ele;
   typedef typename vmpl_reduce_prod_impl<c+1,accu*ele::value,stop,vmpl,exit_::value>::type type;
};
 
 
template<int c, int accu, int stop, typename vmpl>
struct vmpl_reduce_prod_impl<c,accu,stop,vmpl,true>
{
   typedef boost::mpl::int_<accu> type;
};
 
 
template<typename vmpl>
struct vmpl_reduce_prod
{
    typedef typename vmpl_reduce_prod_impl<0,1,(int)vmpl::size::value-1,vmpl,false>::type type;
};
 
template<typename vmpl, int stop>
struct vmpl_reduce_prod_stop
{
    typedef typename vmpl_reduce_prod_impl<0,1,stop,vmpl,false>::type type;
};
 
template<typename vmpl>
struct vmpl_reduce_prod_stop<vmpl,-1>
{
    typedef typename boost::mpl::int_<1> type;
};
 
template<typename vmpl, typename a> __device__ __host__ inline unsigned int Lin_vmpl(a v)
{
    return v*vmpl_reduce_prod_stop<vmpl,((int)vmpl::size::value) - 2>::type::value;
}
 
template<typename vmpl, typename a, typename ...lT>
__device__ __host__ inline unsigned int Lin_vmpl(a v,lT...t)
{
        return v*vmpl_reduce_prod_stop<vmpl,(int)vmpl::size::value - (int)sizeof...(t) - 2>::type::value + Lin_vmpl<vmpl>(t...);
}
 
template<typename vmpl, typename vmpl_off, typename a> __device__ __host__ inline unsigned int Lin_vmpl_off(a v)
{
    return (v + boost::mpl::at<vmpl_off,boost::mpl::int_< ((int)vmpl::size::value) - 1 > >::type::value )*vmpl_reduce_prod_stop<vmpl,(int)vmpl::size::value - 2>::type::value;
}
 
template<typename vmpl, typename vmpl_off, typename a, typename ...lT>
__device__ __host__ inline unsigned int Lin_vmpl_off(a v,lT...t)
{
    return (v + boost::mpl::at<vmpl_off,boost::mpl::int_<((int)vmpl::size::value) - (int)sizeof...(t) - 1> >::type::value)*vmpl_reduce_prod_stop<vmpl,(int)((int)vmpl::size::value - sizeof...(t)  - 2)>::type::value + Lin_vmpl_off<vmpl,vmpl_off>(t...);
}
 
#endif