[615] | 1 | MODULE caldyn_unstructured_mod |
---|
| 2 | USE ISO_C_BINDING |
---|
| 3 | USE OMP_LIB |
---|
[624] | 4 | #ifdef CPP_USING_XIOS |
---|
[615] | 5 | USE xios |
---|
[624] | 6 | #endif |
---|
[615] | 7 | IMPLICIT NONE |
---|
| 8 | PRIVATE |
---|
| 9 | SAVE |
---|
| 10 | |
---|
[624] | 11 | #define BINDC_(thename) BIND(C, name=#thename) |
---|
| 12 | #define BINDC(thename) BINDC_(dynamico_ ## thename) |
---|
[615] | 13 | ! |
---|
| 14 | #define DBL REAL(C_DOUBLE) |
---|
| 15 | #define DOUBLE1(m) DBL, DIMENSION(m) |
---|
| 16 | #define DOUBLE2(m,n) DBL, DIMENSION(m,n) |
---|
| 17 | #define DOUBLE3(m,n,p) DBL, DIMENSION(m,n,p) |
---|
| 18 | #define INDEX INTEGER(C_INT) |
---|
| 19 | #define PHI_BOT(ij) Phi_bot |
---|
[624] | 20 | #define PHI_BOT_VAR 0. |
---|
[615] | 21 | ! |
---|
[624] | 22 | INTEGER, PARAMETER :: eta_mass=1, eta_lag=2, & |
---|
| 23 | thermo_theta=1, thermo_entropy=2, thermo_moist=3, thermo_boussinesq=4, & |
---|
| 24 | caldyn_vert_cons=1 |
---|
| 25 | INTEGER(C_INT), BIND(C) :: caldyn_thermo=thermo_theta, caldyn_eta=eta_lag, & |
---|
[635] | 26 | caldyn_vert_variant=caldyn_vert_cons, nb_threads=0 |
---|
[615] | 27 | LOGICAL(C_BOOL), BIND(C) :: hydrostatic=.TRUE., debug_hevi_solver=.TRUE., rigid=.TRUE. |
---|
| 28 | ! |
---|
| 29 | INDEX, BIND(C) :: llm, nqdyn, edge_num, primal_num, dual_num, & |
---|
| 30 | max_primal_deg, max_dual_deg, max_trisk_deg |
---|
| 31 | INDEX, ALLOCATABLE, PRIVATE :: & ! deg(ij) = nb of vertices = nb of edges of primal/dual cell ij |
---|
| 32 | primal_deg(:), primal_edge(:,:), primal_vertex(:,:), primal_ne(:,:), & |
---|
| 33 | dual_deg(:), dual_edge(:,:), dual_vertex(:,:), dual_ne(:,:), & |
---|
| 34 | trisk_deg(:), trisk(:,:), & |
---|
| 35 | left(:), right(:), up(:), down(:) |
---|
| 36 | ! left and right are adjacent primal cells |
---|
| 37 | ! flux is positive when going from left to right |
---|
| 38 | ! up and down are adjacent dual cells |
---|
| 39 | ! circulation is positive when going from down to up |
---|
| 40 | ! |
---|
| 41 | DBL, BIND(C) :: elapsed, g, ptop, cpp, cppv, Rd, Rv, preff, Treff, & |
---|
| 42 | pbot, Phi_bot, rho_bot |
---|
| 43 | DBL :: kappa |
---|
| 44 | DOUBLE1(:), ALLOCATABLE, PRIVATE :: le_de, fv, Av, Ai |
---|
| 45 | DOUBLE2(:,:), ALLOCATABLE, PRIVATE :: Riv2, wee, ap,bp, mass_bl, mass_dak, mass_dbk |
---|
| 46 | ! |
---|
| 47 | INTEGER(C_INT), BIND(C) :: comm_icosa |
---|
[624] | 48 | |
---|
| 49 | #ifdef CPP_USING_XIOS |
---|
[615] | 50 | TYPE(xios_context) :: ctx_hdl |
---|
[624] | 51 | #endif |
---|
[615] | 52 | |
---|
| 53 | CONTAINS |
---|
| 54 | ! |
---|
| 55 | !-------------------------------------- KERNELS -------------------------------------- |
---|
| 56 | ! |
---|
| 57 | ! |
---|
| 58 | #define DECLARE_INDICES INTEGER ij,l,iq,iedge,edge,ivertex,vertex,ij_left,ij_right,ij_up,ij_down,itrisk,edge_trisk,kup,kdown |
---|
| 59 | |
---|
| 60 | #define FIELD_PS DOUBLE1(primal_num) |
---|
| 61 | #define FIELD_MASS DOUBLE2(llm, primal_num) |
---|
| 62 | #define FIELD_Z DOUBLE2(llm, dual_num) |
---|
| 63 | #define FIELD_U DOUBLE2(llm, edge_num) |
---|
| 64 | #define FIELD_UL DOUBLE2(llm+1, edge_num) |
---|
| 65 | #define FIELD_THETA DOUBLE3(llm, primal_num, nqdyn) |
---|
| 66 | #define FIELD_GEOPOT DOUBLE2(llm+1, primal_num) |
---|
| 67 | ! |
---|
| 68 | #define HASNAN(field) (ANY(.NOT.ABS(field)<1e20)) |
---|
| 69 | ! |
---|
| 70 | SUBROUTINE caldyn_unstructured(tau, mass_col,rhodz,theta_rhodz,u,geopot,w, & ! IN : flow state |
---|
| 71 | theta,ps,pk,hflux,qv, & ! OUT : diags (except surface geopot : IN) |
---|
| 72 | dmass_col,drhodz,dtheta_rhodz,du_fast,du_slow, & |
---|
| 73 | dPhi_fast, dPhi_slow, dW_fast, dW_slow) BINDC(caldyn_unstructured) ! OUT : tendencies |
---|
| 74 | DBL, VALUE :: tau |
---|
| 75 | FIELD_MASS :: rhodz, drhodz, pk, berni ! IN, OUT, DIAG |
---|
| 76 | FIELD_THETA :: theta_rhodz, dtheta_rhodz, theta ! IN, OUT, DIAG |
---|
| 77 | FIELD_GEOPOT :: wflux, w, geopot, & ! DIAG, INOUT |
---|
| 78 | dPhi_fast, dPhi_slow, dW_fast, dW_slow ! OUT |
---|
| 79 | FIELD_U :: u,du_fast,du_slow,hflux,qu ! INOUT,OUT,OUT,DIAG |
---|
| 80 | FIELD_Z :: qv ! DIAG |
---|
| 81 | FIELD_PS :: ps,dmass_col,mass_col ! OUT,OUT,IN (if eta_mass) or OUT,UNUSED,UNUSED (if eta_lag) |
---|
| 82 | DOUBLE2(llm+1, edge_num) :: wwuu |
---|
| 83 | DBL :: time1,time2 |
---|
| 84 | INTEGER :: ij |
---|
| 85 | |
---|
[635] | 86 | ! CALL CPU_TIME(time1) |
---|
[615] | 87 | time1=OMP_GET_WTIME() |
---|
| 88 | |
---|
| 89 | IF(hydrostatic) THEN |
---|
| 90 | |
---|
| 91 | !$OMP PARALLEL NUM_THREADS(nb_threads) |
---|
| 92 | !$OMP DO SCHEDULE(STATIC) |
---|
| 93 | DO ij=1,edge_num |
---|
| 94 | du_fast(:,ij)=0. |
---|
| 95 | du_slow(:,ij)=0. |
---|
| 96 | END DO |
---|
| 97 | !$OMP END DO |
---|
| 98 | CALL compute_theta(mass_col,rhodz,theta_rhodz, theta) |
---|
| 99 | CALL compute_geopot(rhodz,theta, ps,pk,geopot) |
---|
| 100 | |
---|
| 101 | CALL compute_caldyn_fast(tau, pk,berni,theta,geopot, du_fast,u) |
---|
| 102 | |
---|
| 103 | CALL compute_pvort_only(rhodz,u,qv,qu) |
---|
| 104 | CALL compute_caldyn_slow_hydro(rhodz,theta,u, berni,hflux,du_slow) |
---|
| 105 | CALL compute_coriolis(hflux,theta,qu, drhodz,dtheta_rhodz,du_slow) |
---|
| 106 | IF(caldyn_eta == eta_mass) THEN |
---|
| 107 | CALL caldyn_vert(drhodz,rhodz,theta,u, dmass_col,wflux,dtheta_rhodz,du_slow,wwuu) |
---|
| 108 | END IF |
---|
[635] | 109 | !$OMP END PARALLEL |
---|
[615] | 110 | |
---|
| 111 | ELSE ! NH |
---|
| 112 | DO ij=1,edge_num |
---|
| 113 | du_fast(:,ij)=0. |
---|
| 114 | du_slow(:,ij)=0. |
---|
| 115 | END DO |
---|
| 116 | DO ij=1,primal_num |
---|
| 117 | wflux(1,ij)=0. |
---|
| 118 | wflux(llm+1,ij)=0. |
---|
| 119 | END DO |
---|
| 120 | CALL compute_theta(mass_col,rhodz,theta_rhodz, theta) |
---|
| 121 | CALL compute_caldyn_solver(tau,rhodz,theta,pk,geopot,W,dPhi_fast,dW_fast,du_fast) |
---|
| 122 | CALL compute_caldyn_fast(tau, pk,berni,theta,geopot, du_fast,u) |
---|
| 123 | CALL compute_pvort_only(rhodz,u,qv,qu) |
---|
| 124 | CALL compute_caldyn_slow_NH(u,rhodz,geopot,W, hflux,du_slow,dPhi_slow,dW_slow) |
---|
| 125 | CALL compute_coriolis(hflux,theta,qu, drhodz,dtheta_rhodz,du_slow) |
---|
| 126 | IF(caldyn_eta == eta_mass) THEN |
---|
| 127 | CALL caldyn_vert(drhodz,rhodz,theta,u, dmass_col,wflux,dtheta_rhodz,du_slow,wwuu) |
---|
| 128 | CALL compute_caldyn_vert_NH(rhodz,geopot,W,wflux, du_slow,dPhi_slow,dW_slow) |
---|
| 129 | END IF |
---|
| 130 | END IF |
---|
| 131 | |
---|
| 132 | time2=OMP_GET_WTIME() |
---|
| 133 | ! CALL CPU_TIME(time2) |
---|
| 134 | elapsed = elapsed + time2-time1 |
---|
| 135 | END SUBROUTINE caldyn_unstructured |
---|
| 136 | ! |
---|
| 137 | !----------------------------- Non-Hydrostatic ----------------------------- |
---|
| 138 | ! |
---|
| 139 | SUBROUTINE compute_NH_geopot(tau,dummy, m_ik, m_il, theta, W_il, Phi_il) |
---|
| 140 | FIELD_MASS :: m_ik, theta, p_ik, A_ik, C_ik ! IN*2,LOCAL*3 |
---|
| 141 | FIELD_GEOPOT :: m_il, W_il, Phi_il, Phi_star_il, R_il, x_il, B_il, D_il ! IN,INOUT*2, LOCAL*5 |
---|
| 142 | DBL :: tau, dummy, gamma, rho_ij, X_ij, Y_ij, wil, tau2_g, g2, gm2, ml_g2, c2_mik |
---|
| 143 | DECLARE_INDICES |
---|
| 144 | INTEGER :: iter |
---|
| 145 | #include "../kernels_unst/compute_NH_geopot.k90" |
---|
| 146 | END SUBROUTINE compute_NH_geopot |
---|
| 147 | ! |
---|
| 148 | SUBROUTINE compute_caldyn_slow_NH(u,rhodz,Phi,W, hflux,du,dPhi,dW) |
---|
| 149 | FIELD_U :: u, hflux, du ! IN, OUT, OUT |
---|
| 150 | FIELD_MASS :: rhodz, berni ! IN, LOCAL |
---|
| 151 | FIELD_GEOPOT :: Phi,W,dPhi,dW, w_il, gradPhi2 ! IN,IN, OUT,OUT, LOCAL |
---|
| 152 | FIELD_UL :: DePhil, v_el, G_el, F_el ! LOCAL |
---|
| 153 | DECLARE_INDICES |
---|
| 154 | DBL :: W_el, W2_el, gPhi2, dP, divG, u2, uu |
---|
| 155 | #include "../kernels_unst/caldyn_slow_NH.k90" |
---|
| 156 | END SUBROUTINE compute_caldyn_slow_NH |
---|
| 157 | ! |
---|
| 158 | SUBROUTINE compute_caldyn_solver(tau,rhodz,theta,pk,geopot,W,dPhi,dW,du) |
---|
| 159 | DBL, INTENT(IN) :: tau |
---|
| 160 | FIELD_MASS :: rhodz,pk,berni,pres ! IN, OUT, LOCAL |
---|
| 161 | FIELD_THETA :: theta ! IN |
---|
| 162 | FIELD_GEOPOT :: geopot,W,dPhi,dW, m_il ! INOUT,INOUT, OUT,OUT, LOCAL |
---|
| 163 | FIELD_U :: du ! OUT |
---|
| 164 | DECLARE_INDICES |
---|
| 165 | DBL :: X_ij, rho_ij, T_ij, gamma, Cvd, vreff |
---|
| 166 | #include "../kernels_unst/caldyn_solver.k90" |
---|
| 167 | END SUBROUTINE compute_caldyn_solver |
---|
| 168 | ! |
---|
| 169 | SUBROUTINE compute_caldyn_vert_NH(mass,geopot,W,wflux, du,dPhi,dW) |
---|
| 170 | FIELD_MASS :: mass, eta_dot, wcov, W_etadot |
---|
| 171 | FIELD_GEOPOT :: geopot,W,wflux,dPhi,dW |
---|
| 172 | FIELD_U :: du |
---|
| 173 | DECLARE_INDICES |
---|
| 174 | DBL :: w_ij, wflux_ij |
---|
| 175 | #include "../kernels_unst/caldyn_vert_NH.k90" |
---|
| 176 | END SUBROUTINE compute_caldyn_vert_NH |
---|
| 177 | ! |
---|
| 178 | !----------------------------- Hydrostatic ------------------------------- |
---|
| 179 | ! |
---|
| 180 | SUBROUTINE compute_geopot(rhodz,theta,ps,pk,geopot) |
---|
| 181 | FIELD_MASS :: rhodz,pk ! IN, OUT |
---|
| 182 | FIELD_THETA :: theta ! IN |
---|
| 183 | FIELD_GEOPOT :: geopot ! IN(l=1)/OUT(l>1) |
---|
| 184 | FIELD_PS :: ps ! OUT |
---|
| 185 | DECLARE_INDICES |
---|
| 186 | DBL :: gdz, ke, uu, chi, gv, exner_ik, temp_ik, p_ik, qv, Rmix |
---|
| 187 | #include "../kernels_unst/compute_geopot.k90" |
---|
| 188 | END SUBROUTINE compute_geopot |
---|
| 189 | ! |
---|
| 190 | SUBROUTINE compute_caldyn_slow_hydro(rhodz,theta,u, berni,hflux,du) |
---|
| 191 | FIELD_MASS :: rhodz,berni! IN |
---|
| 192 | FIELD_THETA :: theta ! IN |
---|
| 193 | FIELD_U :: u,hflux,du ! IN, OUT, OUT |
---|
| 194 | DECLARE_INDICES |
---|
| 195 | LOGICAL, PARAMETER :: zero=.TRUE. |
---|
| 196 | DBL :: ke, uu |
---|
| 197 | #include "../kernels_unst/caldyn_slow_hydro.k90" |
---|
| 198 | END SUBROUTINE compute_caldyn_slow_hydro |
---|
| 199 | ! |
---|
| 200 | !---------------------------------- Generic ------------------------------ |
---|
| 201 | ! |
---|
| 202 | SUBROUTINE caldyn_vert(convm,rhodz,theta,u, dmass_col,wflux,dtheta_rhodz,du, wwuu) |
---|
| 203 | FIELD_PS :: dmass_col |
---|
| 204 | FIELD_MASS :: convm, rhodz |
---|
| 205 | FIELD_U :: u,du |
---|
| 206 | FIELD_THETA :: dtheta_rhodz, theta |
---|
| 207 | FIELD_GEOPOT :: wflux |
---|
| 208 | DOUBLE2(llm+1, edge_num) :: wwuu |
---|
| 209 | DECLARE_INDICES |
---|
[624] | 210 | DBL :: dF_deta, dFu_deta |
---|
[615] | 211 | wwuu=0. |
---|
[624] | 212 | #include "../kernels_unst/caldyn_wflux.k90" |
---|
| 213 | #include "../kernels_unst/caldyn_dmass.k90" |
---|
[615] | 214 | #include "../kernels_unst/caldyn_vert.k90" |
---|
| 215 | END SUBROUTINE caldyn_vert |
---|
| 216 | ! |
---|
| 217 | SUBROUTINE compute_coriolis(hflux,theta,qu, convm,dtheta_rhodz,du) |
---|
| 218 | FIELD_U :: hflux, Ftheta, qu, du |
---|
| 219 | FIELD_MASS :: convm |
---|
| 220 | FIELD_THETA :: theta, dtheta_rhodz |
---|
| 221 | DECLARE_INDICES |
---|
| 222 | DBL :: divF, du_trisk |
---|
| 223 | #include "../kernels_unst/coriolis.k90" |
---|
| 224 | END SUBROUTINE |
---|
| 225 | ! |
---|
| 226 | SUBROUTINE compute_theta(mass_col,rhodz,theta_rhodz, theta) |
---|
| 227 | FIELD_PS :: mass_col |
---|
| 228 | FIELD_MASS :: rhodz |
---|
| 229 | FIELD_THETA :: theta, theta_rhodz |
---|
| 230 | DECLARE_INDICES |
---|
| 231 | DBL :: m |
---|
| 232 | #include "../kernels_unst/theta.k90" |
---|
| 233 | END SUBROUTINE |
---|
| 234 | ! |
---|
| 235 | SUBROUTINE compute_pvort_only(rhodz,u,qv,qu) |
---|
| 236 | FIELD_MASS :: rhodz |
---|
| 237 | FIELD_U :: u,qu |
---|
| 238 | FIELD_Z :: qv |
---|
| 239 | DECLARE_INDICES |
---|
| 240 | DBL :: etav, hv |
---|
| 241 | #include "../kernels_unst/pvort_only.k90" |
---|
| 242 | END SUBROUTINE compute_pvort_only |
---|
| 243 | ! |
---|
| 244 | SUBROUTINE compute_caldyn_fast(tau, pk,berni,theta,geopot, du,u) |
---|
| 245 | DBL, INTENT(IN) :: tau |
---|
| 246 | FIELD_MASS :: pk,berni ! INOUT, OUT |
---|
| 247 | FIELD_THETA :: theta ! IN |
---|
| 248 | FIELD_GEOPOT :: geopot ! IN |
---|
| 249 | FIELD_U :: u,du ! INOUT,INOUT |
---|
| 250 | DECLARE_INDICES |
---|
| 251 | DBL :: due |
---|
| 252 | |
---|
| 253 | #include "../kernels_unst/caldyn_fast.k90" |
---|
| 254 | |
---|
| 255 | END SUBROUTINE compute_caldyn_fast |
---|
| 256 | ! |
---|
| 257 | !----------------------------- Unused ----------------------------- |
---|
| 258 | ! |
---|
| 259 | SUBROUTINE gradient(b,grad) BINDC(gradient) |
---|
| 260 | DOUBLE2(llm,primal_num) :: b |
---|
| 261 | DOUBLE2(llm,edge_num) :: grad |
---|
| 262 | DECLARE_INDICES |
---|
| 263 | #include "../kernels_unst/gradient.k90" |
---|
| 264 | END SUBROUTINE |
---|
| 265 | ! |
---|
| 266 | SUBROUTINE div(u,divu) BINDC(div) |
---|
| 267 | DOUBLE2(llm,primal_num) :: divu |
---|
| 268 | DOUBLE2(llm,edge_num) :: u |
---|
| 269 | DECLARE_INDICES |
---|
| 270 | DBL :: div_ij |
---|
| 271 | !$OMP PARALLEL NUM_THREADS(nb_threads) |
---|
| 272 | #include "../kernels_unst/div.k90" |
---|
| 273 | !$OMP END PARALLEL |
---|
| 274 | END SUBROUTINE |
---|
| 275 | ! |
---|
| 276 | !---------------------------- CONTEXT INITIALIZATION -------------------------- |
---|
| 277 | ! |
---|
| 278 | #define ALLOC1(v,n1) IF(ALLOCATED(v)) DEALLOCATE(v) ; ALLOCATE(v(n1)) |
---|
| 279 | #define ALLOC2(v,n1,n2) IF(ALLOCATED(v)) DEALLOCATE(v) ; ALLOCATE(v(n1,n2)) |
---|
| 280 | |
---|
| 281 | SUBROUTINE init_mesh( & |
---|
| 282 | primal_deg_, primal_edge_, primal_ne_, & |
---|
| 283 | dual_deg_, dual_edge_, dual_ne_, dual_vertex_, & |
---|
| 284 | left_, right_, up_, down_ ,& |
---|
| 285 | trisk_deg_, trisk_) BINDC(init_mesh) |
---|
| 286 | INDEX :: primal_deg_(primal_num), primal_edge_(max_primal_deg,primal_num), & |
---|
| 287 | primal_ne_(max_primal_deg,primal_num), & |
---|
| 288 | dual_deg_(dual_num), dual_edge_(max_dual_deg,dual_num), & |
---|
| 289 | dual_ne_(max_dual_deg,dual_num), & |
---|
| 290 | dual_vertex_(max_dual_deg,dual_num), & |
---|
| 291 | trisk_deg_(edge_num), trisk_(max_trisk_deg, edge_num) |
---|
| 292 | INDEX, DIMENSION(edge_num) :: left_, right_, down_, up_ |
---|
| 293 | |
---|
| 294 | PRINT *, 'init_mesh ...' |
---|
| 295 | PRINT *, 'Primal mesh : ', primal_num, max_primal_deg |
---|
| 296 | PRINT *, 'Dual mesh : ', dual_num, max_dual_deg |
---|
| 297 | PRINT *, 'Edge mesh : ', edge_num, max_trisk_deg |
---|
| 298 | PRINT *, 'Vertical levels :', llm |
---|
| 299 | ALLOC1(primal_deg, primal_num) |
---|
| 300 | ALLOC2(primal_edge, max_primal_deg,primal_num) |
---|
| 301 | ALLOC2(primal_ne, max_primal_deg,primal_num) |
---|
| 302 | ALLOC1(dual_deg,dual_num) |
---|
| 303 | ALLOC2(dual_edge, max_dual_deg,dual_num) |
---|
| 304 | ALLOC2(dual_ne, max_dual_deg,dual_num) |
---|
| 305 | ALLOC2(dual_vertex, max_dual_deg,dual_num) |
---|
| 306 | ALLOC1(trisk_deg, edge_num) |
---|
| 307 | ALLOC2(trisk, max_trisk_deg, edge_num) |
---|
| 308 | PRINT *, SHAPE(trisk), edge_num |
---|
| 309 | ALLOC1(left, edge_num) |
---|
| 310 | ALLOC1(right, edge_num) |
---|
| 311 | ALLOC1(up, edge_num) |
---|
| 312 | ALLOC1(down, edge_num) |
---|
| 313 | primal_deg(:) = primal_deg_(:) |
---|
| 314 | primal_edge(:,:) = primal_edge_(:,:) |
---|
| 315 | primal_ne(:,:) = primal_ne_(:,:) |
---|
| 316 | dual_deg(:) = dual_deg_(:) |
---|
| 317 | dual_edge(:,:) = dual_edge_(:,:) |
---|
| 318 | dual_ne(:,:) = dual_ne_(:,:) |
---|
| 319 | dual_vertex(:,:) = dual_vertex_(:,:) |
---|
| 320 | IF(MINVAL(dual_deg)<3) THEN |
---|
| 321 | STOP 'At least one dual cell has less than 3 vertices' |
---|
| 322 | END IF |
---|
| 323 | IF(MINVAL(primal_deg)<3) THEN |
---|
| 324 | STOP 'At least one primal cell has less than 3 vertices' |
---|
| 325 | END IF |
---|
| 326 | left(:)=left_(:) |
---|
| 327 | right(:)=right_(:) |
---|
| 328 | down(:)=down_(:) |
---|
| 329 | up=up_(:) |
---|
| 330 | trisk_deg(:)=trisk_deg_(:) |
---|
| 331 | trisk(:,:)=trisk_(:,:) |
---|
| 332 | PRINT *, MAXVAL(primal_edge), edge_num |
---|
| 333 | PRINT *, MAXVAL(dual_edge), edge_num |
---|
| 334 | PRINT *, MAXVAL(dual_vertex), dual_num |
---|
| 335 | PRINT *, MAXVAL(trisk), edge_num |
---|
| 336 | PRINT *, MAX(MAXVAL(left),MAXVAL(right)), primal_num |
---|
| 337 | PRINT *, MAX(MAXVAL(up),MAXVAL(down)), dual_num |
---|
| 338 | PRINT *, SHAPE(trisk), edge_num |
---|
| 339 | PRINT *,' ... Done.' |
---|
| 340 | END SUBROUTINE init_mesh |
---|
| 341 | |
---|
| 342 | SUBROUTINE init_metric(Ai_, Av_, fv_, le_de_, Riv2_, wee_) BINDC(init_metric) |
---|
| 343 | DOUBLE1(primal_num) :: Ai_ |
---|
| 344 | DOUBLE1(dual_num) :: Av_, fv_ |
---|
| 345 | DOUBLE1(edge_num) :: le_de_ |
---|
| 346 | DOUBLE2(max_dual_deg,dual_num) :: Riv2_ |
---|
| 347 | DOUBLE2(max_trisk_deg,edge_num) :: wee_ |
---|
| 348 | PRINT *, 'init_metric ...' |
---|
| 349 | ALLOC1(Ai,primal_num) |
---|
| 350 | ALLOC1(Av,dual_num) |
---|
| 351 | ALLOC1(fv,dual_num) |
---|
| 352 | ALLOC1(le_de,edge_num) |
---|
| 353 | ALLOC2(Riv2, max_dual_deg, dual_num) |
---|
| 354 | ALLOC2(wee, max_trisk_deg, edge_num) |
---|
| 355 | Ai(:) = Ai_(:) |
---|
| 356 | Av(:) = Av_(:) |
---|
| 357 | fv(:) = fv_(:) |
---|
| 358 | le_de(:) = le_de_(:) |
---|
| 359 | Riv2(:,:)=Riv2_(:,:) |
---|
| 360 | wee(:,:) = wee_(:,:) |
---|
| 361 | PRINT *, MAXVAL(ABS(Ai)) |
---|
| 362 | PRINT *, MAXVAL(ABS(Av)) |
---|
| 363 | PRINT *, MAXVAL(ABS(fv)) |
---|
| 364 | PRINT *, MAXVAL(ABS(le_de)) |
---|
| 365 | PRINT *, MAXVAL(ABS(Riv2)) |
---|
| 366 | PRINT *, MAXVAL(ABS(wee)) |
---|
| 367 | PRINT *, MINVAL(right), MAXVAL(right) |
---|
| 368 | PRINT *, MINVAL(right), MAXVAL(left) |
---|
| 369 | PRINT *,' ... Done.' |
---|
[635] | 370 | IF(nb_threads==0) nb_threads=OMP_GET_MAX_THREADS() |
---|
[615] | 371 | PRINT *,'OpenMP : max_threads, num_procs, nb_threads', OMP_GET_MAX_THREADS(), omp_get_num_procs(), nb_threads |
---|
| 372 | |
---|
| 373 | END SUBROUTINE init_metric |
---|
| 374 | ! |
---|
| 375 | SUBROUTINE init_params() BINDC(init_params) |
---|
| 376 | PRINT *, 'Setting physical parameters ...' |
---|
| 377 | IF(hydrostatic) THEN |
---|
| 378 | PRINT *, 'Hydrostatic dynamics (HPE)' |
---|
| 379 | ELSE |
---|
| 380 | PRINT *, 'Non-hydrostatic dynamics (Euler)' |
---|
| 381 | END IF |
---|
| 382 | kappa = Rd/cpp |
---|
| 383 | PRINT *, 'g = ',g |
---|
| 384 | PRINT *, 'preff = ',preff |
---|
| 385 | PRINT *, 'Treff = ',Treff |
---|
| 386 | PRINT *, 'Rd = ',Rd |
---|
| 387 | PRINT *, 'cpp = ',cpp |
---|
| 388 | PRINT *, 'kappa = ',kappa |
---|
| 389 | PRINT *, '... Done' |
---|
| 390 | END SUBROUTINE init_params |
---|
| 391 | ! |
---|
| 392 | SUBROUTINE init_hybrid(bl,dak,dbk) BINDC(init_hybrid) |
---|
| 393 | DOUBLE2(llm+1, primal_num) :: bl |
---|
| 394 | DOUBLE2(llm, primal_num) :: dak,dbk |
---|
| 395 | PRINT *, 'Setting hybrid coefficients ...' |
---|
| 396 | ALLOC2(mass_bl, llm+1, primal_num) |
---|
| 397 | ALLOC2(mass_dak, llm, primal_num) |
---|
| 398 | ALLOC2(mass_dbk, llm, primal_num) |
---|
| 399 | mass_bl(:,:) = bl(:,:) |
---|
| 400 | mass_dak(:,:) = dak(:,:) |
---|
| 401 | mass_dbk(:,:) = dbk(:,:) |
---|
| 402 | PRINT *, '... Done, llm = ', llm |
---|
| 403 | END SUBROUTINE Init_hybrid |
---|
| 404 | |
---|
| 405 | !---------------------------------------------- XIOS ----------------------------------------- |
---|
| 406 | |
---|
[624] | 407 | #ifdef CPP_USING_XIOS |
---|
| 408 | |
---|
[615] | 409 | SUBROUTINE setup_xios() BINDC(setup_xios) |
---|
| 410 | ! MPI_INIT / MPI_finalize are assumed to be called BEFORE/AFTER this routine |
---|
| 411 | ! This is the case when calling from Python after importing mpi4py |
---|
| 412 | INTEGER :: ierr, mpi_threading_mode |
---|
| 413 | |
---|
| 414 | PRINT *, 'Initialize XIOS and obtain our communicator' |
---|
| 415 | CALL xios_initialize("icosagcm",return_comm=comm_icosa) |
---|
| 416 | |
---|
| 417 | PRINT *, 'Initialize our XIOS context' |
---|
| 418 | |
---|
| 419 | CALL xios_context_initialize("icosagcm",comm_icosa) |
---|
| 420 | CALL xios_get_handle("icosagcm",ctx_hdl) |
---|
| 421 | CALL xios_set_current_context(ctx_hdl) |
---|
| 422 | ! CALL xios_set_axis_attr("lev",n_glo=llm ,value=lev_value) ; |
---|
| 423 | ! CALL xios_set_axis_attr("levp1",n_glo=llm+1 ,value=lev_valuep1) ; |
---|
| 424 | ! CALL xios_set_axis_attr("nq",n_glo=nqtot, value=nq_value) ; |
---|
| 425 | ! CALL xios_set_domaingroup_attr("i",ni_glo=ncell_tot, ibegin=displ, ni=ncell) |
---|
| 426 | ! CALL xios_set_domaingroup_attr("i", data_dim=1, type='unstructured' , nvertex=6, i_index=ind_glo) |
---|
| 427 | ! CALL xios_set_domaingroup_attr("i",lonvalue_1d=lon, latvalue_1d=lat, bounds_lon_1d=bounds_lon, bounds_lat_1d=bounds_lat) |
---|
| 428 | ! CALL xios_set_domain_attr("u",ni_glo=ncell_tot, ibegin=displ, ni=ncell) |
---|
| 429 | ! CALL xios_set_domain_attr("u", data_dim=1, type='unstructured' , nvertex=2, i_index=ind_glo) |
---|
| 430 | ! CALL xios_set_domain_attr("u",lonvalue_1d=lon, latvalue_1d=lat, bounds_lon_1d=bounds_lon, bounds_lat_1d=bounds_lat) |
---|
| 431 | ! CALL xios_set_domain_attr("v",ni_glo=ncell_tot, ibegin=displ, ni=ncell) |
---|
| 432 | ! CALL xios_set_domain_attr("v", data_dim=1, type='unstructured' , nvertex=3) |
---|
| 433 | ! CALL xios_set_domain_attr("v",lonvalue_1d=lon, latvalue_1d=lat, bounds_lon_1d=bounds_lon, bounds_lat_1d=bounds_lat) |
---|
| 434 | ! CALL xios_set_timestep(dtime) |
---|
| 435 | ! CALL xios_set_fieldgroup_attr("standard_output", freq_op=itau_out*xios_timestep, freq_offset=(itau_out-1)*xios_timestep) |
---|
| 436 | ! CALL xios_close_context_definition() |
---|
| 437 | |
---|
| 438 | ! PRINT *, 'Read data' |
---|
| 439 | ! CALL xios_recv_field(name,field) |
---|
| 440 | |
---|
| 441 | ! PRINT *,'Write data' |
---|
| 442 | ! CALL xios_send_field(name,field_tmp) |
---|
| 443 | |
---|
| 444 | ! PRINT *, 'Flush to disk, clean up and die' |
---|
| 445 | ! CALL xios_context_finalize |
---|
| 446 | END SUBROUTINE setup_xios |
---|
| 447 | ! |
---|
| 448 | SUBROUTINE call_xios_set_timestep(dt) BINDC(xios_set_timestep) |
---|
| 449 | DBL, VALUE :: dt |
---|
| 450 | TYPE(xios_duration) :: dtime |
---|
| 451 | dtime%second=dt |
---|
| 452 | CALL xios_set_timestep(dtime) |
---|
| 453 | END SUBROUTINE call_xios_set_timestep |
---|
| 454 | |
---|
| 455 | SUBROUTINE call_xios_update_calendar(step) BINDC(xios_update_calendar) |
---|
| 456 | INDEX, value :: step |
---|
| 457 | CALL xios_update_calendar(step) |
---|
| 458 | END SUBROUTINE call_xios_update_calendar |
---|
| 459 | |
---|
[624] | 460 | #endif |
---|
| 461 | |
---|
[615] | 462 | END MODULE caldyn_unstructured_mod |
---|