shear_stress.f90

Go to the documentation of this file.

! Copyright (c) 2024-2025, The Neko Authors
! All rights reserved.
!
! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions
! are met:
!
!   * Redistributions of source code must retain the above copyright
!     notice, this list of conditions and the following disclaimer.
!
!   * 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.
!
!   * Neither the name of the authors 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 OWNER 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 shear_stress
  use num_types, only : rp
  use bc, only : bc_t
  use, intrinsic :: iso_c_binding, only : c_ptr
  use utils, only : neko_error
  use coefs, only : coef_t
  use symmetry, only : symmetry_t
  use neumann, only : neumann_t
  use json_module, only : json_file
  use json_utils, only : json_get
  use vector, only : vector_t
  use time_state, only : time_state_t
  implicit none
  private
 
  type, public, extends(bc_t) :: shear_stress_t
     ! This bc takes care of setting the wall-normal component to zero.
     ! It can be passed to associated bc lists, which take care of masking
     ! changes in residuals and solution increments.
     type(symmetry_t) :: symmetry
 
     type(neumann_t) :: neumann_x
     type(neumann_t) :: neumann_y
     type(neumann_t) :: neumann_z
   contains
     procedure, pass(this) :: apply_scalar => shear_stress_apply_scalar
     procedure, pass(this) :: apply_vector => shear_stress_apply_vector
     procedure, pass(this) :: apply_scalar_dev => shear_stress_apply_scalar_dev
     procedure, pass(this) :: apply_vector_dev => shear_stress_apply_vector_dev
     procedure, pass(this) :: init => shear_stress_init
     procedure, pass(this) :: init_from_components => &
          shear_stress_init_from_components
     procedure, pass(this) :: set_stress_scalar => &
          shear_stress_set_stress_scalar
     procedure, pass(this) :: set_stress_array => &
          shear_stress_set_stress_array
     generic :: set_stress => set_stress_scalar, set_stress_array
     procedure, pass(this) :: free => shear_stress_free
     procedure, pass(this) :: finalize => shear_stress_finalize
  end type shear_stress_t
 
