python - 連立方程式の SymPy codegen を使用して Fortran サブルーチンを生成する

Question

ここで見つけた以前の例に基づいて、私が固執する必要がある特定の形式に対応する Fortran コードを生成する方法を見つけようとします。必要な FORTRAN コードは次のようになります (これは FitzHugh–Nagumo モデルに基づいています)。

  SUBROUTINE FF(NE,U,PAR,F) 
!     ---------- -- 
!     Define the nonlinear term

  IMPLICIT NONE
  INTEGER, INTENT(IN) :: NE
  DOUBLE PRECISION, INTENT(IN) :: U(NE),PAR(*)
  DOUBLE PRECISION, INTENT(OUT) :: F(NE)

  DOUBLE PRECISION u,v,e,a1,a0

    u=U(1)
    v=U(2)
    e=PAR(1)
    a1=PAR(2)
    a0=PAR(3)

    F(1)= u-u**3-v 
    F(2)= e*(u-a1*v-a0)

  END SUBROUTINE FF

SymPy で適切な式を作成できましたが、必要なコードを で生成する方法がわかりませんでしたcodegen。これまでの私の試みは次のとおりです。

from sympy import symbols,latex
from sympy.utilities.codegen import codegen
from sympy.tensor import IndexedBase, Idx
from sympy import Matrix
U, PAR = symbols('U PAR', cls=IndexedBase)

u = U[1]
v = U[2]

e = PAR[1]
a1 = PAR[2]
a0 = PAR[3]

dudt = u-u**3-v 
dvdt = e*(u-a1*v-a0)

print latex(dudt)
print latex(dvdt)

F = Matrix([dudt,dvdt])
print F

result = codegen(('my_function', F), 'f95', 'my_project')
print result[0][1]

しかし、それは私に与えます：

IndexException: 
Range is not defined for all indices in: PAR[3]