[12] | 1 | MODULE kinetic_mod |
---|
[427] | 2 | IMPLICIT NONE |
---|
| 3 | PRIVATE |
---|
[12] | 4 | |
---|
[728] | 5 | PUBLIC :: kinetic, kinetic_new, gradient |
---|
[12] | 6 | |
---|
| 7 | CONTAINS |
---|
| 8 | |
---|
| 9 | SUBROUTINE kinetic(f_ue,f_Ki) |
---|
[19] | 10 | USE icosa |
---|
[12] | 11 | IMPLICIT NONE |
---|
| 12 | TYPE(t_field), POINTER :: f_ue(:) |
---|
| 13 | TYPE(t_field), POINTER :: f_Ki(:) |
---|
| 14 | |
---|
| 15 | REAL(rstd), POINTER :: ue(:,:) |
---|
| 16 | REAL(rstd), POINTER :: Ki(:,:) |
---|
| 17 | INTEGER :: ind |
---|
| 18 | |
---|
[146] | 19 | CALL transfert_request(f_ue,req_e1_vect) |
---|
[12] | 20 | |
---|
| 21 | DO ind=1,ndomain |
---|
[186] | 22 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
[12] | 23 | CALL swap_dimensions(ind) |
---|
| 24 | CALL swap_geometry(ind) |
---|
| 25 | ue=f_ue(ind) |
---|
| 26 | Ki=f_Ki(ind) |
---|
[728] | 27 | CALL compute_kinetic_trisk(ue, Ki) |
---|
[427] | 28 | ENDDO |
---|
| 29 | END SUBROUTINE kinetic |
---|
| 30 | |
---|
[728] | 31 | SUBROUTINE kinetic_new(f_ue,f_Kv,f_Ki) |
---|
[427] | 32 | USE icosa |
---|
| 33 | IMPLICIT NONE |
---|
| 34 | TYPE(t_field), POINTER :: f_ue(:) |
---|
[728] | 35 | TYPE(t_field), POINTER :: f_Kv(:) |
---|
[427] | 36 | TYPE(t_field), POINTER :: f_Ki(:) |
---|
| 37 | |
---|
| 38 | REAL(rstd), POINTER :: ue(:,:) |
---|
[728] | 39 | REAL(rstd), POINTER :: Kv(:,:) |
---|
[427] | 40 | REAL(rstd), POINTER :: Ki(:,:) |
---|
| 41 | INTEGER :: ind |
---|
| 42 | |
---|
| 43 | CALL transfert_request(f_ue,req_e1_vect) |
---|
[728] | 44 | |
---|
| 45 | DO ind=1,ndomain |
---|
| 46 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
| 47 | CALL swap_dimensions(ind) |
---|
| 48 | CALL swap_geometry(ind) |
---|
| 49 | ue=f_ue(ind) |
---|
| 50 | Kv=f_Kv(ind) |
---|
| 51 | CALL compute_kv(ue, Kv) |
---|
| 52 | ENDDO |
---|
[427] | 53 | |
---|
[728] | 54 | CALL transfert_request(f_Kv,req_z1_scal) |
---|
| 55 | |
---|
[427] | 56 | DO ind=1,ndomain |
---|
| 57 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
| 58 | CALL swap_dimensions(ind) |
---|
| 59 | CALL swap_geometry(ind) |
---|
[728] | 60 | Kv=f_Kv(ind) |
---|
[427] | 61 | Ki=f_Ki(ind) |
---|
[728] | 62 | CALL compute_Ki_from_Kv(Kv, Ki) |
---|
[12] | 63 | ENDDO |
---|
[427] | 64 | END SUBROUTINE kinetic_new |
---|
[12] | 65 | |
---|
[728] | 66 | SUBROUTINE compute_kinetic_trisk(ue, Ki) |
---|
[427] | 67 | USE icosa |
---|
[295] | 68 | USE omp_para |
---|
[12] | 69 | IMPLICIT NONE |
---|
| 70 | REAL(rstd),INTENT(IN) :: ue(3*iim*jjm,llm) |
---|
| 71 | REAL(rstd),INTENT(OUT) :: Ki(iim*jjm,llm) |
---|
| 72 | INTEGER :: i,j,ij,l |
---|
| 73 | |
---|
[295] | 74 | DO l=ll_begin,ll_end |
---|
[12] | 75 | DO j=jj_begin,jj_end |
---|
| 76 | DO i=ii_begin,ii_end |
---|
| 77 | ij=(j-1)*iim+i |
---|
| 78 | |
---|
| 79 | Ki(ij,l)=1/(4*Ai(ij))*(le(ij+u_right)*de(ij+u_right)*ue(ij+u_right,l)**2 + & |
---|
| 80 | le(ij+u_rup)*de(ij+u_rup)*ue(ij+u_rup,l)**2 + & |
---|
| 81 | le(ij+u_lup)*de(ij+u_lup)*ue(ij+u_lup,l)**2 + & |
---|
| 82 | le(ij+u_left)*de(ij+u_left)*ue(ij+u_left,l)**2 + & |
---|
| 83 | le(ij+u_ldown)*de(ij+u_ldown)*ue(ij+u_ldown,l)**2 + & |
---|
| 84 | le(ij+u_rdown)*de(ij+u_rdown)*ue(ij+u_rdown,l)**2 ) |
---|
| 85 | |
---|
| 86 | ENDDO |
---|
| 87 | ENDDO |
---|
| 88 | ENDDO |
---|
[728] | 89 | END SUBROUTINE compute_kinetic_trisk |
---|
[427] | 90 | |
---|
| 91 | SUBROUTINE compute_kv(ue, Kv) |
---|
| 92 | USE icosa |
---|
| 93 | USE omp_para |
---|
| 94 | IMPLICIT NONE |
---|
| 95 | REAL(rstd),INTENT(IN) :: ue(3*iim*jjm,llm) |
---|
| 96 | REAL(rstd),INTENT(OUT) :: Kv(2*iim*jjm,llm) |
---|
| 97 | INTEGER :: ij,l, u_up, u_down |
---|
| 98 | |
---|
| 99 | u_up = t_lup + u_right |
---|
| 100 | u_down = t_rdown + u_left |
---|
[12] | 101 | |
---|
[427] | 102 | DO l=ll_begin,ll_end |
---|
[728] | 103 | Kv(:,l)=0. |
---|
[427] | 104 | DO ij=ij_begin,ij_end |
---|
| 105 | Kv(ij+z_up,l) = (radius**2/Av(ij+z_up))*( & |
---|
| 106 | S1(ij,vup)*ue(ij+u_rup,l)**2 + & |
---|
| 107 | S2(ij,vup)*ue(ij+u_lup,l)**2 + & |
---|
| 108 | S2(ij+t_lup,vrdown)*ue(ij+u_up,l)**2) |
---|
| 109 | |
---|
| 110 | Kv(ij+z_down,l) = (radius**2/Av(ij+z_down))*( & |
---|
| 111 | S1(ij,vdown)*ue(ij+u_ldown,l)**2 + & |
---|
| 112 | S2(ij,vdown)*ue(ij+u_rdown,l)**2 + & |
---|
| 113 | S2(ij+t_rdown,vlup)*ue(ij+u_down,l)**2 ) |
---|
| 114 | ENDDO |
---|
| 115 | ENDDO |
---|
| 116 | END SUBROUTINE compute_kv |
---|
[12] | 117 | |
---|
[728] | 118 | SUBROUTINE compute_Ki_from_Kv(Kv, Ki) |
---|
[427] | 119 | USE icosa |
---|
| 120 | USE omp_para |
---|
| 121 | IMPLICIT NONE |
---|
| 122 | REAL(rstd),INTENT(OUT):: Ki(iim*jjm,llm) |
---|
[728] | 123 | REAL(rstd), INTENT(IN) :: Kv(2*iim*jjm,llm) |
---|
| 124 | INTEGER :: ij,l |
---|
[427] | 125 | |
---|
| 126 | DO l=ll_begin,ll_end |
---|
| 127 | DO ij=ij_begin,ij_end |
---|
| 128 | Ki(ij,l) = Riv(ij,vup)*Kv(ij+z_up,l) + & |
---|
| 129 | Riv(ij,vlup) *Kv(ij+z_lup,l) + & |
---|
| 130 | Riv(ij,vldown)*Kv(ij+z_ldown,l) + & |
---|
| 131 | Riv(ij,vdown) *Kv(ij+z_down,l) + & |
---|
| 132 | Riv(ij,vrdown)*Kv(ij+z_rdown,l) + & |
---|
| 133 | Riv(ij,vrup) *Kv(ij+z_rup,l) |
---|
| 134 | END DO |
---|
| 135 | END DO |
---|
[728] | 136 | END SUBROUTINE compute_Ki_from_Kv |
---|
[427] | 137 | |
---|
| 138 | SUBROUTINE gradient(f_berni, f_du) |
---|
| 139 | USE icosa |
---|
| 140 | IMPLICIT NONE |
---|
| 141 | TYPE(t_field), POINTER :: f_berni(:) |
---|
| 142 | TYPE(t_field), POINTER :: f_du(:) |
---|
| 143 | |
---|
| 144 | REAL(rstd), POINTER :: du(:,:) |
---|
| 145 | REAL(rstd), POINTER :: berni(:,:) |
---|
| 146 | INTEGER :: ind |
---|
| 147 | |
---|
| 148 | CALL transfert_request(f_du,req_e1_vect) |
---|
| 149 | CALL transfert_request(f_du,req_e1_vect) |
---|
| 150 | |
---|
| 151 | DO ind=1,ndomain |
---|
| 152 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
| 153 | CALL swap_dimensions(ind) |
---|
| 154 | CALL swap_geometry(ind) |
---|
| 155 | berni=f_berni(ind) |
---|
| 156 | du=f_du(ind) |
---|
| 157 | CALL compute_grad(berni, du) |
---|
| 158 | ENDDO |
---|
| 159 | |
---|
| 160 | END SUBROUTINE gradient |
---|
| 161 | |
---|
| 162 | SUBROUTINE compute_grad(berni, du) |
---|
| 163 | USE icosa |
---|
| 164 | USE omp_para |
---|
| 165 | IMPLICIT NONE |
---|
| 166 | REAL(rstd),INTENT(IN) :: berni(iim*jjm,llm) |
---|
| 167 | REAL(rstd),INTENT(OUT) :: du(3*iim*jjm,llm) |
---|
| 168 | INTEGER :: i,j,ij,l |
---|
| 169 | |
---|
| 170 | DO l=ll_begin,ll_end |
---|
| 171 | DO j=jj_begin,jj_end |
---|
| 172 | DO i=ii_begin,ii_end |
---|
| 173 | ij=(j-1)*iim+i |
---|
| 174 | du(ij+u_right,l) = ne_right*(berni(ij,l)-berni(ij+t_right,l))/de(ij+u_right) |
---|
| 175 | du(ij+u_lup,l) = ne_lup*(berni(ij,l)-berni(ij+t_lup,l))/de(ij+u_right) |
---|
| 176 | du(ij+u_ldown,l) = ne_ldown*(berni(ij,l)-berni(ij+t_ldown,l))/de(ij+u_right) |
---|
| 177 | ENDDO |
---|
| 178 | ENDDO |
---|
| 179 | ENDDO |
---|
| 180 | |
---|
| 181 | END SUBROUTINE compute_grad |
---|
| 182 | |
---|
| 183 | |
---|
[12] | 184 | END MODULE kinetic_mod |
---|