initial commit
This commit is contained in:
+82
@@ -0,0 +1,82 @@
|
||||
// This file is part of Eigen, a lightweight C++ template library
|
||||
// for linear algebra.
|
||||
//
|
||||
// Copyright (C) 2018 Gael Guennebaud <gael.guennebaud@inria.fr>
|
||||
//
|
||||
// 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/.
|
||||
|
||||
#include "main.h"
|
||||
#include <Eigen/SparseCore>
|
||||
|
||||
template <int ExpectedDim, typename Xpr>
|
||||
void check_dim(const Xpr&) {
|
||||
STATIC_CHECK(Xpr::NumDimensions == ExpectedDim);
|
||||
}
|
||||
|
||||
template <template <typename, int, int> class Object>
|
||||
void map_num_dimensions() {
|
||||
typedef Object<double, 1, 1> ArrayScalarType;
|
||||
typedef Object<double, 2, 1> ArrayVectorType;
|
||||
typedef Object<double, 1, 2> TransposeArrayVectorType;
|
||||
typedef Object<double, 2, 2> ArrayType;
|
||||
typedef Object<double, Eigen::Dynamic, 1> DynamicArrayVectorType;
|
||||
typedef Object<double, 1, Eigen::Dynamic> DynamicTransposeArrayVectorType;
|
||||
typedef Object<double, Eigen::Dynamic, Eigen::Dynamic> DynamicArrayType;
|
||||
|
||||
STATIC_CHECK(ArrayScalarType::NumDimensions == 0);
|
||||
STATIC_CHECK(ArrayVectorType::NumDimensions == 1);
|
||||
STATIC_CHECK(TransposeArrayVectorType::NumDimensions == 1);
|
||||
STATIC_CHECK(ArrayType::NumDimensions == 2);
|
||||
STATIC_CHECK(DynamicArrayVectorType::NumDimensions == 1);
|
||||
STATIC_CHECK(DynamicTransposeArrayVectorType::NumDimensions == 1);
|
||||
STATIC_CHECK(DynamicArrayType::NumDimensions == 2);
|
||||
|
||||
typedef Eigen::Map<ArrayScalarType> ArrayScalarMap;
|
||||
typedef Eigen::Map<ArrayVectorType> ArrayVectorMap;
|
||||
typedef Eigen::Map<TransposeArrayVectorType> TransposeArrayVectorMap;
|
||||
typedef Eigen::Map<ArrayType> ArrayMap;
|
||||
typedef Eigen::Map<DynamicArrayVectorType> DynamicArrayVectorMap;
|
||||
typedef Eigen::Map<DynamicTransposeArrayVectorType> DynamicTransposeArrayVectorMap;
|
||||
typedef Eigen::Map<DynamicArrayType> DynamicArrayMap;
|
||||
|
||||
STATIC_CHECK(ArrayScalarMap::NumDimensions == 0);
|
||||
STATIC_CHECK(ArrayVectorMap::NumDimensions == 1);
|
||||
STATIC_CHECK(TransposeArrayVectorMap::NumDimensions == 1);
|
||||
STATIC_CHECK(ArrayMap::NumDimensions == 2);
|
||||
STATIC_CHECK(DynamicArrayVectorMap::NumDimensions == 1);
|
||||
STATIC_CHECK(DynamicTransposeArrayVectorMap::NumDimensions == 1);
|
||||
STATIC_CHECK(DynamicArrayMap::NumDimensions == 2);
|
||||
}
|
||||
|
||||
template <typename Scalar, int Rows, int Cols>
|
||||
using TArray = Array<Scalar, Rows, Cols>;
|
||||
|
||||
template <typename Scalar, int Rows, int Cols>
|
||||
using TMatrix = Matrix<Scalar, Rows, Cols>;
|
||||
|
||||
EIGEN_DECLARE_TEST(num_dimensions) {
|
||||
int n = 10;
|
||||
ArrayXXd A(n, n);
|
||||
CALL_SUBTEST(check_dim<2>(A));
|
||||
CALL_SUBTEST(check_dim<2>(A.block(1, 1, 2, 2)));
|
||||
CALL_SUBTEST(check_dim<1>(A.col(1)));
|
||||
CALL_SUBTEST(check_dim<1>(A.row(1)));
|
||||
|
||||
MatrixXd M(n, n);
|
||||
CALL_SUBTEST(check_dim<0>(M.row(1) * M.col(1)));
|
||||
|
||||
SparseMatrix<double> S(n, n);
|
||||
CALL_SUBTEST(check_dim<2>(S));
|
||||
CALL_SUBTEST(check_dim<2>(S.block(1, 1, 2, 2)));
|
||||
CALL_SUBTEST(check_dim<1>(S.col(1)));
|
||||
CALL_SUBTEST(check_dim<1>(S.row(1)));
|
||||
|
||||
SparseVector<double> s(n);
|
||||
CALL_SUBTEST(check_dim<1>(s));
|
||||
CALL_SUBTEST(check_dim<1>(s.head(2)));
|
||||
|
||||
CALL_SUBTEST(map_num_dimensions<TArray>());
|
||||
CALL_SUBTEST(map_num_dimensions<TMatrix>());
|
||||
}
|
||||
Reference in New Issue
Block a user