Mercurial > hg > octave-nkf
view scripts/linear-algebra/logm.m @ 10825:cace99cb01ab
rewrite logm (M. Caliari, R.T. Guy)
author | Jaroslav Hajek <highegg@gmail.com> |
---|---|
date | Wed, 28 Jul 2010 09:22:41 +0200 (2010-07-28) |
parents | a1dbe9d80eee |
children | 228cd18455a6 |
line wrap: on
line source
## Copyright (C) 2010 Richard T. Guy <guyrt7@wfu.edu> ## Copyright (C) 2010 Marco Caliari <marco.caliari@univr.it> ## Copyright (C) 2008 N.J. Higham ## ## This file is part of Octave. ## ## <http://www.gnu.org/licenses/>. ## -*- texinfo -*- ## @deftypefn {Function File} {[@var{s}, @var{iters}] =} logm (@var{a}, @var{opt_iters}) ## Compute the matrix logarithm of the square matrix @var{a}. Utilizes a Pade ## approximant and the identity ## ## @code{ logm(@var{a}) = 2^k * logm(@var{a}^(1 / 2^k)) }. ## ## Optional argument @var{opt_iters} is the number of square roots computed ## and defaults to 100. Optional output @var{iters} is the number of square ## roots actually computed. ## ## @end deftypefn ## Reference: N. J. Higham, Functions of Matrices: Theory and Computation ## (SIAM, 2008.) ## function [s, iters] = logm (a, opt_iters) if (nargin == 0) print_usage (); elseif (nargin < 2) opt_iters = 100; elseif (nargin > 2) print_usage (); endif if (! issquare (a)) error ("logm: argument must be a square matrix."); endif [u, s] = schur (a); if (isreal (a)) [u, s] = rsf2csf (u, s); endif if (any (diag (s) < 0)) warning ("Octave:logm:non-principal", ["logm: Matrix has nonegative eigenvalues. Principal matrix logarithm is not defined.", ... "Computing non-principal logarithm."]); endif k = 0; ## Algorithm 11.9 in "Function of matrices", by N. Higham theta = [0, 0, 1.61e-2, 5.38e-2, 1.13e-1, 1.86e-1, 2.6429608311114350e-1]; p = 0; m = 7; while (k < opt_iters) tau = norm (s - eye (size (s)),1); if (tau <= theta (7)) p = p + 1; j(1) = find (tau <= theta, 1); j(2) = find (tau / 2 <= theta, 1); if (j(1) - j(2) <= 1 || p == 2) m = j(1); break endif endif k = k + 1; s = sqrtm (s); endwhile if (k >= opt_iters) warning ("logm: Maximum number of square roots exceeded. Results may still be accurate."); endif s = logm_pade_pf (s - eye (size (s)), m); s = 2^k * u * s * u'; if (nargout == 2) iters = k; endif endfunction ################## ANCILLARY FUNCTIONS ################################ ###### Taken from the mfttoolbox (GPL 3) by D. Higham. ###### Reference: ###### D. Higham, Functions of Matrices: Theory and Computation ###### (SIAM, 2008.). ####################################################################### ##LOGM_PADE_PF Evaluate Pade approximant to matrix log by partial fractions. ## Y = LOGM_PADE_PF(a,M) evaluates the [M/M] Pade approximation to ## LOG(EYE(SIZE(a))+a) using a partial fraction expansion. function s = logm_pade_pf(a,m) [nodes,wts] = gauss_legendre(m); ## Convert from [-1,1] to [0,1]. nodes = (nodes + 1)/2; wts = wts/2; n = length(a); s = zeros(n); for j=1:m s = s + wts(j)*(a/(eye(n) + nodes(j)*a)); endfor endfunction ###################################################################### ## GAUSS_LEGENDRE Nodes and weights for Gauss-Legendre quadrature. ## [X,W] = GAUSS_LEGENDRE(N) computes the nodes X and weights W ## for N-point Gauss-Legendre quadrature. ## Reference: ## G. H. Golub and J. H. Welsch, Calculation of Gauss quadrature ## rules, Math. Comp., 23(106):221-230, 1969. function [x,w] = gauss_legendre(n) i = 1:n-1; v = i./sqrt((2*i).^2-1); [V,D] = eig( diag(v,-1)+diag(v,1) ); x = diag(D); w = 2*(V(1,:)'.^2); endfunction %!assert(norm(logm([1 -1;0 1]) - [0 -1; 0 0]) < 1e-5); %!assert(norm(expm(logm([-1 2 ; 4 -1])) - [-1 2 ; 4 -1]) < 1e-5); %! %!error logm (); %!error logm (1, 2, 3); %!error <logm: argument must be a square matrix.> logm([1 0;0 1; 2 2]);