diagonal_matrix.pro

来自「IDL语言编写的用于天文自适应光学仿真的软件CAOS V6.0的第一部分。」· PRO 代码 · 共 21 行

; $Id: diagonal_matrix.pro,v 1.2 2002/12/11 18:39:09 riccardi Exp $
;+
;    DIAGONAL_MARIX
;
;  ret = diagonal_matrix(diag)
;
;  Returns a diagonal matrix having the same data type of diag and
;  diag as diagonal elements.
;
; 21 Oct, 2002 Written by A. Riccardi, INAF-Osservatorio Astrofisico di Arcetri (Italy)
;              [e-mail]
;
;-
function diagonal_matrix, diagonal

	s = size(diagonal)
	n = s[s[0]+2]
	mat = make_array(TYPE=s[s[0]+1], n, n)
	mat[lindgen(n),lindgen(n)] = diagonal
	return, mat
end