scale_localmesh_3d.F90

Go to the documentation of this file.

!-------------------------------------------------------------------------------
#include "scaleFElib.h"
module scale_localmesh_3d
 
  !-----------------------------------------------------------------------------
  !
  !++ used modules
  !
  use scale_precision
  use scale_localmesh_base, only: &
    localmeshbase, localmeshbase_init, localmeshbase_final
  use scale_localmesh_2d, only: localmesh2d
  use scale_element_base, only: elementbase, elementbase3d
 
  !-----------------------------------------------------------------------------
  implicit none
  private
 
  !-----------------------------------------------------------------------------
  !
  !++ Public type & procedure
  ! 
  type, extends(localmeshbase), public :: localmesh3d
 
    type(elementbase3d), pointer :: refelem3d
    real(rp) :: xmin, xmax
    real(rp) :: ymin, ymax
    real(rp) :: zmin, zmax
    integer :: nex
    integer :: ney
    integer :: nez
    integer :: ne2d
    integer :: ne2da
 
    real(rp), allocatable :: sz(:,:)
    real(rp), allocatable :: zs(:,:)
    real(rp), allocatable :: gi3(:,:,:)
    real(rp), allocatable :: gsqrth(:,:)
    real(rp), allocatable :: zlev(:,:)
    real(rp), allocatable :: gam(:,:)
 
    class(localmesh2d), pointer :: lcmesh2d
    real(rp), allocatable :: lon2d(:,:)     
    real(rp), allocatable :: lat2d(:,:)
 
    integer, allocatable :: emap3dto2d(:)  
  contains
    procedure :: setlocalmesh2d => localmesh3d_setlocalmesh2d 
    procedure :: getvmapz1d => localmesh3d_getvmapz1d 
  end type localmesh3d
 
  public :: localmesh3d_init, localmesh3d_final
 
  !-----------------------------------------------------------------------------
  !
  !++ Public parameters & variables
  !
  
  !-----------------------------------------------------------------------------
  !
  !++ Private procedure
  !
  
  !-----------------------------------------------------------------------------
  !
  !++ Private parameters & variables
  !
 
contains
!OCL SERIAL
  subroutine localmesh3d_init( this, &
    lcdomID, refElem, myrank )
    implicit none
 
    class(localmesh3d), intent(inout) :: this
    integer, intent(in) :: lcdomid
    class(elementbase3d), intent(in), target :: refelem
    integer, intent(in), optional :: myrank
    !-------------------------------------------------
 
    this%refElem3D    => refelem
    nullify( this%lcmesh2D )
 
    call localmeshbase_init( this, lcdomid, refelem, 3, myrank )
 
    return
  end subroutine localmesh3d_init
 
!OCL SERIAL
  subroutine localmesh3d_final( this, is_generated )
    implicit none
    type(localmesh3d), intent(inout) :: this
    logical, intent(in) :: is_generated
    !-------------------------------------------------
 
    call localmeshbase_final( this, is_generated )
    if (is_generated) then
      deallocate( this%zS, this%Sz )
      deallocate( this%GI3, this%GsqrtH )
      deallocate( this%zlev )
      deallocate( this%gam )
      deallocate( this%lon2D, this%lat2D )
      deallocate( this%EMap3Dto2D )
    end if
 
    return
  end subroutine localmesh3d_final
  
!OCL SERIAL
  subroutine localmesh3d_setlocalmesh2d( this, lcmesh2D )
    implicit none
    class(localmesh3d), intent(inout) :: this
    class(localmesh2d), intent(in), target :: lcmesh2d
    !-------------------------------------------------
 
    this%lcmesh2D => lcmesh2d
 
    return
  end subroutine localmesh3d_setlocalmesh2d
 
!OCL SERIAL
  subroutine localmesh3d_getvmapz1d( this, vmapM, vmapP )
 
    implicit none
    class(localmesh3d), intent(in), target :: this
    integer, intent(out) :: vmapm(this%refelem3d%nfptot,this%nez)
    integer, intent(out) :: vmapp(this%refelem3d%nfptot,this%nez)    
 
    integer :: ke_z
    integer :: f
    integer :: vs, ve
 
    class(elementbase3d), pointer :: elem
    !------------------------------
 
    elem => this%refElem3D
 
    do ke_z=1, this%NeZ
      do f=1, elem%Nfaces_h
        vs = 1 + (f-1)*elem%Nfp_h
        ve = vs + elem%Nfp_h - 1
        vmapm(vs:ve,ke_z) = elem%Fmask_h(:,f) + (ke_z-1)*elem%Np
      end do
      do f=1, elem%Nfaces_v
        vs = elem%Nfp_h*elem%Nfaces_h + 1 + (f-1)*elem%Nfp_v
        ve = vs + elem%Nfp_v - 1
        vmapm(vs:ve,ke_z) = elem%Fmask_v(:,f) + (ke_z-1)*elem%Np
      end do
      vmapp(:,ke_z) = vmapm(:,ke_z)
    end do
 
    do ke_z=1, this%NeZ
      vs = elem%Nfp_h*elem%Nfaces_h + 1
      ve = vs + elem%Nfp_v - 1
      if (ke_z > 1) &
        vmapp(vs:ve,ke_z) = elem%Fmask_v(:,2) + (ke_z-2)*elem%Np
 
      vs = elem%Nfp_h*elem%Nfaces_h + elem%Nfp_v + 1
      ve = vs + elem%Nfp_v - 1
      if (ke_z < this%NeZ) &
        vmapp(vs:ve,ke_z) = elem%Fmask_v(:,1) + ke_z*elem%Np
    end do
 
    return
  end subroutine localmesh3d_getvmapz1d
 
end module scale_localmesh_3d