#include "MersenneTwister.h"
#include "Python.h"

static MTRand mt;

static PyObject *
pymt_seed(PyObject *self, PyObject *args) {
	long int seed;
	
	if (!PyArg_ParseTuple(args, "l", &seed))
		return NULL;
	
	mt.seed(seed);
	
	Py_INCREF(Py_None);
	return Py_None;
}

static PyObject *
pymt_randint(PyObject *self, PyObject *args) {
	return PyLong_FromUnsignedLong(mt.randInt());
}

static PyObject *
pymt_random(PyObject *self, PyObject *args) {
	return PyFloat_FromDouble(mt.randExc());
}

static PyMethodDef pymtMethods[] = {
	{"seed",		pymt_seed,		METH_VARARGS, "Set seed."},
	{"randint",		pymt_randint,	METH_VARARGS, "Returns random integer."},
	{"random",	pymt_random,	METH_VARARGS, "Returns random number in the interval [0, 1)."},
	{NULL,				NULL}
};

extern "C" void initpymt() {
	(void) Py_InitModule3("pymt", pymtMethods, "Mersenne Twister");
}