JGonzalez/vlaplex
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add Z_list

Browse source
This commit is contained in:
JHendrikx 2025-02-04 14:24:15 +01:00
commit 74f011adf4
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
vlaplex.f90
View file

@ -67,6 +67,7 @@ program VlaPlEx
real(dp), allocatable, dimension(:,:):: T_i real(dp), allocatable, dimension(:,:):: T_i
real(dp), allocatable, dimension(:):: n_e real(dp), allocatable, dimension(:):: n_e
real(dp), allocatable, dimension(:):: Zave real(dp), allocatable, dimension(:):: Zave
real(dp), allocatable, dimension(:):: Z_list
real(dp), allocatable, dimension(:):: diag, diag_low, diag_high real(dp), allocatable, dimension(:):: diag, diag_low, diag_high
real(dp), allocatable, dimension(:,:):: A real(dp), allocatable, dimension(:,:):: A
real(dp), allocatable, dimension(:):: Res real(dp), allocatable, dimension(:):: Res
@ -157,13 +158,14 @@ program VlaPlEx
write(*, '(A,ES0.4e3)') 'CFL: ', dt*vf/dr write(*, '(A,ES0.4e3)') 'CFL: ', dt*vf/dr
nz = 2 nz = 16
! Allocate vectors ! Allocate vectors
allocate(f_i(1:nz,1:nr,1:nv), f_i_old(1:nz,1:nr,1:nv)) allocate(f_i(1:nz,1:nr,1:nv), f_i_old(1:nz,1:nr,1:nv))
allocate(n_i(1:nz,1:nr)) allocate(n_i(1:nz,1:nr))
allocate(sum_ni(1:nr)) allocate(sum_ni(1:nr))
allocate(u_i(1:nz,1:nr), E_i(1:nr), T_i(1:nz,1:nr)) allocate(u_i(1:nz,1:nr), E_i(1:nr), T_i(1:nz,1:nr))
allocate(Zave(1:nr)) allocate(Zave(1:nr))
allocate(Z_list(1:nz))
allocate(n_e(1:nr)) allocate(n_e(1:nr))
allocate(phi(1:nr), phi_old(1:nr), E(1:nr)) allocate(phi(1:nr), phi_old(1:nr), E(1:nr))
allocate(fCum_i(1:nz,1:nv)) allocate(fCum_i(1:nz,1:nv))
@ -177,6 +179,7 @@ program VlaPlEx
n_e = 0.0_dp n_e = 0.0_dp
T_e = 0.0_dp T_e = 0.0_dp
Zave = 0.0_dp Zave = 0.0_dp
Z_list = (/(I, I = 1, nz) /)
phi = 0.0_dp phi = 0.0_dp
phi_old = 0.0_dp phi_old = 0.0_dp
E = 0.0_dp E = 0.0_dp
@ -268,7 +271,6 @@ program VlaPlEx
f_i_old(:,:,nv) = 0.0_dp f_i_old(:,:,nv) = 0.0_dp
sum_ni = 0.0_dp sum_ni = 0.0_dp
! Advect in the r direction ! Advect in the r direction
do iz = 1, nz do iz = 1, nz
!$omp parallel do !$omp parallel do
do i = 1, nr do i = 1, nr
@ -300,7 +302,6 @@ program VlaPlEx
!$omp end parallel do !$omp end parallel do
sum_ni = sum_ni + Zave * n_i(iz,:) sum_ni = sum_ni + Zave * n_i(iz,:)
end do end do
! Assume quasi-neutrality to start iterating ! Assume quasi-neutrality to start iterating
n_e = 1.0_dp/nz * sum_ni n_e = 1.0_dp/nz * sum_ni