diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2015-07-30 16:07:50 +0200 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2015-07-30 16:07:50 +0200 |
| commit | dec32e1516a05db805ff75a9e016e8bea04d1d8e (patch) | |
| tree | 1087efc189d5ca533ac01fec6746347212d66202 /taus.c | |
| parent | added readme (diff) | |
| download | SyncRNG-dec32e1516a05db805ff75a9e016e8bea04d1d8e.tar.gz SyncRNG-dec32e1516a05db805ff75a9e016e8bea04d1d8e.zip | |
name change to SyncRNG and documentation in the .c file
Diffstat (limited to 'taus.c')
| -rw-r--r-- | taus.c | 189 |
1 files changed, 0 insertions, 189 deletions
@@ -1,189 +0,0 @@ -#ifdef TARGETPYTHON -#include "Python.h" -#endif - -#ifndef TARGETPYTHON -#define STRICT_R_HEADERS -#include <stdio.h> -#include <stdlib.h> -#include <R.h> -#include <Rinternals.h> -#endif - -int lfsr113(int **state) -{ - unsigned long z1, z2, z3, z4, b; - - z1 = (*state)[0]; - z2 = (*state)[1]; - z3 = (*state)[2]; - z4 = (*state)[3]; - - b = (((z1 << 6) ^ z1) >> 13); - z1 = (((z1 & 4294967294) << 18) ^ b); - - b = (((z2 << 2) ^ z2) >> 27); - z2 = (((z2 & 4294967288) << 2) ^ b); - - b = (((z3 << 13) ^ z3) >> 21); - z3 = (((z3 & 4294967280) << 7) ^ b); - - b = (((z4 << 3) ^ z4) >> 12); - z4 = (((z4 & 4294967168) << 13) ^ b); - - b = (z1 ^ z2 ^ z3 ^ z4); - - (*state)[0] = z1 & 0x7FFFFFFF; - (*state)[1] = z2 & 0x7FFFFFFF; - (*state)[2] = z3 & 0x7FFFFFFF; - (*state)[3] = z4 & 0x7FFFFFFF; - b = b & 0x7FFFFFFF; - - return(b); -} - -void lfsr113_seed(unsigned long seed, int **state) -{ - unsigned long z1 = 2, - z2 = 8, - z3 = 16, - z4 = 128; - - z1 = (z1 * (seed + 1)) & 0x7FFFFFFF; - z2 = (z2 * (seed + 1)) & 0x7FFFFFFF; - z3 = (z3 * (seed + 1)) & 0x7FFFFFFF; - z4 = (z4 * (seed + 1)) & 0x7FFFFFFF; - - if (*state == NULL) { - (*state) = malloc(sizeof(int)*4); - } - - (*state)[0] = (int) z1; - (*state)[1] = (int) z2; - (*state)[2] = (int) z3; - (*state)[3] = (int) z4; -} - -#ifdef TARGETPYTHON -/* - * - * Start of Python code - * - */ - -static char module_docstring[] = -"This module provides the Tausworthe RNG for R and Python simultaneously"; - -static PyObject *taus_seed(PyObject *self, PyObject *args) -{ - int seed, *state = NULL; - - if (!PyArg_ParseTuple(args, "i", &seed)) - return NULL; - - lfsr113_seed(seed, &state); - PyObject *pystate = Py_BuildValue("[i, i, i, i]", - state[0], state[1], state[2], state[3]); - free(state); - return pystate; -} - -static PyObject *taus_rand(PyObject *self, PyObject *args) -{ - int i, value, numints, *localstate; - - PyObject *listObj; - PyObject *intObj; - - if (!PyArg_ParseTuple(args, "O!", &PyList_Type, &listObj)) - return NULL; - - // we're just assuming you would never pass more than 4 values - localstate = malloc(sizeof(int)*5); - numints = PyList_Size(listObj); - for (i=0; i<numints; i++) { - intObj = PyList_GetItem(listObj, i); - value = (int) PyLong_AsLong(intObj); - localstate[i] = value; - } - - int rand = lfsr113(&localstate); - localstate[4] = rand; - - PyObject *pystate = Py_BuildValue("[i, i, i, i, i]", - localstate[0], localstate[1], localstate[2], - localstate[3], rand); - free(localstate); - return pystate; -} - -static PyMethodDef TausMethods[] = { - {"seed", taus_seed, METH_VARARGS, - "Seed the Tausworthe RNG."}, - {"rand", taus_rand, METH_VARARGS, - "Generate a single random integer."}, - {NULL, NULL, 0, NULL} -}; - -PyMODINIT_FUNC inittaus(void) -{ - PyObject *m = Py_InitModule3("taus", TausMethods, module_docstring); - if (m == NULL) - return; -} -#endif - -#ifndef TARGETPYTHON -/* - * - * Start of R code - * - */ -SEXP R_tausworthe_seed(SEXP seed) -{ - int i, *pstate = NULL, *state = NULL; - int *pseed = INTEGER(seed); - - SEXP Rstate = PROTECT(allocVector(INTSXP, 5)); - pstate = INTEGER(Rstate); - - lfsr113_seed(*pseed, &state); - - for (i=0; i<4; i++) { - pstate[i] = state[i]; - } - pstate[4] = -1; - free(state); - - UNPROTECT(1); - return Rstate; -} - -SEXP R_tausworthe_rand(SEXP state) -{ - int *localstate = malloc(sizeof(int)*4); - int *pstate = INTEGER(state); - int i; - for (i=0; i<4; i++) - localstate[i] = pstate[i]; - - int rand = lfsr113(&localstate); - - SEXP Rstate = PROTECT(allocVector(INTSXP, 5)); - pstate = INTEGER(Rstate); - - for (i=0; i<4; i++) - pstate[i] = localstate[i]; - pstate[4] = rand; - UNPROTECT(1); - - free(localstate); - - return Rstate; -} -/* - * - * End of R code - * - */ -#endif |
