subroutine auxval ! Set auxiliary values, at beginning of simulation USE basic, ONLY: str, speak USE grid, ONLY: local_np, local_np_offset, total_np, local_nj, local_nj_offset, total_nj,& local_nky, local_nky_offset, total_nky, local_nkx, local_nkx_offset, dmax,& local_nz, local_nz_offset, total_nz, init_grids_data, set_grids !USE array USE model, ONLY: Na, EM, LINEARITY, N_HD USE fourier, ONLY: init_grid_distr_and_plans use MPI, ONLY: MPI_COMM_WORLD USE numerics, ONLY: build_dnjs_table, build_dv4Hp_table, compute_lin_coeff, & evaluate_EM_op, evaluate_kernels USE geometry, ONLY: Npol, shear, eval_magnetic_geometry USE closure, ONLY: set_closure_model, hierarchy_closure USE parallel, ONLY: init_parallel_var, my_id, num_procs, & num_procs_p, num_procs_z, num_procs_ky, rank_p, rank_ky, rank_z USE processing, ONLY: init_process USE ExB_shear_flow, ONLY: Setup_ExB_shear_flow #ifdef TEST_SVD USE CLA, ONLY: init_CLA #endif IMPLICIT NONE INTEGER :: i_, ierr CALL speak('=== Set auxiliary values ===') ! Init the grids CALL init_grids_data(Na,EM,LINEARITY) CALL set_grids(shear,Npol,LINEARITY,N_HD,EM,Na) ! Allocate memory for global arrays CALL memory ! Initialize displacement and receive arrays CALL init_parallel_var(local_np,total_np,local_nky,total_nky,local_nz) ! Initialize heatflux variables CALL init_process ! precompute coeff for lin equation CALL eval_magnetic_geometry ! precompute coeff for lin equation and geometry CALL compute_lin_coeff ! precompute the kernels CALL evaluate_kernels ! compute inverse of poisson and ampere operators CALL evaluate_EM_op ! precompute the Laguerre nonlin product coeffs IF ( LINEARITY .NE. 'linear' ) & CALL build_dnjs_table ! precompute the hermite fourth derivative table CALL build_dv4Hp_table ! set the closure scheme in use CALL set_closure_model ! Setup ExB shear variables CALL Setup_ExB_shear_flow #ifdef TEST_SVD ! If we want to test SVD decomposition etc. CALL init_CLA(local_nky,local_np*local_nj) #endif !! Display parallel settings CALL mpi_barrier(MPI_COMM_WORLD, ierr) DO i_ = 0,num_procs-1 CALL mpi_barrier(MPI_COMM_WORLD, ierr) IF (my_id .EQ. i_) THEN IF (my_id .EQ. 0) WRITE(*,*) '' IF (my_id .EQ. 0) WRITE(*,*) '--------- Parallel environement ----------' IF (my_id .EQ. 0) WRITE(*,'(A12,I3)') 'n_procs ', num_procs IF (my_id .EQ. 0) WRITE(*,'(A12,I3,A14,I3,A14,I3)') 'num_procs_p = ', num_procs_p, ', num_procs_ky = ', num_procs_ky, ', num_procs_z = ', num_procs_z IF (my_id .EQ. 0) WRITE(*,*) '' WRITE(*,'(A9,I3,A10,I3,A10,I3,A9,I3)')& 'my_id = ', my_id, ', rank_p = ', rank_p, ', rank_ky = ', rank_ky,', rank_z = ', rank_z WRITE(*,'(A22,I3,A11,I3)')& ' local_np = ', local_np , ', offset = ', local_np_offset WRITE(*,'(A22,I3,A11,I3)')& ' local_nj = ', local_nj , ', offset = ', local_nj_offset WRITE(*,'(A22,I3,A11,I3)')& ' local_nkx = ', local_nkx , ', offset = ', local_nkx_offset WRITE(*,'(A22,I3,A11,I3)')& ' local_nky = ', local_nky , ', offset = ', local_nky_offset WRITE(*,'(A22,I3,A11,I3)')& ' local_nz = ', local_nz , ', offset = ', local_nz_offset IF (my_id .NE. num_procs-1) WRITE (*,*) '' IF (my_id .EQ. num_procs-1) WRITE(*,*) '------------------------------------------' ENDIF ENDDO CALL mpi_barrier(MPI_COMM_WORLD, ierr) SELECT CASE(hierarchy_closure) CASE('truncation') CALL speak('Truncation closure') CASE('max_degree') CALL speak('Max degree closure -> Maximal Napj degree is D = '// str(dmax)) END SELECT END SUBROUTINE auxval