1 files changed, 189 insertions, 0 deletions
diff --git a/taus.c b/taus.c
new file mode 100644
index 0000000..9eba229
--- /dev/null
+++ b/taus.c
@@ -0,0 +1,189 @@
+#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