Matrizes em Fortran 90

Transcrição

1 Matrizes em Fortran 90 Flavio Aristone Juvenal Muniz 1 de maio de 2010 Resumo Discutiremos neste documento a implementação de operações básicas sobre matrizes na linguagem de programação Fortran 90: adição, multiplicação, transposta e inversa. Calcularemos o determinante de uma matriz pela sua decomposição LU. Sumário 1 Introdução 1 2 Teoria Arrays, vetores e matrizes Alocação dinâmica de arrays Funções e subrotinas 3 3 Implementação 4 Referências 11 1 Introdução Uma matriz é um arranjo retangular de números. Por exemplo, A = As operações básicas que podem ser aplicadas a matrizes são: adição, multiplicação, transposição e inversão. Nossa preocupação maior neste documento é a implementação destas operações na linguagem de programação Fortran 90. Para uma discussão teórica sobre matrizes, sugerimos o livro [3]. 2 Teoria Antes de implementarmos as operações matriciais, vamos fornecer um breve resumo de alguns conceitos da linguagem Fortran 90 que iremos utilizar mais adiante neste documento. 1

2 2.1 Arrays, vetores e matrizes Para implementar matrizes na linguagem Fortran 90 iremos utilizar uma array 1. Uma array é uma estrutura de dados que consiste de uma coleção de valores (variáveis ou constantes) de um mesmo tipo que são referidos por um mesmo nome. Um valor individual dentro da array é chamado um elemento da array e é identificado pelo nome da array juntamente com um índice indicando a localização do elemento dentro da array. Por exemplo, a seguinte coleção de 5 números inteiros poderá ser representada em Fortran 90 por {5, 7, 9, 1, 8} i n t e g e r, dimension ( 5 ) : : numeros = (/ 5, 7, 9, 1, 8 /) O trecho de código acima declara uma array chamada numeros contendo 5 números inteiros. No código, a array numeros é inicializada já na sua própria declaração pelo construtor de arrays (/ /). Outra forma de declarar e inicializar esta array seria i n t e g e r, dimension ( 5 ) : : numeros numeros ( 1 ) = 5 numeros ( 2 ) = 7 numeros ( 3 ) = 9 numeros ( 4 ) = 1 numeros ( 5 ) = 8 O número entre os parênteses é o índice que localiza a posição dentro da array; o índice 1 localiza a primeira posição na array, o 2 a segunda e assim em diante. Portanto, o código acima atribui o valor 5 à posição 1 da array, o valor 7 à posição 2, etc. O sinal de igual = é a forma de atribuir valores a variáveis em Fortran 90. As arrays em Fortran 90 podem ter mais de um índice, podendo ser organizadas em múltiplas dimensões. Essas arrays são convenientes para representar dados organizados, por exemplo, em forma matricial (linhas e colunas). Utilizaremos arrays bidimensionais (dois índices) para representar matrizes. Chamaremos de matrizes as arrays bidimensionais e de vetores as arrays unidimensionais. Por exemplo, se quiséssemos representar a matriz identidade de ordem Id 3 = poderíamos utilizar o seguinte código em Fortran 90 i n t e g e r : : i, j i n t e g e r, dimension ( 3, 3 ) : : identidade do i = 1, 3 do j = 1, 3 i f ( i /= j ) then identidade ( i, j ) = 0 identidade ( i, j ) = 1 1 Por falta de uma melhor tradução da palavra, iremos utilizar o termo em inglês neste documento. 2

