! //////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// ! ! ! Maintainers : support@fluidnumerics.com ! Official Repository : https://github.com/FluidNumerics/self/ ! ! Copyright © 2024 Fluid Numerics LLC ! ! Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: ! ! 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. ! ! 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in ! the documentation and/or other materials provided with the distribution. ! ! 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from ! this software without specific prior written permission. ! ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS “AS IS” AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT ! LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT ! HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT ! LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY ! THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF ! THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ! ! //////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// ! module SELF_MappedScalar_3D_t use SELF_Constants use SELF_Lagrange use SELF_Scalar_3D use SELF_Tensor_3D use SELF_Mesh_3D use SELF_Geometry_3D use SELF_DomainDecomposition use FEQParse use iso_c_binding implicit none type,extends(Scalar3D),public :: MappedScalar3D_t logical :: geometry_associated = .false. type(SEMHex),pointer :: geometry => null() contains procedure,public :: AssociateGeometry => AssociateGeometry_MappedScalar3D_t procedure,public :: DissociateGeometry => DissociateGeometry_MappedScalar3D_t procedure,public :: SideExchange => SideExchange_MappedScalar3D_t generic,public :: MappedGradient => MappedGradient_MappedScalar3D_t procedure,private :: MappedGradient_MappedScalar3D_t generic,public :: MappedDGGradient => MappedDGGradient_MappedScalar3D_t procedure,private :: MappedDGGradient_MappedScalar3D_t procedure,private :: MPIExchangeAsync => MPIExchangeAsync_MappedScalar3D_t procedure,private :: ApplyFlip => ApplyFlip_MappedScalar3D_t procedure,public :: SetInteriorFromEquation => SetInteriorFromEquation_MappedScalar3D_t ! procedure,public :: WriteTecplot => WriteTecplot_MappedScalar3D_t endtype MappedScalar3D_t contains subroutine AssociateGeometry_MappedScalar3D_t(this,geometry) implicit none class(MappedScalar3D_t),intent(inout) :: this type(SEMHex),target,intent(in) :: geometry if(.not. associated(this%geometry)) then this%geometry => geometry this%geometry_associated = .true. endif endsubroutine AssociateGeometry_MappedScalar3D_t subroutine DissociateGeometry_MappedScalar3D_t(this) implicit none class(MappedScalar3D_t),intent(inout) :: this if(associated(this%geometry)) then this%geometry => null() this%geometry_associated = .false. endif endsubroutine DissociateGeometry_MappedScalar3D_t subroutine SetInteriorFromEquation_MappedScalar3D_t(this,geometry,time) !! Sets the this % interior attribute using the eqn attribute, !! geometry (for physical positions), and provided simulation time. implicit none class(MappedScalar3D_t),intent(inout) :: this type(SEMHex),intent(in) :: geometry real(prec),intent(in) :: time ! Local integer :: i,j,k,iEl,iVar real(prec) :: x real(prec) :: y real(prec) :: z do iVar = 1,this%nVar do iEl = 1,this%nElem do k = 1,this%interp%N+1 do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 ! Get the mesh positions x = geometry%x%interior(i,j,k,iEl,1,1) y = geometry%x%interior(i,j,k,iEl,1,2) z = geometry%x%interior(i,j,k,iEl,1,3) this%interior(i,j,k,iEl,iVar) = & this%eqn(iVar)%Evaluate((/x,y,z,time/)) enddo enddo enddo enddo enddo endsubroutine SetInteriorFromEquation_MappedScalar3D_t subroutine MPIExchangeAsync_MappedScalar3D_t(this,mesh) implicit none class(MappedScalar3D_t),intent(inout) :: this type(Mesh3D),intent(inout) :: mesh ! Local integer :: e1,s1,e2,s2,ivar integer :: globalSideId,r2,tag integer :: iError integer :: msgCount msgCount = 0 do ivar = 1,this%nvar do e1 = 1,this%nElem do s1 = 1,6 e2 = mesh%sideInfo(3,s1,e1) ! Neighbor Element if(e2 > 0) then r2 = mesh%decomp%elemToRank(e2) ! Neighbor Rank if(r2 /= mesh%decomp%rankId) then s2 = mesh%sideInfo(4,s1,e1)/10 globalSideId = abs(mesh%sideInfo(2,s1,e1)) tag = globalsideid+mesh%nUniqueSides*(ivar-1) msgCount = msgCount+1 call MPI_IRECV(this%extBoundary(:,:,s1,e1,ivar), & (this%interp%N+1)*(this%interp%N+1), & mesh%decomp%mpiPrec, & r2,globalSideId, & mesh%decomp%mpiComm, & mesh%decomp%requests(msgCount),iError) msgCount = msgCount+1 call MPI_ISEND(this%boundary(:,:,s1,e1,ivar), & (this%interp%N+1)*(this%interp%N+1), & mesh%decomp%mpiPrec, & r2,globalSideId, & mesh%decomp%mpiComm, & mesh%decomp%requests(msgCount),iError) endif endif enddo enddo enddo mesh%decomp%msgCount = msgCount endsubroutine MPIExchangeAsync_MappedScalar3D_t subroutine ApplyFlip_MappedScalar3D_t(this,mesh) ! Apply side flips to sides where MPI exchanges took place. implicit none class(MappedScalar3D_t),intent(inout) :: this type(Mesh3D),intent(inout) :: mesh ! Local integer :: e1,s1,e2,s2 integer :: i,i2,j,j2 integer :: r2,flip,ivar integer :: bcid real(prec) :: extBuff(1:this%interp%N+1,1:this%interp%N+1) do ivar = 1,this%nvar do e1 = 1,this%nElem do s1 = 1,6 e2 = mesh%sideInfo(3,s1,e1) ! Neighbor Element s2 = mesh%sideInfo(4,s1,e1)/10 bcid = mesh%sideInfo(5,s1,e1) if(e2 > 0) then ! Interior Element r2 = mesh%decomp%elemToRank(e2) ! Neighbor Rank if(r2 /= mesh%decomp%rankId) then flip = mesh%sideInfo(4,s1,e1)-s2*10 ! Need to update extBoundary with flip applied if(flip == 0) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 extBuff(i,j) = this%extBoundary(i,j,s1,e1,ivar) enddo enddo else if(flip == 1) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-i j2 = j extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo else if(flip == 2) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-i j2 = this%interp%N+2-j extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo else if(flip == 3) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = i j2 = this%interp%N+2-j extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo else if(flip == 4) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 extBuff(i,j) = this%extBoundary(j,i,s1,e1,ivar) enddo enddo else if(flip == 5) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-j j2 = i extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo else if(flip == 6) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-j j2 = this%interp%N+2-i extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo else if(flip == 7) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = j j2 = this%interp%N+2-i extBuff(i,j) = this%extBoundary(i2,j2,s1,e1,ivar) enddo enddo endif do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 this%extBoundary(i,j,s1,e1,ivar) = extBuff(i,j) enddo enddo endif endif enddo enddo enddo endsubroutine ApplyFlip_MappedScalar3D_t subroutine SideExchange_MappedScalar3D_t(this,mesh) implicit none class(MappedScalar3D_t),intent(inout) :: this type(Mesh3D),intent(inout) :: mesh ! Local integer :: e1,e2,s1,s2,e2Global integer :: flip integer :: i,i2,j,j2,ivar integer :: r2 integer :: rankId,offset integer,pointer :: elemtorank(:) elemtorank => mesh%decomp%elemToRank(:) rankId = mesh%decomp%rankId offset = mesh%decomp%offsetElem(rankId+1) if(mesh%decomp%mpiEnabled) then call this%MPIExchangeAsync(mesh) endif do concurrent(s1=1:6,e1=1:mesh%nElem,ivar=1:this%nvar) e2Global = mesh%sideInfo(3,s1,e1) s2 = mesh%sideInfo(4,s1,e1)/10 flip = mesh%sideInfo(4,s1,e1)-s2*10 if(e2Global > 0) then r2 = elemToRank(e2Global) if(r2 == rankId) then e2 = e2Global-offset if(flip == 0) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i,j,s2,e2,ivar) enddo enddo else if(flip == 1) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-i j2 = j this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo else if(flip == 2) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-i j2 = this%interp%N+2-j this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo else if(flip == 3) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = i j2 = this%interp%N+2-j this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo else if(flip == 4) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 this%extBoundary(i,j,s1,e1,ivar) = this%boundary(j,i,s2,e2,ivar) enddo enddo else if(flip == 5) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-j j2 = i this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo else if(flip == 6) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = this%interp%N+2-j j2 = this%interp%N+2-i this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo else if(flip == 7) then do j = 1,this%interp%N+1 do i = 1,this%interp%N+1 i2 = j j2 = this%interp%N+2-i this%extBoundary(i,j,s1,e1,ivar) = this%boundary(i2,j2,s2,e2,ivar) enddo enddo endif endif endif enddo if(mesh%decomp%mpiEnabled) then call mesh%decomp%FinalizeMPIExchangeAsync() ! Apply side flips for data exchanged with MPI call this%ApplyFlip(mesh) endif endsubroutine SideExchange_MappedScalar3D_t subroutine MappedGradient_MappedScalar3D_t(this,df) !! Calculates the gradient of a function using the strong form of the gradient !! in mapped coordinates. implicit none class(MappedScalar3D_t),intent(in) :: this real(prec),intent(out) :: df(1:this%N+1,1:this%N+1,1:this%N+1,1:this%nelem,1:this%nvar,1:3) ! Local integer :: iEl,iVar,i,j,k,ii,idir real(prec) :: dfdx,ja do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 ! dsdx(j,i) is contravariant vector i, component j ja = this%geometry%dsdx%interior(ii,j,k,iel,1,idir,1) dfdx = dfdx+this%interp%dMatrix(ii,i)* & this%interior(ii,j,k,iel,ivar)*ja enddo df(i,j,k,iel,ivar,idir) = dfdx enddo do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 ja = this%geometry%dsdx%interior(i,ii,k,iel,1,idir,2) dfdx = dfdx+this%interp%dMatrix(ii,j)* & this%interior(i,ii,k,iel,ivar)*ja enddo df(i,j,k,iel,ivar,idir) = (df(i,j,k,iel,ivar,idir)+dfdx) enddo do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 ja = this%geometry%dsdx%interior(i,j,ii,iel,1,idir,3) dfdx = dfdx+this%interp%dMatrix(ii,k)* & this%interior(i,j,ii,iel,ivar)*ja enddo df(i,j,k,iel,ivar,idir) = (df(i,j,k,iel,ivar,idir)+dfdx)/ & this%geometry%J%interior(i,j,k,iEl,1) enddo endsubroutine MappedGradient_MappedScalar3D_t subroutine MappedDGGradient_MappedScalar3D_t(this,df) !! Calculates the gradient of a function using the weak form of the gradient !! and the average boundary state. !! This method will compute the average boundary state from the !! and attributes of implicit none class(MappedScalar3D_t),intent(in) :: this real(prec),intent(out) :: df(1:this%N+1,1:this%N+1,1:this%N+1,1:this%nelem,1:this%nvar,1:3) ! Local integer :: iEl,iVar,i,j,k,ii,idir real(prec) :: dfdx,jaf,bfl,bfr do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 ! dsdx(j,i) is contravariant vector i, component j jaf = this%geometry%dsdx%interior(ii,j,k,iel,1,idir,1)* & this%interior(ii,j,k,iel,ivar) dfdx = dfdx+this%interp%dgMatrix(ii,i)*jaf enddo bfl = this%avgboundary(j,k,5,iel,ivar)* & this%geometry%dsdx%boundary(j,k,5,iel,1,idir,1) ! west bfr = this%avgboundary(j,k,3,iel,ivar)* & this%geometry%dsdx%boundary(j,k,3,iel,1,idir,1) ! east df(i,j,k,iel,ivar,idir) = dfdx+ & (this%interp%bMatrix(i,1)*bfl+ & this%interp%bMatrix(i,2)*bfr)/this%interp%qweights(i) enddo do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 jaf = this%geometry%dsdx%interior(i,ii,k,iel,1,idir,2)* & this%interior(i,ii,k,iel,ivar) dfdx = dfdx+this%interp%dgMatrix(ii,j)*jaf enddo bfl = this%avgboundary(i,k,2,iel,ivar)* & this%geometry%dsdx%boundary(i,k,2,iel,1,idir,2) ! south bfr = this%avgboundary(i,k,4,iel,ivar)* & this%geometry%dsdx%boundary(i,k,4,iel,1,idir,2) ! north dfdx = dfdx+(this%interp%bMatrix(j,1)*bfl+ & this%interp%bMatrix(j,2)*bfr)/this%interp%qweights(j) df(i,j,k,iel,ivar,idir) = (df(i,j,k,iel,ivar,idir)+dfdx) enddo do concurrent(i=1:this%N+1,j=1:this%N+1, & k=1:this%N+1,iel=1:this%nelem,ivar=1:this%nvar,idir=1:3) dfdx = 0.0_prec do ii = 1,this%N+1 jaf = this%geometry%dsdx%interior(i,j,ii,iel,1,idir,3)* & this%interior(i,j,ii,iel,ivar) dfdx = dfdx+this%interp%dgMatrix(ii,k)*jaf enddo bfl = this%avgboundary(i,j,1,iel,ivar)* & this%geometry%dsdx%boundary(i,j,1,iel,1,idir,3) ! bottom bfr = this%avgboundary(i,j,6,iel,ivar)* & this%geometry%dsdx%boundary(i,j,6,iel,1,idir,3) ! top dfdx = dfdx+(this%interp%bMatrix(k,1)*bfl+ & this%interp%bMatrix(k,2)*bfr)/this%interp%qweights(k) df(i,j,k,iel,ivar,idir) = (df(i,j,k,iel,ivar,idir)+dfdx)/ & this%geometry%J%interior(i,j,k,iEl,1) enddo endsubroutine MappedDGGradient_MappedScalar3D_t ! subroutine WriteTecplot_MappedScalar3D_t(this,geometry,filename) ! implicit none ! class(MappedScalar3D_t),intent(inout) :: this ! type(SEMHex),intent(in) :: geometry ! character(*),intent(in),optional :: filename ! ! Local ! character(8) :: zoneID ! integer :: fUnit ! integer :: iEl,i,j,k,iVar ! character(LEN=self_FileNameLength) :: tecFile ! character(LEN=self_TecplotHeaderLength) :: tecHeader ! character(LEN=self_FormatLength) :: fmat ! character(13) :: timeStampString ! character(5) :: rankString ! real(prec) :: f(1:this%M+1,1:this%M+1,1:this%M+1,1:this%nelem,1:this%nvar) ! real(prec) :: x(1:this%M+1,1:this%M+1,1:this%M+1,1:this%nelem,1:this%nvar,1:3) ! if(present(filename)) then ! tecFile = filename ! else ! tecFile = "mappedscalar.tec" ! endif ! ! Map the mesh positions to the target grid ! x = geometry%x%GridInterp() ! ! Map the solution to the target grid ! f = this%GridInterp() ! open(UNIT=NEWUNIT(fUnit), & ! FILE=trim(tecFile), & ! FORM='formatted', & ! STATUS='replace') ! tecHeader = 'VARIABLES = "X", "Y", "Z"' ! do iVar = 1,this%nVar ! tecHeader = trim(tecHeader)//', "'//trim(this%meta(iVar)%name)//'"' ! enddo ! write(fUnit,*) trim(tecHeader) ! ! Create format statement ! write(fmat,*) this%nvar+3 ! fmat = '('//trim(fmat)//'(ES16.7E3,1x))' ! do iEl = 1,this%nElem ! write(zoneID,'(I8.8)') iEl ! write(fUnit,*) 'ZONE T="el'//trim(zoneID)//'", I=',this%interp%M+1, & ! ', J=',this%interp%M+1,', K=',this%interp%M+1 ! do k = 1,this%interp%M+1 ! do j = 1,this%interp%M+1 ! do i = 1,this%interp%M+1 ! write(fUnit,fmat) x(i,j,k,iEl,1,1), & ! x(i,j,k,iEl,1,2), & ! x(i,j,k,iEl,1,3), & ! f(i,j,k,iEl,1:this%nvar) ! enddo ! enddo ! enddo ! enddo ! close(UNIT=fUnit) ! endsubroutine WriteTecplot_MappedScalar3D_t endmodule SELF_MappedScalar_3D_t