Homec4science
Diffusion COSOlver (improved_sugama) Browse

cosolver/src/output_mod.f90
cd1286e2d410improved_sugama

Last Change
Raw File

output_mod.f90
View Options

MODULE output
! module for output parameters
USE prec_const
USE array
USE basic
USE timer
USE futils, ONLY:closef, putarr,attach,putarrnd
use grid
USE basic_mpi
IMPLICIT NONE
! HDF5 file
CHARACTER(len=250),PUBLIC :: SUFFIX_resfileCei ! main result file
CHARACTER(len=250),PUBLIC :: SUFFIX_resfileCie ! main result file
CHARACTER(len=250),PUBLIC :: SUFFIX_resfileCaa ! main result file
CHARACTER(len=20),PUBLIC :: suffix_resfile
! outputdir
CHARACTER(len=256), PUBLIC :: outdir ! output dir
! FID for resfile
INTEGER, PUBLIC :: selffidres
INTEGER, PUBLIC :: eifidres
INTEGER, PUBLIC :: iefidres
public :: write_output,read_output
CONTAINS
!
!================================================================================
!
SUBROUTINE read_output
! read output namelist
use prec_const
IMPLICIT NONE
logical :: outdir_exist
NAMELIST /OUTPUT_PAR/ suffix_resfile, outdir
! READ INPUT OUTPUT
IF( me .ne. 0 ) GOTO 110
READ(lu_in,output_par)
WRITE(*,output_par)
110 CONTINUE
!
CALL MPI_bcast(suffix_resfile,100,MPI_CHARACTER,0,MPI_COMM_WORLD,ierr)
CALL MPI_bcast(outdir,256,MPI_CHARACTER,0,MPI_COMM_WORLD,ierr)
! Check if diagdir exist
INQUIRE(directory=trim(adjustl(outdir)),exist=outdir_exist)
!
IF( me .eq. 0 ) THEN
IF(.not. outdir_exist) THEN
print*, "output dir not found...."
print*, "Create " // trim(adjustl(outdir )) // ' ...'
call system( 'mkdir ' // trim(adjustl(outdir)) )
ENDIF
ENDIF
!
end SUBROUTINE read_output
!
!----------------------------------------------------------
!
SUBROUTINE write_output
! write collisional matrices to iCa/ab/ab_resfile h5 file
USE model
IMPLICIT NONE
CHARACTER(len = 64) :: dst_name
! OUTPUT electron-ion if EICOLLS
IF(eicolls .and. is_e) THEN
!___________________________________________________
!
! Output Electron-ion Test Operator
!
IF( ETEST .ne. 0) THEN
!
IF( me_e .eq. 0) WRITE(*,*) 'Write CeiT in ',trim(adjustl(outdir))//'/ei.h5'
!
IF( IFCOULOMB .and. GKE) THEN
dst_name = "/Ceipj/CeipjTpjnn"
CALL putarr(eifidres, trim(adjustl(dst_name)), CeiTdp, 'CeiTpjnn',pardim=1)
CALL putarr(eifidres,'/Ceipj/rows_eT',rows_e_4dim,'row label',ionode=0)
ELSE
dst_name = "/Ceipj/CeipjT"
CALL putarr(eifidres,trim(adjustl(dst_name)),CeiTdp,'CeiT ',pardim=1)
CALL putarr(eifidres,'/Ceipj/rows_eT',rows_e,'row label',ionode=0)
ENDIF
!
! Add basic data
CALL attach(eifidres, trim(adjustl(dst_name)), "Pmaxe", Pmaxe)
CALL attach(eifidres, trim(adjustl(dst_name)), "Jmaxe",Jmaxe)
CALL attach(eifidres, trim(adjustl(dst_name)), "Pmaxi",Pmaxi)
CALL attach(eifidres, trim(adjustl(dst_name)), "Jmaxi",Jmaxi)
CALL attach(eifidres, trim(adjustl(dst_name)), "neFLR",neFLR)
CALL attach(eifidres, trim(adjustl(dst_name)), "niFLR",niFLR)
CALL attach(eifidres, trim(adjustl(dst_name)), "ETEST",ETEST)
!
! Add model parameters
!
CALL attach(eifidres, trim(adjustl(dst_name)),"mu",mu)
CALL attach(eifidres, trim(adjustl(dst_name)),"tau",tau)
CALL attach(eifidres, trim(adjustl(dst_name)),"nu",nu)
CALL attach(eifidres, trim(adjustl(dst_name)),"kperp",kperp)
!
! Add CPU time as attribu
!
CALL ATTACH(eifidres,trim(adjustl(dst_name)), "system_clock",system_ellapsed_time)
!
ENDIF
!___________________________________________________
!
! Output Electron-ion Field/Back-reaction
!
IF( EBACK .ne. 0) THEN
!
!
IF( me_e .eq. 0) WRITE(*,*) 'Write CeiF in ',trim(adjustl(outdir))//'/ei.h5'
!
IF(IFCOULOMB .and. GKE) THEN
dst_name = "/Ceipj/CeipjFpjnn"
CALL putarr(eifidres, trim(adjustl(dst_name)),CeiFdp,'CeiFpjnn',pardim=1)
CALL putarr(eifidres,'/Ceipj/rows_eF',rows_e_4dim,'row label',ionode=0)
ELSE
dst_name = "/Ceipj/CeipjF"
CALL putarr(eifidres, trim(adjustl(dst_name)),CeiFdp,'CeiF',pardim=1)
CALL putarr(eifidres,'/Ceipj/rows_eF',rows_e,'row label',ionode=0)
ENDIF
!
!
! Add basic data
CALL attach(eifidres, trim(adjustl(dst_name)), "Pmaxe", Pmaxe)
CALL attach(eifidres, trim(adjustl(dst_name)), "Jmaxe",Jmaxe)
CALL attach(eifidres, trim(adjustl(dst_name)), "Pmaxi",Pmaxi)
CALL attach(eifidres, trim(adjustl(dst_name)), "Jmaxi",Jmaxi)
CALL attach(eifidres, trim(adjustl(dst_name)), "neFLR",neFLR)
CALL attach(eifidres, trim(adjustl(dst_name)), "niFLR",niFLR)
CALL attach(eifidres, trim(adjustl(dst_name)), "EBACK",EBACK)
!
! Add model parameters
!
CALL attach(eifidres, trim(adjustl(dst_name)),"mu",mu)
CALL attach(eifidres, trim(adjustl(dst_name)),"tau",tau)
CALL attach(eifidres, trim(adjustl(dst_name)),"nu",nu)
CALL attach(eifidres, trim(adjustl(dst_name)),"kperp",kperp)
!
! Add CPU time as attribu
!
CALL ATTACH(eifidres,trim(adjustl(dst_name)), "system_clock",system_ellapsed_time)
!
IF( me_e .eq. 0) WRITE(*,*) '... DONE'
ENDIF
ENDIF
IF(eecolls .and. is_self) THEN
!___________________________________________________
!
! Output Electron-Electron Operator
!
!
IF(me_self .eq. 0) WRITE(*,*) 'Write Cee in ',trim(adjustl(outdir))//'/self.h5'
!
! Collisional Matrix, Ceepj
IF(IFCOULOMB .and. GKE) THEN
dst_name = "/Caapj/Ceepjpjnn"
CALL putarr(selffidres, trim(adjustl( dst_name)),Ceedp,"Ceepjnn",pardim=1)
CALL putarr(selffidres,'/Caapj/rows_e',rows_e_4dim,'row label',ionode=0)
ELSE
dst_name = "/Caapj/Ceepj"
CALL putarr(selffidres, trim(adjustl( dst_name)),Ceedp,"Cee",pardim=1)
CALL putarr(selffidres,'/Caapj/rows_e',rows_e,'row label',ionode=0)
ENDIF
!
! Attributes
CALL attach(selffidres, trim(adjustl( dst_name)),"Pmaxe",Pmaxe)
CALL attach(selffidres,trim(adjustl( dst_name)),"Jmaxe",Jmaxe)
CALL attach(selffidres,trim(adjustl( dst_name)),"nu",nu)
CALL attach(selffidres, trim(adjustl( dst_name)),"kperp",kperp)
CALL attach(selffidres,trim(adjustl( dst_name)),"neFLR",neFLR)
CALL attach(selffidres,trim(adjustl( dst_name)),"ESELF",ESELF)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"system_clock",system_ellapsed_time)
!
IF(me_self .eq. 0) WRITE(*,*)'... DONE'
ENDIF
!_____________________________________________
!
! Output Ion-Ion Operator
!
! Note: output ion-ion collision matrix only if the ion and electron Hermite-Laguerre res. are different. Otherwise Cii = Cee.
!
IF(iicolls .and. is_self) THEN
!
IF(me_self .eq. 0) WRITE(*,*) 'Write Cii in ',trim(adjustl(outdir))//'/self.h5'
!
IF(IFCOULOMB .and. GKI) THEN
dst_name = "/Caapj/Ciipjpjnn"
CALL putarr(selffidres, trim(adjustl( dst_name)),Ciidp,"Ciipjnn",pardim=1)
CALL putarr(selffidres,'/Caapj/rows_i',rows_i_4dim,'row label',ionode=0)
ELSE
dst_name = "/Caapj/Ciipj"
CALL putarr(selffidres, trim(adjustl( dst_name)),Ciidp,"Cii",pardim=1)
CALL putarr(selffidres,'/Caapj/rows_i',rows_i,'row label',ionode=0)
ENDIF
!
IF( me .eq. 0) WRITE(*,*)'... DONE'
!
! Attributes
CALL attach(selffidres,trim(adjustl( dst_name)),"Pmaxi",Pmaxi)
CALL attach(selffidres,trim(adjustl( dst_name)),"Jmaxi",Jmaxi)
CALL attach(selffidres,trim(adjustl( dst_name)),"nu",nu)
CALL attach(selffidres,trim(adjustl( dst_name)),"kperp",kperp)
CALL attach(selffidres,trim(adjustl( dst_name)),"niFLR",niFLR)
CALL attach(selffidres,trim(adjustl( dst_name)),"npiFLR",npiFLR)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"system_clock",system_ellapsed_time)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"NCPU",Caanproc)
ENDIF
!
! OUTPUT ion-electrons if iecolls
IF(iecolls .and. is_i ) THEN
!___________________________________________________
!
! Output Ion-Electron Test Operator
!
IF( ITEST .ne. 0 ) THEN
IF( me_i .eq. 0) WRITE(*,*) 'Write CieT in ie.h5'
!
IF(IFCOULOMB .and. GKI) THEN
dst_name = '/Ciepj/CiepjTpjnn'
CALL putarr(iefidres,trim(adjustl( dst_name)), CieTdp, 'CiepjTpjnn',pardim=1)
CALL putarr(iefidres,'/Ciepj/rows_iT',rows_i_4dim,'row label',ionode=0)
ELSE
dst_name = '/Ciepj/CiepjT'
CALL putarr(iefidres,trim(adjustl( dst_name)),CieTdp,'CiepjT',pardim=1)
CALL putarr(iefidres,'/Ciepj/rows_iT',rows_i,'row label',ionode=0)
ENDIF
!
! Add basic data
CALL attach(iefidres, trim(adjustl( dst_name)), "Pmaxe", Pmaxe)
CALL attach(iefidres, trim(adjustl( dst_name)), "Jmaxe",Jmaxe)
CALL attach(iefidres, trim(adjustl( dst_name)), "Pmaxi",Pmaxi)
CALL attach(iefidres, trim(adjustl( dst_name)), "Jmaxi",Jmaxi)
CALL attach(iefidres, trim(adjustl( dst_name)),"neFLR",neFLR)
CALL attach(iefidres, trim(adjustl( dst_name)),"niFLR",niFLR)
CALL attach(iefidres, trim(adjustl( dst_name)),"ITEST",ITEST)
!
! Add model parameters
!
CALL attach(iefidres,trim(adjustl( dst_name)),"mu",mu)
CALL attach(iefidres,trim(adjustl( dst_name)),"tau",tau)
CALL attach(iefidres,trim(adjustl( dst_name)),"nu",nu)
CALL attach(iefidres,trim(adjustl( dst_name)),"kperp",kperp)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"system_clock",system_ellapsed_time)
!
ENDIF
!___________________________________________________
!
! Output Ion-Electron Field/Back-reaction
!
IF(IBACK .ne. 0) THEN
IF( me_i .eq. 0) WRITE(*,*) 'Write CieF in ie.h5'
!
IF(IFCOULOMB .and. GKI) THEN
dst_name = '/Ciepj/CiepjFpjnn'
CALL putarr(iefidres, trim(adjustl( dst_name)), CieFdp,'CieFpjnn',pardim=1)
CALL putarr(iefidres, '/Ciepj/rows_iF', rows_i_4dim,'row label',ionode=0)
ELSE
dst_name = '/Ciepj/CiepjF'
CALL putarr(iefidres, trim(adjustl( dst_name)), CieFdp,'CieF',pardim=1)
CALL putarr(iefidres, '/Ciepj/rows_iF', rows_i,'row label',ionode=0)
ENDIF
!
! Add basic data
CALL attach(iefidres, trim(adjustl( dst_name)), "Pmaxe", Pmaxe)
CALL attach(iefidres, trim(adjustl( dst_name)), "Jmaxe",Jmaxe)
CALL attach(iefidres, trim(adjustl( dst_name)), "Pmaxi", Pmaxi)
CALL attach(iefidres, trim(adjustl( dst_name)), "Jmaxi",Jmaxi)
CALL attach(iefidres, trim(adjustl( dst_name)),"neFLR",neFLR)
CALL attach(iefidres, trim(adjustl( dst_name)),"niFLR",niFLR)
CALL attach(iefidres, trim(adjustl( dst_name)),"IBACK",IBACK)
!
! Add model parameters
!
CALL attach(iefidres,trim(adjustl( dst_name)),"mu",mu)
CALL attach(iefidres,trim(adjustl( dst_name)),"tau",tau)
CALL attach(iefidres,trim(adjustl( dst_name)),"nu",nu)
CALL attach(iefidres,trim(adjustl( dst_name)),"kperp",kperp)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"system_clock",system_ellapsed_time)
!
IF( me .eq. 0) WRITE(*,*) '... DONE'
!
ENDIF
ENDIF
! Close *.h5 files
IF(eicolls .and. is_e ) CALL closef(eifidres)
IF(iecolls .and. is_i ) CALL closef(iefidres)
IF((eecolls .or. iicolls) .and. is_self) CALL closef(selffidres)
END SUBROUTINE write_output
end module output
c4science · Help