initial_conditions.f90

module user_initial_conditions
  use field, only: field_t
  use json_file_module, only: json_file
  use json_utils, only: json_get_or_default
  use device, only: host_to_device
  use num_types, only: rp
  implicit none
 
contains
 
  subroutine scalar_z_split_ic(s, split_value, value_below, value_above)
    type(field_t), intent(inout) :: s
    real(kind=rp), intent(in) :: split_value
    real(kind=rp), intent(in) :: value_below, value_above
 
    real(kind=rp) :: z_value
    integer :: i
 
    do i = 1, s%dof%size()
       z_value = s%dof%z(i, 1, 1, 1)
 
       if (z_value .gt. split_value) then
          s%x(i, 1, 1, 1) = value_above
       else
          s%x(i, 1, 1, 1) = value_below
       end if
 
    end do
 
    call s%copy_from(host_to_device, .true.)
 
  end subroutine scalar_z_split_ic
 
end module user_initial_conditions