idamnjanovic@10: function id = diag_ids(x,k)
idamnjanovic@10: %DIAG_IDS Indices of matrix diagonal elements.
idamnjanovic@10: %  ID = DIAG_IDS(X) returns the indices of the main diagonal of X.
idamnjanovic@10: %
idamnjanovic@10: %  ID = DIAG_IDS(X,K) returns the indices of the K-th diagonal. K=0
idamnjanovic@10: %  represents the main diagonal, positive values are above the main
idamnjanovic@10: %  diagonal and negative values are below the main diagonal.
idamnjanovic@10: 
idamnjanovic@10: 
idamnjanovic@10: %  Ron Rubinstein
idamnjanovic@10: %  Computer Science Department
idamnjanovic@10: %  Technion, Haifa 32000 Israel
idamnjanovic@10: %  ronrubin@cs
idamnjanovic@10: %
idamnjanovic@10: %  September 2006
idamnjanovic@10: 
idamnjanovic@10: 
idamnjanovic@10: if (nargin==1), k=0; end
idamnjanovic@10: 
idamnjanovic@10: [m,n] = size(x);
idamnjanovic@10: l = max(m,n);
idamnjanovic@10: 
idamnjanovic@10: if (k<=0)
idamnjanovic@10:   id = (0:l-1)*m + (1:l) - k;
idamnjanovic@10: else
idamnjanovic@10:   id = (0:l-1)*m + (1:l) + k*m;
idamnjanovic@10: end
idamnjanovic@10: 
idamnjanovic@10: if (l-k>m), id = id(1:end-(l-k-m)); end
idamnjanovic@10: if (l+k>n), id = id(1:end-(l+k-n)); end