diff options
Diffstat (limited to 'gsl-1.9/rng/rng.c')
-rw-r--r-- | gsl-1.9/rng/rng.c | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/gsl-1.9/rng/rng.c b/gsl-1.9/rng/rng.c new file mode 100644 index 0000000..eafdefb --- /dev/null +++ b/gsl-1.9/rng/rng.c @@ -0,0 +1,211 @@ +/* rng/rng.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 <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <gsl/gsl_errno.h> +#include <gsl/gsl_rng.h> + +gsl_rng * +gsl_rng_alloc (const gsl_rng_type * T) +{ + + gsl_rng *r = (gsl_rng *) malloc (sizeof (gsl_rng)); + + if (r == 0) + { + GSL_ERROR_VAL ("failed to allocate space for rng struct", + GSL_ENOMEM, 0); + }; + + r->state = malloc (T->size); + + if (r->state == 0) + { + free (r); /* exception in constructor, avoid memory leak */ + + GSL_ERROR_VAL ("failed to allocate space for rng state", + GSL_ENOMEM, 0); + }; + + r->type = T; + + gsl_rng_set (r, gsl_rng_default_seed); /* seed the generator */ + + return r; +} + +int +gsl_rng_memcpy (gsl_rng * dest, const gsl_rng * src) +{ + if (dest->type != src->type) + { + GSL_ERROR ("generators must be of the same type", GSL_EINVAL); + } + + memcpy (dest->state, src->state, src->type->size); + + return GSL_SUCCESS; +} + +gsl_rng * +gsl_rng_clone (const gsl_rng * q) +{ + gsl_rng *r = (gsl_rng *) malloc (sizeof (gsl_rng)); + + if (r == 0) + { + GSL_ERROR_VAL ("failed to allocate space for rng struct", + GSL_ENOMEM, 0); + }; + + r->state = malloc (q->type->size); + + if (r->state == 0) + { + free (r); /* exception in constructor, avoid memory leak */ + + GSL_ERROR_VAL ("failed to allocate space for rng state", + GSL_ENOMEM, 0); + }; + + r->type = q->type; + + memcpy (r->state, q->state, q->type->size); + + return r; +} + +void +gsl_rng_set (const gsl_rng * r, unsigned long int seed) +{ + (r->type->set) (r->state, seed); +} + +#ifndef HIDE_INLINE_STATIC +unsigned long int +gsl_rng_get (const gsl_rng * r) +{ + return (r->type->get) (r->state); +} + +double +gsl_rng_uniform (const gsl_rng * r) +{ + return (r->type->get_double) (r->state); +} + +double +gsl_rng_uniform_pos (const gsl_rng * r) +{ + double x ; + do + { + x = (r->type->get_double) (r->state) ; + } + while (x == 0) ; + + return x ; +} + +/* Note: to avoid integer overflow in (range+1) we work with scale = + range/n = (max-min)/n rather than scale=(max-min+1)/n, this reduces + efficiency slightly but avoids having to check for the out of range + value. Note that range is typically O(2^32) so the addition of 1 + is negligible in most usage. */ + +unsigned long int +gsl_rng_uniform_int (const gsl_rng * r, unsigned long int n) +{ + unsigned long int offset = r->type->min; + unsigned long int range = r->type->max - offset; + unsigned long int scale; + unsigned long int k; + + if (n > range || n == 0) + { + GSL_ERROR_VAL ("invalid n, either 0 or exceeds maximum value of generator", + GSL_EINVAL, 0) ; + } + + scale = range / n; + + do + { + k = (((r->type->get) (r->state)) - offset) / scale; + } + while (k >= n); + + return k; +} +#endif + +unsigned long int +gsl_rng_max (const gsl_rng * r) +{ + return r->type->max; +} + +unsigned long int +gsl_rng_min (const gsl_rng * r) +{ + return r->type->min; +} + +const char * +gsl_rng_name (const gsl_rng * r) +{ + return r->type->name; +} + +size_t +gsl_rng_size (const gsl_rng * r) +{ + return r->type->size; +} + +void * +gsl_rng_state (const gsl_rng * r) +{ + return r->state; +} + +void +gsl_rng_print_state (const gsl_rng * r) +{ + size_t i; + unsigned char *p = (unsigned char *) (r->state); + const size_t n = r->type->size; + + for (i = 0; i < n; i++) + { + /* FIXME: we're assuming that a char is 8 bits */ + printf ("%.2x", *(p + i)); + } + +} + +void +gsl_rng_free (gsl_rng * r) +{ + free (r->state); + free (r); +} |