| 1 | // This file is part of Eigen, a lightweight C++ template library |
| 2 | // for linear algebra. |
| 3 | // |
| 4 | // Copyright (C) 2008-2010 Gael Guennebaud <gael.guennebaud@inria.fr> |
| 5 | // Copyright (C) 2006-2008 Benoit Jacob <jacob.benoit.1@gmail.com> |
| 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_VECTORBLOCK_H |
| 12 | #define EIGEN_VECTORBLOCK_H |
| 13 | |
| 14 | namespace Eigen { |
| 15 | |
| 16 | namespace internal { |
| 17 | template<typename VectorType, int Size> |
| 18 | struct traits<VectorBlock<VectorType, Size> > |
| 19 | : public traits<Block<VectorType, |
| 20 | traits<VectorType>::Flags & RowMajorBit ? 1 : Size, |
| 21 | traits<VectorType>::Flags & RowMajorBit ? Size : 1> > |
| 22 | { |
| 23 | }; |
| 24 | } |
| 25 | |
| 26 | /** \class VectorBlock |
| 27 | * \ingroup Core_Module |
| 28 | * |
| 29 | * \brief Expression of a fixed-size or dynamic-size sub-vector |
| 30 | * |
| 31 | * \tparam VectorType the type of the object in which we are taking a sub-vector |
| 32 | * \tparam Size size of the sub-vector we are taking at compile time (optional) |
| 33 | * |
| 34 | * This class represents an expression of either a fixed-size or dynamic-size sub-vector. |
| 35 | * It is the return type of DenseBase::segment(Index,Index) and DenseBase::segment<int>(Index) and |
| 36 | * most of the time this is the only way it is used. |
| 37 | * |
| 38 | * However, if you want to directly maniputate sub-vector expressions, |
| 39 | * for instance if you want to write a function returning such an expression, you |
| 40 | * will need to use this class. |
| 41 | * |
| 42 | * Here is an example illustrating the dynamic case: |
| 43 | * \include class_VectorBlock.cpp |
| 44 | * Output: \verbinclude class_VectorBlock.out |
| 45 | * |
| 46 | * \note Even though this expression has dynamic size, in the case where \a VectorType |
| 47 | * has fixed size, this expression inherits a fixed maximal size which means that evaluating |
| 48 | * it does not cause a dynamic memory allocation. |
| 49 | * |
| 50 | * Here is an example illustrating the fixed-size case: |
| 51 | * \include class_FixedVectorBlock.cpp |
| 52 | * Output: \verbinclude class_FixedVectorBlock.out |
| 53 | * |
| 54 | * \sa class Block, DenseBase::segment(Index,Index,Index,Index), DenseBase::segment(Index,Index) |
| 55 | */ |
| 56 | template<typename VectorType, int Size> class VectorBlock |
| 57 | : public Block<VectorType, |
| 58 | internal::traits<VectorType>::Flags & RowMajorBit ? 1 : Size, |
| 59 | internal::traits<VectorType>::Flags & RowMajorBit ? Size : 1> |
| 60 | { |
| 61 | typedef Block<VectorType, |
| 62 | internal::traits<VectorType>::Flags & RowMajorBit ? 1 : Size, |
| 63 | internal::traits<VectorType>::Flags & RowMajorBit ? Size : 1> Base; |
| 64 | enum { |
| 65 | IsColVector = !(internal::traits<VectorType>::Flags & RowMajorBit) |
| 66 | }; |
| 67 | public: |
| 68 | EIGEN_DENSE_PUBLIC_INTERFACE(VectorBlock) |
| 69 | |
| 70 | using Base::operator=; |
| 71 | |
| 72 | /** Dynamic-size constructor |
| 73 | */ |
| 74 | EIGEN_DEVICE_FUNC |
| 75 | inline VectorBlock(VectorType& vector, Index start, Index size) |
| 76 | : Base(vector, |
| 77 | IsColVector ? start : 0, IsColVector ? 0 : start, |
| 78 | IsColVector ? size : 1, IsColVector ? 1 : size) |
| 79 | { |
| 80 | EIGEN_STATIC_ASSERT_VECTOR_ONLY(VectorBlock); |
| 81 | } |
| 82 | |
| 83 | /** Fixed-size constructor |
| 84 | */ |
| 85 | EIGEN_DEVICE_FUNC |
| 86 | inline VectorBlock(VectorType& vector, Index start) |
| 87 | : Base(vector, IsColVector ? start : 0, IsColVector ? 0 : start) |
| 88 | { |
| 89 | EIGEN_STATIC_ASSERT_VECTOR_ONLY(VectorBlock); |
| 90 | } |
| 91 | }; |
| 92 | |
| 93 | |
| 94 | } // end namespace Eigen |
| 95 | |
| 96 | #endif // EIGEN_VECTORBLOCK_H |
| 97 | |