mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
Allow user to specify max number of iterations (bug #479).
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
// This file is part of Eigen, a lightweight C++ template library
|
||||
// for linear algebra.
|
||||
//
|
||||
// Copyright (C) 2010 Jitse Niesen <jitse@maths.leeds.ac.uk>
|
||||
// Copyright (C) 2010,2012 Jitse Niesen <jitse@maths.leeds.ac.uk>
|
||||
//
|
||||
// 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
|
||||
@@ -47,6 +47,22 @@ template<typename MatrixType> void schur(int size = MatrixType::ColsAtCompileTim
|
||||
VERIFY_IS_EQUAL(cs1.matrixT(), cs2.matrixT());
|
||||
VERIFY_IS_EQUAL(cs1.matrixU(), cs2.matrixU());
|
||||
|
||||
// Test maximum number of iterations
|
||||
ComplexSchur<MatrixType> cs3;
|
||||
cs3.compute(A, true, ComplexSchur<MatrixType>::m_maxIterations * size);
|
||||
VERIFY_IS_EQUAL(cs3.info(), Success);
|
||||
VERIFY_IS_EQUAL(cs3.matrixT(), cs1.matrixT());
|
||||
VERIFY_IS_EQUAL(cs3.matrixU(), cs1.matrixU());
|
||||
cs3.compute(A, true, 1);
|
||||
VERIFY_IS_EQUAL(cs3.info(), size > 1 ? NoConvergence : Success);
|
||||
|
||||
MatrixType Atriangular = A;
|
||||
Atriangular.template triangularView<StrictlyLower>().setZero();
|
||||
cs3.compute(Atriangular, true, 1); // triangular matrices do not need any iterations
|
||||
VERIFY_IS_EQUAL(cs3.info(), Success);
|
||||
VERIFY_IS_EQUAL(cs3.matrixT(), Atriangular.template cast<ComplexScalar>());
|
||||
VERIFY_IS_EQUAL(cs3.matrixU(), ComplexMatrixType::Identity(size, size));
|
||||
|
||||
// Test computation of only T, not U
|
||||
ComplexSchur<MatrixType> csOnlyT(A, false);
|
||||
VERIFY_IS_EQUAL(csOnlyT.info(), Success);
|
||||
|
||||
Reference in New Issue
Block a user