rng.c

来自「General Hidden Markov Model Library 一个通用」· C语言 代码 · 共 129 行

/*******************************************************************************
*
*       This file is part of the General Hidden Markov Model Library,
*       GHMM version 0.8_beta1, see http://ghmm.org
*
*       Filename: ghmm/ghmm/rng.c
*       Authors:  Alexander Schliep
*
*       Copyright (C) 1998-2004 Alexander Schliep
*       Copyright (C) 1998-2001 ZAIK/ZPR, Universitaet zu Koeln
*       Copyright (C) 2002-2004 Max-Planck-Institut fuer Molekulare Genetik,
*                               Berlin
*
*       Contact: [e-mail]
*
*       This library is free software; you can redistribute it and/or
*       modify it under the terms of the GNU Library General Public
*       License as published by the Free Software Foundation; either
*       version 2 of the License, or (at your option) any later version.
*
*       This library 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
*       Library General Public License for more details.
*
*       You should have received a copy of the GNU Library General Public
*       License along with this library; if not, write to the Free
*       Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*
*       This file is version $Revision: 1639 $
*                       from $Date: 2006-07-05 23:18:57 +0200 (Wed, 05 Jul 2006) $
*             last change by $Author: grunau $.
*
*******************************************************************************/

#ifdef WIN32
#  include "win_config.h"
#endif

#ifdef HAVE_CONFIG_H
#  include "../config.h"
#endif
#include <stdlib.h>
#include <stdio.h>
#include "rng.h"
#include "math.h"
#include "time.h"

/* The global RNG */
GHMM_RNG *RNG;

/* ----- Mersenne Twister -------------------------------------------------- */         
#ifdef GHMM_RNG_MERSENNE_TWISTER
#include "mt19937ar.c"

static ghmm_rng_state_t ighmm_rng_state;
static char ighmm_rng_name[] = "Mersenne Twister";

void ghmm_rng_set(GHMM_RNG * r, unsigned long int seed)
{ init_genrand(seed); }

double ghmm_rng_uniform(GHMM_RNG * r)
{ return genrand_real2(); }

const char *ghmm_rng_name(GHMM_RNG * r)
{ return ighmm_rng_name; }

void ghmm_rng_init(void)
{
  initstate(1, ighmm_rng_state, sizeof(ghmm_rng_state_t));
  RNG = &ighmm_rng_state;
}
#endif /* GHMM_RNG_MERSENNE_TWISTER */


/* ----- BSD --------------------------------------------------------------- */
#ifdef GHMM_RNG_BSD

static ghmm_rng_state_t ighmm_rng_state;
static char ighmm_rng_name[] = "random";

void ghmm_rng_set(GHMM_RNG * r, unsigned long int seed)
{
  srandom(seed);
}

double ghmm_rng_uniform(GHMM_RNG * r)
{
  return ((double)random()) / (RAND_MAX + 1.0);
}

const char *ghmm_rng_name(GHMM_RNG * r)
{
  return ighmm_rng_name;
}

void ghmm_rng_init(void)
{
  initstate(1, ighmm_rng_state, sizeof(ghmm_rng_state_t));
  RNG = &ighmm_rng_state;
}
#endif /* "GHMM_RNG_BSD */


/* ----- GSL --------------------------------------------------------------- */
#ifdef GHMM_RNG_GSL

void ghmm_rng_init(void)
{
  gsl_rng_env_setup();
  RNG = gsl_rng_alloc(gsl_rng_default);
}
#endif /* GHMM_RNG_GSL */


void ghmm_rng_timeseed(GHMM_RNG * r)
{
  unsigned long tm;             /* Time seed */
  unsigned int timeseed;

  timeseed = time(NULL);
  srand(timeseed);
  tm = rand();
  GHMM_RNG_SET(r, tm);
  /*printf("# using rng '%s' seed=%ld\n", GHMM_RNG_NAME(r), tm);  */
  fflush(stdout);
}