3 Neste trecho de código utilizamos as variáveis i e j para representar os valores das linhas e colunas, respectivamente. O laço externo percorre as linhas da matriz, equanto que o laço interno percorre as colunas. Para cada elemento da matriz é atribuído o valor 0, caso a posição deste elemento esteja fora da diagonal (i j), ou o valor 1, caso o elemento esteja na diagonal (i = j) Alocação dinâmica de arrays Em todos os trechos de código acima, as declarações das arrays são do tipo denominado alocação estática da memória, isso porque o tamanho de cada array é escolhido no momento da declaração e não poderá mudar. Desta maneira, o tamanho da array deverá ser grande o suficiente para que todos os valores possam ser armazenados sem problemas. A alocação estática de memória pode apresentar sérias limitações. Por exemplo, suponha que você declare uma array de 1000 posições. Essa array irá ocupar parte da memória do computador para armazenar 1000 números, mesmo que você só utilize 2 posições dela, ou seja, você estaria desperdiçando 998 posições de memória. Uma solução para este problema é a chamada alocação dinâmica da memória, exemplificada no seguinte código i n t e g e r, a l l o c a t a b l e, dimension ( :, : ) : : identidade O código acima declara a matriz identidade com alocação dinâmica da memória. Observe que os sinais de dois pontos, :, entre os parênteses, são obrigatórios, assim como a palavra allocatable. Observe também que a vírgula após o primeiro sinal de dois pontos foi necessária, pois estamos declarando uma array bidimensional (uma matriz). Com este tipo de declaração, nenhuma memória é reservada. Para utilizarmos a matriz identidade, temos que alocar (reservar) memória antes. Veja o seguinte código i n t e g e r, a l l o c a t a b l e, dimension ( :, : ) : : identidade a l l o c a t e ( identidade ( 3, 3 ) ) Para alocarmos memória, utilizamos a função allocate com o nome da matriz e o tamanho que quisermos 2. Feito isto, podemos atribuir valores aos elementos da matriz normalmente. Após utilizarmos a matriz, poderemos liberar a memória para ela reservada, utilizando a função deallocate, como a seguir d e a l l o c a t e ( identidade ) 2.2 Funções e subrotinas Utilizamos certos procedimentos em Fortran 90, tais como funções e subrotinas, para organizar melhor nossos programas. Essa organização facilita a correção de erros, estruturação do programa em tarefas e subtarefas, etc. Iremos apenas mostrar como utilizar funções e subrotinas em Fortran 90. Para mais informações sobre este assunto, sugerimos o livro [1]. Basicamente, subrotinas são procedimentos que são executados por uma instrução chamada CALL. Subrotinas podem retornar múltiplos resultados através da sua lista de argumentos de chamada. Funções são procedimentos que são executados pelos próprios 2 Isto é, caso haja memória suficiente para alocar a matriz. 3

4 nomes em expressões e só podem retornar um único resultado a ser usado na expressão. Os exemplos a seguir dão a forma geral de uma subrotina e de uma função, respectivamente s u b r o u t i n e minha_subrotina ( lista de argumentos ) (declarações de variáveis) (código a ser executado) r e t u r n f u n c t i o n minha_funcao ( lista de argumentos ) (declarações de variáveis) (código a ser executado) minha_funcao = expressão r e t u r n end f u n c t i o n Tanto para subrotinas quanto para funções, é possível estabelecer uma lista de argumentos que serão utilizados pela subrotina ou função. Vale destacar que na função, é necessário que haja uma atribuição de valor ao nome da função, indicando o valor que será retornado pela mesma. 3 Implementação O código a seguir é a implementação em Fortran 90 das operações básicas sobre matrizes e o cálculo do determinante de uma matriz pela sua decomposição em matrizes triangulares inferior e superior, conhecida como decomposição LU. O algoritmo utilizado é conhecido como algoritmo de Doolittle. Utilizamos um algoritmo simples para o cálculo da inversa de uma matriz. Esse algoritmo é baseado no processo de eliminação gaussiana e tem um caráter mais didático, pois seu desempenho é baixo. Tentamos, exageradamente, comentar o código para facilitar a compreensão do material. Utilizamos várias técnicas e construções sintáticas para mostrar várias formas de implementação. O código é extenso devido ao seu caráter didático, mas você poderá modificá-lo à vontade, buscando melhores formas de expor os algoritmos ou, até mesmo, utilizar algoritmos mais eficientes e precisos É aconselhável que você digite (ao invés de copiar-e-colar) todo o código e tente executá-lo, pois assim, estará treinando suas habilidades para detectar e corrigir erros, caso ocorram. Vale ressaltar que este é um código com caráter didático e não deve ser utilizado em outros ambientes que não os de sala de aula. Algumas instruções em Fortran 90 possuem opções até agora não utilizadas, como por exemplo, a opção advance= no na instrução write. Não detalharemos o significado dessas opções, mas todas elas são facilmente encontradas em [1]. 4

