Homec4science
Diffusion COSOlver (master) Browse

cosolver/src/basis_transformation_mod.f90
5b7d24338f83master

Last Change
Raw File

basis_transformation_mod.f90
View Options

MODULE basis_transformation
! Basis transformation contains routines to compute/read the basis transformation coefficients
! T4 (drift-kinetic) and T5 (gyrokinetic).
!
! T4: Legendre/AssociatedLaguerrre -> Hermite-Laguerre
! T5: associatedLegendre/AssociatedLaguerre -> Hermite-Laguerre
USE prec_const
USE model
USE basic
USE array
USE T4T5
USE FMZM
IMPLICIT NONE
PRIVATE
CHARACTER(len=*),PARAMETER :: T4CSV = 'T4.csv' ! T4 (drift-kinetic basis transformation) CSV filename
! Upper bound basis transformation
INTEGER, SAVE :: INDT4MAX
!! INTEGER, SAVE :: INDT5MAX
! I/O
INTEGER, SAVE:: T4FID
!! INTEGER, SAVE:: T5FID
! Files containing the list of T4 coefficients
!! CHARACTER(len=*), PARAMETER :: T5in ='T5.in'
CHARACTER(len=*), PARAMETER :: T4in ='T4.in'
PUBLIC :: GET_NNT4, GET_T4, LOAD_BASIS_TRANSFORMATION
CONTAINS
!
include 'numerical_coeffs.h'
!
!______________________________________________________________________________________________
!
SUBROUTINE LOAD_BASIS_TRANSFORMATION
! loads T4 or T5 basis transformation coefficients
IMPLICIT NONE
!
!
! Note: T5 coefficients are computed on the fly
!!! IF(IFT5) CALL GET_NNT5
! T5 coefficients are computed on the fly
!
! Note : T4 coefficients are load from csv files
IF(IFT4) CALL GET_NNT4
!
END SUBROUTINE LOAD_BASIS_TRANSFORMATION
!
!______________________________________________________________________________________________
!
SUBROUTINE GET_NNT4
!
! Inquire Non-Null T4 coefficients and store them as array of character strings
!
IMPLICIT NONE
!
! local variables
INTEGER :: LMAXX,KMAXX,NMAXX
INTEGER :: FMAXX,GMAXX,HMAXX
INTEGER :: NLINE,ENDOF,FID
INTEGER :: Ntables,iTable
CHARACTER(len=100),DIMENSION(:),ALLOCATABLE :: T4TableNames ! string array containing the names of T4 tables in srcT4 dir
CHARACTER(len=100) :: buff_,filename_
LOGICAL :: file_T4_exist
!
INTEGER :: IT1,IT2,IT3,IT4
INTEGER :: NN ! ... number of NN T4 coeff with indices less than FMAXX,GMAXX ...
INTEGER :: iNN
!
CHARACTER*2 :: SIT1,SIT2,SIT3,SIT4
CHARACTER(len =T4len) :: T4STR_
! Maximum indices inquired
!
LMAXX = max(Pmaxe,Pmaxi)
KMAXX = max(Jmaxe,Jmaxi)
NMAXX = max(neFLR,niFLR)
FMAXX = KMAXX + NMAXX
GMAXX = LMAXX + 2*FMAXX
HMAXX = FMAXX + LMAXX/2
!
IF(idxT4max .eq. 0) THEN
INDT4MAX = max(GMAXX,JEmaxx,JEmaxx) +2
ELSE
INDT4MAX = idxT4max
ENDIF
!
IF( me .eq. 0) WRITE(*,*) 'T4 max index: ', INDT4MAX
!
!
! 1.1 open T4TablesList.in files containg the namees of the T4 tables
!
INQUIRE(FILE=T4in,EXIST=file_T4_exist)
IF(.not. file_T4_exist) THEN
ERROR STOP("T4.in file not found... ")
ENDIF
OPEN(FID,FILE=T4in)
! Count the number of files
NTables = 0
ENDOF = 0
!
DO WHILE (ENDOF .eq. 0 )
READ(FID,*,IOSTAT=ENDOF) buff_
NTables = NTables+1
ENDDO
!
NTables = NTables -1
!
CLOSE(FID)
!
! Allocate array of T4 tables
ALLOCATE(T4TableNames(1:NTables))
!
! Get_table names
OPEN(FID,FILE=T4in)
ENDOF = 0
iTable = 1
!
! Get the name of the table containing T4 coeff.
!
DO WHILE (ENDOF .eq. 0 )
READ(FID,*,IOSTAT=ENDOF) buff_
IF (iTable .le. NTables) THEN
T4TableNames(iTable) = adjustl(trim(buff_))
iTable = iTable +1
ENDIF
ENDDO
CLOSE(FID)
!
! Get the number of NN T4 coeff contained in T4TableNames.csv files
NN = 0
DO iTable=1,NTables
filename_ = trim(adjustl(trim(adjustl(T4dir))//'/'//T4TableNames(iTable)))
OPEN(FID,FILE = filename_)
ENDOF = 0
DO WHILE (ENDOF .eq. 0 )
READ(FID,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4
IF (IT1 .le. INDT4MAX .and. IT2 .le. INDT4MAX .and. IT3 .le. INDT4MAX .and. IT4 .le. INDT4MAX) THEN
NN = NN+1
ENDIF
ENDDO
CLOSE(FID)
ENDDO
!
! Get the T4 coefficients
allocate(NNLKPJ(1:NN))
allocate(T4STR(1:NN))
!
! Inquire NN indices
!
iNN = 1
!
DO iTable =1, NTables
filename_ = trim(adjustl(trim(adjustl(T4dir))// '/' // T4TableNames(iTable)))
OPEN(FID,FILE = filename_)
ENDOF = 0
DO WHILE (ENDOF .eq. 0 )
READ(FID,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4,T4STR_
IF (IT1 .le. INDT4MAX .and. IT2 .le. INDT4MAX .and. IT3 .le. INDT4MAX .and. IT4 .le. INDT4MAX ) THEN
!
CALL GET_NNLKPJ(IT1,IT2,IT3,IT4,NNLKPJ(iNN))
T4STR(iNN) = T4STR_
IF(ENDOF .eq. 0) iNN = iNN+1
ENDIF
ENDDO
!
CLOSE(FID)
!
ENDDO
!
DEALLOCATE(T4TableNames)
!
IF( me .eq. 0) WRITE(*,*) NN, 'non-null (NN) Basis Transformation Coefficients loaded up to (l,k,p,j) <=',INDT4MAX,'...'
!
END SUBROUTINE GET_NNT4
!
!----------------------------------------------------------------
!
SUBROUTINE GET_NNLKPJ(l,k,p,j,NNLKPJ_)
!
! construct character string l//$//k//$//p//$//j, called str id
!
IMPLICIT NONE
!
INTEGER,intent(in) :: l,k,p,j
CHARACTER(len=NNLKPJlen),intent(inout) :: NNLKPJ_
!
CHARACTER*2 :: SIT1,SIT2,SIT3,SIT4
!
! Convert int to string
WRITE(SIT1,'(i2)') l
WRITE(SIT2,'(i2)') k
WRITE(SIT3,'(i2)') p
WRITE(SIT4,'(i2)') j
!
NNLKPJ_ = trim(adjustl(SIT1))//'$'//trim(adjustl(SIT2))//'$'//trim(adjustl(SIT3))//'$'//trim(adjustl(SIT4))
!
END SUBROUTINE GET_NNLKPJ
!
FUNCTION GET_T4(l,k,p,j) RESULT(T4STR_)
!
! Returns the T4 string from character string array T4STR
!
! Note: (l,k,p,j) in the table is T_{lk}^{pj}
!
use T4T5
IMPLICIT NONE
!
INTEGER,intent(in) :: l,k,p,j
CHARACTER(len=T4len) :: T4STR_
! Local variables
CHARACTER(len=NNLKPJlen) :: NNLKPJ_
INTEGER :: POS,i,inquireMaxInd
LOGICAL :: IS_NULL
!
!
IF ( l .ge. INDT4MAX .or. k .ge. INDT4MAX .or. p .ge. INDT4MAX .or. j .ge. INDT4MAX) THEN
inquireMaxInd = max(l,j,p,k)
write(*,*) 'WARNING !!!! NEED TO LOAD MORE T4 coeff ..... !!!!!!!!!! Inquire', inquireMaxInd
STOP
ENDIF
!
POS =0
T4STR_ = '0.0'
IS_NULL = T4zero( l, k, p, j )
IF( .not. IS_NULL) THEN
! Get string id
CALL GET_NNLKPJ(l,k,p,j,NNLKPJ_)
! Get T4 position
DO i=1,size(NNLKPJ)
IF( NNLKPJ(i) == NNLKPJ_) THEN
POS = i
GOTO 55
ENDIF
ENDDO
55 CONTINUE
! Retrieve T4 string from array
IF( POS .gt. 0) THEN
T4STR_ = T4str(POS)
!! WRITE(*,*),NNLKPJ_,T4STR_
ELSE
T4STR_ = '0.0'
ENDIF
!
ENDIF
END FUNCTION GET_T4
!
!______________________________________________________________________________________________
!
!
!______________________________________________________________________________________________
!
! SUBROUTINE GET_NNLKMPJ(l,k,m,p,j,NNLKMPJ_)
! !
! ! construct character string l//$//k//$//m//$//p//$//j, called str id
! !
! IMPLICIT NONE
! !
! INTEGER,intent(in) :: l,k,p,j,m
! CHARACTER(len=NNLKMPJlen),intent(inout) :: NNLKMPJ_
! !
! CHARACTER*2 :: SIT1,SIT2,SIT3,SIT4,SIT5
! !
! ! Convert int to string
! WRITE(SIT1,'(i2)') l
! WRITE(SIT2,'(i2)') k
! WRITE(SIT3,'(i2)') m
! WRITE(SIT4,'(i2)') p
! WRITE(SIT5,'(i2)') j
! !
! NNLKMPJ_ = trim(adjustl(SIT1))//'$'//trim(adjustl(SIT2))//'$'//trim(adjustl(SIT3))//'$'//trim(adjustl(SIT4))//'$'//trim(adjustl(SIT5))
! !
! END SUBROUTINE GET_NNLKMPJ
!______________________________________________________________________________________________
!
!___________________________________________________________
!
! SUBROUTINE GET_NNT5
! !
! ! Inquire Non-Null T5 coefficients and store them as array of character strings f
! ! Inquire T5 coefficients in files *.csv
! !
! IMPLICIT NONE
! !
! ! local variables
! !
! INTEGER :: LMAXX,KMAXX,NMAXX
! INTEGER :: NTables, iTable
! CHARACTER(len=100),DIMENSION(:),ALLOCATABLE :: T5TableNames ! string array containing the names of T5 tables in srcT5 dir
! CHARACTER(len=100) :: buff_,filename_
! INTEGER :: ENDOF,FID,ierr
! !
! INTEGER :: IT1,IT2,IT3,IT4,IT5
! INTEGER :: NN ! ... number of NN T5 coeff with indices less than FMAXX,GMAXX ...
! INTEGER :: iNN
! !
! INTEGER :: FMAXX,GMAXX,HMAXX,PMAXX
! !
! CHARACTER*2 :: SIT1,SIT2,SIT3,SIT4,SIT5
! CHARACTER(len =FMlen) :: T5STR_
! !
! FMAXX = KMAXX + NMAXX
! PMAXX = LMAXX + 2*FMAXX
! GMAXX = PMAXX + 2*JEmaxx
! HMAXX = JEmaxx + floor(real(PMAXX)/2)
! !
! ! Max T5 index
! INDT5MAX = idxT5max
! ! !
! IF(me .eq. 0) WRITE(*,*) 'Load basis T5 coeff. up to (p,j,k,m,l) <= ', INDT5MAX
! !
! !
! ! open T5TablesList.txt files containg the namees of the T5 tables
! !
! OPEN(FID,FILE=T5in)
! ! Count the number of files
! NTables = 0
! ENDOF = 0
! !
! DO WHILE (ENDOF .eq. 0 )
! READ(FID,*,IOSTAT=ENDOF) buff_
! NTables = NTables+1
! ENDDO
! !
! NTables = NTables -1
! !
! CLOSE(FID)
! !
! ! Allocate array of T5 tables
! ALLOCATE(T5TableNames(1:NTables))
! !
! OPEN(FID,FILE=T5in)
! ENDOF = 0
! iTable = 1
! !
! ! Get the name of the table containing T5 coeff.
! !
! DO WHILE (ENDOF .eq. 0 )
! READ(FID,*,IOSTAT=ENDOF) buff_
! IF (iTable .le. NTables) THEN
! T5TableNames(iTable) = adjustl(trim(buff_))
! iTable = iTable +1
! ENDIF
! ENDDO
! CLOSE(FID)
! !
! ! Get the number of NN T5 coeff contained in T5TableNames.csv files
! NN = 0
! DO iTable=1,NTables
! filename_ = trim(adjustl(trim(adjustl(T5dir))//T5TableNames(iTable)))
! OPEN(FID,FILE = filename_)
! ENDOF = 0
! DO WHILE (ENDOF .eq. 0 )
! READ(FID,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4,IT5
! IF (IT1 .le. INDT5MAX .and. IT2 .le. INDT5MAX .and. IT3 .le. INDT5MAX .and. IT4 .le. INDT5MAX .and. IT5 .le. INDT5MAX) THEN
! NN = NN+1
! ENDIF
! ENDDO
! CLOSE(FID)
! ENDDO
! !
! ! Get the T5 coefficients
! allocate(NNLKMPJ(1:NN))
! allocate(T5STR(1:NN))
! !
! ! Inquire NN indices
! !
! iNN = 1
! !
! DO iTable =1, NTables
! filename_ = trim(adjustl(trim(adjustl(T5dir))//T5TableNames(iTable)))
! OPEN(FID,FILE = filename_)
! ENDOF = 0
! DO WHILE (ENDOF .eq. 0 )
! READ(FID,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4,IT5,T5STR_
! IF (IT1 .le. INDT5MAX .and. IT2 .le. INDT5MAX .and. IT3 .le. INDT5MAX .and. IT4 .le. INDT5MAX .and. IT5 .le. INDT5MAX ) THEN
! !
! CALL GET_NNLKMPJ(IT1,IT2,IT3,IT4,IT5,NNLKMPJ(iNN))
! T5STR(iNN) = T5STR_
! !! write(*,*) NNLKMPJ(iNN),T5STR(iNN)
! IF(ENDOF .eq. 0) iNN = iNN+1
! ENDIF
! ENDDO
! !
! CLOSE(FID)
! ENDDO
! !
! DEALLOCATE(T5TableNames)
! !
! END SUBROUTINE GET_NNT5
!
!______________________________________________________________________________________________
!
END MODULE basis_transformation
c4science · Help