diff options
Diffstat (limited to 'gsl-1.9/randist/bernoulli.c')
-rw-r--r-- | gsl-1.9/randist/bernoulli.c | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/gsl-1.9/randist/bernoulli.c b/gsl-1.9/randist/bernoulli.c new file mode 100644 index 0000000..3bed22b --- /dev/null +++ b/gsl-1.9/randist/bernoulli.c @@ -0,0 +1,61 @@ +/* randist/bernoulli.c + * + * Copyright (C) 1996, 1997, 1998, 1999, 2000 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 2 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> + +/* The bernoulli distribution has the form, + + prob(0) = 1-p, prob(1) = p + + */ + +unsigned int +gsl_ran_bernoulli (const gsl_rng * r, double p) +{ + double u = gsl_rng_uniform (r) ; + + if (u < p) + { + return 1 ; + } + else + { + return 0 ; + } +} + +double +gsl_ran_bernoulli_pdf (const unsigned int k, double p) +{ + if (k == 0) + { + return 1 - p ; + } + else if (k == 1) + { + return p ; + } + else + { + return 0 ; + } +} |