contains
 
  subroutine shear_stress_apply_scalar(this, x, n, time, strong)
    class(shear_stress_t), intent(inout) :: this
    integer, intent(in) :: n
    real(kind=rp), intent(inout), dimension(n) :: x
    type(time_state_t), intent(in), optional :: time
    logical, intent(in), optional :: strong
    integer :: i, m, k, facet
    ! Store non-linear index
    integer :: idx(4)
 
    call neko_error("The shear stress bc is not applicable to scalar fields.")
 
  end subroutine shear_stress_apply_scalar
 
  subroutine shear_stress_apply_vector(this, x, y, z, n, time, strong)
    class(shear_stress_t), intent(inout) :: this
    integer, intent(in) :: n
    real(kind=rp), intent(inout), dimension(n) :: x
    real(kind=rp), intent(inout), dimension(n) :: y
    real(kind=rp), intent(inout), dimension(n) :: z
    type(time_state_t), intent(in), optional :: time
    logical, intent(in), optional :: strong
    logical :: strong_
 
    if (present(strong)) then
       strong_ = strong
    else
       strong_ = .true.
    end if
 
    if (strong_) then
       call this%symmetry%apply_vector(x, y, z, n, strong = .true.)
    else
       call this%neumann_x%apply_scalar(x, n, strong = .false.)
       call this%neumann_y%apply_scalar(y, n, strong = .false.)
       call this%neumann_z%apply_scalar(z, n, strong = .false.)
    end if
 
  end subroutine shear_stress_apply_vector
 
  subroutine shear_stress_apply_scalar_dev(this, x_d, time, strong, strm)
    class(shear_stress_t), intent(inout), target :: this
    type(c_ptr), intent(inout) :: x_d
    type(time_state_t), intent(in), optional :: time
    logical, intent(in), optional :: strong
    type(c_ptr), intent(inout) :: strm
 
    call neko_error("The shear stress bc is not applicable to scalar fields.")
 
  end subroutine shear_stress_apply_scalar_dev
 
  subroutine shear_stress_apply_vector_dev(this, x_d, y_d, z_d, time, &
       strong, strm)
    class(shear_stress_t), intent(inout), target :: this
    type(c_ptr), intent(inout) :: x_d
    type(c_ptr), intent(inout) :: y_d
    type(c_ptr), intent(inout) :: z_d
    type(time_state_t), intent(in), optional :: time
    logical, intent(in), optional :: strong
    type(c_ptr), intent(inout) :: strm
    logical :: strong_
 
    if (present(strong)) then
       strong_ = strong
    else
       strong_ = .true.
    end if
 
    if (strong_) then
       call this%symmetry%apply_vector_dev(x_d, y_d, z_d, strong = .true., &
            strm = strm)
    else
       call this%neumann_x%apply_scalar_dev(x_d, strong = .false., strm = strm)
       call this%neumann_y%apply_scalar_dev(y_d, strong = .false., strm = strm)
       call this%neumann_z%apply_scalar_dev(z_d, strong = .false., strm = strm)
    end if
 
  end subroutine shear_stress_apply_vector_dev
 
  subroutine shear_stress_init(this, coef, json)
    class(shear_stress_t), target, intent(inout) :: this
    type(coef_t), target, intent(in) :: coef
    type(json_file), intent(inout) ::json
    real(kind=rp), allocatable :: value(:)
 
    call json_get(json, 'value', value)
 
    if (size(value) .ne. 3) then
       call neko_error ("The shear stress vector provided for the shear stress &
       & boundary condition should have 3 components.")
    end if
 
    call this%init_from_components(coef, value)
  end subroutine shear_stress_init
 
  subroutine shear_stress_init_from_components(this, coef, value)
    class(shear_stress_t), target, intent(inout) :: this
    type(coef_t), intent(in) :: coef
    real(kind=rp), intent(in) :: value(3)
 
    call this%init_base(coef)
    this%strong = .false.
 
    call this%symmetry%free()
    call this%symmetry%init_from_components(this%coef)
 
    call this%neumann_x%free()
    call this%neumann_y%free()
    call this%neumann_z%free()
 
    call this%neumann_x%init_from_components(this%coef, value(1))
    call this%neumann_y%init_from_components(this%coef, value(2))
    call this%neumann_z%init_from_components(this%coef, value(3))
 
  end subroutine shear_stress_init_from_components
 
  subroutine shear_stress_finalize(this, only_facets)
    class(shear_stress_t), target, intent(inout) :: this
    logical, optional, intent(in) :: only_facets
    logical :: only_facets_
 
    if (present(only_facets)) then
       only_facets_ = only_facets
    else
       only_facets_ = .false.
    end if
 
    call this%finalize_base(only_facets_)
 
    call this%symmetry%mark_facets(this%marked_facet)
    call this%symmetry%finalize()
 
 
    call this%neumann_x%mark_facets(this%marked_facet)
    call this%neumann_y%mark_facets(this%marked_facet)
    call this%neumann_z%mark_facets(this%marked_facet)
 
    call this%neumann_x%finalize(only_facets_)
    call this%neumann_y%finalize(only_facets_)
    call this%neumann_z%finalize(only_facets_)
 
  end subroutine shear_stress_finalize
 
  subroutine shear_stress_set_stress_scalar(this, tau_x, tau_y, tau_z)
    class(shear_stress_t), intent(inout) :: this
    real(kind=rp), intent(in) :: tau_x
    real(kind=rp), intent(in) :: tau_y
    real(kind=rp), intent(in) :: tau_z
 
    ! Calls finalize and allocates the flux arrays
    call this%neumann_x%set_flux(tau_x)
    call this%neumann_y%set_flux(tau_y)
    call this%neumann_z%set_flux(tau_z)
 
 
  end subroutine shear_stress_set_stress_scalar
 
  subroutine shear_stress_set_stress_array(this, tau_x, tau_y, tau_z)
    class(shear_stress_t), intent(inout) :: this
    type(vector_t), intent(in) :: tau_x
    type(vector_t), intent(in) :: tau_y
    type(vector_t), intent(in) :: tau_z
 
    call this%neumann_x%set_flux(tau_x)
    call this%neumann_y%set_flux(tau_y)
    call this%neumann_z%set_flux(tau_z)
 
  end subroutine shear_stress_set_stress_array
 
  subroutine shear_stress_free(this)
    class(shear_stress_t), target, intent(inout) :: this
    call this%free_base
    call this%symmetry%free
 
    call this%neumann_x%free
    call this%neumann_y%free
    call this%neumann_z%free
 
  end subroutine shear_stress_free
end module shear_stress