[1] | 1 | SUBROUTINE init_advec |
---|
| 2 | |
---|
| 3 | !------------------------------------------------------------------------------! |
---|
[484] | 4 | ! Current revisions: |
---|
[1] | 5 | ! ----------------- |
---|
| 6 | ! |
---|
| 7 | ! |
---|
| 8 | ! Former revisions: |
---|
| 9 | ! ----------------- |
---|
[3] | 10 | ! $Id: init_advec.f90 484 2010-02-05 07:36:54Z raasch $ |
---|
| 11 | ! RCS Log replace by Id keyword, revision history cleaned up |
---|
| 12 | ! |
---|
[1] | 13 | ! Revision 1.6 2004/04/30 11:59:31 raasch |
---|
| 14 | ! impulse_advec renamed momentum_advec |
---|
| 15 | ! |
---|
| 16 | ! Revision 1.1 1999/02/05 09:07:38 raasch |
---|
| 17 | ! Initial revision |
---|
| 18 | ! |
---|
| 19 | ! |
---|
| 20 | ! Description: |
---|
| 21 | ! ------------ |
---|
| 22 | ! Initialize constant coefficients and parameters for certain advection schemes. |
---|
| 23 | !------------------------------------------------------------------------------! |
---|
| 24 | |
---|
| 25 | USE advection |
---|
| 26 | USE arrays_3d |
---|
| 27 | USE indices |
---|
| 28 | USE control_parameters |
---|
| 29 | |
---|
| 30 | IMPLICIT NONE |
---|
| 31 | |
---|
| 32 | INTEGER :: i, intervals, j, k |
---|
| 33 | REAL :: delt, dn, dnneu, ex1, ex2, ex3, ex4, ex5, ex6, spl_alpha, & |
---|
| 34 | spl_beta, sterm |
---|
| 35 | REAL, DIMENSION(:), ALLOCATABLE :: spl_u, temp |
---|
| 36 | |
---|
| 37 | |
---|
| 38 | IF ( scalar_advec == 'bc-scheme' ) THEN |
---|
| 39 | |
---|
| 40 | ! |
---|
| 41 | !-- Compute exponential coefficients for the Bott-Chlond scheme |
---|
| 42 | intervals = 1000 |
---|
| 43 | ALLOCATE( aex(intervals), bex(intervals), dex(intervals), eex(intervals) ) |
---|
| 44 | |
---|
| 45 | delt = 1.0 / REAL( intervals ) |
---|
| 46 | sterm = delt * 0.5 |
---|
| 47 | |
---|
| 48 | DO i = 1, intervals |
---|
| 49 | |
---|
| 50 | IF ( sterm > 0.5 ) THEN |
---|
| 51 | dn = -5.0 |
---|
| 52 | ELSE |
---|
| 53 | dn = 5.0 |
---|
| 54 | ENDIF |
---|
| 55 | |
---|
| 56 | DO j = 1, 15 |
---|
| 57 | ex1 = dn * EXP( -dn ) - EXP( 0.5 * dn ) + EXP( -0.5 * dn ) |
---|
| 58 | ex2 = EXP( dn ) - EXP( -dn ) |
---|
| 59 | ex3 = EXP( -dn ) * ( 1.0 - dn ) - 0.5 * EXP( 0.5 * dn ) & |
---|
| 60 | - 0.5 * EXP( -0.5 * dn ) |
---|
| 61 | ex4 = EXP( dn ) + EXP( -dn ) |
---|
| 62 | ex5 = dn * sterm + ex1 / ex2 |
---|
| 63 | ex6 = sterm + ( ex3 * ex2 - ex4 * ex1 ) / ( ex2 * ex2 ) |
---|
| 64 | dnneu = dn - ex5 / ex6 |
---|
| 65 | dn = dnneu |
---|
| 66 | ENDDO |
---|
| 67 | |
---|
| 68 | IF ( sterm < 0.5 ) dn = MAX( 2.95E-2, dn ) |
---|
| 69 | IF ( sterm > 0.5 ) dn = MIN( -2.95E-2, dn ) |
---|
| 70 | ex1 = EXP( -dn ) |
---|
| 71 | ex2 = EXP( dn ) - ex1 |
---|
| 72 | aex(i) = -ex1 / ex2 |
---|
| 73 | bex(i) = 1.0 / ex2 |
---|
| 74 | dex(i) = dn |
---|
| 75 | eex(i) = EXP( dex(i) * 0.5 ) |
---|
| 76 | sterm = sterm + delt |
---|
| 77 | |
---|
| 78 | ENDDO |
---|
| 79 | |
---|
| 80 | ENDIF |
---|
| 81 | |
---|
| 82 | IF ( momentum_advec == 'ups-scheme' .OR. scalar_advec == 'ups-scheme' ) & |
---|
| 83 | THEN |
---|
| 84 | |
---|
| 85 | ! |
---|
| 86 | !-- Provide the constant parameters for the Upstream-Spline advection scheme. |
---|
| 87 | !-- In x- und y-direction the Sherman-Morrison formula is applied |
---|
| 88 | !-- (cf. Press et al, 1986 (Numerical Recipes)). |
---|
| 89 | ! |
---|
| 90 | !-- Allocate nonlocal arrays |
---|
| 91 | ALLOCATE( spl_z_x(0:nx), spl_z_y(0:ny), spl_tri_x(5,0:nx), & |
---|
| 92 | spl_tri_y(5,0:ny), spl_tri_zu(5,nzb:nzt+1), & |
---|
| 93 | spl_tri_zw(5,nzb:nzt+1) ) |
---|
| 94 | |
---|
| 95 | ! |
---|
| 96 | !-- Provide diagonal elements of the tridiagonal matrices for all |
---|
| 97 | !-- directions |
---|
| 98 | spl_tri_x(1,:) = 2.0 |
---|
| 99 | spl_tri_y(1,:) = 2.0 |
---|
| 100 | spl_tri_zu(1,:) = 2.0 |
---|
| 101 | spl_tri_zw(1,:) = 2.0 |
---|
| 102 | |
---|
| 103 | ! |
---|
| 104 | !-- Elements of the cyclic tridiagonal matrix |
---|
| 105 | !-- (same for all horizontal directions) |
---|
| 106 | spl_alpha = 0.5 |
---|
| 107 | spl_beta = 0.5 |
---|
| 108 | |
---|
| 109 | ! |
---|
| 110 | !-- Sub- and superdiagonal elements, x-direction |
---|
| 111 | spl_tri_x(2,0:nx) = 0.5 |
---|
| 112 | spl_tri_x(3,0:nx) = 0.5 |
---|
| 113 | |
---|
| 114 | ! |
---|
| 115 | !-- mMdify the diagonal elements (Sherman-Morrison) |
---|
| 116 | spl_gamma_x = -spl_tri_x(1,0) |
---|
| 117 | spl_tri_x(1,0) = spl_tri_x(1,0) - spl_gamma_x |
---|
| 118 | spl_tri_x(1,nx) = spl_tri_x(1,nx) - spl_alpha * spl_beta / spl_gamma_x |
---|
| 119 | |
---|
| 120 | ! |
---|
| 121 | !-- Split the tridiagonal matrix for Thomas algorithm |
---|
| 122 | spl_tri_x(4,0) = spl_tri_x(1,0) |
---|
| 123 | DO i = 1, nx |
---|
| 124 | spl_tri_x(5,i) = spl_tri_x(2,i) / spl_tri_x(4,i-1) |
---|
| 125 | spl_tri_x(4,i) = spl_tri_x(1,i) - spl_tri_x(5,i) * spl_tri_x(3,i-1) |
---|
| 126 | ENDDO |
---|
| 127 | |
---|
| 128 | ! |
---|
| 129 | !-- Allocate arrays required locally |
---|
| 130 | ALLOCATE( temp(0:nx), spl_u(0:nx) ) |
---|
| 131 | |
---|
| 132 | ! |
---|
| 133 | !-- Provide "corrective vector", x-direction |
---|
| 134 | spl_u(0) = spl_gamma_x |
---|
| 135 | spl_u(1:nx-1) = 0.0 |
---|
| 136 | spl_u(nx) = spl_alpha |
---|
| 137 | |
---|
| 138 | ! |
---|
| 139 | !-- Solve the system of equations for the corrective vector |
---|
| 140 | !-- (Sherman-Morrison) |
---|
| 141 | temp(0) = spl_u(0) |
---|
| 142 | DO i = 1, nx |
---|
| 143 | temp(i) = spl_u(i) - spl_tri_x(5,i) * temp(i-1) |
---|
| 144 | ENDDO |
---|
| 145 | spl_z_x(nx) = temp(nx) / spl_tri_x(4,nx) |
---|
| 146 | DO i = nx-1, 0, -1 |
---|
| 147 | spl_z_x(i) = ( temp(i) - spl_tri_x(3,i) * spl_z_x(i+1) ) / & |
---|
| 148 | spl_tri_x(4,i) |
---|
| 149 | ENDDO |
---|
| 150 | |
---|
| 151 | ! |
---|
| 152 | !-- Deallocate local arrays, for they are allocated in a different way for |
---|
| 153 | !-- operations in y-direction |
---|
| 154 | DEALLOCATE( temp, spl_u ) |
---|
| 155 | |
---|
| 156 | ! |
---|
| 157 | !-- Provide sub- and superdiagonal elements, y-direction |
---|
| 158 | spl_tri_y(2,0:ny) = 0.5 |
---|
| 159 | spl_tri_y(3,0:ny) = 0.5 |
---|
| 160 | |
---|
| 161 | ! |
---|
| 162 | !-- Modify the diagonal elements (Sherman-Morrison) |
---|
| 163 | spl_gamma_y = -spl_tri_y(1,0) |
---|
| 164 | spl_tri_y(1,0) = spl_tri_y(1,0) - spl_gamma_y |
---|
| 165 | spl_tri_y(1,ny) = spl_tri_y(1,ny) - spl_alpha * spl_beta / spl_gamma_y |
---|
| 166 | |
---|
| 167 | ! |
---|
| 168 | !-- Split the tridiagonal matrix for Thomas algorithm |
---|
| 169 | spl_tri_y(4,0) = spl_tri_y(1,0) |
---|
| 170 | DO j = 1, ny |
---|
| 171 | spl_tri_y(5,j) = spl_tri_y(2,j) / spl_tri_y(4,j-1) |
---|
| 172 | spl_tri_y(4,j) = spl_tri_y(1,j) - spl_tri_y(5,j) * spl_tri_y(3,j-1) |
---|
| 173 | ENDDO |
---|
| 174 | |
---|
| 175 | ! |
---|
| 176 | !-- Allocate arrays required locally |
---|
| 177 | ALLOCATE( temp(0:ny), spl_u(0:ny) ) |
---|
| 178 | |
---|
| 179 | ! |
---|
| 180 | !-- Provide "corrective vector", y-direction |
---|
| 181 | spl_u(0) = spl_gamma_y |
---|
| 182 | spl_u(1:ny-1) = 0.0 |
---|
| 183 | spl_u(ny) = spl_alpha |
---|
| 184 | |
---|
| 185 | ! |
---|
| 186 | !-- Solve the system of equations for the corrective vector |
---|
| 187 | !-- (Sherman-Morrison) |
---|
| 188 | temp = 0.0 |
---|
| 189 | spl_z_y = 0.0 |
---|
| 190 | temp(0) = spl_u(0) |
---|
| 191 | DO j = 1, ny |
---|
| 192 | temp(j) = spl_u(j) - spl_tri_y(5,j) * temp(j-1) |
---|
| 193 | ENDDO |
---|
| 194 | spl_z_y(ny) = temp(ny) / spl_tri_y(4,ny) |
---|
| 195 | DO j = ny-1, 0, -1 |
---|
| 196 | spl_z_y(j) = ( temp(j) - spl_tri_y(3,j) * spl_z_y(j+1) ) / & |
---|
| 197 | spl_tri_y(4,j) |
---|
| 198 | ENDDO |
---|
| 199 | |
---|
| 200 | ! |
---|
| 201 | !-- deallocate local arrays, for they are no longer required |
---|
| 202 | DEALLOCATE( temp, spl_u ) |
---|
| 203 | |
---|
| 204 | ! |
---|
| 205 | !-- provide sub- and superdiagonal elements, z-direction |
---|
| 206 | spl_tri_zu(2,nzb) = 0.0 |
---|
| 207 | spl_tri_zu(2,nzt+1) = 1.0 |
---|
| 208 | spl_tri_zw(2,nzb) = 0.0 |
---|
| 209 | spl_tri_zw(2,nzt+1) = 1.0 |
---|
| 210 | |
---|
| 211 | spl_tri_zu(3,nzb) = 1.0 |
---|
| 212 | spl_tri_zu(3,nzt+1) = 0.0 |
---|
| 213 | spl_tri_zw(3,nzb) = 1.0 |
---|
| 214 | spl_tri_zw(3,nzt+1) = 0.0 |
---|
| 215 | |
---|
| 216 | DO k = nzb+1, nzt |
---|
| 217 | spl_tri_zu(2,k) = dzu(k) / ( dzu(k) + dzu(k+1) ) |
---|
| 218 | spl_tri_zw(2,k) = dzw(k) / ( dzw(k) + dzw(k+1) ) |
---|
| 219 | spl_tri_zu(3,k) = 1.0 - spl_tri_zu(2,k) |
---|
| 220 | spl_tri_zw(3,k) = 1.0 - spl_tri_zw(2,k) |
---|
| 221 | ENDDO |
---|
| 222 | |
---|
| 223 | spl_tri_zu(4,nzb) = spl_tri_zu(1,nzb) |
---|
| 224 | spl_tri_zw(4,nzb) = spl_tri_zw(1,nzb) |
---|
| 225 | DO k = nzb+1, nzt+1 |
---|
| 226 | spl_tri_zu(5,k) = spl_tri_zu(2,k) / spl_tri_zu(4,k-1) |
---|
| 227 | spl_tri_zw(5,k) = spl_tri_zw(2,k) / spl_tri_zw(4,k-1) |
---|
| 228 | spl_tri_zu(4,k) = spl_tri_zu(1,k) - spl_tri_zu(5,k) * & |
---|
| 229 | spl_tri_zu(3,k-1) |
---|
| 230 | spl_tri_zw(4,k) = spl_tri_zw(1,k) - spl_tri_zw(5,k) * & |
---|
| 231 | spl_tri_zw(3,k-1) |
---|
| 232 | ENDDO |
---|
| 233 | |
---|
| 234 | ENDIF |
---|
| 235 | |
---|
| 236 | END SUBROUTINE init_advec |
---|