Homec4science
Diffusion COSOlver (master) Browse

cosolver/src/output_mod.f90
5a16512d28damaster

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
use model
IMPLICIT NONE
! HDF5 file
CHARACTER(len=llen),PUBLIC :: SUFFIX_resfileCei ! main result file
CHARACTER(len=llen),PUBLIC :: SUFFIX_resfileCie ! main result file
CHARACTER(len=llen),PUBLIC :: SUFFIX_resfileCaa ! main result file
CHARACTER(len=llen),PUBLIC :: suffix_resfile
! outputdir
CHARACTER(len=llen), PUBLIC :: outdir ! output dir
! fullpath to output files
CHARACTER(len=llen), PUBLIC :: fullpath_self ! output dir
CHARACTER(len=llen), PUBLIC :: fullpath_ei ! output dir
CHARACTER(len=llen), PUBLIC :: fullpath_ie ! 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,llen,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
! copy input file to outdir
call system( 'cp fort.' // trim(adjustl(strlu_)) // ' ' // trim(adjustl(outdir))//'/parameters.in' )
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
integer :: ikp, kpfid, fid_
! OUTPUT electron-ion if EICOLLS
IF(eicolls .and. is_e) THEN
!___________________________________________________
!
! Output Electron-ion Test Operator
!
!
IF( me_e .eq. 0) WRITE(*,*) 'Write CeiT in ',trim(adjustl(fullpath_ei))
!
dst_name = "/Ceipj/CeipjT"
IF( IFGKCOULOMB) THEN
CALL putarr(eifidres,trim(adjustl(dst_name)),CeiTdp_finalized,'CeiT ', ionode = 0)
ELSE
CALL putarr(eifidres,trim(adjustl(dst_name)),CeiTdp,'CeiT ',pardim=1)
ENDIF
CALL putarr(eifidres,'/Ceipj/rows_eT',rows_e,'row label',ionode=0)
!
! 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)
CALL ATTACH(eifidres,trim(adjustl( dst_name)),"NCPU", nprocs_ei)
CALL ATTACH(eifidres,trim(adjustl( dst_name)),"model", etest_coll )
!
! Add model parameters
!
CALL attach(eifidres, trim(adjustl(dst_name)),"sigmaei", sigmaei)
CALL attach(eifidres, trim(adjustl(dst_name)),"tauie",tauie)
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)
!
!___________________________________________________
!
! Output Electron-ion Field/Back-reaction
!
!
!
IF( me_e .eq. 0) WRITE(*,*) 'Write CeiF in ',trim(adjustl(fullpath_ei))
!
dst_name = "/Ceipj/CeipjF"
IF(IFGKCOULOMB) THEN
CALL putarr(eifidres, trim(adjustl(dst_name)),CeiFdp_finalized,'CeiF',ionode = 0)
ELSE
CALL putarr(eifidres, trim(adjustl(dst_name)),CeiFdp,'CeiF',pardim=1)
ENDIF
CALL putarr(eifidres,'/Ceipj/rows_eF',rows_e,'row label',ionode=0)
!
!
! 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)
CALL ATTACH(eifidres,trim(adjustl( dst_name)),"NCPU", nprocs_ei)
CALL ATTACH(eifidres,trim(adjustl( dst_name)),"model", eback_coll )
!
! Add model parameters
!
CALL attach(eifidres, trim(adjustl(dst_name)),"sigmaei",sigmaei)
CALL attach(eifidres, trim(adjustl(dst_name)),"tauie",tauie)
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
IF(eecolls .and. is_self) THEN
!___________________________________________________
!
! Output Electron-Electron Operator
!
!
IF(me_self .eq. 0) WRITE(*,*) 'Write Cee in ',trim(fullpath_self)
!
! Collisional Matrix, Ceepj
dst_name = "/Caapj/Ceepj"
IF(IFGKCOULOMB) THEN
CALL putarr(selffidres, trim(adjustl( dst_name)),Ceedp_finalized,"Cee", ionode =0)
ELSE
CALL putarr(selffidres, trim(adjustl( dst_name)),Ceedp,"Cee",pardim=1)
ENDIF
CALL putarr(selffidres,'/Caapj/rows_e',rows_e,'row label',ionode=0)
!
! 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)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"NCPU", nprocs_self)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"model", eself_coll )
!
IF(me_self .eq. 0) WRITE(*,*)'... DONE'
ENDIF
!_____________________________________________
!
! Output Ion-Ion Operator
!
!
IF(iicolls .and. is_self) THEN
!
IF(me_self .eq. 0) WRITE(*,*) 'Write Cii in ',trim(adjustl(fullpath_self))
!
! ELSE
dst_name = "/Caapj/Ciipj"
IF( IFGKCOULOMB) THEN
CALL putarr(selffidres, trim(adjustl( dst_name)),Ciidp_finalized &
,"Cii", ionode =0)
ELSE
CALL putarr(selffidres, trim(adjustl( dst_name)),Ciidp,"Cii",pardim=1)
ENDIF
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", nprocs_self)
CALL ATTACH(selffidres,trim(adjustl( dst_name)),"model", iself_coll )
ENDIF
!
! OUTPUT ion-electrons if iecolls
IF(iecolls .and. is_i ) THEN
!___________________________________________________
!
! Output Ion-Electron Test Operator
!
IF( me_i .eq. 0) WRITE(*,*) 'Write CieT in ', trim(adjustl(fullpath_ie))
!
dst_name = '/Ciepj/CiepjT'
IF(IFGKCOULOMB) THEN
CALL putarr(iefidres,trim(adjustl( dst_name)),CieTdp_finalized,'CiepjT',ionode =0)
ELSE
CALL putarr(iefidres,trim(adjustl( dst_name)),CieTdp,'CiepjT',pardim=1)
ENDIF
CALL putarr(iefidres,'/Ciepj/rows_iT',rows_i,'row label',ionode=0)
! 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)),"sigmaei", sigmaei)
CALL attach(iefidres,trim(adjustl( dst_name)),"tauie",tauie)
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)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"NCPU", nprocs_ie)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"model", itest_coll )
!
!___________________________________________________
!
! Output Ion-Electron Field/Back-reaction
!
IF( me_i .eq. 0) WRITE(*,*) 'Write CieF in ', trim(adjustl(fullpath_ie))
!
dst_name = '/Ciepj/CiepjF'
IF(IFGKCOULOMB) THEN
CALL putarr(iefidres, trim(adjustl( dst_name)), CieFdp_finalized,'CieF', ionode = 0)
ELSE
CALL putarr(iefidres, trim(adjustl( dst_name)), CieFdp,'CieF',pardim=1)
ENDIF
CALL putarr(iefidres, '/Ciepj/rows_iF', rows_i,'row label',ionode=0)
!
! 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)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"NCPU", nprocs_ie)
CALL ATTACH(iefidres,trim(adjustl( dst_name)),"model", iback_coll )
!
! Add model parameters
!
CALL attach(iefidres,trim(adjustl( dst_name)),"sigmaei",sigmaei)
CALL attach(iefidres,trim(adjustl( dst_name)),"tauie",tauie)
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
! 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)
! flush out kperp
if( kpscan .and. me .eq. 0 ) then
open(unit = kpfid, file = trim(adjustl(outdir)) // '/kperp.in')
do ikp = iks, ike
write(kpfid,"(F10.6)"), kperp_array(ikp)
enddo
close(kpfid)
endif
if( me .eq. 0) print*, 'Results found in dir :', trim(adjustl(outdir))
open(unit = fid_, file = trim(adjustl(outdir)) // '/cosolver.finished')
END SUBROUTINE write_output
SUBROUTINE create_filenames_output
! defines the ouput filesname
implicit none
CHARACTER(len = 6 ) :: kp_current_str
! self colls
IF (((eecolls .or. iicolls) ) .and. is_self) then
fullpath_self = trim(adjustl(outdir))//'/self.h5'
! IF( IFGKCOULOMB) fullpath_self =trim(adjustl(outdir))//'/self.gkc.h5'
if( kpscan ) THEN
write(kp_current_str, '(I5.5)') ik_current
fullpath_self = trim( adjustl(outdir)) // '/self.' // adjustl(trim( kp_current_str)) //'.h5'
! IF( IFGKCOULOMB) fullpath_self =trim(adjustl(outdir))//'/self.gkc.' // adjustl(trim( kp_current_str)) // '.h5'
ENDIF
ENDIF
! electron-ion coll.
IF ( eicolls .and. is_e) THEN
!
fullpath_ei = trim(adjustl(outdir))//'/ei.h5'
! IF( IFGKCOULOMB) fullpath_ei =trim(adjustl(outdir))//'/ei.gkc.h5'
if( kpscan ) THEN
write(kp_current_str, '(I5.5)') ik_current
fullpath_ei = trim( adjustl(outdir)) // '/ei.' // adjustl(trim( kp_current_str)) //'.h5'
! IF( IFGKCOULOMB) fullpath_ei =trim(adjustl(outdir))//'/ei.gkc.'//adjustl(trim( kp_current_str)) // '.h5'
ENDIF
ENDIF
! ion-electron coll.
IF ( iecolls .and. is_i) THEN
!
fullpath_ie =trim(adjustl(outdir))//'/ie.h5'
! IF( IFGKCOULOMB) fullpath_ie =trim(adjustl(outdir))//'/ie.gkc.h5'
if( kpscan ) THEN
write(kp_current_str, '(I5.5)') ik_current
fullpath_ie = trim( adjustl(outdir)) // '/ie.' // adjustl(trim( kp_current_str)) //'.h5'
! IF( IFGKCOULOMB) fullpath_ie =trim(adjustl(outdir))//'/ie.gkc.'//adjustl(trim( kp_current_str)) // '.h5'
ENDIF
ENDIF
!
END SUBROUTINE create_filenames_output
!
end module output
c4science · Help