[614] | 1 | !-------------------------------------------------------------------------- |
---|
| 2 | !---------------------------- caldyn_vert_NH ---------------------------------- |
---|
| 3 | !$OMP DO SCHEDULE(STATIC) |
---|
| 4 | DO ij = 1, primal_num |
---|
| 5 | l=1 |
---|
| 6 | w_ij = ( W(l,ij)+.5*W(l+1,ij) )/mass(l,ij) |
---|
| 7 | wflux_ij = .5*(wflux(l,ij)+wflux(l+1,ij)) |
---|
| 8 | W_etadot(l,ij) = wflux_ij*w_ij |
---|
| 9 | eta_dot(l,ij) = wflux_ij / mass(l,ij) |
---|
| 10 | wcov(l,ij) = w_ij*(geopot(l+1,ij)-geopot(l,ij)) |
---|
[658] | 11 | !DIR$ SIMD |
---|
[614] | 12 | DO l = 2, llm-1 |
---|
| 13 | w_ij = .5*( W(l,ij)+W(l+1,ij) )/mass(l,ij) |
---|
| 14 | wflux_ij = .5*(wflux(l,ij)+wflux(l+1,ij)) |
---|
| 15 | W_etadot(l,ij) = wflux_ij*w_ij |
---|
| 16 | eta_dot(l,ij) = wflux_ij / mass(l,ij) |
---|
| 17 | wcov(l,ij) = w_ij*(geopot(l+1,ij)-geopot(l,ij)) |
---|
| 18 | END DO |
---|
| 19 | l=llm |
---|
| 20 | w_ij = ( .5*W(l,ij)+W(l+1,ij) )/mass(l,ij) |
---|
| 21 | wflux_ij = .5*(wflux(l,ij)+wflux(l+1,ij)) |
---|
| 22 | W_etadot(l,ij) = wflux_ij*w_ij |
---|
| 23 | eta_dot(l,ij) = wflux_ij / mass(l,ij) |
---|
| 24 | wcov(l,ij) = w_ij*(geopot(l+1,ij)-geopot(l,ij)) |
---|
| 25 | END DO |
---|
| 26 | !$OMP END DO |
---|
| 27 | ! add NH term to du |
---|
| 28 | !$OMP DO SCHEDULE(STATIC) |
---|
| 29 | DO edge = 1, edge_num |
---|
| 30 | ij_left = left(edge) |
---|
| 31 | ij_right = right(edge) |
---|
[658] | 32 | !DIR$ SIMD |
---|
[614] | 33 | DO l = 1, llm |
---|
| 34 | du(l,edge) = du(l,edge) - .5*(wcov(l,ij_left)+wcov(l,ij_right))*1.*(eta_dot(l,ij_right)-eta_dot(l,ij_left)) |
---|
| 35 | END DO |
---|
| 36 | END DO |
---|
| 37 | !$OMP END DO |
---|
| 38 | ! add NH terms to dW, dPhi |
---|
| 39 | ! FIXME : TODO top and bottom |
---|
| 40 | !$OMP DO SCHEDULE(STATIC) |
---|
| 41 | DO ij = 1, primal_num |
---|
[658] | 42 | !DIR$ SIMD |
---|
[614] | 43 | DO l = 2, llm |
---|
| 44 | dPhi(l,ij)=dPhi(l,ij)-wflux(l,ij)*(geopot(l+1,ij)-geopot(l-1,ij))/(mass(l-1,ij)+mass(l,ij)) |
---|
| 45 | END DO |
---|
| 46 | END DO |
---|
| 47 | !$OMP END DO |
---|
| 48 | ! We need a barrier here because we compute W_etadot above and do a vertical difference below |
---|
| 49 | !$OMP BARRIER |
---|
| 50 | !$OMP DO SCHEDULE(STATIC) |
---|
| 51 | DO ij = 1, primal_num |
---|
| 52 | l=1 |
---|
| 53 | dW(l,ij) = dW(l,ij) - W_etadot(l,ij) |
---|
[658] | 54 | !DIR$ SIMD |
---|
[614] | 55 | DO l = 2, llm |
---|
| 56 | dW(l,ij) = dW(l,ij) + W_etadot(l-1,ij) - W_etadot(l,ij) |
---|
| 57 | END DO |
---|
| 58 | l=llm+1 |
---|
| 59 | dW(l,ij) = dW(l,ij) + W_etadot(l-1,ij) |
---|
| 60 | END DO |
---|
| 61 | !$OMP END DO |
---|
| 62 | !---------------------------- caldyn_vert_NH ---------------------------------- |
---|
| 63 | !-------------------------------------------------------------------------- |
---|