% POLY POLY Convert Roots To Polynomial Coefficients
% 
% Usage
% 
% This function calculates the polynomial coefficients for given roots
% 
%     p = poly(r)
% 
% when r is a vector, is a vector whose elements are the coefficients 
% of the polynomial whose roots are the elements of r.  Alternately,
% you can provide a matrix
% 
%     p = poly(A)
% 
% when A is an N x N square matrix, is a row vector with 
% N+1 elements which are the coefficients of the
% characteristic polynomial, det(lambda*eye(size(A))-A).
% 
% Contributed by Paulo Xavier Candeias under GPL.
function p = poly(r)
   if (nargin < 1) | (nargout > 1)
      error('wrong use (see help poly)')
   end
   
   % Identify arguments
   [l,c] = size(r);
   if l == c
      r = eig(r);
   elseif l*c == max(l,c)
      r = r(:).';
   else
      error('wrong argument (see help poly)')
   end
   
   % Strip out infinities
   r = r( ~isinf(r) );
   
   % Expand recursion formula
   n = length(r);
   p = [1,zeros(1,n)];
   for v = 1:n
      p(2:v+1) = p(2:v+1)-r(v)*p(1:v);
   end

   % The result should be real if the roots are complex conjugates.
   if sort(r(imag(r)>0))-sort(conj(r(imag(r)<0)))<teps(r)
      p = real(p);
   end