5 A variável seed será compartilhada com a subrotina que cria os elementos das matrizes A e B aleatoriamente. Isto é necessário para que os valores das matrizes não sejam iguais. Módulos devem sempre vir definidos antes do seu uso no programa. module global i n t e g e r : : seed end module Início do programa program matrizes use global Propósito: Criar procedimentos para somar, multiplicar, transpor, inverter e decompor matrizes e calcular o determinante de uma matriz. Data: Programador: Juvenal Muniz Revisões : Código original Declarações das variáveis e das subrotinas utilizadas Descrição das variáveis e/ou matrizes n Dimensão da matriz i Contador condicao Indica se houve erro ao abrir o arquivo opcao Armazena a opcao selecionada pelo usuário determinante Valor do determinante da matriz a, b, c Matrizes a, b e c arquivo Constante utilizada para acessar o arquivo resposta Armazenará a resposta à pergunta (S ou N) nome arquivo Nome do arquivo contendo os elementos das matrizes Descrição das subrotinas matriz aleatoria Gera uma matriz quadrada aleatória de ordem n matriz adicao Efetua a adição de duas matrizes matriz mul Efetua a multiplicação de duas matrizes matriz transposta Obtém a transposta de uma matriz matriz inversa Obtém a inversa de uma matriz matriz lu Decompõe a matriz em matrizes triangulares inferior e superior matriz imprime Exibe os elementos da matriz i n t e g e r : : n, i, j, condicao, opcao r e a l : : determinante r e a l, a l l o c a t a b l e, dimension ( :, : ) : : a, b, c i n t e g e r, parameter : : arquivo = 17 c h a r a c t e r ( 1 ) : : resposta c h a r a c t e r ( 2 0 ) : : nome_arquivo Valor inicial para a função RAN() seed = Leitura da dimensão das matrizes quadradas w r i t e (, ' (A) ', advance= ' no ' ) ' Qual e a dimensao das m a t r i z e s? ' read (, ' ( I ) ' ) n Verifica se a dimensão é maior ou igual a 2. Caso contrário, interrompe o programa. i f ( n >= 2) then a l l o c a t e ( a ( n, n ), b ( n, n ), c ( n, n ) ) w r i t e (, ) ' As m a t r i z e s devem p o s s u i r dimensao maior ou i g u a l a 2 ' Pede confirmação para gerar as matrizes aleatoriamente w r i t e (, ' (A) ', advance= ' no ' ) ' Deseja que as m a t r i z e s a e b sejam geradas? ( S ou N) ' read (, ' (A1) ' ) resposta Verifica a resposta dada pelo usuário i f ( ( resposta == ' S ' ). or. ( resposta == ' s ' ) ) then Usuário prefere que as matrizes sejam geradas aleatoriamente c a l l matriz_aleatoria ( a, n ) 5

