| 1 | // This file is part of Eigen, a lightweight C++ template library |
| 2 | // for linear algebra. |
| 3 | // |
| 4 | // Copyright (C) 2007-2010 Benoit Jacob <jacob.benoit.1@gmail.com> |
| 5 | // Copyright (C) 2008 Gael Guennebaud <gael.guennebaud@inria.fr> |
| 6 | // |
| 7 | // This Source Code Form is subject to the terms of the Mozilla |
| 8 | // Public License v. 2.0. If a copy of the MPL was not distributed |
| 9 | // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. |
| 10 | |
| 11 | #ifndef EIGEN_MAP_H |
| 12 | #define EIGEN_MAP_H |
| 13 | |
| 14 | namespace Eigen { |
| 15 | |
| 16 | namespace internal { |
| 17 | template<typename PlainObjectType, int MapOptions, typename StrideType> |
| 18 | struct traits<Map<PlainObjectType, MapOptions, StrideType> > |
| 19 | : public traits<PlainObjectType> |
| 20 | { |
| 21 | typedef traits<PlainObjectType> TraitsBase; |
| 22 | enum { |
| 23 | PlainObjectTypeInnerSize = ((traits<PlainObjectType>::Flags&RowMajorBit)==RowMajorBit) |
| 24 | ? PlainObjectType::ColsAtCompileTime |
| 25 | : PlainObjectType::RowsAtCompileTime, |
| 26 | |
| 27 | InnerStrideAtCompileTime = StrideType::InnerStrideAtCompileTime == 0 |
| 28 | ? int(PlainObjectType::InnerStrideAtCompileTime) |
| 29 | : int(StrideType::InnerStrideAtCompileTime), |
| 30 | OuterStrideAtCompileTime = StrideType::OuterStrideAtCompileTime == 0 |
| 31 | ? (InnerStrideAtCompileTime==Dynamic || PlainObjectTypeInnerSize==Dynamic |
| 32 | ? Dynamic |
| 33 | : int(InnerStrideAtCompileTime) * int(PlainObjectTypeInnerSize)) |
| 34 | : int(StrideType::OuterStrideAtCompileTime), |
| 35 | Alignment = int(MapOptions)&int(AlignedMask), |
| 36 | Flags0 = TraitsBase::Flags & (~NestByRefBit), |
| 37 | Flags = is_lvalue<PlainObjectType>::value ? int(Flags0) : (int(Flags0) & ~LvalueBit) |
| 38 | }; |
| 39 | private: |
| 40 | enum { Options }; // Expressions don't have Options |
| 41 | }; |
| 42 | } |
| 43 | |
| 44 | /** \class Map |
| 45 | * \ingroup Core_Module |
| 46 | * |
| 47 | * \brief A matrix or vector expression mapping an existing array of data. |
| 48 | * |
| 49 | * \tparam PlainObjectType the equivalent matrix type of the mapped data |
| 50 | * \tparam MapOptions specifies the pointer alignment in bytes. It can be: \c #Aligned128, , \c #Aligned64, \c #Aligned32, \c #Aligned16, \c #Aligned8 or \c #Unaligned. |
| 51 | * The default is \c #Unaligned. |
| 52 | * \tparam StrideType optionally specifies strides. By default, Map assumes the memory layout |
| 53 | * of an ordinary, contiguous array. This can be overridden by specifying strides. |
| 54 | * The type passed here must be a specialization of the Stride template, see examples below. |
| 55 | * |
| 56 | * This class represents a matrix or vector expression mapping an existing array of data. |
| 57 | * It can be used to let Eigen interface without any overhead with non-Eigen data structures, |
| 58 | * such as plain C arrays or structures from other libraries. By default, it assumes that the |
| 59 | * data is laid out contiguously in memory. You can however override this by explicitly specifying |
| 60 | * inner and outer strides. |
| 61 | * |
| 62 | * Here's an example of simply mapping a contiguous array as a \ref TopicStorageOrders "column-major" matrix: |
| 63 | * \include Map_simple.cpp |
| 64 | * Output: \verbinclude Map_simple.out |
| 65 | * |
| 66 | * If you need to map non-contiguous arrays, you can do so by specifying strides: |
| 67 | * |
| 68 | * Here's an example of mapping an array as a vector, specifying an inner stride, that is, the pointer |
| 69 | * increment between two consecutive coefficients. Here, we're specifying the inner stride as a compile-time |
| 70 | * fixed value. |
| 71 | * \include Map_inner_stride.cpp |
| 72 | * Output: \verbinclude Map_inner_stride.out |
| 73 | * |
| 74 | * Here's an example of mapping an array while specifying an outer stride. Here, since we're mapping |
| 75 | * as a column-major matrix, 'outer stride' means the pointer increment between two consecutive columns. |
| 76 | * Here, we're specifying the outer stride as a runtime parameter. Note that here \c OuterStride<> is |
| 77 | * a short version of \c OuterStride<Dynamic> because the default template parameter of OuterStride |
| 78 | * is \c Dynamic |
| 79 | * \include Map_outer_stride.cpp |
| 80 | * Output: \verbinclude Map_outer_stride.out |
| 81 | * |
| 82 | * For more details and for an example of specifying both an inner and an outer stride, see class Stride. |
| 83 | * |
| 84 | * \b Tip: to change the array of data mapped by a Map object, you can use the C++ |
| 85 | * placement new syntax: |
| 86 | * |
| 87 | * Example: \include Map_placement_new.cpp |
| 88 | * Output: \verbinclude Map_placement_new.out |
| 89 | * |
| 90 | * This class is the return type of PlainObjectBase::Map() but can also be used directly. |
| 91 | * |
| 92 | * \sa PlainObjectBase::Map(), \ref TopicStorageOrders |
| 93 | */ |
| 94 | template<typename PlainObjectType, int MapOptions, typename StrideType> class Map |
| 95 | : public MapBase<Map<PlainObjectType, MapOptions, StrideType> > |
| 96 | { |
| 97 | public: |
| 98 | |
| 99 | typedef MapBase<Map> Base; |
| 100 | EIGEN_DENSE_PUBLIC_INTERFACE(Map) |
| 101 | |
| 102 | typedef typename Base::PointerType PointerType; |
| 103 | typedef PointerType PointerArgType; |
| 104 | EIGEN_DEVICE_FUNC |
| 105 | inline PointerType cast_to_pointer_type(PointerArgType ptr) { return ptr; } |
| 106 | |
| 107 | EIGEN_DEVICE_FUNC |
| 108 | inline Index innerStride() const |
| 109 | { |
| 110 | return StrideType::InnerStrideAtCompileTime != 0 ? m_stride.inner() : 1; |
| 111 | } |
| 112 | |
| 113 | EIGEN_DEVICE_FUNC |
| 114 | inline Index outerStride() const |
| 115 | { |
| 116 | return int(StrideType::OuterStrideAtCompileTime) != 0 ? m_stride.outer() |
| 117 | : int(internal::traits<Map>::OuterStrideAtCompileTime) != Dynamic ? Index(internal::traits<Map>::OuterStrideAtCompileTime) |
| 118 | : IsVectorAtCompileTime ? (this->size() * innerStride()) |
| 119 | : (int(Flags)&RowMajorBit) ? (this->cols() * innerStride()) |
| 120 | : (this->rows() * innerStride()); |
| 121 | } |
| 122 | |
| 123 | /** Constructor in the fixed-size case. |
| 124 | * |
| 125 | * \param dataPtr pointer to the array to map |
| 126 | * \param stride optional Stride object, passing the strides. |
| 127 | */ |
| 128 | EIGEN_DEVICE_FUNC |
| 129 | explicit inline Map(PointerArgType dataPtr, const StrideType& stride = StrideType()) |
| 130 | : Base(cast_to_pointer_type(dataPtr)), m_stride(stride) |
| 131 | { |
| 132 | PlainObjectType::Base::_check_template_params(); |
| 133 | } |
| 134 | |
| 135 | /** Constructor in the dynamic-size vector case. |
| 136 | * |
| 137 | * \param dataPtr pointer to the array to map |
| 138 | * \param size the size of the vector expression |
| 139 | * \param stride optional Stride object, passing the strides. |
| 140 | */ |
| 141 | EIGEN_DEVICE_FUNC |
| 142 | inline Map(PointerArgType dataPtr, Index size, const StrideType& stride = StrideType()) |
| 143 | : Base(cast_to_pointer_type(dataPtr), size), m_stride(stride) |
| 144 | { |
| 145 | PlainObjectType::Base::_check_template_params(); |
| 146 | } |
| 147 | |
| 148 | /** Constructor in the dynamic-size matrix case. |
| 149 | * |
| 150 | * \param dataPtr pointer to the array to map |
| 151 | * \param rows the number of rows of the matrix expression |
| 152 | * \param cols the number of columns of the matrix expression |
| 153 | * \param stride optional Stride object, passing the strides. |
| 154 | */ |
| 155 | EIGEN_DEVICE_FUNC |
| 156 | inline Map(PointerArgType dataPtr, Index rows, Index cols, const StrideType& stride = StrideType()) |
| 157 | : Base(cast_to_pointer_type(dataPtr), rows, cols), m_stride(stride) |
| 158 | { |
| 159 | PlainObjectType::Base::_check_template_params(); |
| 160 | } |
| 161 | |
| 162 | EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Map) |
| 163 | |
| 164 | protected: |
| 165 | StrideType m_stride; |
| 166 | }; |
| 167 | |
| 168 | |
| 169 | } // end namespace Eigen |
| 170 | |
| 171 | #endif // EIGEN_MAP_H |
| 172 | |