调用功能

时间:2012-02-17 20:47:42

标签: fortran computation

我不知道这个问题是否正确。如果不是,我很抱歉。 我是fortran的新用户,并花了很多时间来处理以下事情。

我构建了一个名为“loglike”的函数,它根据两个参数返回一个实数。我想用这个函数来构造一个mcmc算法

就是这样。

psi = min(0, loglike(propalpha,propbeta) - loglike(currentalpha,currentbeta))

其中propalpha = currentalpha + noisepropbeta = currentbeta + noise,噪音是来自某些发行版的随机样本。

现在我想通过调用先前构造的函数“loglike”来使用这个算法。

1)如何为称为主程序的新程序调用函数'loglike' 2)如何将其用于子程序?

任何帮助对我来说都非常好。 提前致谢

编辑:

module mcmc

implicit none

contains

    subroutine subru(A,B, alphaprop, betaprop)
        real, intent(in)::A,B
        real, intent(out)::alphaprop, betaprop
    end subroutine subru

    real function aloglike(A,B)
        real:: A,B,U, aloglike
        aloglike = U
    end function aloglike

end module mcmc


program  likelihood
    use mcmc

    implicit none

    real :: alpha,beta,dist1,dist2,prob1,prob2
    real:: psi,a(1000),b(1000), u1, u2,u, alphaprop, betaprop
    real, dimension(1:1):: loglike
    integer :: t,i,j,k,l
    real, dimension(1:625):: x
    real, dimension(1:625):: y
    integer, dimension(1:625):: inftime

    alpha = 0.5
    beta = 2.0

    open(10, file = 'epidemic.txt', form = 'formatted')
    do l = 1,625
       read(10,*,end = 200) x(l), y(l), inftime(l)
    enddo
200 continue

    loglike = 0.0
    do t=1,10
        do i=1,625
            if(inftime(i)==0 .or. t < (inftime(i)-1)) then
                dist1 = 0.0
                do  j = 1, 625
                    if(t >= inftime(j) .and. inftime(j)/=0)then
                        dist1 = dist1 + sqrt((x(i) - x(j))**2 + (y(i) - y(j))**2)**(-beta)
                    endif
                enddo
                prob1 = 1 - exp(-alpha * dist1)
                loglike = loglike + log(1 - prob1)
            endif

            if(inftime(i) .eq. (t+1)) then
                dist2 = 0.0
                    do k=1, 625
                    if(t >= inftime(k) .and. inftime(k)/=0) then
                        dist2 = dist2 + sqrt((x(i) - x(k))**2 + (y(i) - y(k))**2)**(-beta)
                    endif
                enddo
                prob2 = 1 - exp(-alpha * dist2)
                loglike = loglike + log(prob2)
            endif
        enddo
    enddo

    do i = 2, 1000
        a(1)= 0.0
        b(1) = 0.0
        call subru(a(i),b(i), alphaprop, betaprop)
        call random_number(u1)
        call random_number(u2)

        alphaprop = a(i-1) + (u1*0.4)-0.2
        betaprop= b(i-1) + (u2*0.4)-0.2

        if(alphaprop> 0 .and. alphaprop < 0.2 .and. betaprop > 0 .and. betaprop < 0.2)then
            psi = min(0.0,aloglike(alphaprop,betaprop)- aloglike(a(i-1),b(i-1)))
            call random_number(u)

            if(u < psi)then
                a(i)= alphaprop
                b(i) = betaprop
            else
                a(i) = a(i-1)
                b(i) = b(i-1)
            endif
        endif
    enddo

    do j = 1, 1000
        print *, A(j), A(j), LOGLIKE
    enddo

end program

2 个答案:

答案 0 :(得分:4)

最简单,最可靠的技术是将您的函数和子程序放在一个模块中,并从主程序中“使用”该模块。这可以在一个文件中完成。此方法使得过程(函数和子例程)的接口已知,以便编译器可以检查调用(实际参数)和调用(伪参数)中的参数之间的一致性。草图:

module mysubs

implicit none

contains

subroutine sub1 (xyz)
   declarations
   code
end subroutine sub1

function func2 (u)
   declarations
   code
   func2 = ...
end func2

end module mysubs

program myprog

use mysubs

implicit none

declarations...

call sub1 (xyz)

q = func2 (z)

end program myprog

ADDED:“implicit none”用于禁用隐式输入,这在我看来很危险。因此,您需要键入所有变量,包括函数中的函数名称。您可以从模块的其他过程调用子例程和函数 - 它们将自动被知道。因此,如果您愿意,可以使用“sub1”中的“func2”。对于模块外部的实体,例如主程序,您必须“使用”模块。

答案 1 :(得分:2)

这是它看起来的一般方式。请注意,函数通过将结果分配给自己的名称来返回值。

return