6 c a l l matriz_aleatoria ( b, n ) Usuário prefere fornecer as matrizes em um arquivo w r i t e (, ' ( /, x,a) ' ) 'ATENCAO Termine o arquivo com uma l i n h a em branco. ' w r i t e (, ' ( x,a) ', advance= ' no ' ) 'Nome do arquivo contendo as m a t r i z e s (max. 20 c a r a c t e r e s ) : ' read (, ' ( A20 ) ' ) nome_arquivo Abrindo o nome arquivo para leitura. status= old força que nome arquivo já exista iostat=condicao armazena informações sobre erros ao abrir o arquivo open ( u n i t=arquivo, f i l e =nome_arquivo, s t a t u s= ' old ', a c t i o n= ' read ', i o s t a t=condicao ) O nome arquivo abriu sem problemas? i f ( condicao == 0) then Arquivo aberto Agora vamos ler os valores das matrizes a e b read ( arquivo, ) ( ( a ( i, j ), j = 1, n ), i = 1, n ) read ( arquivo, ) ( ( b ( i, j ), j = 1, n ), i = 1, n ) w r i t e (, ' ( /, x,a) ' ) ' Matrizes l i d a s com s u c e s s o ' Erro ao abrir o arquivo. Terminar o programa. w r i t e (, ' ( /, x,a) ' ) ' Erro ao a b r i r o arquivo V e r i f i q u e o arquivo de entrada. ' c l o s e ( arquivo ) Fechar o arquivo, pois já temos os valores armazenados nas variáveis c l o s e ( arquivo ) Menu para seleção das operações matriciais implementadas Este laço se repetirá até que o usuário escolha a opção 0. do w r i t e (, ' ( 3 /, x,a) ' ) ' ( 1 ) Somar as m a t r i z e s A e B ' w r i t e (, ) ' ( 2 ) M u l t i p l i c a r as m a t r i z e s A e B ' w r i t e (, ) ' ( 3 ) Obter a t r a n s p o s t a de A ' w r i t e (, ) ' ( 4 ) Obter a i n v e r s a de A ' w r i t e (, ) ' ( 5 ) C a l c u l a r o determinante de A ' w r i t e (, ) ' ( 6 ) E x i b i r as m a t r i z e s A e B ' w r i t e (, ) ' ( 0 ) S a i r ' w r i t e (, ' ( /,A) ', advance= ' no ' ) ' D i g i t e a opcao d e s e j a d a (0 6) : ' read (, ) opcao i f ( opcao == 1) then C = A + B c a l l matriz_soma ( a, b, c, n ) w r i t e (, ' ( /, x,a) ' ) 'A soma da matriz A com B e i g u a l a ' c a l l matriz_imprime ( c, n ) i f ( opcao == 2) then C = A * B c a l l matriz_mul ( a, b, c, n ) w r i t e (, ' ( /, x,a) ' ) 'O produto da matriz A com B e i g u a l a ' c a l l matriz_imprime ( c, n ) i f ( opcao == 3) then Matriz transposta de A w r i t e (, ' ( /, x,a) ' ) 'A matriz t r a n s p o s t a de A e i g u a l a ' w r i t e (, ' ( /, x,a) ' ) 'MATRIX A ' c a l l matriz_imprime ( a, n ) c a l l matriz_transposta ( a, n ) w r i t e (, ' ( /, x,a) ' ) 'MATRIX A ( t r a n s p o s t a ) ' c a l l matriz_imprime ( a, n ) i f ( opcao == 4) then Matriz inversa de A w r i t e (, ' ( /, x,a) ' ) 'A matriz i n v e r s a de A e i g u a l a ' w r i t e (, ' ( /, x,a) ' ) 'MATRIX A ' c a l l matriz_imprime ( a, n ) c a l l matriz_inversa ( a, b, n ) w r i t e (, ' ( /, x,a) ' ) 'MATRIX B ( i n v e r s a ) ' c a l l matriz_imprime ( b, n ) c a l l matriz_inversa_verifica ( a, b, n ) 6

7 i f ( opcao == 5) then Decomposição de A em LU c a l l matriz_lu ( a, n ) Cálculo do determinante para uma matriz LU determinante = 1. 0 determinante = determinante a ( i, i ) w r i t e (, ' ( /, x,a) ' ) 'MATRIX A ' c a l l matriz_imprime ( a, n ) w r i t e (, ' ( /, x,a, F) ' ) 'O determinante da matriz A v a l e : ', determinante i f ( opcao == 6) then Exibe os elementos das matrizes A e B w r i t e (, ' ( /, x,a) ' ) 'MATRIX A ' c a l l matriz_imprime ( a, n ) w r i t e (, ' ( /, x,a) ' ) 'MATRIX B ' c a l l matriz_imprime ( b, n ) i f ( opcao == 0) then Interrompe o programa w r i t e (, ' ( /, x,a, / ) ' ) ' Programa terminado. Ate mais ' e x i t Libera a memória ocupada pelas matrizes A, B, C d e a l l o c a t e ( a, b, c ) end program Esta subrotina exibe os valores dos elementos de uma matriz Parâmetros: a Matriz cujos elementos serão exibidos n Dimensão da matriz (apenas matrizes quadradas) s u b r o u t i n e matriz_imprime ( a, n ) i n t e g e r : : n, i, j r e a l : : a ( n, n ) w r i t e (, ' ( f, 2 x ) ', advance= ' no ' ) a ( i, j ) p r i n t Esta subrotina gera uma matriz com valores aleatórios para seus elementos Parâmetros: a Matriz que armazenará os elementos gerados n Dimensão da matriz (apenas matrizes quadradas) s u b r o u t i n e matriz_aleatoria ( a, n ) use global i n t e g e r : : n, i, j r e a l : : a ( n, n ) Calcula valores aleatórios de -5 à +5 A função ran() retorna valores entre 0.0 e 1.0 a ( i, j ) = ( 1.0) ( i+j ) 5. 0 ran ( seed ) Esta subrotina faz a adição de duas matrizes quadradas (a + b) Parâmetros: a e b Matrizes de entrada 7

