python - 使用 f2py 在 (Fortran) 模块错误中停止 python 代码？

Question

我正在使用 f2py 在 Fortran 中创建一个 Python 模块。如果在 Fortran 模块中遇到错误，我想在 Python 程序中产生错误(包括错误消息)。考虑以下示例:

Fortran 代码(test.f):

subroutine foo(a,m)

  integer :: m,i
  integer, dimension(m) :: a
  !f2py intent(in) :: m
  !f2py intent(in,out) :: a
  !f2py intent(hide), depend(a) :: m=shape(a)

  do i = 1,m
    if ( a(i) .eq. 0 ) then
      print*, 'ERROR HERE..?'
    end if 
    a(i) = a(i)+1
  end do

end subroutine

这个非常简单的程序将 1 添加到 a 的每个元素。但如果 a(i) 等于零，则应该产生错误。附带的 Python 代码:

import test

print test.foo(np.array([1,2],dtype='uint32'))
print test.foo(np.array([0,2],dtype='uint32'))

现在的输出是:

[2 3]
ERROR HERE..?
[1 3]

但是我希望 Python 程序保留错误。请帮忙。

回答

Fortran 中的stop 命令就是这样做的。考虑更新后的 Fortran 代码:

subroutine foo(a,m)

  integer :: m,i
  integer, dimension(m) :: a
  !f2py intent(in) :: m
  !f2py intent(in,out) :: a
  !f2py intent(hide), depend(a) :: m=shape(a)

  do i = 1,m
    if ( a(i) .eq. 0 ) then
      print*, 'Error from Fortran'
      stop
    end if 
    a(i) = a(i)+1
  end do

end subroutine

现在的输出是:

[2 3]
Error from Fortran

即错误后 Python 代码不会继续。

Answer 1

我已经向 numpy 社区建议添加一个额外的 f2py“增强”(raise_python_exception)，这使得在 Fortran 中定义一个字符串变量成为可能，如果一旦函数返回，非空将导致 Python 引发异常。

所以在 Fortran 中你会这样写:

subroutine calc_dq(q, temp, dq, error_mesg)
  !f2py raise_python_exception error_mesg

  real, intent(in) :: q, temp
  real, intent(out) :: dq

  character(len=100), intent(out) :: error_mesg

  if (.not. init_called()) then
     error_mesg = "`init` hasn't been called."
  else
     call q_flux_function(q, temp, dq)
  endif
end subroutine calc_dq

并从 Python 中调用 error_mesg 变量的内容作为异常的内容:

In [2]: calc_dq(1.0, 300.)
---------------------------------------------------------------------------
Exception                                 Traceback (most recent call last)
<ipython-input-8-c0ce0cb9cda1> in <module>()
----> 1 calc_dq(1.0, 300.)

Exception: `init` hasn't been called.

我认为这是从 Fortran 引发异常的一种非常方便的方法，因为它允许轻松定义异常消息。我把我的suggestion on github .