time_state.f90

Go to the documentation of this file.

! Copyright (c) 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 time_state
  use num_types, only : rp
  use logger, only : neko_log, log_size, neko_log_quiet
  use checkpoint, only : chkp_t
  use json_module, only : json_file
  use json_utils, only : json_get_or_lookup, json_get_or_default, &
       json_get_or_lookup_or_default
  implicit none
  private
 
  type, public :: time_state_t
     real(kind=rp), dimension(10) :: tlag = 0.0_rp 
     real(kind=rp), dimension(10) :: dtlag = 0.0_rp 
     real(kind=rp) :: t = 0.0_rp 
     real(kind=rp) :: dt = 0.0_rp 
     real(kind=rp) :: start_time = 0.0_rp 
     real(kind=rp) :: end_time = 0.0_rp 
     integer :: tstep = 0
 
   contains
     generic :: init => init_from_components, init_from_json
     procedure, pass(this) :: init_from_components => &
          time_state_init_from_components
     procedure, pass(this) :: init_from_json => time_state_init_from_json
     procedure, pass(this) :: reset => time_state_reset
     procedure, pass(this) :: restart => time_state_restart
     procedure, pass(this) :: status => time_state_status
     procedure, pass(this) :: is_done => time_state_is_done
  end type time_state_t
 
contains
 
  subroutine time_state_init_from_json(this, params)
    class(time_state_t), intent(inout) :: this
    type(json_file), intent(inout) :: params
 
    real(kind=rp) :: time_step
    real(kind=rp) :: start_time
    real(kind=rp) :: end_time
    logical :: is_variable
 
    call json_get_or_lookup_or_default(params, 'start_time', start_time, 0.0_rp)
    call json_get_or_lookup(params, 'end_time', end_time)
    call json_get_or_default(params, 'variable_timestep', is_variable, .false.)
    if (.not. is_variable) then
       call json_get_or_lookup(params, 'timestep', time_step)
    else
       ! randomly set an initial dt to get cfl when dt is variable
       time_step = 1.0_rp
    end if
 
    call this%init_from_components(start_time, end_time, time_step)
 
  end subroutine time_state_init_from_json
 
  subroutine time_state_init_from_components(this, start_time, end_time, dt)
    class(time_state_t), intent(inout) :: this
    real(kind=rp), intent(in) :: start_time
    real(kind=rp), intent(in) :: end_time
    real(kind=rp), intent(in) :: dt
 
    if (dt .gt. 0.0_rp .and. start_time .gt. end_time .or. &
         dt .lt. 0.0_rp .and. start_time .lt. end_time) then
       call neko_log%error('Time step size must match direction of time.')
    end if
 
    this%start_time = start_time
    this%end_time = end_time
    this%dt = dt
 
    this%t = start_time
    this%tstep = 0
    this%tlag = start_time
    this%dtlag = dt
 
  end subroutine time_state_init_from_components
 
  subroutine time_state_reset(this)
    class(time_state_t), intent(inout) :: this
 
    this%t = this%start_time
    this%tstep = 0
    this%tlag = this%start_time
    this%dtlag = this%dt
 
  end subroutine time_state_reset
 
  subroutine time_state_restart(this, chkp)
    class(time_state_t), intent(inout) :: this
    type(chkp_t), intent(in) :: chkp
 
    this%t = chkp%t
    this%dtlag = chkp%dtlag
    this%tlag = chkp%tlag
  end subroutine time_state_restart
 
  subroutine time_state_status(this)
    class(time_state_t), intent(in) :: this
    character(len=LOG_SIZE) :: log_buf
    character(len=38) :: log_fmt
    real(kind=rp) :: t_prog
 
    t_prog = 100.0_rp * (this%t - this%start_time) / &
         (this%end_time - this%start_time)
 
    write(log_fmt, '(A,I2,A)') &
         '(A7,1X,I10,1X,A4,E15.7,', log_size - 50, 'X,A2,F6.2,A3)'
    write(log_buf, log_fmt) 'Step = ', this%tstep, 't = ', this%t, &
         '[ ', t_prog, '% ]'
 
    call neko_log%message(repeat('-', log_size - 1))
    call neko_log%message(log_buf, neko_log_quiet)
    call neko_log%message(repeat('-', log_size - 1))
 
  end subroutine time_state_status
 
  pure function time_state_is_done(this) result(is_done)
    class(time_state_t), intent(in) :: this
    logical :: is_done
 
    is_done = this%t - this%start_time .ge. this%end_time - this%start_time &
         .and. this%tstep .gt. 0
 
  end function time_state_is_done
 
end module time_state