source: codes/icosagcm/devel/src/kernels_unst/caldyn_vert.k90 @ 913

Last change on this file since 913 was 658, checked in by dubos, 7 years ago

devel/unstructured : updated kernels
File size: 2.5 KB
Line 
1   !--------------------------------------------------------------------------
2   !---------------------------- caldyn_vert ----------------------------------
3   DO iq=1,nqdyn
4      !$OMP DO SCHEDULE(STATIC)
5      DO ij = 1, primal_num
6         !DIR$ SIMD
7         DO l = 2, llm
8            dtheta_rhodz(l,ij,iq) = dtheta_rhodz(l,ij,iq) + 0.5*(theta(l,ij,iq)+theta(l-1,ij,iq))*wflux(l,ij)
9         END DO
10      END DO
11      !$OMP END DO
12      !$OMP DO SCHEDULE(STATIC)
13      DO ij = 1, primal_num
14         !DIR$ SIMD
15         DO l = 1, llm-1
16            dtheta_rhodz(l,ij,iq) = dtheta_rhodz(l,ij,iq) - 0.5*(theta(l,ij,iq)+theta(l+1,ij,iq))*wflux(l+1,ij)
17         END DO
18      END DO
19      !$OMP END DO
20   END DO
21   IF(caldyn_vert_variant == caldyn_vert_cons) THEN
22      ! conservative vertical transport of momentum : (F/m)du/deta = 1/m (d/deta(Fu)-u.dF/deta)
23      !$OMP DO SCHEDULE(STATIC)
24      DO edge = 1, edge_num
25         ij_left = left(edge)
26         ij_right = right(edge)
27         !DIR$ SIMD
28         DO l = 2, llm
29            wwuu(l,edge) = .25*(wflux(l,ij_left)+wflux(l,ij_right))*(u(l,edge)+u(l-1,edge)) ! Fu
30         END DO
31      END DO
32      !$OMP END DO
33      ! make sure wwuu is up to date
34      !$OMP BARRIER
35      !$OMP DO SCHEDULE(STATIC)
36      DO edge = 1, edge_num
37         ij_left = left(edge)
38         ij_right = right(edge)
39         !DIR$ SIMD
40         DO l = 1, llm
41            dFu_deta = wwuu(l+1,edge)-wwuu(l,edge) ! d/deta (F*u)
42            dF_deta = .5*(wflux(l+1,ij_left)+wflux(l+1,ij_right)-(wflux(l,ij_left)+wflux(l,ij_right))) ! d/deta(F)
43            du(l,edge) = du(l,edge) - (dFu_deta-u(l,edge)*dF_deta) / (.5*(rhodz(l,ij_left)+rhodz(l,ij_right))) ! (F/m)du/deta
44         END DO
45      END DO
46      !$OMP END DO
47   ELSE
48      !$OMP DO SCHEDULE(STATIC)
49      DO edge = 1, edge_num
50         ij_left = left(edge)
51         ij_right = right(edge)
52         !DIR$ SIMD
53         DO l = 2, llm
54            wwuu(l,edge) = .5*(wflux(l,ij_left)+wflux(l,ij_right))*(u(l,edge)-u(l-1,edge))
55         END DO
56      END DO
57      !$OMP END DO
58      ! make sure wwuu is up to date
59      !$OMP BARRIER
60      !$OMP DO SCHEDULE(STATIC)
61      DO edge = 1, edge_num
62         ij_left = left(edge)
63         ij_right = right(edge)
64         !DIR$ SIMD
65         DO l = 1, llm
66            du(l,edge) = du(l,edge) - (wwuu(l,edge)+wwuu(l+1,edge)) / (rhodz(l,ij_left)+rhodz(l,ij_right))
67         END DO
68      END DO
69      !$OMP END DO
70   END IF
71   !---------------------------- caldyn_vert ----------------------------------
72   !--------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.