view liboctave/cruft/ranlib/tstgmn.for @ 18794:03c2671493f9

maint: Periodic merge of gui-release to default.
author Rik <rik@octave.org>
date Mon, 02 Jun 2014 11:17:59 -0700
parents 648dabbb4c6b
children 446c46af4b42
line wrap: on
line source

C     JJV changed name to ONECOV to avoid confusion with array COVAR
C     JJV this was also changed in the body of the function
C      REAL FUNCTION covar(x,y,n)
      REAL FUNCTION onecov(x,y,n)
C     .. Scalar Arguments ..
      INTEGER n
C     ..
C     .. Array Arguments ..
      REAL x(n),y(n)
C     ..
C     .. Local Scalars ..
      REAL avx,avy,varx,vary,xmax,xmin
      INTEGER i
C     ..
C     .. External Subroutines ..
      EXTERNAL stat
C     ..
C     .. Intrinsic Functions ..
      INTRINSIC real
C     ..
C     .. Executable Statements ..
      CALL stat(x,n,avx,varx,xmin,xmax)
      CALL stat(y,n,avy,vary,xmin,xmax)
C      covar = 0.0
      onecov = 0.0
      DO 10,i = 1,n
C      covar = covar + (x(i)-avx)* (y(i)-avy)
         onecov = onecov + (x(i)-avx)* (y(i)-avy)
 10   CONTINUE
C      covar = covar/real(n-1)
      onecov = onecov/real(n-1)
      RETURN
      
      END

C     JJV Added argument LDXCOV (leading dimension of XCOVAR) to be
C     JJV consistent with the program TSTGMN, see comments below.
C     JJV This change necessitated changes in the declarations.
C      SUBROUTINE prcomp(p,mean,xcovar,answer)
      SUBROUTINE prcomp(p,mean,xcovar,ldxcov,answer)

C      INTEGER p,maxp
      INTEGER p,maxp,ldxcov
      PARAMETER (maxp=10)
C      REAL mean(p),xcovar(p,p),rcovar(maxp,maxp)
      REAL mean(p),xcovar(ldxcov,p),rcovar(maxp,maxp)
      REAL answer(1000,maxp)
C     JJV added ONECOV because of name change to function COVAR
C      REAL rmean(maxp),rvar(maxp)
      REAL rmean(maxp),rvar(maxp),onecov
      INTEGER maxobs
      PARAMETER (maxobs=1000)

      DO 10,i = 1,p
          CALL stat(answer(1,i),maxobs,rmean(i),rvar(i),dum1,dum2)
          WRITE (*,*) ' Variable Number',i
          WRITE (*,*) ' Mean ',mean(i),' Generated ',rmean(i)
          WRITE (*,*) ' Variance ',xcovar(i,i),' Generated',rvar(i)
   10 CONTINUE
      WRITE (*,*) '                   Covariances'
      DO 30,i = 1,p
          DO 20,j = 1,i - 1
              WRITE (*,*) ' I = ',i,' J = ',j
C     JJV changed COVAR to match new name
C              rcovar(i,j) = covar(answer(1,i),answer(1,j),maxobs)
              rcovar(i,j) = onecov(answer(1,i),answer(1,j),maxobs)
              WRITE (*,*) ' Covariance ',xcovar(i,j),' Generated ',
     +          rcovar(i,j)
   20     CONTINUE
   30 CONTINUE
      RETURN

      END

C     JJV added LDCOV (leading dimension of COVAR) to be
C     JJV consistent with the program TSTGMN, see comments below.
C     JJV This change necessitated changes in the declarations.
C      SUBROUTINE setcov(p,var,corr,covar)
      SUBROUTINE setcov(p,var,corr,covar,ldcov)
C     Set covariance matrix from variance and common correlation
C     .. Scalar Arguments ..
      REAL corr
C      INTEGER p
      INTEGER p,ldcov
C     ..
C     .. Array Arguments ..
C      REAL covar(p,p),var(p)
      REAL covar(ldcov,p),var(p)
C     ..
C     .. Local Scalars ..
      INTEGER i,j
C     ..
C     .. Intrinsic Functions ..
      INTRINSIC sqrt
C     ..
C     .. Executable Statements ..
      DO 40,i = 1,p
          DO 30,j = 1,p
              IF (.NOT. (i.EQ.j)) GO TO 10
              covar(i,j) = var(i)
              GO TO 20

   10         covar(i,j) = corr*sqrt(var(i)*var(j))
   20         CONTINUE
   30     CONTINUE
   40 CONTINUE
      RETURN

      END

      SUBROUTINE stat(x,n,av,var,xmin,xmax)
