/[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 165 by guez, Wed Jul 29 09:52:33 2015 UTC revision 166 by guez, Wed Jul 29 14:32:55 2015 UTC
# Line 7  module inifilr_m Line 7  module inifilr_m
7    INTEGER jfiltnu, jfiltnv    INTEGER jfiltnu, jfiltnv
8    ! index of the last scalar line filtered in northern hemisphere    ! index of the last scalar line filtered in northern hemisphere
9    
10    real, allocatable:: matriceun(:, :, :) ! (iim, iim, 2:jfiltnu)    real, pointer:: matriceun(:, :, :) ! (iim, iim, jfiltnu - 1)
11    ! matrice filtre pour les champs situes sur la grille scalaire    ! matrice filtre pour les champs situes sur la grille scalaire
12    
13    real, allocatable:: matrinvn(:, :, :) ! (iim, iim, 2:jfiltnu)    real, pointer:: matrinvn(:, :, :) ! (iim, iim, jfiltnu - 1)
14    ! matrice filtre pour les champs situes sur la grille scalaire, pour    ! matrice filtre pour les champs situes sur la grille scalaire, pour
15    ! le filtre inverse    ! le filtre inverse
16    
17    real, allocatable:: matricevn(:, :, :) ! (iim, iim, jfiltnv)    real, pointer:: matricevn(:, :, :) ! (iim, iim, jfiltnv)
18    ! matrice filtre pour les champs situes sur la grille de V ou de Z    ! matrice filtre pour les champs situes sur la grille de V ou de Z
19    
20    ! South:    ! South:
# Line 22  module inifilr_m Line 22  module inifilr_m
22    integer jfiltsu, jfiltsv    integer jfiltsu, jfiltsv
23    ! index of the first line filtered in southern hemisphere    ! index of the first line filtered in southern hemisphere
24    
25    real, allocatable:: matriceus(:, :, :) ! (iim, iim, jfiltsu:jjm)    real, pointer:: matriceus(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
26    ! matrice filtre pour les champs situes sur la grille scalaire    ! matrice filtre pour les champs situes sur la grille scalaire
27    
28    real, allocatable:: matrinvs(:, :, :) ! (iim, iim, jfiltsu:jjm)    real, pointer:: matrinvs(:, :, :) ! (iim, iim, jjm - jfiltsu + 1)
29    ! matrice filtre pour les champs situes sur la grille scalaire, pour    ! matrice filtre pour les champs situes sur la grille scalaire, pour
30    ! le filtre inverse    ! le filtre inverse
31    
32    real, allocatable:: matricevs(:, :, :) ! (iim, iim, jfiltsv:jjm)    real, pointer:: matricevs(:, :, :) ! (iim, iim, jjm - jfiltsv + 1)
33    ! matrice filtre pour les champs situes sur la grille de V ou de Z    ! matrice filtre pour les champs situes sur la grille de V ou de Z
34    
35  contains  contains
# Line 49  contains Line 49  contains
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 (<= 0)      real eignvl(iim) ! eigenvalues sorted in descending order (<= 0)
61      INTEGER i, j, unit      INTEGER j, unit
62      REAL colat0 ! > 0      REAL colat0 ! > 0
63      REAL eignft(iim, iim)      integer j1 ! index of smallest positive latitude
64    
65      real eignfnu(iim, iim), eignfnv(iim, iim)      real eignfnu(iim, iim), eignfnv(iim, iim)
66      ! eigenvectors of the discrete second derivative with respect to longitude      ! eigenvectors of the discrete second derivative with respect to longitude
67    
     ! Filtering coefficients (lamda_max * cos(rlat) / lamda):  
     real coefil(iim)  
   
     ! 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)  
   
68      !-----------------------------------------------------------      !-----------------------------------------------------------
69    
70      print *, "Call sequence information: inifilr"      print *, "Call sequence information: inifilr"
# Line 85  contains Line 77  contains
77      PRINT *, 'colat0 = ', colat0      PRINT *, 'colat0 = ', colat0
78    
79      rlamda = iim / (pi * colat0 / grossismx) / sqrt(- eignvl(2: iim))      rlamda = iim / (pi * colat0 / grossismx) / sqrt(- eignvl(2: iim))
   
     ! D\'etermination de jfilt[ns][uv] :  
   
     jfiltnu = (jjm + 1) / 2  
     do while (cos(rlatu(jfiltnu)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatu(jfiltnu)) >= 1.)  
        jfiltnu = jfiltnu - 1  
     end do  
   
     jfiltsu = jjm / 2 + 2  
     do while (cos(rlatu(jfiltsu)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatu(jfiltsu)) >= 1.)  
        jfiltsu = jfiltsu + 1  
     end do  
   
     jfiltnv = jjm / 2  
     do while ((cos(rlatv(jfiltnv)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatv(jfiltnv)) >= 1.) .and. jfiltnv >= 2)  
        jfiltnv = jfiltnv - 1  
     end do  
   
     if (cos(rlatv(jfiltnv)) >= colat0 &  
          .or. rlamda(iim) * cos(rlatv(jfiltnv)) >= 1.) then  
        ! {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  
   
     PRINT *, 'jfiltnu =', jfiltnu  
     PRINT *, 'jfiltsu =', jfiltsu  
     PRINT *, 'jfiltnv =', jfiltnv  
     PRINT *, 'jfiltsv =', jfiltsv  
   
     ! D\'etermination de modfrst[ns][uv] :  
   
     allocate(modfrstnu(2:jfiltnu), modfrstsu(jfiltsu:jjm))  
     allocate(modfrstnv(jfiltnv), modfrstsv(jfiltsv:jjm))  
   
     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  
     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  
     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  
     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  
     END DO  
   
80      call new_unit(unit)      call new_unit(unit)
   
81      open(unit, file = "inifilr_out.txt", status = "replace", action = "write")      open(unit, file = "inifilr_out.txt", status = "replace", action = "write")
82      write(unit, fmt = *) '"EIGNVL"', eignvl      write(unit, fmt = *) '"EIGNVL"', eignvl
83      close(unit)      close(unit)
84        open(unit, file = "modfrst.csv", status = "replace", action = "write")
85        write(unit, fmt = *) '"rlat (degrees)" modfrst' ! title line
86    
87      open(unit, file = "modfrstnu.csv", status = "replace", action = "write")     ! D\'etermination de jfilt[ns][uv] :
     write(unit, fmt = *) '"rlatu (degrees north)" modfrstnu ' &  
          // '"rlamda(modfrstnu) * cos(rlatu) < 1"'  
     DO j = 2, jfiltnu  
        write(unit, fmt = *) rlatu(j) / pi * 180., modfrstnu(j), &  
             rlamda(modfrstnu(j)) * cos(rlatu(j)) < 1  
     end DO  
     close(unit)  
88    
89      open(unit, file = "modfrstnv.csv", status = "replace", action = "write")      j1 = jjm + 1 - ifirstloc(rlatu(jjm:1:- 1) >= 0.)
     write(unit, fmt = *) '"rlatv (degrees north)" modfrstnv ' &  
          // '"rlamda(modfrstnv) * cos(rlatv) < 1"'  
     DO j = 1, jfiltnv  
        write(unit, fmt = *) rlatv(j) / pi * 180., modfrstnv(j), &  
             rlamda(modfrstnv(j)) * cos(rlatv(j)) < 1  
     end DO  
     close(unit)  
90    
91      open(unit, file = "modfrstsu.csv", status = "replace", action = "write")      call inifilr_hemisph(rlatu(j1:2:- 1), colat0, rlamda, unit, eignfnv, &
92       write(unit, fmt = *) '"rlatu (degrees north)" modfrstsu ' &           jfiltnu, matriceun, matrinvn)
93           // '"rlamda(modfrstsu) * cos(rlatu) < 1"'      jfiltnu = j1 + 1 - jfiltnu
94     DO j = jfiltsu, jjm      matriceun = matriceun(:, :, jfiltnu - 1:1:- 1)
95         write(unit, fmt = *) rlatu(j) / pi * 180., modfrstsu(j), &      matrinvn = matrinvn(:, :, jfiltnu - 1:1:- 1)
             rlamda(modfrstsu(j)) * cos(rlatu(j)) < 1  
     end DO  
     close(unit)  
96    
97      open(unit, file = "modfrstsv.csv", status = "replace", action = "write")      call inifilr_hemisph(- rlatu(j1 + 1:jjm), colat0, rlamda, unit, eignfnv, &
98      write(unit, fmt = *) '"rlatv (degrees north)" modfrstsv ' &           jfiltsu, matriceus, matrinvs)
99           // '"rlamda(modfrstsv) * cos(rlatv) < 1"'      jfiltsu = j1 + jfiltsu
100      DO j = jfiltsv, jjm  
101         write(unit, fmt = *) rlatv(j) / pi * 180., modfrstsv(j), &      j1 = jjm + 1 - ifirstloc(rlatv(jjm:1:- 1) >= 0.)
102              rlamda(modfrstsv(j)) * cos(rlatv(j)) < 1  
103      end DO      call inifilr_hemisph(rlatv(j1:1:- 1), colat0, rlamda, unit, eignfnu, &
104      close(unit)           jfiltnv, matricevn)
105        jfiltnv = j1 + 1 - jfiltnv
106        matricevn = matricevn(:, :, jfiltnv:1:- 1)
107    
108      allocate(matriceun(iim, iim, 2:jfiltnu), matrinvn(iim, iim, 2:jfiltnu))      call inifilr_hemisph(- rlatv(j1 + 1:jjm), colat0, rlamda, unit, eignfnu, &
109      allocate(matricevn(iim, iim, jfiltnv))           jfiltsv, matricevs)
110      allocate(matricevs(iim, iim, jfiltsv:jjm))      jfiltsv = j1 + jfiltsv
111      allocate(matriceus(iim, iim, jfiltsu:jjm), matrinvs(iim, iim, jfiltsu:jjm))  
112        close(unit)
113      ! Calcul de matriceu et matrinv      PRINT *, 'jfiltnu =', jfiltnu
114        PRINT *, 'jfiltsu =', jfiltsu
115      DO j = 2, jfiltnu      PRINT *, 'jfiltnv =', jfiltnv
116         if (rlamda(modfrstnu(j)) * cos(rlatu(j)) < 1.) then      PRINT *, 'jfiltsv =', jfiltsv
           DO i = modfrstnu(j), iim  
              coefil(i) = rlamda(i) * cos(rlatu(j)) - 1.  
           end DO  
   
           eignft(:modfrstnu(j) - 1, :) = 0.  
   
           forall (i = modfrstnu(j):iim) eignft(i, :) = eignfnv(:, i) * coefil(i)  
           matriceun(:, :, j) = matmul(eignfnv, eignft)  
   
           forall (i = modfrstnu(j):iim) eignft(i, :) = eignfnv(:, i) &  
                * coefil(i) / (1. + coefil(i))  
           matrinvn(:, :, j) = matmul(eignfnv, eignft)  
        else  
           matriceun(:, :, j) = 0.  
           matrinvn(:, :, j) = 0.  
        end if  
     END DO  
   
     DO j = jfiltsu, jjm  
        if (rlamda(modfrstsu(j)) * cos(rlatu(j)) < 1.) then  
           DO i = modfrstsu(j), iim  
              coefil(i) = rlamda(i) * cos(rlatu(j)) - 1.  
           end DO  
   
           eignft(:modfrstsu(j) - 1, :) = 0.  
   
           forall (i = modfrstsu(j):iim) eignft(i, :) = eignfnv(:, i) * coefil(i)  
           matriceus(:, :, j) = matmul(eignfnv, eignft)  
   
           forall (i = modfrstsu(j):iim) eignft(i, :) = eignfnv(:, i) &  
                * coefil(i) / (1. + coefil(i))  
           matrinvs(:, :, j) = matmul(eignfnv, eignft)  
        else  
           matriceus(:, :, j) = 0.  
           matrinvs(:, :, j) = 0.  
        end if  
     END DO  
   
     ! Calcul de matricev  
   
     DO j = 1, jfiltnv  
        if (rlamda(modfrstnv(j)) * cos(rlatv(j)) < 1.) then  
           DO i = modfrstnv(j), iim  
              coefil(i) = rlamda(i) * cos(rlatv(j)) - 1.  
           end DO  
   
           eignft(:modfrstnv(j) - 1, :) = 0.  
           forall (i = modfrstnv(j):iim) eignft(i, :) = eignfnu(:, i) * coefil(i)  
           matricevn(:, :, j) = matmul(eignfnu, eignft)  
        else  
           matricevn(:, :, j) = 0.  
        end if  
     END DO  
   
     DO j = jfiltsv, jjm  
        if (rlamda(modfrstsv(j)) * cos(rlatv(j)) < 1.) then  
           DO i = modfrstsv(j), iim  
              coefil(i) = rlamda(i) * cos(rlatv(j)) - 1.  
           end DO  
   
           eignft(:modfrstsv(j) - 1, :) = 0.  
           forall (i = modfrstsv(j):iim) eignft(i, :) = eignfnu(:, i) * coefil(i)  
           matricevs(:, :, j) = matmul(eignfnu, eignft)  
        else  
           matricevs(:, :, j) = 0.  
        end if  
     END DO  
117    
118    END SUBROUTINE inifilr    END SUBROUTINE inifilr
119    
Legend:
Removed from v.165  
changed lines
  Added in v.166
  ViewVC Help
Powered by ViewVC 1.1.21