1c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This file is part of Eigen, a lightweight C++ template library 2c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// for linear algebra. 3c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 4c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Copyright (C) 2010 Benoit Jacob <jacob.benoit.1@gmail.com> 5c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 6c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This Source Code Form is subject to the terms of the Mozilla 7c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Public License v. 2.0. If a copy of the MPL was not distributed 8c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 10c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#ifndef EIGEN_STRIDE_H 11c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#define EIGEN_STRIDE_H 12c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 13c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace Eigen { 14c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 15c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath/** \class Stride 16c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \ingroup Core_Module 17c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 18c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \brief Holds strides information for Map 19c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 20c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * This class holds the strides information for mapping arrays with strides with class Map. 21c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 22c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * It holds two values: the inner stride and the outer stride. 23c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 24c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * The inner stride is the pointer increment between two consecutive entries within a given row of a 25c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * row-major matrix or within a given column of a column-major matrix. 26c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 27c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * The outer stride is the pointer increment between two consecutive rows of a row-major matrix or 28c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * between two consecutive columns of a column-major matrix. 29c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 30c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * These two values can be passed either at compile-time as template parameters, or at runtime as 31c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * arguments to the constructor. 32c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 33c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * Indeed, this class takes two template parameters: 34c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \param _OuterStrideAtCompileTime the outer stride, or Dynamic if you want to specify it at runtime. 35c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \param _InnerStrideAtCompileTime the inner stride, or Dynamic if you want to specify it at runtime. 36c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 37c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * Here is an example: 38c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \include Map_general_stride.cpp 39c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * Output: \verbinclude Map_general_stride.out 40c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * 41c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * \sa class InnerStride, class OuterStride, \ref TopicStorageOrders 42c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath */ 43c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<int _OuterStrideAtCompileTime, int _InnerStrideAtCompileTime> 44c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass Stride 45c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 46c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 47c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef DenseIndex Index; 48c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath enum { 49c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath InnerStrideAtCompileTime = _InnerStrideAtCompileTime, 50c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath OuterStrideAtCompileTime = _OuterStrideAtCompileTime 51c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath }; 52c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 53c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath /** Default constructor, for use when strides are fixed at compile time */ 54c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Stride() 55c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : m_outer(OuterStrideAtCompileTime), m_inner(InnerStrideAtCompileTime) 56c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 57c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath eigen_assert(InnerStrideAtCompileTime != Dynamic && OuterStrideAtCompileTime != Dynamic); 58c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 59c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 60c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath /** Constructor allowing to pass the strides at runtime */ 61c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Stride(Index outerStride, Index innerStride) 62c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : m_outer(outerStride), m_inner(innerStride) 63c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 64c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath eigen_assert(innerStride>=0 && outerStride>=0); 65c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 66c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 67c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath /** Copy constructor */ 68c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Stride(const Stride& other) 69c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : m_outer(other.outer()), m_inner(other.inner()) 70c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath {} 71c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 72c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath /** \returns the outer stride */ 73c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath inline Index outer() const { return m_outer.value(); } 74c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath /** \returns the inner stride */ 75c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath inline Index inner() const { return m_inner.value(); } 76c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 77c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath protected: 78c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath internal::variable_if_dynamic<Index, OuterStrideAtCompileTime> m_outer; 79c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath internal::variable_if_dynamic<Index, InnerStrideAtCompileTime> m_inner; 80c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 81c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 82c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath/** \brief Convenience specialization of Stride to specify only an inner stride 83c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * See class Map for some examples */ 84c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<int Value = Dynamic> 85c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass InnerStride : public Stride<0, Value> 86c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 87c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef Stride<0, Value> Base; 88c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 89c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef DenseIndex Index; 90c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath InnerStride() : Base() {} 91c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath InnerStride(Index v) : Base(0, v) {} 92c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 93c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 94c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath/** \brief Convenience specialization of Stride to specify only an outer stride 95c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath * See class Map for some examples */ 96c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<int Value = Dynamic> 97c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass OuterStride : public Stride<Value, 0> 98c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 99c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef Stride<Value, 0> Base; 100c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 101c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef DenseIndex Index; 102c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath OuterStride() : Base() {} 103c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath OuterStride(Index v) : Base(v,0) {} 104c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 105c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 106c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} // end namespace Eigen 107c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 108c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#endif // EIGEN_STRIDE_H 109