/[lmdze]/trunk/filtrez/inifilr.f
ViewVC logotype

Diff of /trunk/filtrez/inifilr.f

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 154 by guez, Tue Jul 7 17:49:23 2015 UTC revision 168 by guez, Wed Sep 9 10:41:47 2015 UTC
# Line 2  module inifilr_m Line 2  module inifilr_m
2    
3    IMPLICIT NONE    IMPLICIT NONE
4    
   INTEGER jfiltnu, jfiltsu, jfiltnv, jfiltsv  
   ! jfiltn index of the last scalar line filtered in NH  
   ! jfilts index of the first line filtered in SH  
   
5    ! North:    ! North:
   real, allocatable:: matriceun(:, :, :), matrinvn(:, :, :)  
   ! (iim, iim, 2:jfiltnu)  
6    
7    real, allocatable:: matricevn(:, :, :) ! (iim, iim, jfiltnv)    INTEGER jfiltnu, jfiltnv
8      ! index of the last scalar line filtered in northern hemisphere
9    
10      real, pointer:: matriceun(:, :, :) ! (iim, iim, jfiltnu - 1)
11      ! matrice filtre pour les champs situes sur la grille scalaire
12    
13      real, pointer:: matrinvn(:, :, :) ! (iim, iim, jfiltnu - 1)
14      ! matrice filtre pour les champs situes sur la grille scalaire, pour
15      ! le filtre inverse
16    
17      real, pointer:: matricevn(:, :, :) ! (iim, iim, jfiltnv)
18      ! matrice filtre pour les champs situes sur la grille de V ou de Z
19    
20    ! South:    ! South:
   real, allocatable:: matriceus(:, :, :), matrinvs(:, :, :)  
   ! (iim, iim, jfiltsu:jjm)  
