Fortran内层子程序实现主循环回退与时间步调整方案咨询

关于你的两个问题 #

1. 直接从sub_abc减半dt并回退main_loop：无法直接实现。Fortran的子程序调用基于栈结构，没有跨子程序跳转回外层循环的原生语法，强行通过非标准手段实现会破坏代码逻辑的可维护性，还可能引发不可预测的错误。
2. 自动退出多层子程序回到sub_main：没有原生的“自动栈展开”机制，但可以通过错误标志逐层传递的方式模拟这个效果，这是Fortran处理此类问题的标准做法。

可行解决方案 #

方案1：逐层传递错误标志（推荐） #

核心思路是通过模块变量或子程序输出参数，把深层子程序的异常状态逐层传递回sub_main，再进行时间步调整。

1. 定义全局错误标志模块

   module mod_error_state
       logical :: step_failure = .false.  ! 标记当前时间步是否失败
       logical :: in_main_loop = .false.  ! 标记是否处于主时间步循环中
   end module mod_error_state
   

2. 在sub_main中维护上下文状态

   subroutine sub_main(args)
       use mod_error_state
       use mod1, only: sub11, sub12
       use mod2, only: sub21, sub22
       real :: dt, t
       ...
       main_loop: do
           step_failure = .false.
           in_main_loop = .true.  ! 进入主循环时标记上下文
           
           ! 执行各物理过程子程序
           call sub11(...)
           if (step_failure) goto handle_step_error
           
           call sub21(...)
           if (step_failure) goto handle_step_error
           
           ! 其他子程序调用...
           
           ! 无异常则推进时间
           t = t + dt
           in_main_loop = .false.
           cycle main_loop
           
       handle_step_error:
           dt = dt / 2.0
           in_main_loop = .false.
           cycle main_loop
       enddo main_loop
   end subroutine sub_main
   

3. 深层子程序中设置错误标志

   subroutine sub_abc(...)
       use mod_error_state
       ...
       ! 检测到收敛异常
       if (convergence_failed) then
           ! 仅在主循环上下文内触发时间步调整
           if (in_main_loop) step_failure = .true.
           return  ! 直接返回上一层子程序
       endif
   end subroutine sub_abc
   

4. 中间调用层检查标志并快速返回

   subroutine sub_ghi(...)
       use mod_error_state
       ...
       call sub_abc(...)
       if (step_failure) return  ! 发现异常立即返回，不执行后续逻辑
       ! 其他正常处理代码
   end subroutine sub_ghi
   

   同理，sub_def、sub21都需要添加相同的标志检查逻辑。

方案2：封装时间步控制模块 #

把时间步调整的请求逻辑封装到专用模块，让所有子程序可以直接调用请求接口，简化标志传递的代码：

module mod_time_control
       real :: dt, t
       logical :: redo_current_step = .false.
       logical :: in_time_loop = .false.
contains
       subroutine request_redo_step()
           if (in_time_loop) redo_current_step = .true.
       end subroutine request_redo_step
end module mod_time_control

在sub_abc中只需调用call request_redo_step()，sub_main的循环中检查redo_current_step即可处理时间步调整，本质是方案1的封装优化。

关键注意事项 #

• 必须严格区分sub_abc的调用上下文：通过in_main_loop或in_time_loop变量，确保只有在主时间步循环内的异常才会触发时间步调整。
• 逐层检查标志虽然繁琐，但能保证代码的可控性，避免隐式控制流跳转带来的调试困难。
• 禁止使用跨子程序的goto跳转，这不符合Fortran标准，会导致代码逻辑混乱。

内容的提问来源于stack exchange，提问作者narsonalin