8 c Matriz que armazenará a soma de a com b n Dimensão da matriz (apenas matrizes quadradas) s u b r o u t i n e matriz_soma ( a, b, c, n ) i n t e g e r : : n, i, j r e a l : : a ( n, n ), b ( n, n ), c ( n, n ) c ( i, j ) = a ( i, j ) + b ( i, j ) Esta subrotina faz a multiplicação de duas matrizes quadradas (a * b) Parâmetros: a e b Matrizes de entrada c Matriz que armazenará o produto de a por b n Dimensão da matriz (apenas matrizes quadradas) s u b r o u t i n e matriz_mul ( a, b, c, n ) i n t e g e r : : n, i, j, k r e a l : : a ( n, n ), b ( n, n ), c ( n, n ) c = 0. 0 do k = 1, n c ( i, j ) = c ( i, j ) + a ( i, k ) b ( k, j ) Esta subrotina cria a transposta de uma matriz Parâmetros: a Matriz de entrada n Dimensão da matriz (apenas matrizes quadradas) AVISO: A matriz a será modificada no processo s u b r o u t i n e matriz_transposta ( a, n ) i n t e g e r : : n, i, j r e a l : : a ( n, n ), temp i f ( i < j ) then temp = a ( i, j ) a ( i, j ) = a ( j, i ) a ( j, i ) = temp Esta subrotina calcula a inversa de uma matriz por eliminação gaussiana Parâmetros: matriz Matriz de entrada inversa Matriz que armazenará a inversa n Dimensão da matriz (apenas matrizes quadradas) AVISO: A matriz a será modificada no processo Referência : mcfarlat/inverse.htm inverse.html s u b r o u t i n e matriz_inversa ( matriz, inversa, n ) i n t e g e r : : n 8

9 r e a l, dimension ( n, n ) : : matriz r e a l, dimension ( n, n ) : : inversa l o g i c a l : : invertivel =. true. i n t e g e r : : i, j, k, l r e a l : : m r e a l, dimension ( n, 2 n ) : : matriz_aumentada Matriz aumentada com uma matriz identidade do j = 1, 2 n i f ( j <= n ) then matriz_aumentada ( i, j ) = matriz ( i, j ) i f ( ( i+n ) == j ) then matriz_aumentada ( i, j ) = 1 matriz_aumentada ( i, j ) = 0 e n d i f Reduzir a matriz aumentada a uma matriz triangular superior pela eliminação gaussiana do k = 1, n 1 Verifica se algum elemento da diagonal é zero i f ( abs ( matriz_aumentada ( k, k ) ) <= 1. 0 E 6) then invertivel =. false. do i = k + 1, n Verifica se os elementos são maiores que zero i f ( abs ( matriz_aumentada ( i, k ) ) > 1. 0 E 6) then do j = 1, 2 n matriz_aumentada ( k, j ) = matriz_aumentada ( k, j ) + matriz_aumentada ( i, j ) invertivel =. true. e x i t e n d i f Se algum elemento da diagonal for zero, não podemos calcular a inversa i f ( invertivel ==. false. ) then w r i t e (, ' ( /, x,a, / ) ' ) ' A matriz nao e i n v e r s t i v e l ' inversa = 0 e n d i f e n d i f Eliminação gaussiana do j = k + 1, n m = matriz_aumentada ( j, k ) / matriz_aumentada ( k, k ) do i = k, 2 n matriz_aumentada ( j, i ) = matriz_aumentada ( j, i ) m matriz_aumentada ( k, i ) Teste para invertibilidade Elementos da diagonal não podem ser zero i f ( abs ( matriz_aumentada ( i, i ) ) <= 1. 0 E 6) then w r i t e (, ' ( /, x,a, / ) ' ) ' A matriz nao e i n v e r s t i v e l ' inversa = 0 e n d i f Elementos da diagonal iguais a 1 m = matriz_aumentada ( i, i ) do j = i, 2 n matriz_aumentada ( i, j ) = matriz_aumentada ( i, j ) / m Reduzir o lado esquerdo da matriz aumentada a matriz identidade do k = n 1, 1, 1 do i = 1, k 9

