fortran - fortran中的函数，传入数组，接收数组

Question

我有这个功能，如下图所示。它传入两个向量，每个向量具有三个值，并且应该传递一个具有三个值的向量。我这样调用函数:

Fr       = Flux(W(:,i),W(:,i+1))

通过弄乱代码、尝试纯函数和模块以及研究错误语句(我将在底部包含)，我意识到，fortran 正在读取我的函数 Flux，并认为输入向量是尝试从数组中调用一个条目。这是我对正在发生的事情的最佳猜测。我在实验室里四处询问，大多数人建议使用子程序，但这似乎很笨拙，我想应该有更优雅的方法，但我还没有找到。我试图通过说来定义一个结果:

 DOUBLE PRECISION FUNCTION Flux(W1,W2) Result(FluxArray(3))

然后返回fluxarray，但这不起作用，因为fortran无法理解语法

实际功能是这样的:

DOUBLE PRECISION FUNCTION Flux(W1,W2)
USE parameters
IMPLICIT NONE
DOUBLE PRECISION, DIMENSION(3), INTENT(IN)::W1, W2
DOUBLE PRECISION, DIMENSION(3), INTENT(OUT):: Flux
DOUBLE PRECISION, DIMENSION(3):: F1, F2
DOUBLE PRECISION::U1,U2,Rh1,Rh2,P1,P2,E1,E2,Rh,P,u,c,Lambda
INTEGER:: k
U1=W1(2)/W1(1)
U2=W2(2)/W2(1)

Rh1=W1(1)
Rh2=W2(1)

P1=(gamma_constant-1.d0)*(W1(3)-.5d0*Rh1*U1**2)
P2=(gamma_constant-1.d0)*(W2(3)-.5d0*Rh2*U2**2)

E1=W1(3)
E2=W2(3)

F1=[Rh1*U1,Rh1*U1**2+P1,(E1+P1)*U1]
F2=[Rh2*U2,Rh2*U2**2+P2,(E2+P2)*U2]

Rh=.5d0*(Rh1+Rh2)
P=.5d0*(P1+P2)
u=.5d0*(U1+U2)
c=sqrt(gamma_constant*P/Rh)

Lambda=max(u, u+c, u-c)
do k=1,3,1
    Flux(k)=.5d0*(F1(k)+F2(k))-.5d0*eps*Lambda*(W2(k)-W1(k))
end do
RETURN
END FUNCTION Flux

这是错误声明:

Quasi1DEuler.f90:191.51:

DOUBLE PRECISION, DIMENSION(3), INTENT(OUT):: Flux
                                               1
Error: Symbol 'flux' at (1) already has basic type of REAL
Quasi1DEuler.f90:217.58:

Flux(k)=.5d0*(F1(k)+F2(k))-.5d0*eps*Lambda*(W2(k)-W1(k))
                                                      1
Error: Unexpected STATEMENT FUNCTION statement at (1)
Quasi1DEuler.f90:76.18:

Fr      = Flux(W(:,i),W(:,i+1))

最后一个错误发生在 Fr 和 Fl 上。感谢您的宝贵时间以及您可以提供的任何帮助或考虑!

编辑/跟进::
感谢您的帮助，我不知道更好的方式来呈现这个，所以我要编辑最初的问题。

我按照你的建议做了，它解决了这个问题，现在它说:

Fr      = Flux(W(:,i),W(:,i+1))
         1
Error: The reference to function 'flux' at (1) either needs an explicit INTERFACE or the rank is incorrect  

我在这个链接上看到了一个类似的问题:

Computing the cross product of two vectors in Fortran 90

他们建议他将所有功能放入模块中。有没有更好/更简单的方法来解决这个错误？

Answer 1

与 RESULT(FluxArray) , fluxArray是结果变量的名称。因此，您在结果子句中声明特征的尝试是错误的。

相反，结果变量应该在函数体中指定:

function Flux(W1,W2) result(fluxArray)
  double precision, dimension(3), intent(in)::W1, W2
  double precision, dimension(3) :: fluxArray  ! Note, no intent for result.
end function Flux

是的，可以在 function 中声明结果变量的类型。声明，但不能在那里声明数组。我不建议使用独特的 dimension结果变量的函数体中的语句。

请注意，当引用返回数组的函数时，需要有一个显式接口(interface)可供调用者使用。一种方法是将函数放在模块中，即 use。 d。有关更多详细信息，请参阅 SO 或语言教程的其他地方。

在没有 result 的情况下从您的问题中得出错误.

DOUBLE PRECISION FUNCTION Flux(W1,W2)
  DOUBLE PRECISION, DIMENSION(3), INTENT(OUT):: Flux

这里是 Flux 的类型已宣布两次。此外，它不是函数的虚拟参数，因此如上所述它不需要 intent属性。

可以写

FUNCTION Flux(W1,W2)
  DOUBLE PRECISION, DIMENSION(3) :: Flux  ! Deleting intent

或(颤抖)

DOUBLE PRECISION FUNCTION Flux(W1,W2)
  DIMENSION :: Flux(3)

在错误声明之后，关于语句功能的提示在这里并不重要。