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

Diff of /trunk/filtrez/inifilr.f90

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

revision 164 by guez, Tue Jul 28 14:53:31 2015 UTC revision 175 by guez, Fri Feb 5 16:02:34 2016 UTC
# Line 2  module inifilr_m Line 2  module inifilr_m
2    
3    IMPLICIT NONE    IMPLICIT NONE
4    
   ! North:  
   
5    INTEGER jfiltnu, jfiltnv    INTEGER jfiltnu, jfiltnv
6    ! index of the last scalar line filtered in northern hemisphere    ! index of the last line filtered in northern hemisphere at rlat[uv]
7      ! latitudes
   real, allocatable:: matriceun(:, :, :), matrinvn(:, :, :)  
   ! (iim, iim, 2:jfiltnu)  
   
   real, allocatable:: matricevn(:, :, :) ! (iim, iim, jfiltnv)  
   
   ! South:  
8    
9    integer jfiltsu, jfiltsv    integer jfiltsu, jfiltsv
10    ! index of the first line filtered in southern hemisphere    ! index of the first line filtered in southern hemisphere at
11      ! rlat[uv] latitudes
12    
13    real, allocatable:: matriceus(:, :, :), matrinvs(:, :, :)    ! Filtre pour les champs situes sur la grille scalaire (longitudes
14    ! (iim, iim, jfiltsu:jjm)    ! rlonv, latitudes rlatu) :
15      real, pointer:: matriceun(:, :, :) ! (iim, iim, jfiltnu - 1)
16    real, allocatable:: matricevs(:, :, :) ! (iim, iim, jfiltsv:jjm)    real, pointer:: matriceus(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
17    
18      ! Filtre pour les champs situes sur la grille scalaire (longitudes
19      ! rlonv, latitudes rlatu), pour le filtre inverse :
20      real, pointer:: matrinvn(:, :, :) ! (iim, iim, jfiltnu - 1)
21      real, pointer:: matrinvs(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
22    
23      ! Filtre pour les champs situes sur la grille de la vorticit\'e
24      ! (longitudes rlonu, latitudes rlatv)
25      real, pointer:: matricevn(:, :, :) ! (iim, iim, jfiltnv) matrice
26      real, pointer:: matricevs(:, :, :) ! (iim, iim, jjm - jfiltsv + 1)
27    
28  contains  contains
29    
# Line 39  contains Line 42  contains
42      USE dimens_m, ONLY : iim, jjm      USE dimens_m, ONLY : iim, jjm
43      USE dynetat0_m, ONLY : rlatu, rlatv, xprimu, grossismx      USE dynetat0_m, ONLY : rlatu, rlatv, xprimu, grossismx
44      use inifgn_m, only: inifgn      use inifgn_m, only: inifgn
45        use inifilr_hemisph_m, only: inifilr_hemisph
46      use jumble, only: new_unit      use jumble, only: new_unit
47      use nr_util, only: pi      use nr_util, only: pi, ifirstloc, assert
48    
49      ! Local:      ! Local:
50    
51      REAL dlatu(jjm)      REAL dlatu(jjm)
52      REAL rlamda(2: iim)      REAL rlamda(2:iim) ! > 0, in descending order
53      real eignvl(iim) ! eigenvalues sorted in descending order (<= 0)      real eignvl(iim) ! eigenvalues (<= 0) sorted in descending order
54      INTEGER i, j, unit      INTEGER j, unit
55      REAL colat0 ! > 0      REAL colat0 ! > 0
56      REAL eignft(iim, iim)      integer j1 ! index of negative latitude closest to the equator
57    
58      real eignfnu(iim, iim), eignfnv(iim, iim)      real eignfnu(iim, iim), eignfnv(iim, iim)
59      ! eigenvectors of the discrete second derivative with respect to longitude      ! eigenvectors of the discrete second derivative with respect to
60        ! longitude, at rlon[uv] longitudes
     ! Filtering coefficients (lamda_max * cos(rlat) / lamda):  
     real, allocatable:: coefilnu(:, :) ! (iim, 2:jfiltnu)  
     real, allocatable:: coefilsu(:, :) ! (iim, jfiltsu:jjm)  
     real, allocatable:: coefilnv(:, :) ! (iim, jfiltnv)  
     real, allocatable:: coefilsv(:, :) ! (iim, jfiltsv:jjm)  
   
     ! Index of the mode from where modes are filtered:  
     integer, allocatable:: modfrstnu(:) ! (2:jfiltnu)  
     integer, allocatable:: modfrstsu(:) ! (jfiltsu:jjm)  
     integer, allocatable:: modfrstnv(:) ! (jfiltnv)  
     integer, allocatable:: modfrstsv(:) ! (jfiltsv:jjm)  
61    
62      !-----------------------------------------------------------      !-----------------------------------------------------------
63    
# Line 77  contains Line 70  contains
70      colat0 = min(0.5, minval(dlatu) / minval(xprimu(:iim)))      colat0 = min(0.5, minval(dlatu) / minval(xprimu(:iim)))
71      PRINT *, 'colat0 = ', colat0      PRINT *, 'colat0 = ', colat0
72    
73      rlamda = iim / (pi * colat0 / grossismx) / sqrt(- eignvl(2: iim))      rlamda = iim / pi / colat0 * grossismx / sqrt(- eignvl(2: iim))
74        print *, "1 / rlamda(iim) = ", 1. / rlamda(iim)
75        ! This is demonstrated in the notes but just to be sure:
76        call assert(rlamda(iim) * colat0 >= 1. - 2. * epsilon(0.), &
77             "inifilr rlamda(iim) * colat0")
78    
79        call new_unit(unit)
80        open(unit, file = "modfrst.csv", status = "replace", action = "write")
81        write(unit, fmt = *) '"rlat (degrees)" modfrst' ! title line
82    
83      ! Determination de jfiltnu, jfiltsu, jfiltnv, jfiltsv      j1 = ifirstloc(rlatu <= 0.)
84    
85      jfiltnu = (jjm + 1) / 2      call inifilr_hemisph(rlatu(j1 - 1:2:- 1), rlamda, unit, eignfnv, jfiltnu, &
86      do while (cos(rlatu(jfiltnu)) >= colat0 &           matriceun, matrinvn)
87           .or. rlamda(iim) * cos(rlatu(jfiltnu)) >= 1.)      jfiltnu = j1 - jfiltnu
88         jfiltnu = jfiltnu - 1      matriceun = matriceun(:, :, jfiltnu - 1:1:- 1)
89      end do      matrinvn = matrinvn(:, :, jfiltnu - 1:1:- 1)
90    
91      jfiltsu = jjm / 2 + 2      call inifilr_hemisph(- rlatu(j1:jjm), rlamda, unit, eignfnv, jfiltsu, &
92      do while (cos(rlatu(jfiltsu)) >= colat0 &           matriceus, matrinvs)
93           .or. rlamda(iim) * cos(rlatu(jfiltsu)) >= 1.)      jfiltsu = j1 - 1 + jfiltsu
94         jfiltsu = jfiltsu + 1  
95      end do      j1 = ifirstloc(rlatv <= 0.)
96    
97      jfiltnv = jjm / 2      call inifilr_hemisph(rlatv(j1 - 1:1:- 1), rlamda, unit, eignfnu, jfiltnv, &
98      do while ((cos(rlatv(jfiltnv)) >= colat0 &           matricevn)
99           .or. rlamda(iim) * cos(rlatv(jfiltnv)) >= 1.) .and. jfiltnv >= 2)      jfiltnv = j1 - jfiltnv
100         jfiltnv = jfiltnv - 1      matricevn = matricevn(:, :, jfiltnv:1:- 1)
101      end do  
102        call inifilr_hemisph(- rlatv(j1:jjm), rlamda, unit, eignfnu, jfiltsv, &
103      if (cos(rlatv(jfiltnv)) >= colat0 &           matricevs)
104           .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  
105    
106        close(unit)
107      PRINT *, 'jfiltnu =', jfiltnu      PRINT *, 'jfiltnu =', jfiltnu
108      PRINT *, 'jfiltsu =', jfiltsu      PRINT *, 'jfiltsu =', jfiltsu
109      PRINT *, 'jfiltnv =', jfiltnv      PRINT *, 'jfiltnv =', jfiltnv
110      PRINT *, 'jfiltsv =', jfiltsv      PRINT *, 'jfiltsv =', jfiltsv
111    
     ! D\'etermination de coefil[ns][uv], modfrst[ns][uv]:  
   
     allocate(modfrstnu(2:jfiltnu), modfrstsu(jfiltsu:jjm))  
     allocate(modfrstnv(jfiltnv), modfrstsv(jfiltsv:jjm))  
     allocate(coefilnu(iim, 2:jfiltnu), coefilsu(iim, jfiltsu:jjm))  
     allocate(coefilnv(iim, jfiltnv), coefilsv(iim, jfiltsv:jjm))  
   
     coefilnu = 0.  
     coefilnv = 0.  
     coefilsu = 0.  
     coefilsv = 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 i = modfrstnu(j), iim  
              coefilnu(i, j) = rlamda(i) * cos(rlatu(j)) - 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 i = modfrstnv(j), iim  
              coefilnv(i, j) = rlamda(i) * cos(rlatv(j)) - 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 i = modfrstsu(j), iim  
              coefilsu(i, j) = rlamda(i) * cos(rlatu(j)) - 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 i = modfrstsv(j), iim  
              coefilsv(i, j) = rlamda(i) * cos(rlatv(j)) - 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  
        eignft(:modfrstnu(j) - 1, :) = 0.  
        forall (i = modfrstnu(j):iim) eignft(i, :) = eignfnv(:, i) &  
             * coefilnu(i, j)  
        matriceun(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
     DO j = jfiltsu, jjm  
        eignft(:modfrstsu(j) - 1, :) = 0.  
        forall (i = modfrstsu(j):iim) eignft(i, :) = eignfnv(:, i) &  
             * coefilsu(i, j)  
        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  
        eignft(:modfrstnv(j) - 1, :) = 0.  
        forall (i = modfrstnv(j): iim) eignft(i, :) = eignfnu(:, i) &  
             * coefilnv(i, j)  
        matricevn(:, :, j) = matmul(eignfnu, eignft)  
     END DO  
   
     DO j = jfiltsv, jjm  
        eignft(:modfrstsv(j) - 1, :) = 0.  
        forall (i = modfrstsv(j):iim) eignft(i, :) = eignfnu(:, i) &  
             * coefilsv(i, j)  
        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  
        eignft(:modfrstnu(j) - 1, :) = 0.  
        forall (i = modfrstnu(j):iim) eignft(i, :) = eignfnv(:, i) &  
             * coefilnu(i, j) / (1. + coefilnu(i, j))  
        matrinvn(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
     DO j = jfiltsu, jjm  
        eignft(:modfrstsu(j) - 1, :) = 0.  
        forall (i = modfrstsu(j):iim) eignft(i, :) = eignfnv(:, i) &  
             * coefilsu(i, j) / (1. + coefilsu(i, j))  
        matrinvs(:, :, j) = matmul(eignfnv, eignft)  
     END DO  
   
112    END SUBROUTINE inifilr    END SUBROUTINE inifilr
113    
114  end module inifilr_m  end module inifilr_m
Legend:
Removed from v.164  
changed lines
  Added in v.175
  ViewVC Help
Powered by ViewVC 1.1.21