Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Simplifying ASA digama function and adding R_INTERUPTs
- Loading branch information
Peter Glaus
committed
Oct 9, 2013
1 parent
aad5d6d
commit c277c9a
Showing
6 changed files
with
103 additions
and
119 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,96 @@ | ||
double digama ( double x, int *ifault ); | ||
# include <cmath> | ||
|
||
//****************************************************************************80 | ||
|
||
double digama ( double x, int *ifault ) | ||
|
||
//****************************************************************************80 | ||
// | ||
// Purpose: | ||
// | ||
// DIGAMA calculates DIGAMMA ( X ) = d ( LOG ( GAMMA ( X ) ) ) / dX | ||
// | ||
// Licensing: | ||
// | ||
// This code is distributed under the GNU LGPL license. | ||
// | ||
// Modified: | ||
// | ||
// 18 January 2008 | ||
// | ||
// Author: | ||
// | ||
// Original FORTRAN77 version by Jose Bernardo. | ||
// C++ version by John Burkardt. | ||
// | ||
// Reference: | ||
// | ||
// Jose Bernardo, | ||
// Algorithm AS 103: | ||
// Psi ( Digamma ) Function, | ||
// Applied Statistics, | ||
// Volume 25, Number 3, 1976, pages 315-317. | ||
// | ||
// Parameters: | ||
// | ||
// Input, double X, the argument of the digamma function. | ||
// 0 < X. | ||
// | ||
// Output, int *IFAULT, error flag. | ||
// 0, no error. | ||
// 1, X <= 0. | ||
// | ||
// Output, double DIGAMA, the value of the digamma function at X. | ||
// | ||
{ | ||
double c = 8.5; | ||
double d1 = -0.5772156649; | ||
double r; | ||
double s = 0.00001; | ||
double s3 = 0.08333333333; | ||
double s4 = 0.0083333333333; | ||
double s5 = 0.003968253968; | ||
double value; | ||
double y; | ||
// | ||
// Check the input. | ||
// | ||
if ( x <= 0.0 ) | ||
{ | ||
value = 0.0; | ||
*ifault = 1; | ||
return value; | ||
} | ||
// | ||
// Initialize. | ||
// | ||
*ifault = 0; | ||
y = x; | ||
value = 0.0; | ||
// | ||
// Use approximation if argument <= S. | ||
// | ||
if ( y <= s ) | ||
{ | ||
value = d1 - 1.0 / y; | ||
return value; | ||
} | ||
// | ||
// Reduce to DIGAMA(X + N) where (X + N) >= C. | ||
// | ||
while ( y < c ) | ||
{ | ||
value = value - 1.0 / y; | ||
y = y + 1.0; | ||
} | ||
// | ||
// Use Stirling's (actually de Moivre's) expansion if argument > C. | ||
// | ||
r = 1.0 / y; | ||
value = value + log ( y ) - 0.5 * r; | ||
r = r * r; | ||
value = value - r * ( s3 - r * ( s4 - r * s5 ) ); | ||
|
||
return value; | ||
} | ||
//****************************************************************************80 |