function可以返回一个数组吗

vvt

[Fortran] 纯文本查看 复制代码

  R0 = position0()
  V = Velocity()

 Function position0() result(R0)
    implicit none
    real*8::R0(N,3),x,y
    integer ::i
    do i=1,N
      do
        x=ran()
        y=ran()
        if((2*x-1)**2+(2*y-1)**2<=1) exit
      end do
      R0(i,1)=wx*(2.0*x-1.0)
      R0(i,2)=wy*(2.0*y-1.0)
    end do
    do i=1,N
      R0(i,3)=normal(0.D0,wz/2.0)
    end do
  end Function position0

  Function Velocity() result(V)
    implicit none
    real*8::V(N,3),r1,r2,r3
    integer::i
    do i=1,N
      r1=ran()*2.*pi
      r2=ran()*2.-1.0
      r3=normal(0.d0,dsqrt(1.3806504E4*T/(87*1.65053878)))
      V(i,1)=r3*sqrt(1-r2*r2)*cos(r1)
      V(i,2)=r3*sqrt(1-r2*r2)*sin(r1)
      V(i,3)=r3*r2
    end do
  end Function Velocity