You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
3.8 KiB
121 lines
3.8 KiB
// |
|
// Copyright (c) 2002 Kresimir Fresl |
|
// Copyright (c) 2010 Rutger ter Borg |
|
// |
|
// Distributed under the Boost Software License, Version 1.0. |
|
// (See accompanying file LICENSE_1_0.txt or copy at |
|
// http://www.boost.org/LICENSE_1_0.txt) |
|
// |
|
|
|
#ifndef BOOST_NUMERIC_BINDINGS_UBLAS_BANDED_HPP |
|
#define BOOST_NUMERIC_BINDINGS_UBLAS_BANDED_HPP |
|
|
|
#include <boost/numeric/bindings/begin.hpp> |
|
#include <boost/numeric/bindings/detail/adaptor.hpp> |
|
#include <boost/numeric/bindings/detail/if_row_major.hpp> |
|
#include <boost/numeric/bindings/end.hpp> |
|
#include <boost/numeric/bindings/ublas/detail/basic_ublas_adaptor.hpp> |
|
#include <boost/numeric/bindings/ublas/detail/convert_to.hpp> |
|
#include <boost/numeric/bindings/ublas/matrix_expression.hpp> |
|
#include <boost/numeric/bindings/value_type.hpp> |
|
#include <boost/numeric/ublas/banded.hpp> |
|
|
|
namespace boost { |
|
namespace numeric { |
|
namespace bindings { |
|
namespace detail { |
|
|
|
template< typename T, typename F, typename A, typename Id, typename Enable > |
|
struct adaptor< ublas::banded_matrix< T, F, A >, Id, Enable > { |
|
|
|
// The ublas banded row_major format corresponds to the LAPACK band format. |
|
// LAPACK is column_major; so we flip the data order reported by uBLAS. |
|
typedef typename copy_const< Id, T >::type value_type; |
|
typedef typename if_row_major< |
|
typename convert_to< tag::data_order, F >::type, |
|
tag::column_major, |
|
tag::row_major |
|
>::type data_order; |
|
typedef mpl::map< |
|
mpl::pair< tag::value_type, value_type >, |
|
mpl::pair< tag::entity, tag::matrix >, |
|
mpl::pair< tag::size_type<1>, std::ptrdiff_t >, |
|
mpl::pair< tag::size_type<2>, std::ptrdiff_t >, |
|
mpl::pair< tag::matrix_type, tag::band >, |
|
mpl::pair< tag::data_structure, tag::band_array >, |
|
mpl::pair< tag::data_order, data_order >, |
|
mpl::pair< tag::bandwidth_type<1>, std::ptrdiff_t >, |
|
mpl::pair< tag::bandwidth_type<2>, std::ptrdiff_t >, |
|
mpl::pair< tag::stride_type<1>, |
|
typename if_row_major< data_order, std::ptrdiff_t, tag::contiguous >::type >, |
|
mpl::pair< tag::stride_type<2>, |
|
typename if_row_major< data_order, tag::contiguous, std::ptrdiff_t >::type > |
|
> property_map; |
|
|
|
static std::ptrdiff_t size1( const Id& id ) { |
|
return id.size1(); |
|
} |
|
|
|
static std::ptrdiff_t size2( const Id& id ) { |
|
return id.size2(); |
|
} |
|
|
|
static value_type* begin_value( Id& id ) { |
|
return bindings::begin_value( id.data() ); |
|
} |
|
|
|
static value_type* end_value( Id& id ) { |
|
return bindings::end_value( id.data() ); |
|
} |
|
|
|
// A.k.a. left half-bandwidth |
|
static std::ptrdiff_t bandwidth1( const Id& id ) { |
|
return id.lower(); |
|
} |
|
|
|
// A.k.a. right half-bandwidth |
|
static std::ptrdiff_t bandwidth2( const Id& id ) { |
|
return id.upper(); |
|
} |
|
|
|
// These strides are over the band array structure; not over |
|
// the band matrix representation of this structure |
|
static std::ptrdiff_t stride1( const Id& id ) { |
|
return id.lower() + id.upper() + 1; |
|
} |
|
|
|
static std::ptrdiff_t stride2( const Id& id ) { |
|
return id.lower() + id.upper() + 1; |
|
} |
|
|
|
}; |
|
|
|
|
|
template< typename T, typename Id, typename Enable > |
|
struct adaptor< ublas::banded_adaptor< T >, Id, Enable >: |
|
basic_ublas_adaptor< |
|
T, |
|
Id, |
|
mpl::pair< tag::matrix_type, tag::band >, |
|
mpl::pair< tag::bandwidth_type<1>, std::ptrdiff_t >, |
|
mpl::pair< tag::bandwidth_type<2>, std::ptrdiff_t > |
|
> { |
|
|
|
// A.k.a. left half-bandwidth |
|
static std::ptrdiff_t bandwidth1( const Id& id ) { |
|
return id.lower(); |
|
} |
|
|
|
// A.k.a. right half-bandwidth |
|
static std::ptrdiff_t bandwidth2( const Id& id ) { |
|
return id.upper(); |
|
} |
|
|
|
}; |
|
|
|
} // detail |
|
} // bindings |
|
} // numeric |
|
} // boost |
|
|
|
#endif
|
|
|