diff --git a/src/solve_EM_fields.F90 b/src/solve_EM_fields.F90 index d1de15b04290c319288be55aa6d456a4f1b48f3c..0c506c6d6ea77430ba2a2694a67057e763d5ac22 100644 --- a/src/solve_EM_fields.F90 +++ b/src/solve_EM_fields.F90 @@ -20,7 +20,7 @@ CONTAINS USE grid USE calculus, ONLY : simpson_rule_z USE parallel, ONLY : manual_3D_bcast - use model, ONLY : qe2_taue, qi2_taui, q_e, q_i, lambdaD, KIN_E + use model, ONLY : qe2_taue, qi2_taui, q_e, q_i, lambdaD, KIN_E, sigma_e, sigma_i USE processing, ONLY : compute_density USE geometry, ONLY : iInt_Jacobian, Jacobian IMPLICIT NONE @@ -33,9 +33,7 @@ CONTAINS ! Execution time start CALL cpu_time(t0_poisson) !! Poisson can be solved only for process containing p=0 - ! IF ( SOLVE_POISSON ) THEN - IF ( (ips_e .EQ. ip0_e) .AND. (ips_i .EQ. ip0_i) ) THEN - + IF ( SOLVE_POISSON ) THEN kxloop: DO ikx = ikxs,ikxe kyloop: DO iky = ikys,ikye phi(iky,ikx,izs:ize) = 0._dp @@ -45,14 +43,13 @@ CONTAINS rho_i(izs:ize) = rho_i(izs:ize) & +q_i*kernel_i(ini,iky,ikx,izs:ize,0)*moments_i(ip0_i,ini,iky,ikx,izs:ize,updatetlevel) END DO - !!!! Compute electron particle charge density q_e n_e rho_e = 0._dp IF (KIN_E) THEN ! Kinetic electrons - DO ine=1,jmaxe+1 - rho_e(izs:ize) = rho_e(izs:ize) & - +q_e*kernel_e(ine,iky,ikx,izs:ize,0)*moments_e(ip0_e,ine,iky,ikx,izs:ize,updatetlevel) - END DO + DO ine=1,jmaxe+1 + rho_e(izs:ize) = rho_e(izs:ize) & + +q_e*kernel_e(ine,iky,ikx,izs:ize,0)*moments_e(ip0_e,ine,iky,ikx,izs:ize,updatetlevel) + END DO ELSE ! Adiabatic electrons ! Adiabatic charge density (linked to flux surface averaged phi) ! We compute the flux surface average solving a flux surface averaged @@ -70,13 +67,10 @@ CONTAINS ENDIF !!!!!!!!!!!!!!! Inverting the poisson equation !!!!!!!!!!!!!!!!!!!!!!!!!! phi(iky,ikx,izs:ize) = (rho_e(izs:ize) + rho_i(izs:ize))*inv_poisson_op(iky,ikx,izs:ize) - END DO kyloop END DO kxloop - ! Cancel origin singularity IF (contains_kx0 .AND. contains_ky0) phi(iky_0,ikx_0,:) = 0._dp - ENDIF ! Transfer phi to all the others process along p @@ -85,7 +79,6 @@ CONTAINS ! Execution time end CALL cpu_time(t1_poisson) tc_poisson = tc_poisson + (t1_poisson - t0_poisson) - !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! END SUBROUTINE poisson SUBROUTINE ampere @@ -104,34 +97,24 @@ CONTAINS ! Execution time start CALL cpu_time(t0_poisson) - !! Ampere can be solved only with beta > 0 and for process containing p=1 + !! Ampere can be solved only with beta > 0 and for process containing p=1 moments IF ( SOLVE_AMPERE ) THEN - kxloop: DO ikx = ikxs,ikxe - kyloop: DO iky = ikys,ikye - psi(iky,ikx,izs:ize) = 0._dp - !!!! Compute ion particle current density q_i u_i - I_i = 0._dp - DO ini=1,jmaxi+1 - I_i(izs:ize) = I_i(izs:ize) & - +kernel_i(ini,iky,ikx,izs:ize,0)*moments_i(ip1_i,ini,iky,ikx,izs:ize,updatetlevel) - END DO - I_i = q_i * sqrt_tau_o_sigma_i * I_i - !!!! Compute electron particle charge density q_e n_e - I_e = 0._dp - DO ine=1,jmaxe+1 - I_e(izs:ize) = I_e(izs:ize) & - +q_e*kernel_e(ine,iky,ikx,izs:ize,0)*moments_e(ip1_e,ine,iky,ikx,izs:ize,updatetlevel) - END DO - I_e = q_e * sqrt_tau_o_sigma_e * I_e - !!!!!!!!!!!!!!! Inverting the poisson equation !!!!!!!!!!!!!!!!!!!!!!!!!! - psi(iky,ikx,izs:ize) = beta*(I_e(izs:ize) + I_i(izs:ize))*inv_ampere_op(iky,ikx,izs:ize) - - END DO kyloop - END DO kxloop + psi(ikys:ikye,ikxs:ikxe,izs:ize) = 0._dp + !!!! ion particle current density contribution "q_i u_i" + DO ini=1,jmaxi+1 + psi(ikys:ikye,ikxs:ikxe,izs:ize) = psi(ikys:ikye,ikxs:ikxe,izs:ize) & + +q_i*sqrt_tau_o_sigma_i*kernel_i(ini,ikys:ikye,ikxs:ikxe,izs:ize,0)*moments_i(ip1_i,ini,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel) + END DO + !!!! electron particle current density contribution "q_e u_e" + DO ine=1,jmaxe+1 + psi(ikys:ikye,ikxs:ikxe,izs:ize) = psi(ikys:ikye,ikxs:ikxe,izs:ize) & + +q_e*sqrt_tau_o_sigma_e*kernel_e(ine,ikys:ikye,ikxs:ikxe,izs:ize,0)*moments_e(ip1_e,ine,ikys:ikye,ikxs:ikxe,izs:ize,updatetlevel) + END DO + !!!!!!!!!!!!!!! Inverting the poisson equation !!!!!!!!!!!!!!!!!!!!!!!!!! + psi(ikys:ikye,ikxs:ikxe,izs:ize) = beta*psi(ikys:ikye,ikxs:ikxe,izs:ize)*inv_ampere_op(ikys:ikye,ikxs:ikxe,izs:ize) ! Cancel origin singularity IF (contains_kx0 .AND. contains_ky0) psi(iky_0,ikx_0,:) = 0._dp - ENDIF ! Transfer phi to all the others process along p @@ -140,7 +123,6 @@ CONTAINS ! Execution time end CALL cpu_time(t1_poisson) tc_poisson = tc_poisson + (t1_poisson - t0_poisson) - !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! END SUBROUTINE ampere END SUBROUTINE solve_EM_fields