|
Packit |
67cb25 |
/* specfunc/beta_inc.c
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* Copyright (C) 2007 Brian Gough
|
|
Packit |
67cb25 |
* Copyright (C) 1996, 1997, 1998, 1999, 2000 Gerard Jungman
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
67cb25 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
67cb25 |
* the Free Software Foundation; either version 3 of the License, or (at
|
|
Packit |
67cb25 |
* your option) any later version.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is distributed in the hope that it will be useful, but
|
|
Packit |
67cb25 |
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
67cb25 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
67cb25 |
* General Public License for more details.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
67cb25 |
* along with this program; if not, write to the Free Software
|
|
Packit |
67cb25 |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* Author: G. Jungman */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
#include <config.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_math.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_errno.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_sf_log.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_sf_exp.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_sf_gamma.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_sf_hyperg.h>
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
#include "error.h"
|
|
Packit |
67cb25 |
#include "check.h"
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double
|
|
Packit |
67cb25 |
isnegint (const double x)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
return (x < 0) && (x == floor(x));
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static
|
|
Packit |
67cb25 |
int
|
|
Packit |
67cb25 |
beta_cont_frac(
|
|
Packit |
67cb25 |
const double a,
|
|
Packit |
67cb25 |
const double b,
|
|
Packit |
67cb25 |
const double x,
|
|
Packit |
67cb25 |
gsl_sf_result * result
|
|
Packit |
67cb25 |
)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
const unsigned int max_iter = 512; /* control iterations */
|
|
Packit |
67cb25 |
const double cutoff = 2.0 * GSL_DBL_MIN; /* control the zero cutoff */
|
|
Packit |
67cb25 |
unsigned int iter_count = 0;
|
|
Packit |
67cb25 |
double cf;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* standard initialization for continued fraction */
|
|
Packit |
67cb25 |
double num_term = 1.0;
|
|
Packit |
67cb25 |
double den_term = 1.0 - (a+b)*x/(a+1.0);
|
|
Packit |
67cb25 |
if (fabs(den_term) < cutoff) den_term = cutoff;
|
|
Packit |
67cb25 |
den_term = 1.0/den_term;
|
|
Packit |
67cb25 |
cf = den_term;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
while(iter_count < max_iter) {
|
|
Packit |
67cb25 |
const int k = iter_count + 1;
|
|
Packit |
67cb25 |
double coeff = k*(b-k)*x/(((a-1.0)+2*k)*(a+2*k));
|
|
Packit |
67cb25 |
double delta_frac;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* first step */
|
|
Packit |
67cb25 |
den_term = 1.0 + coeff*den_term;
|
|
Packit |
67cb25 |
num_term = 1.0 + coeff/num_term;
|
|
Packit |
67cb25 |
if(fabs(den_term) < cutoff) den_term = cutoff;
|
|
Packit |
67cb25 |
if(fabs(num_term) < cutoff) num_term = cutoff;
|
|
Packit |
67cb25 |
den_term = 1.0/den_term;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
delta_frac = den_term * num_term;
|
|
Packit |
67cb25 |
cf *= delta_frac;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
coeff = -(a+k)*(a+b+k)*x/((a+2*k)*(a+2*k+1.0));
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* second step */
|
|
Packit |
67cb25 |
den_term = 1.0 + coeff*den_term;
|
|
Packit |
67cb25 |
num_term = 1.0 + coeff/num_term;
|
|
Packit |
67cb25 |
if(fabs(den_term) < cutoff) den_term = cutoff;
|
|
Packit |
67cb25 |
if(fabs(num_term) < cutoff) num_term = cutoff;
|
|
Packit |
67cb25 |
den_term = 1.0/den_term;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
delta_frac = den_term*num_term;
|
|
Packit |
67cb25 |
cf *= delta_frac;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if(fabs(delta_frac-1.0) < 2.0*GSL_DBL_EPSILON) break;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
++iter_count;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
result->val = cf;
|
|
Packit |
67cb25 |
result->err = iter_count * 4.0 * GSL_DBL_EPSILON * fabs(cf);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if(iter_count >= max_iter)
|
|
Packit |
67cb25 |
GSL_ERROR ("error", GSL_EMAXITER);
|
|
Packit |
67cb25 |
else
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/*-*-*-*-*-*-*-*-*-*-*-* Functions with Error Codes *-*-*-*-*-*-*-*-*-*-*-*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
int
|
|
Packit |
67cb25 |
gsl_sf_beta_inc_e(
|
|
Packit |
67cb25 |
const double a,
|
|
Packit |
67cb25 |
const double b,
|
|
Packit |
67cb25 |
const double x,
|
|
Packit |
67cb25 |
gsl_sf_result * result
|
|
Packit |
67cb25 |
)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
if(x < 0.0 || x > 1.0) {
|
|
Packit |
67cb25 |
DOMAIN_ERROR(result);
|
|
Packit |
67cb25 |
} else if (isnegint(a) || isnegint(b)) {
|
|
Packit |
67cb25 |
DOMAIN_ERROR(result);
|
|
Packit |
67cb25 |
} else if (isnegint(a+b)) {
|
|
Packit |
67cb25 |
DOMAIN_ERROR(result);
|
|
Packit |
67cb25 |
} else if(x == 0.0) {
|
|
Packit |
67cb25 |
result->val = 0.0;
|
|
Packit |
67cb25 |
result->err = 0.0;
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else if(x == 1.0) {
|
|
Packit |
67cb25 |
result->val = 1.0;
|
|
Packit |
67cb25 |
result->err = 0.0;
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
} else if (a <= 0 || b <= 0) {
|
|
Packit |
67cb25 |
gsl_sf_result f, beta;
|
|
Packit |
67cb25 |
int stat;
|
|
Packit |
67cb25 |
const int stat_f = gsl_sf_hyperg_2F1_e(a, 1-b, a+1, x, &f);
|
|
Packit |
67cb25 |
const int stat_beta = gsl_sf_beta_e(a, b, &beta);
|
|
Packit |
67cb25 |
double prefactor = (pow(x, a) / a);
|
|
Packit |
67cb25 |
result->val = prefactor * f.val / beta.val;
|
|
Packit |
67cb25 |
result->err = fabs(prefactor) * f.err/ fabs(beta.val) + fabs(result->val/beta.val) * beta.err;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
stat = GSL_ERROR_SELECT_2(stat_f, stat_beta);
|
|
Packit |
67cb25 |
if(stat == GSL_SUCCESS) {
|
|
Packit |
67cb25 |
CHECK_UNDERFLOW(result);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
return stat;
|
|
Packit |
67cb25 |
} else {
|
|
Packit |
67cb25 |
gsl_sf_result ln_beta;
|
|
Packit |
67cb25 |
gsl_sf_result ln_x;
|
|
Packit |
67cb25 |
gsl_sf_result ln_1mx;
|
|
Packit |
67cb25 |
gsl_sf_result prefactor;
|
|
Packit |
67cb25 |
const int stat_ln_beta = gsl_sf_lnbeta_e(a, b, &ln_beta);
|
|
Packit |
67cb25 |
const int stat_ln_1mx = gsl_sf_log_1plusx_e(-x, &ln_1mx);
|
|
Packit |
67cb25 |
const int stat_ln_x = gsl_sf_log_e(x, &ln_x);
|
|
Packit |
67cb25 |
const int stat_ln = GSL_ERROR_SELECT_3(stat_ln_beta, stat_ln_1mx, stat_ln_x);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
const double ln_pre_val = -ln_beta.val + a * ln_x.val + b * ln_1mx.val;
|
|
Packit |
67cb25 |
const double ln_pre_err = ln_beta.err + fabs(a*ln_x.err) + fabs(b*ln_1mx.err);
|
|
Packit |
67cb25 |
const int stat_exp = gsl_sf_exp_err_e(ln_pre_val, ln_pre_err, &prefactor);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if(stat_ln != GSL_SUCCESS) {
|
|
Packit |
67cb25 |
result->val = 0.0;
|
|
Packit |
67cb25 |
result->err = 0.0;
|
|
Packit |
67cb25 |
GSL_ERROR ("error", GSL_ESANITY);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if(x < (a + 1.0)/(a+b+2.0)) {
|
|
Packit |
67cb25 |
/* Apply continued fraction directly. */
|
|
Packit |
67cb25 |
gsl_sf_result cf;
|
|
Packit |
67cb25 |
const int stat_cf = beta_cont_frac(a, b, x, &cf);
|
|
Packit |
67cb25 |
int stat;
|
|
Packit |
67cb25 |
result->val = prefactor.val * cf.val / a;
|
|
Packit |
67cb25 |
result->err = (fabs(prefactor.err * cf.val) + fabs(prefactor.val * cf.err))/a;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
stat = GSL_ERROR_SELECT_2(stat_exp, stat_cf);
|
|
Packit |
67cb25 |
if(stat == GSL_SUCCESS) {
|
|
Packit |
67cb25 |
CHECK_UNDERFLOW(result);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
return stat;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else {
|
|
Packit |
67cb25 |
/* Apply continued fraction after hypergeometric transformation. */
|
|
Packit |
67cb25 |
gsl_sf_result cf;
|
|
Packit |
67cb25 |
const int stat_cf = beta_cont_frac(b, a, 1.0-x, &cf);
|
|
Packit |
67cb25 |
int stat;
|
|
Packit |
67cb25 |
const double term = prefactor.val * cf.val / b;
|
|
Packit |
67cb25 |
result->val = 1.0 - term;
|
|
Packit |
67cb25 |
result->err = fabs(prefactor.err * cf.val)/b;
|
|
Packit |
67cb25 |
result->err += fabs(prefactor.val * cf.err)/b;
|
|
Packit |
67cb25 |
result->err += 2.0 * GSL_DBL_EPSILON * (1.0 + fabs(term));
|
|
Packit |
67cb25 |
/* since the prefactor term is subtracted from 1 we need to
|
|
Packit |
67cb25 |
ignore underflow */
|
|
Packit |
67cb25 |
if (stat_exp != GSL_EUNDRFLW) {
|
|
Packit |
67cb25 |
stat = GSL_ERROR_SELECT_2(stat_exp, stat_cf);
|
|
Packit |
67cb25 |
} else {
|
|
Packit |
67cb25 |
stat = stat_cf;
|
|
Packit |
67cb25 |
};
|
|
Packit |
67cb25 |
if(stat == GSL_SUCCESS) {
|
|
Packit |
67cb25 |
CHECK_UNDERFLOW(result);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
return stat;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/*-*-*-*-*-*-*-*-*-* Functions w/ Natural Prototypes *-*-*-*-*-*-*-*-*-*-*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
#include "eval.h"
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
double gsl_sf_beta_inc(const double a, const double b, const double x)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
EVAL_RESULT(gsl_sf_beta_inc_e(a, b, x, &result));
|
|
Packit |
67cb25 |
}
|