traits.hpp

Go to the documentation of this file.

// mds_utils/python/numpy/traits.hpp
//
// Copyright (c) 2014 - Michele De Stefano (micdestefano@users.sourceforge.net)
//
// Distributed under the MIT License (See accompanying file LICENSE)

#ifndef MDS_UTILS_PYTHON_NUMPY_TRAITS_HPP_INCLUDED
#define MDS_UTILS_PYTHON_NUMPY_TRAITS_HPP_INCLUDED

#include <numpy/arrayobject.h>
#include <boost/preprocessor/seq/for_each.hpp>
#include <boost/preprocessor/tuple/elem.hpp>
#include <complex>

namespace mds_utils {
    namespace python {
        namespace numpy {
template<class T>
struct numpy_dtype_traits;


#define SEQ_TYPE_NPYDTYPE_TYPENUM_TUPLES    \
    ((bool,npy_bool,NPY_BOOL))((char,npy_byte,NPY_BYTE))    \
    ((short,npy_short,NPY_SHORT))((int,npy_int,NPY_INT))    \
    ((long,npy_long,NPY_LONG))((long long,npy_longlong,NPY_LONGLONG))   \
    ((unsigned char,npy_ubyte,NPY_UBYTE))   \
    ((unsigned short,npy_short,NPY_USHORT)) \
    ((unsigned int,npy_uint,NPY_UINT))  \
    ((unsigned long,npy_ulong,NPY_ULONG))\
    ((unsigned long long,npy_ulonglong,NPY_ULONGLONG))  \
    ((float,npy_float,NPY_FLOAT))((double,npy_double,NPY_DOUBLE))   \
    ((long double,npy_longdouble,NPY_LONGDOUBLE))   \
    ((std::complex<float>,npy_cfloat,NPY_CFLOAT))   \
    ((std::complex<double>,npy_cdouble,NPY_CDOUBLE))    \
    ((std::complex<long double>,npy_clongdouble,NPY_CLONGDOUBLE))

#define numpy_dtype_traits_DEFSPEC_GEN(r,data,elem) \
    template<>  \
    struct numpy_dtype_traits< BOOST_PP_TUPLE_ELEM(3,0,elem) > {    \
        typedef BOOST_PP_TUPLE_ELEM(3,1,elem)   type;   \
        static const int    typenum = BOOST_PP_TUPLE_ELEM(3,2,elem);    \
    };

BOOST_PP_SEQ_FOR_EACH(numpy_dtype_traits_DEFSPEC_GEN,~,SEQ_TYPE_NPYDTYPE_TYPENUM_TUPLES)


        } // namespace numpy
    } // namespace python
} // namespace mds_utils


#endif /* MDS_UTILS_PYTHON_NUMPY_TRAITS_HPP_INCLUDED */