21    
22    real, allocatable:: matricevs(:, :, :) ! (iim, iim, jfiltsv:jjm)    integer jfiltsu, jfiltsv
23      ! index of the first line filtered in southern hemisphere
24    
25      real, pointer:: matriceus(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
26      ! matrice filtre pour les champs situes sur la grille scalaire
27    
28      real, pointer:: matrinvs(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
29      ! matrice filtre pour les champs situes sur la grille scalaire, pour
30      ! le filtre inverse
31    
32      real, pointer:: matricevs(:, :, :) ! (iim, iim, jjm - jfiltsv + 1)
33      ! matrice filtre pour les champs situes sur la grille de V ou de Z
34    
35  contains  contains
36    
37    SUBROUTINE inifilr    SUBROUTINE inifilr
38    
39      ! From filtrez/inifilr.F, version 1.1.1.1 2004/05/19 12:53:09      ! From filtrez/inifilr.F, version 1.1.1.1, 2004/05/19 12:53:09
40      ! H. Upadhyaya, O. Sharma      ! H. Upadhyaya, O. Sharma
41    
42      ! This routine computes the eigenfunctions of the laplacian on the      ! This procedure computes the filtering coefficients for scalar
43      ! stretched grid, and the filtering coefficients. The modes are      ! lines and meridional wind v lines. The modes are filtered from
44      ! filtered from modfrst to iim.      ! modfrst to iim. We filter all those latitude lines where coefil
45        ! < 1. No filtering at poles. colat0 is to be used when alpha
46        ! (stretching coefficient) is set equal to zero for the regular
47        ! grid case.
48    
49      USE dimens_m, ONLY : iim, jjm      USE dimens_m, ONLY : iim, jjm
50      USE dynetat0_m, ONLY : rlatu, rlatv, xprimu, grossismx      USE dynetat0_m, ONLY : rlatu, rlatv, xprimu, grossismx
51      use inifgn_m, only: inifgn      use inifgn_m, only: inifgn
52        use inifilr_hemisph_m, only: inifilr_hemisph
53      use jumble, only: new_unit      use jumble, only: new_unit
54      use nr_util, only: pi      use nr_util, only: pi, ifirstloc
55    
56      ! Local:      ! Local:
57    
58      REAL dlatu(jjm)      REAL dlatu(jjm)
59      REAL rlamda(2: iim)      REAL rlamda(2:iim) ! > 0, in descending order
60      real eignvl(iim) ! eigenvalues sorted in descending order      real eignvl(iim) ! eigenvalues (<= 0) sorted in descending order
61      REAL cof      INTEGER j, unit
     INTEGER i, j, k, unit  
62      REAL colat0 ! > 0      REAL colat0 ! > 0
63      REAL eignft(iim, iim), coff      integer j1 ! index of smallest positive latitude
64    
65      real eignfnu(iim, iim), eignfnv(iim, iim)      real eignfnu(iim, iim), eignfnv(iim, iim)
66      ! eigenfunctions of the discrete laplacian      ! eigenvectors of the discrete second derivative with respect to longitude
   
     ! Filtering coefficients (lamda_max * cos(rlat) / lamda):  
     real coefilu(iim, jjm), coefilv(iim, jjm)  
     real coefilu2(iim, jjm), coefilv2(iim, jjm)  
   
     ! Index of the mode from where modes are filtered:  
     integer, allocatable:: modfrstnu(:), modfrstsu(:)  
     integer, allocatable:: modfrstnv(:), modfrstsv(:)  
67    
68      !-----------------------------------------------------------      !-----------------------------------------------------------
69    
# Line 61  contains Line 71  contains
71    
72      CALL inifgn(eignvl, eignfnu, eignfnv)      CALL inifgn(eignvl, eignfnu, eignfnv)
73    
     ! compute eigenvalues and eigenfunctions  
     ! compute the filtering coefficients for scalar lines and  
     ! meridional wind v-lines  
     ! we filter all those latitude lines where coefil < 1  
     ! NO FILTERING AT POLES  
     ! colat0 is to be used when alpha (stretching coefficient)  
     ! is set equal to zero for the regular grid case  
   
74      ! Calcul de colat0      ! Calcul de colat0
75      forall (j = 1:jjm) dlatu(j) = rlatu(j) - rlatu(j + 1)      forall (j = 1:jjm) dlatu(j) = rlatu(j) - rlatu(j + 1)
76      colat0 = min(0.5, minval(dlatu) / minval(xprimu(:iim)))      colat0 = min(0.5, minval(dlatu) / minval(xprimu(:iim)))
77      PRINT *, 'colat0 = ', colat0      PRINT *, 'colat0 = ', colat0
78    
79      rlamda = iim / (pi * colat0 / grossismx) / sqrt(abs(eignvl(2: iim)))      rlamda = iim / (pi * colat0 / grossismx) / sqrt(- eignvl(2: iim))
80        print *, "1 / rlamda(iim) = ", 1. / rlamda(iim)
81        call new_unit(unit)
82        open(unit, file = "modfrst.csv", status = "replace", action = "write")
83        write(unit, fmt = *) '"rlat (degrees)" modfrst' ! title line
84    
85      ! Determination de jfiltnu, jfiltsu, jfiltnv, jfiltsv      j1 = ifirstloc(rlatu <= 0.)
86    
87      jfiltnu = (jjm + 1) / 2      call inifilr_hemisph(rlatu(j1 - 1:2:- 1), colat0, rlamda, unit, eignfnv, &
88      do while (cos(rlatu(jfiltnu)) >= colat0 &           jfiltnu, matriceun, matrinvn)
89           .or. rlamda(iim) * cos(rlatu(jfiltnu)) >= 1.)      jfiltnu = j1 - jfiltnu
90         jfiltnu = jfiltnu - 1      matriceun = matriceun(:, :, jfiltnu - 1:1:- 1)
91      end do      matrinvn = matrinvn(:, :, jfiltnu - 1:1:- 1)
92    
93      jfiltsu = jjm / 2 + 2      call inifilr_hemisph(- rlatu(j1:jjm), colat0, rlamda, unit, eignfnv, &
94      do while (cos(rlatu(jfiltsu)) >= colat0 &           jfiltsu, matriceus, matrinvs)
95           .or. rlamda(iim) * cos(rlatu(jfiltsu)) >= 1.)      jfiltsu = j1 - 1 + jfiltsu
96         jfiltsu = jfiltsu + 1  
97      end do      j1 = ifirstloc(rlatv <= 0.)
98    
99      jfiltnv = jjm / 2      call inifilr_hemisph(rlatv(j1 - 1:1:- 1), colat0, rlamda, unit, eignfnu, &
100      do while ((cos(rlatv(jfiltnv)) >= colat0 &           jfiltnv, matricevn)
101           .or. rlamda(iim) * cos(rlatv(jfiltnv)) >= 1.) .and. jfiltnv >= 2)      jfiltnv = j1 - jfiltnv
102         jfiltnv = jfiltnv - 1      matricevn = matricevn(:, :, jfiltnv:1:- 1)
103      end do  
104        call inifilr_hemisph(- rlatv(j1:jjm), colat0, rlamda, unit, eignfnu, &
105      if (cos(rlatv(jfiltnv)) >= colat0 &           jfiltsv, matricevs)
106           .or. rlamda(iim) * cos(rlatv(jfiltnv)) >= 1.) then      jfiltsv = j1 - 1 + jfiltsv
        ! {jfiltnv == 1}  
        PRINT *, 'Could not find jfiltnv.'  
        STOP 1  
     END IF  
   
     jfiltsv = (jjm + 1)/ 2 + 1  
     do while ((cos(rlatv(jfiltsv)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatv(jfiltsv)) >= 1.) .and. jfiltsv <= jjm - 1)  
        jfiltsv = jfiltsv + 1  
     end do  
   
     IF (cos(rlatv(jfiltsv)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatv(jfiltsv)) >= 1.) THEN  
        ! {jfiltsv == jjm}  
        PRINT *, 'Could not find jfiltsv.'  
        STOP 1  
     END IF  
107    
108        close(unit)
109      PRINT *, 'jfiltnu =', jfiltnu      PRINT *, 'jfiltnu =', jfiltnu
110      PRINT *, 'jfiltsu =', jfiltsu      PRINT *, 'jfiltsu =', jfiltsu
111      PRINT *, 'jfiltnv =', jfiltnv      PRINT *, 'jfiltnv =', jfiltnv
112      PRINT *, 'jfiltsv =', jfiltsv      PRINT *, 'jfiltsv =', jfiltsv
113    
     ! Determination de coefilu, coefilv, modfrst[ns][uv]:  
   
     allocate(modfrstnu(2:jfiltnu), modfrstsu(jfiltsu:jjm))  
     allocate(modfrstnv(jfiltnv), modfrstsv(jfiltsv:jjm))  
     coefilu = 0.  
     coefilv = 0.  
     coefilu2 = 0.  
     coefilv2 = 0.  
   
     DO j = 2, jfiltnu  
        modfrstnu(j) = 2  
        do while (rlamda(modfrstnu(j)) * cos(rlatu(j)) >= 1. &  
             .and. modfrstnu(j) <= iim - 1)  
           modfrstnu(j) = modfrstnu(j) + 1  
        end do  
   
        if (rlamda(modfrstnu(j)) * cos(rlatu(j)) < 1.) then  
           DO k = modfrstnu(j), iim  
              cof = rlamda(k) * cos(rlatu(j))  
              coefilu(k, j) = cof - 1.  
              coefilu2(k, j) = cof**2 - 1.  
           end DO  
        end if  
     END DO  
   
     DO j = 1, jfiltnv  
        modfrstnv(j) = 2  
        do while (rlamda(modfrstnv(j)) * cos(rlatv(j)) >= 1. &  
             .and. modfrstnv(j) <= iim - 1)  
           modfrstnv(j) = modfrstnv(j) + 1  
        end do  
   
        if (rlamda(modfrstnv(j)) * cos(rlatv(j)) < 1.) then  
           DO k = modfrstnv(j), iim  
              cof = rlamda(k) * cos(rlatv(j))  
              coefilv(k, j) = cof - 1.  
              coefilv2(k, j) = cof**2 - 1.  
           end DO  
        end if  
     end DO  
   
     DO j = jfiltsu, jjm  
        modfrstsu(j) = 2  
        do while (rlamda(modfrstsu(j)) * cos(rlatu(j)) >= 1. &  
             .and. modfrstsu(j) <= iim - 1)  
           modfrstsu(j) = modfrstsu(j) + 1  
        end do  
   
        if (rlamda(modfrstsu(j)) * cos(rlatu(j)) < 1.) then  
           DO k = modfrstsu(j), iim  
              cof = rlamda(k) * cos(rlatu(j))  
              coefilu(k, j) = cof - 1.  
              coefilu2(k, j) = cof**2 - 1.  
           end DO  
        end if  
     end DO  
   
     DO j = jfiltsv, jjm  
        modfrstsv(j) = 2  
        do while (rlamda(modfrstsv(j)) * cos(rlatv(j)) >= 1. &  
             .and. modfrstsv(j) <= iim - 1)  
           modfrstsv(j) = modfrstsv(j) + 1  
        end do  
   
        if (rlamda(modfrstsv(j)) * cos(rlatv(j)) < 1.) then  
           DO k = modfrstsv(j), iim  
              cof = rlamda(k) * cos(rlatv(j))  
              coefilv(k, j) = cof - 1.  
              coefilv2(k, j) = cof**2 - 1.  
           end DO  
        end if  
     END DO  
   
     call new_unit(unit)  
     open(unit, file = "inifilr_out.txt", status = "replace", action = "write")  
     write(unit, fmt = *) '"EIGNVL"', eignvl  
     write(unit, fmt = *) '"modfrstnu"', modfrstnu  
     write(unit, fmt = *) '"modfrstsu"', modfrstsu  
     write(unit, fmt = *) '"modfrstnv"', modfrstnv  
     write(unit, fmt = *) '"modfrstsv"', modfrstsv  
     close(unit)  
   
     allocate(matriceun(iim, iim, 2:jfiltnu), matrinvn(iim, iim, 2:jfiltnu))  
     allocate(matricevn(iim, iim, jfiltnv))  
     allocate(matricevs(iim, iim, jfiltsv:jjm))  
     allocate(matriceus(iim, iim, jfiltsu:jjm), matrinvs(iim, iim, jfiltsu:jjm))  
   
     ! Calcul de la matrice filtre 'matriceu' pour les champs situes  
     ! sur la grille scalaire  
   
     DO j = 2, jfiltnu  
        DO i = 1, iim  
           IF (i < modfrstnu(j)) then  
              coff = 0.  
           else  
              coff = coefilu(i, j)  
           end IF  
           eignft(i, :) = eignfnv(:, i) * coff  
        END DO  
        matriceun(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
     DO j = jfiltsu, jjm  
        DO i = 1, iim  
           IF (i < modfrstsu(j)) then  
              coff = 0.  
           else  
              coff = coefilu(i, j)  
           end IF  
           eignft(i, :) = eignfnv(:, i) * coff  
        END DO  
        matriceus(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
     ! Calcul de la matrice filtre 'matricev' pour les champs situes  
     ! sur la grille de V ou de Z  
   
     DO j = 1, jfiltnv  
        DO i = 1, iim  
           IF (i < modfrstnv(j)) then  
              coff = 0.  
           else  
              coff = coefilv(i, j)  
           end IF  
           eignft(i, :) = eignfnu(:, i) * coff  
        END DO  
        matricevn(:, :, j) = matmul(eignfnu, eignft)  
     END DO  
   
     DO j = jfiltsv, jjm  
        DO i = 1, iim  
           IF (i < modfrstsv(j)) then  
              coff = 0.  
           else  
              coff = coefilv(i, j)  
           end IF  
           eignft(i, :) = eignfnu(:, i) * coff  
        END DO  
        matricevs(:, :, j) = matmul(eignfnu, eignft)  
     END DO  
   
     ! Calcul de la matrice filtre 'matrinv' pour les champs situes  
     ! sur la grille scalaire , pour le filtre inverse  
   
     DO j = 2, jfiltnu  
        DO i = 1, iim  
           IF (i < modfrstnu(j)) then  
              coff = 0.  
           else  
              coff = coefilu(i, j) / (1. + coefilu(i, j))  
           end IF  
           eignft(i, :) = eignfnv(:, i) * coff  
        END DO  
        matrinvn(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
     DO j = jfiltsu, jjm  
        DO i = 1, iim  
           IF (i < modfrstsu(j)) then  
              coff = 0.  
           else  
              coff = coefilu(i, j) / (1. + coefilu(i, j))  
           end IF  
           eignft(i, :) = eignfnv(:, i) * coff  
        END DO  
        matrinvs(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
114    END SUBROUTINE inifilr    END SUBROUTINE inifilr
115    
116  end module inifilr_m  end module inifilr_m
Legend:
Removed from v.154  
changed lines
  Added in v.168
  ViewVC Help
Powered by ViewVC 1.1.21