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

Question

我正在使用Fortran创建一个Python模块f2py.如果在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程序能够抓住错误.请帮忙.

回答

stopFortran中的命令就是这样做的.考虑更新的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

f2py确实提供了一些可用于引发异常的语句。请参阅此处的详细信息。

特别是看看callstatementwhich描述了如何添加f2py_success = 0which会触发异常。

我不确定这是否会帮助您调试 fortran 库的内部结构，但至少这是一个开始。