common.hpp

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

#ifndef MDS_UTILS_PYTHON_UBLAS_DETAIL_MATRIX_HPP_INCLUDED
#define MDS_UTILS_PYTHON_UBLAS_DETAIL_MATRIX_HPP_INCLUDED


#include <mds_utils/python/numpy/array_iterator.hpp>
#include <type_traits>
#include <algorithm>
#include <boost/type_traits/is_complex.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <boost/numeric/ublas/fwd.hpp>



namespace mds_utils { namespace python { namespace ublas { namespace detail {

template<class ublas_orientation>
struct ublas_to_npy_storage;

template<>
struct ublas_to_npy_storage<boost::numeric::ublas::row_major_tag> {

    typedef mds_utils::python::numpy::c_storage tag;
};

template<>
struct ublas_to_npy_storage<boost::numeric::ublas::row_major> {
    
    typedef mds_utils::python::numpy::c_storage tag;
};

template<>
struct ublas_to_npy_storage<boost::numeric::ublas::column_major_tag> {

    typedef mds_utils::python::numpy::fortran_storage   tag;
};

template<>
struct ublas_to_npy_storage<boost::numeric::ublas::column_major> {

    typedef mds_utils::python::numpy::fortran_storage   tag;
};


template<class orientation_category>
int numpy_flags();

template<>
inline int numpy_flags<boost::numeric::ublas::row_major_tag>() {
    return 0;
}

template<>
inline int numpy_flags<boost::numeric::ublas::column_major_tag>() {
    return 1;
}


//template<class py_T,class M>
//inline typename std::enable_if<
//  !boost::is_complex<typename M::value_type>::value
//          >::type
//  copy_arr(const PyArrayObject *po,size_t nel,M& m) {
//
//  py_T    *data(reinterpret_cast<py_T*>(PyArray_DATA(const_cast<PyArrayObject*>(po))));
//  std::transform(data,data+nel,m.data().begin(),
//      [](py_T x) {
//          return boost::numeric_cast<typename M::value_type>(x);
//      });
//}
//
//
//template<class py_T,class M>
//inline typename std::enable_if<
//  boost::is_complex<typename M::value_type>::value
//          >::type
//      copy_arr(const PyArrayObject *po,size_t nel,M& m) {
//  py_T    *data(reinterpret_cast<py_T*>(PyArray_DATA(const_cast<PyArrayObject*>(po))));
//
//  std::transform(data,data+nel,m.data().begin(),
//      [](py_T x) {
//          return typename M::value_type(
//              boost::numeric_cast<typename M::value_type::value_type>(x.real),
//              boost::numeric_cast<typename M::value_type::value_type>(x.imag));
//      });
//}


}}}}

#endif