1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
/* randist/logarithmic.c
*
* Copyright (C) 1996, 1997, 1998, 1999, 2000, 2007 James Theiler, Brian Gough
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or (at
* your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include <config.h>
#include <math.h>
#include <gsl/gsl_rng.h>
#include <gsl/gsl_randist.h>
/* Logarithmic distribution
prob(n) = p^n / (n log(1/(1-p)) for n = 1, 2, 3, ...
We use Kemp's second accelerated generator, from Luc Devroye's book
on "Non-Uniform Random Variate Generation", Springer */
unsigned int
gsl_ran_logarithmic (const gsl_rng * r, const double p)
{
double c = log (1-p) ;
double v = gsl_rng_uniform_pos (r);
if (v >= p)
{
return 1 ;
}
else
{
double u = gsl_rng_uniform_pos (r);
double q = 1 - exp (c * u);
if (v <= q*q)
{
double x = 1 + log(v)/log(q) ;
return x ;
}
else if (v <= q)
{
return 2;
}
else
{
return 1 ;
}
}
}
double
gsl_ran_logarithmic_pdf (const unsigned int k, const double p)
{
if (k == 0)
{
return 0 ;
}
else
{
double P = pow(p, (double)k) / (double) k / log(1/(1-p)) ;
return P;
}
}
|