Il Linguaggio Fortran 90/95

292 Unità di Programma
FUNCTION vector_mul(Y,Z) RESULT(X)
! Esegue il prodotto elemento x elemento di due array 1D di tipo "matrix"
TYPE(matrix), INTENT(IN), DIMENSION(:) :: Y
TYPE(matrix), INTENT(IN), DIMENSION(SIZE(Y,1)) :: Z
REAL :: X
X = DOT_PRODUCT(Y(:)%elem, Z(:)%elem)
END FUNCTION vector_mul
FUNCTION matrix_vector_mul(Y,Z) RESULT(X)
! Esegue il prodotto righe x colonne fra un array 2D
! ed un array 1D di tipo "matrix"
TYPE(matrix), INTENT(IN), DIMENSION(:,:) :: Y
TYPE(matrix), INTENT(IN), DIMENSION(SIZE(Y,2)) :: Z
TYPE(matrix), DIMENSION(SIZE(Y,1)) :: X
X(:)%elem = MATMUL(Y(:,:)%elem, Z(:)%elem)
END FUNCTION matrix_vector_mul
!
! Operatore "divisione": X = Y / Z = INV(Z) * Y
!
FUNCTION matrix_div(Y,Z) RESULT(X)
! Esegue il "rapporto" fra due array 2D di tipo "matrix"
TYPE(matrix), INTENT(IN), DIMENSION(:,:) :: Y
TYPE(matrix), INTENT(IN), DIMENSION(:,:) :: Z
TYPE(matrix), DIMENSION(SIZE(Y,1),SIZE(Y,2)) :: X
REAL, DIMENSION(SIZE(Z,1),SIZE(Z,2)) :: W
INTEGER :: i, j, k, n
! effettua una copia di riserva degli argomenti
W(:,:) = Z(:,:)%elem
X(:,:)%elem = Y(:,:)%elem
! eliminazione di Gauss sulla matrice argomento (W|X)
n = SIZE(Z,2)
DO k = 1,n-1
DO i=k+1,n
W(i,k) = W(i,k)/W(k,k)
X(i,:)%elem = X(i,:)%elem - W(i,k) * X(k,:)%elem
END DO
DO j=k+1,n
DO i=k+1,n
W(i,j) = W(i,j) - W(i,k) * W(k,j)
END DO
END DO
END DO
! "back substitution" su X