VectorBlock.h

// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2008-2010 Gael Guennebaud <gael.guennebaud@inria.fr>
// Copyright (C) 2006-2008 Benoit Jacob <jacob.benoit.1@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
// SPDX-License-Identifier: MPL-2.0
 
#ifndef EIGEN_VECTORBLOCK_H
#define EIGEN_VECTORBLOCK_H
 
// IWYU pragma: private
#include "./InternalHeaderCheck.h"
 
namespace Eigen {
 
namespace internal {
template <typename VectorType, int Size>
struct traits<VectorBlock<VectorType, Size> >
    : public traits<Block<VectorType, traits<VectorType>::Flags & RowMajorBit ? 1 : Size,
                          traits<VectorType>::Flags & RowMajorBit ? Size : 1> > {};
}  // namespace internal

template <typename VectorType, int Size>
class VectorBlock : public Block<VectorType, internal::traits<VectorType>::Flags & RowMajorBit ? 1 : Size,
                                 internal::traits<VectorType>::Flags & RowMajorBit ? Size : 1> {
  using Base = Block<VectorType, internal::traits<VectorType>::Flags & RowMajorBit ? 1 : Size,
                     internal::traits<VectorType>::Flags & RowMajorBit ? Size : 1>;
  enum { IsColVector = !(internal::traits<VectorType>::Flags & RowMajorBit) };
 
 public:
  EIGEN_DENSE_PUBLIC_INTERFACE(VectorBlock)
  EIGEN_STATIC_ASSERT_VECTOR_ONLY(VectorBlock)
  EIGEN_INHERIT_ASSIGNMENT_OPERATORS(VectorBlock)
 
  
  EIGEN_DEVICE_FUNC constexpr EIGEN_STRONG_INLINE VectorBlock(VectorType& vector, Index start, Index size)
      : Base(vector, IsColVector ? start : 0, IsColVector ? 0 : start, IsColVector ? size : 1, IsColVector ? 1 : size) {
  }

  EIGEN_DEVICE_FUNC constexpr EIGEN_STRONG_INLINE VectorBlock(VectorType& vector, Index start)
      : Base(vector, IsColVector ? start : 0, IsColVector ? 0 : start) {}
};
 
}  // end namespace Eigen
 
#endif  // EIGEN_VECTORBLOCK_H