extract_diagonal.pro

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

; $Id: extract_diagonal.pro,v 1.2 2002/03/14 11:49:12 riccardi Exp $

;+
; EXTRACT_DIAGONAL
;
; Result = EXTRACT_DIAGONAL(A)
;
; INPUT
; A:	matrice quadrata
;
; OUTPUT
; vettore che contiene gli elementi della diagonale di A
;-
function extract_diagonal, mat

	on_error, 2
	s_mat = size(mat)

	case s_mat[0] of
		2: begin
			if s_mat(1) eq s_mat(2) then begin
				return, mat(lindgen(s_mat(1))*(s_mat(1)+1))
			endif
		end

		1: begin
			if s_mat(1) eq 1 then begin
				return, mat
			endif
		end

		0: begin
			return, mat
		end

		else: begin
		end
	endcase

	message, "The input matrix must be square"
end