10 m = matriz_aumentada ( i, k+1) do j = k, 2 n matriz_aumentada ( i, j ) = matriz_aumentada ( i, j ) matriz_aumentada ( k+1, j ) m Armazene o resultado do i =1, n inversa ( i, j ) = matriz_aumentada ( i, j + n ) end s u b r o u t i n e Esta subrotina verifica se a matriz inversa é válida (a * b = identidade) Parâmetros: a Matriz original b Matriz inversa de a n Dimensão da matriz (apenas matrizes quadradas) s u b r o u t i n e matriz_inversa_verifica ( a, b, n ) i n t e g e r : : n, i, j r e a l, dimension ( n, n ) : : a, b, identidade l o g i c a l : : ok ok =. true. c a l l matriz_mul ( a, b, identidade, n ) Elementos fora da diagonal principal não podem ser diferentes de zero i f ( ( i /= j ). and. ( abs ( identidade ( i, j ) ) >= 1. 0 E 6) ) then ok =. false. Elementos na diagonal principal não podem ser diferentes de um i f ( ( i == j ). and. ( abs ( identidade ( i, i ) 1. 0 ) >= 1. 0 E 6) ) then ok =. false. i f ( ok ==. false. ) then w r i t e (, ' ( /, x,a, / ) ' ) 'A v e r f i c a c a o da matriz i n v e r s a f a l h o u ' w r i t e (, ' ( /, x,a, / ) ' ) 'A v e r f i c a c a o da matriz i n v e r s a f o i c o n c l u i d a com s u c e s s o ' end s u b r o u t i n e Esta subrotina decompõe uma matriz em matrizes triangulares inferior e superior (LU) pelo algoritmo de Doolittle (diagonal principal da matriz L igual a 1). Para uma descrição teórica sobre o algoritmo de Doolittle, veja [2]. Parâmetros: a Matriz que armazenará as matrizes LU n Dimensão da matriz (apenas matrizes quadradas) AVISO: A matriz a será modificada no processo s u b r o u t i n e matriz_lu ( a, n ) i n t e g e r : : n, i, j, k r e a l : : a ( n, n ), s, ss Os elementos da diagonal não podem ser zero i f ( abs ( a ( 1, 1 ) ) <= 1. 0 E 6) then w r i t e (, ' ( /, x,a, / ) ' ) ' I m p o s s i v e l f a t o r a r a matriz ' do i = 2, n a ( i, 1 ) = a ( i, 1 ) / a ( 1, 1 ) 10

11 end do do i = 2, n 1 s = 0. 0 do k = 1, i 1 s = s a ( i, k ) a ( k, i ) a ( i, i ) = ( a ( i, i ) + s ) Os elementos da diagonal não podem ser zero i f ( abs ( a ( i, i ) ) <= 1. 0 E 6) then w r i t e (, ' ( /, x,a, / ) ' ) ' i m p o s s i v e l f a t o r a r a matriz ' do j = i + 1, ss = 0. 0 s = 0. 0 n do k = 1, i 1 ss = ss a ( i, k ) a ( k, j ) s = s a ( j, k ) a ( k, i ) a ( i, j ) = a ( i, j ) + ss a ( j, i ) = ( a ( j, i ) + s ) / a ( i, i ) s = 0. 0 do k = 1, n 1 s = s a ( n, k ) a ( k, n ) a ( n, n ) = a ( n, n ) + s Referências [1] Stephen J. Chapman, Fortran 95/2003 for scientists and engineers, McGrall-Hill Companies, New York, NY, [2] R.L. Burden e J.D. Faires, Numerical analysis, Brooks Cole, California, CA, [3] J.L. Boldrini et al., Álgebra linear, Editora Harbra, Ltda., São Paulo, SP,