C     .. Scalar Arguments ..
      REAL av,var,xmax,xmin
      INTEGER n
C     ..
C     .. Array Arguments ..
      REAL x(n)
C     ..
C     .. Local Scalars ..
      REAL sum
      INTEGER i
C     ..
C     .. Intrinsic Functions ..
      INTRINSIC real
C     ..
C     .. Executable Statements ..
      xmin = x(1)
      xmax = x(1)
      sum = 0.0
      DO 10,i = 1,n
          sum = sum + x(i)
          IF (x(i).LT.xmin) xmin = x(i)
          IF (x(i).GT.xmax) xmax = x(i)
   10 CONTINUE
      av = sum/real(n)
      sum = 0.0
      DO 20,i = 1,n
          sum = sum + (x(i)-av)**2
   20 CONTINUE
      var = sum/real(n-1)
      RETURN

      END

      PROGRAM tstgmn
C     Test Generation of Multivariate Normal Data
C     JJV SETGMN was: SUBROUTINE setgmn(meanv,covm,p,parm)
C     JJV         is: SUBROUTINE setgmn(meanv,covm,ldcovm,p,parm)
C     JJV So the covariance matrices have been changed to 2-dim'l
C     JJV matrices, and the additional argument has been added to
C     JJV the subroutine call.  Additional changes have been made
C     JJV to reflect this.  (in declarations, the matrix copy routine,
C     JJV and in subroutine calls.)
C     .. Parameters ..
      INTEGER maxp
      PARAMETER (maxp=10)
      INTEGER maxobs
      PARAMETER (maxobs=1000)
C     JJV this parameter is no longer needed
C      INTEGER p2
C      PARAMETER (p2=maxp*maxp)
C     ..
C     .. Local Scalars ..
      REAL corr
      INTEGER i,iobs,is1,is2,j,p
      CHARACTER phrase*100
C     ..
C     .. Local Arrays ..
C      REAL answer(1000,maxp),ccovar(p2),covar(p2),mean(maxp),param(500),
C     +     temp(maxp),var(maxp),work(maxp)
      REAL answer(1000,maxp),ccovar(maxp,maxp),covar(maxp,maxp),
     +     mean(maxp),param(500),temp(maxp),var(maxp),work(maxp)
C     ..
C     .. External Subroutines ..
      EXTERNAL genmn,phrtsd,prcomp,setall,setcov,setgmn
C     ..
C     .. Executable Statements ..
      WRITE (*,9000)

 9000 FORMAT (
     +     ' Tests Multivariate Normal Generator for Up to 10 Variables'
     +       /
     +  ' User inputs means, variances, one correlation that is applied'
     +       /'     to all pairs of variables'/
     +       ' 1000 multivariate normal deviates are generated'/
     +     ' Means, variances and covariances are calculated for these.'
     +       )

   10 WRITE (*,*) 'Enter number of variables for normal generator'
      READ (*,*) p
      WRITE (*,*) 'Enter mean vector of length ',p
      READ (*,*) (mean(i),i=1,p)
      WRITE (*,*) 'Enter variance vector of length ',p
      READ (*,*) (var(i),i=1,p)
      WRITE (*,*) 'Enter correlation of all variables'
      READ (*,*) corr
C      CALL setcov(p,var,corr,covar)
      CALL setcov(p,var,corr,covar,maxp)
      WRITE (*,*) ' Enter phrase to initialize rn generator'
      READ (*,'(a)') phrase
      CALL phrtsd(phrase,is1,is2)
      CALL setall(is1,is2)
C      DO 20,i = 1,p2
C          ccovar(i) = covar(i)
C 20   CONTINUE
      DO 25,i = 1,maxp
         DO 20,j = 1,maxp
            ccovar(i,j) = covar(i,j)
 20      CONTINUE
 25   CONTINUE
C
C     Generate Variables
C
C      CALL setgmn(mean,ccovar,p,param)
      CALL setgmn(mean,ccovar,maxp,p,param)
      DO 40,iobs = 1,maxobs
          CALL genmn(param,work,temp)
          DO 30,j = 1,p
              answer(iobs,j) = work(j)
   30     CONTINUE
   40 CONTINUE
C      CALL prcomp(p,mean,covar,answer)
      CALL prcomp(p,mean,covar,maxp,answer)
C
C     Print Comparison of Generated and Reconstructed Values
C
      GO TO 10

      END