--- trunk/libf/filtrez/filtreg.f90 2013/07/08 18:12:18 71 +++ trunk/Sources/filtrez/filtreg.f 2015/04/30 14:22:32 135 @@ -4,106 +4,75 @@ contains - SUBROUTINE filtreg(champ, nlat, nbniv, ifiltre, iaire, griscal) + SUBROUTINE filtreg(champ, direct, intensive) ! From filtrez/filtreg.F, version 1.1.1.1, 2004/05/19 12:53:09 ! Author: P. Le Van - ! Objet : filtre matriciel longitudinal, avec les matrices précalculées - ! pour l'opérateur filtre. + ! Objet : filtre matriciel longitudinal, avec les matrices pr\'ecalcul\'ees + ! pour l'op\'erateur filtre. USE coefils, ONLY: sddu, sddv, unsddu, unsddv USE dimens_m, ONLY: iim, jjm use inifilr_m, only: jfiltnu, jfiltnv, jfiltsu, jfiltsv, matriceun, & matriceus, matricevn, matricevs, matrinvn, matrinvs + use nr_util, only: assert - INTEGER, intent(in):: nlat ! nombre de latitudes a filtrer - integer, intent(in):: nbniv ! nombre de niveaux verticaux a filtrer + REAL, intent(inout):: champ(:, :, :) ! (iim + 1, nlat, nbniv) + ! en entr\'ee : champ \`a filtrer, en sortie : champ filtr\'e - REAL, intent(inout):: champ(iim + 1, nlat, nbniv) - ! en entrée : champ à filtrer, en sortie : champ filtré + logical, intent(in):: direct ! filtre direct ou inverse - integer, intent(in):: ifiltre - ! +1 Transformee directe - ! -1 Transformee inverse - ! +2 Filtre directe - ! -2 Filtre inverse - ! Variable Intensive - ! ifiltre = 1 filtre directe - ! ifiltre =-1 filtre inverse - ! Variable Extensive - ! ifiltre = 2 filtre directe - ! ifiltre =-2 filtre inverse - - integer, intent(in):: iaire - ! 1 si champ intensif - ! 2 si champ extensif (pondere par les aires) - - LOGICAL, intent(in):: griscal - - ! Variables local to the procedure: + logical, intent(in):: intensive + ! champ intensif ou extensif (pond\'er\'e par les aires) + ! Local: + LOGICAL griscal + INTEGER nlat ! nombre de latitudes \`a filtrer + integer nbniv ! nombre de niveaux verticaux \`a filtrer INTEGER jdfil1, jdfil2, jffil1, jffil2, jdfil, jffil - INTEGER i, j, l, k + INTEGER j, l REAL eignq(iim), sdd1(iim), sdd2(iim) INTEGER hemisph !----------------------------------------------------------- - IF (ifiltre==1 .OR. ifiltre==-1) then - print *, 'Pas de transformee simple dans cette version' - STOP 1 - end IF - - IF (ifiltre==-2 .AND. .NOT. griscal) THEN - PRINT *, ' Cette routine ne calcule le filtre inverse que ', & - ' sur la grille des scalaires !' - STOP 1 - END IF + call assert(size(champ, 1) == iim + 1, "filtreg iim + 1") + nlat = size(champ, 2) + nbniv = size(champ, 3) + call assert(nlat == jjm .or. nlat == jjm + 1, "filtreg nlat") + griscal = nlat == jjm + 1 - IF (ifiltre/=2 .AND. ifiltre/=-2) THEN - PRINT *, ' Probleme dans filtreg car ifiltre NE 2 et NE -2', & - ' corriger et repasser !' + IF (.not. direct .AND. nlat == jjm) THEN + PRINT *, 'filtreg: inverse filter on scalar grid only' STOP 1 END IF IF (griscal) THEN - IF (nlat /= jjm + 1) THEN - PRINT *, 'Erreur dans le dimensionnement du tableau CHAMP a ' & - // 'filtrer, sur la grille des scalaires' - STOP 1 + IF (intensive) THEN + sdd1 = sddv + sdd2 = unsddv ELSE - IF (iaire==1) THEN - sdd1 = sddv - sdd2 = unsddv - ELSE - sdd1 = unsddv - sdd2 = sddv - END IF - - jdfil1 = 2 - jffil1 = jfiltnu - jdfil2 = jfiltsu - jffil2 = jjm + sdd1 = unsddv + sdd2 = sddv END IF + + jdfil1 = 2 + jffil1 = jfiltnu + jdfil2 = jfiltsu + jffil2 = jjm ELSE - IF (nlat /= jjm) THEN - PRINT *, 'Erreur dans le dimensionnement du tableau CHAMP a ' & - // 'filtrer, sur la grille de V ou de Z' - STOP 1 + IF (intensive) THEN + sdd1 = sddu + sdd2 = unsddu ELSE - IF (iaire==1) THEN - sdd1 = sddu - sdd2 = unsddu - ELSE - sdd1 = unsddu - sdd2 = sddu - END IF - - jdfil1 = 1 - jffil1 = jfiltnv - jdfil2 = jfiltsv - jffil2 = jjm + sdd1 = unsddu + sdd2 = sddu END IF + + jdfil1 = 1 + jffil1 = jfiltnv + jdfil2 = jfiltsv + jffil2 = jjm END IF DO hemisph = 1, 2 @@ -115,90 +84,40 @@ jffil = jffil2 END IF - loop_vertical: DO l = 1, nbniv - loop_latitude: DO j = jdfil, jffil - DO i = 1, iim - champ(i, j, l) = champ(i, j, l)*sdd1(i) - END DO + DO l = 1, nbniv + DO j = jdfil, jffil + champ(:iim, j, l) = champ(:iim, j, l) * sdd1 IF (hemisph==1) THEN - IF (ifiltre==-2) THEN - DO k = 1, iim - eignq(k) = 0. - END DO - DO k = 1, iim - DO i = 1, iim - eignq(k) = eignq(k) + matrinvn(k, i, j)*champ(i, j, l) - END DO - END DO + IF (.not. direct) THEN + eignq = matmul(matrinvn(:, :, j), champ(:iim, j, l)) ELSE IF (griscal) THEN - DO k = 1, iim - eignq(k) = 0. - END DO - DO i = 1, iim - DO k = 1, iim - eignq(k) = eignq(k) + matriceun(k, i, j) & - * champ(i, j, l) - END DO - END DO + eignq = matmul(matriceun(:, :, j), champ(:iim, j, l)) ELSE - DO k = 1, iim - eignq(k) = 0. - END DO - DO i = 1, iim - DO k = 1, iim - eignq(k) = eignq(k) + matricevn(k, i, j) & - * champ(i, j, l) - END DO - END DO + eignq = matmul(matricevn(:, :, j), champ(:iim, j, l)) END IF ELSE - IF (ifiltre==-2) THEN - DO k = 1, iim - eignq(k) = 0. - END DO - DO i = 1, iim - DO k = 1, iim - eignq(k) = eignq(k) + matrinvs(k, i, j-jfiltsu+1) & - *champ(i, j, l) - END DO - END DO + IF (.not. direct) THEN + eignq = matmul(matrinvs(:, :, j - jfiltsu + 1), & + champ(:iim, j, l)) ELSE IF (griscal) THEN - DO k = 1, iim - eignq(k) = 0. - END DO - DO i = 1, iim - DO k = 1, iim - eignq(k) = eignq(k) + matriceus(k, i, j-jfiltsu+1) & - *champ(i, j , l) - END DO - END DO + eignq = matmul(matriceus(:, :, j - jfiltsu + 1), & + champ(:iim, j, l)) ELSE - DO k = 1, iim - eignq(k) = 0. - END DO - DO i = 1, iim - DO k = 1, iim - eignq(k) = eignq(k) + matricevs(k, i, j-jfiltsv+1) & - *champ(i, j , l) - END DO - END DO + eignq = matmul(matricevs(:, :, j - jfiltsv + 1), & + champ(:iim, j, l)) END IF END IF - IF (ifiltre==2) THEN - DO i = 1, iim - champ(i, j, l) = (champ(i, j, l)+eignq(i))*sdd2(i) - end DO + IF (direct) THEN + champ(:iim, j, l) = (champ(:iim, j, l) + eignq) * sdd2 ELSE - DO i = 1, iim - champ(i, j, l) = (champ(i, j, l)-eignq(i))*sdd2(i) - end DO + champ(:iim, j, l) = (champ(:iim, j, l) - eignq) * sdd2 END IF champ(iim + 1, j, l) = champ(1, j, l) - END DO loop_latitude - END DO loop_vertical + END DO + END DO end DO END SUBROUTINE filtreg