mrqmin.pl
author Andreas Thurnherr <ant@ldeo.columbia.edu>
Mon, 13 Apr 2020 11:06:22 -0400
changeset 40 c1803ae2540f
parent 0 a5233793bf69
permissions -rw-r--r--
.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     1
#======================================================================
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     2
#                    M R Q M I N . P L 
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     3
#                    doc: Wed Feb 24 15:10:22 1999
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     4
#                    dlm: Tue Aug 22 22:05:43 2006
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     5
#                    (c) 1999 A.M. Thurnherr
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     6
#                    uE-Info: 15 67 NIL 0 0 72 2 2 4 NIL ofnI
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     7
#======================================================================
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     8
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
     9
# MRQMIN routine from Numerical Recipes adapted to ANTS
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    10
# NB: based on 1st edtion of NR!!!!
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    11
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    12
# HISTORY:
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    13
#	Mar 11, 1999: - created
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    14
#	Sep 27, 1999: - adapted to allow for new version of covsrt.pl as well
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    15
#	Aug 22, 2006: - changed require from covsrt_old.pl to covsrt.pl
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    16
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    17
# Notes:
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    18
#	- x,y,sig are field numbers for data in $ants_
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    19
#	- if sig is a negative number, -sig is used as constant input stddev
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    20
#	- @A, @listA, @alpha, @covar, $chiSq, &funcs, $alamda passed as refs
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    21
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    22
require "$ANTS/mrqcof.pl";
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    23
require "$ANTS/gaussj.pl";
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    24
require "$ANTS/covsrt.pl";
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    25
require "$ANTS/nrutil.pl";
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    26
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    27
{													# static scope
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    28
	my(@da,@atry,@oneda,@beta,$oChiSq);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    29
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    30
	sub mrqmin($$$$$$$$$$)
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    31
	{
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    32
		my($xfnr,$yfnr,$sig,$AR,$listAR,$covarR,
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    33
		   $alphaR,$chiSqR,$funcsR,$alamdaR) = @_;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    34
	
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    35
		my($k,$kk,$j,$ihit);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    36
	
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    37
		if ($$alamdaR < 0.0) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    38
			&matrix(\@oneda,1,$#{$AR},1,1);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    39
			&vector(\@atry,1,$#{$AR});
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    40
			&vector(\@da,1,$#{$AR});
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    41
			&vector(\@beta,1,$#{$AR});
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    42
			$kk = $#{$listAR}+1;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    43
			for ($j=1; $j<=$#{$AR}; $j++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    44
				$ihit = 0;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    45
				for ($k=1; $k<=$#{$listAR}; $k++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    46
					if ($listAR->[$k] == $j) { $ihit++; }
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    47
				}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    48
				if ($ihit == 0) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    49
					$listAR->[$kk++] = $j;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    50
				} elsif ($ihit > 1) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    51
					croak("Bad listA permutation in MRQMIN-1");
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    52
				}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    53
			}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    54
			if ($kk != $#{$AR}+1) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    55
				for ($ei=1; $ei<=$#{$listAR}; $ei++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    56
					print(STDERR "listA[$ei] = $listAR->[$ei]\n");
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    57
				}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    58
				croak("Bad listA permutation in MRQMIN-2 " .
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    59
					"($kk != $#{$AR}+1)");
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    60
			}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    61
			$$alamdaR = 0.001;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    62
			&mrqcof($xfnr,$yfnr,$sig,$AR,$listAR,$alphaR,
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    63
					\@beta,$chiSqR,$funcsR);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    64
			$oChiSq = $$chiSqR;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    65
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    66
		for ($j=1; $j<=$#{$listAR}; $j++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    67
			for ($k=1; $k<=$#{$listAR}; $k++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    68
				$covarR->[$j][$k] = $alphaR->[$j][$k];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    69
#				print(STDERR "covar[$j][$k] = $covarR->[$j][$k]\n");
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    70
			}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    71
			$covarR->[$j][$j] = $alphaR->[$j][$j]*(1.0+$$alamdaR);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    72
			$oneda[$j][1] = $beta[$j];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    73
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    74
		&gaussj($covarR,\@oneda);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    75
		for ($j=1; $j<=$#{$listAR}; $j++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    76
			$da[$j] = $oneda[$j][1];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    77
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    78
		if ($$alamdaR == 0.0) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    79
			&covsrt($covarR,$listAR);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    80
			return;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    81
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    82
		for ($j=1; $j<=$#{$AR}; $j++) { $atry[$j] = $AR->[$j]; }
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    83
		for ($j=1; $j<=$#{$listAR}; $j++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    84
			$atry[$listAR->[$j]] = $AR->[$listAR->[$j]]+$da[$j];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    85
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    86
		&mrqcof($xfnr,$yfnr,$sig,\@atry,$listAR,$covarR,\@da,$chiSqR,$funcsR);
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    87
		if ($$chiSqR < $oChiSq) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    88
			$$alamdaR *= 0.1;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    89
			$oChiSq = $$chiSqR;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    90
			for ($j=1; $j<=$#{$listAR}; $j++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    91
				for ($k=1; $k<=$#{$listAR}; $k++) {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    92
					$alphaR->[$j][$k] = $covarR->[$j][$k];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    93
				}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    94
				$beta[$j] = $da[$j];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    95
				$AR->[$listAR->[$j]] = $atry[$listAR->[$j]];
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    96
			}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    97
		} else {
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    98
			$$alamdaR *= 10.0;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
    99
			$$chiSqR = $oChiSq;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   100
		}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   101
		return;
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   102
	}
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   103
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   104
} # end of static scope
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   105
A.M. Thurnherr <athurnherr@yahoo.com>
parents:
diff changeset
   106
1;