/*
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * 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.
 *
 * ***** END GPL LICENSE BLOCK *****
 */

/** \file source/blender/freestyle/intern/python/BPy_IntegrationType.cpp
 *  \ingroup freestyle
 */

#include "BPy_IntegrationType.h"

#include "BPy_Convert.h"
#include "UnaryFunction0D/BPy_UnaryFunction0DDouble.h"
#include "UnaryFunction0D/BPy_UnaryFunction0DFloat.h"
#include "UnaryFunction0D/BPy_UnaryFunction0DUnsigned.h"
#include "Iterator/BPy_Interface0DIterator.h"

#ifdef __cplusplus
extern "C" {
#endif

///////////////////////////////////////////////////////////////////////////////////////////

//------------------------ MODULE FUNCTIONS ----------------------------------

PyDoc_STRVAR(Integrator_integrate_doc,
".. function:: integrate(func, it, it_end, integration_type)\n"
"\n"
"   Returns a single value from a set of values evaluated at each 0D\n"
"   element of this 1D element.\n"
"\n"
"   :arg func: The UnaryFunction0D used to compute a value at each\n"
"      Interface0D.\n"
"   :type func: :class:`UnaryFunction0D`\n"
"   :arg it: The Interface0DIterator used to iterate over the 0D\n"
"      elements of this 1D element. The integration will occur over\n"
"      the 0D elements starting from the one pointed by it.\n"
"   :type it: :class:`Interface0DIterator`\n"
"   :arg it_end: The Interface0DIterator pointing the end of the 0D\n"
"      elements of the 1D element.\n"
"   :type it_end: :class:`Interface0DIterator`\n"
"   :arg integration_type: The integration method used to compute a\n"
"      single value from a set of values.\n"
"   :type integration_type: :class:`IntegrationType`\n"
"   :return: The single value obtained for the 1D element.  The return\n"
"      value type is float if func is of the :class:`UnaryFunction0DDouble`\n"
"      or :class:`UnaryFunction0DFloat` type, and int if func is of the\n"
"      :class:`UnaryFunction0DUnsigned` type.\n"
"   :rtype: int or float");

static PyObject * Integrator_integrate(PyObject *self, PyObject *args, PyObject *kwds)
{
	static const char *kwlist[] = {"func", "it", "it_end", "integration_type", NULL};
	PyObject *obj1, *obj4 = 0;
	BPy_Interface0DIterator *obj2, *obj3;

	if (!PyArg_ParseTupleAndKeywords(args, kwds, "O!O!O!|O!", (char **)kwlist,
	                                 &UnaryFunction0D_Type, &obj1, &Interface0DIterator_Type, &obj2,
	                                 &Interface0DIterator_Type, &obj3, &IntegrationType_Type, &obj4))
	{
		return NULL;
	}

	Interface0DIterator it(*(obj2->if0D_it)), it_end(*(obj3->if0D_it));
	IntegrationType t = (obj4) ? IntegrationType_from_BPy_IntegrationType(obj4) : MEAN;

	if (BPy_UnaryFunction0DDouble_Check(obj1)) {
		UnaryFunction0D<double> *fun = ((BPy_UnaryFunction0DDouble *)obj1)->uf0D_double;
		double res = integrate(*fun, it, it_end, t);
		return PyFloat_FromDouble(res);
	}
	else if (BPy_UnaryFunction0DFloat_Check(obj1)) {
		UnaryFunction0D<float> *fun = ((BPy_UnaryFunction0DFloat *)obj1)->uf0D_float;
		float res = integrate(*fun, it, it_end, t);
		return PyFloat_FromDouble(res);
	}
	else if (BPy_UnaryFunction0DUnsigned_Check(obj1)) {
		UnaryFunction0D<unsigned int> *fun = ((BPy_UnaryFunction0DUnsigned *)obj1)->uf0D_unsigned;
		unsigned int res = integrate(*fun, it, it_end, t);
		return PyLong_FromLong(res);
	}
	else {
		string class_name(Py_TYPE(obj1)->tp_name);
		PyErr_SetString(PyExc_TypeError, ("unsupported function type: " + class_name).c_str());
		return NULL;
	}
}

/*-----------------------Integrator module docstring---------------------------------------*/

PyDoc_STRVAR(module_docstring, "The Blender Freestyle.Integrator submodule\n\n");

/*-----------------------Integrator module functions definitions---------------------------*/

static PyMethodDef module_functions[] = {
	{"integrate", (PyCFunction) Integrator_integrate, METH_VARARGS | METH_KEYWORDS, Integrator_integrate_doc},
	{NULL, NULL, 0, NULL}
};

/*-----------------------Integrator module definition--------------------------------------*/

static PyModuleDef module_definition = {
    PyModuleDef_HEAD_INIT,
    "Freestyle.Integrator",
    module_docstring,
    -1,
    module_functions
};

/*-----------------------BPy_IntegrationType type definition ------------------------------*/

PyDoc_STRVAR(IntegrationType_doc,
"Class hierarchy: int > :class:`IntegrationType`\n"
"\n"
"Different integration methods that can be invoked to integrate into a\n"
"single value the set of values obtained from each 0D element of an 1D\n"
"element:\n"
"\n"
"* IntegrationType.MEAN: The value computed for the 1D element is the\n"
"  mean of the values obtained for the 0D elements.\n"
"* IntegrationType.MIN: The value computed for the 1D element is the\n"
"  minimum of the values obtained for the 0D elements.\n"
"* IntegrationType.MAX: The value computed for the 1D element is the\n"
"  maximum of the values obtained for the 0D elements.\n"
"* IntegrationType.FIRST: The value computed for the 1D element is the\n"
"  first of the values obtained for the 0D elements.\n"
"* IntegrationType.LAST: The value computed for the 1D element is the\n"
"  last of the values obtained for the 0D elements.");

PyTypeObject IntegrationType_Type = {
	PyVarObject_HEAD_INIT(NULL, 0)
	"IntegrationType",              /* tp_name */
	sizeof(PyLongObject),           /* tp_basicsize */
	0,                              /* tp_itemsize */
	0,                              /* tp_dealloc */
	0,                              /* tp_print */
	0,                              /* tp_getattr */
	0,                              /* tp_setattr */
	0,                              /* tp_reserved */
	0,                              /* tp_repr */
	0,                              /* tp_as_number */
	0,                              /* tp_as_sequence */
	0,                              /* tp_as_mapping */
	0,                              /* tp_hash  */
	0,                              /* tp_call */
	0,                              /* tp_str */
	0,                              /* tp_getattro */
	0,                              /* tp_setattro */
	0,                              /* tp_as_buffer */
	Py_TPFLAGS_DEFAULT,             /* tp_flags */
	IntegrationType_doc,            /* tp_doc */
	0,                              /* tp_traverse */
	0,                              /* tp_clear */
	0,                              /* tp_richcompare */
	0,                              /* tp_weaklistoffset */
	0,                              /* tp_iter */
	0,                              /* tp_iternext */
	0,                              /* tp_methods */
	0,                              /* tp_members */
	0,                              /* tp_getset */
	&PyLong_Type,                   /* tp_base */
	0,                              /* tp_dict */
	0,                              /* tp_descr_get */
	0,                              /* tp_descr_set */
	0,                              /* tp_dictoffset */
	0,                              /* tp_init */
	0,                              /* tp_alloc */
	0,                              /* tp_new */
};

/*-----------------------BPy_IntegrationType instance definitions -------------------------*/

static PyLongObject _IntegrationType_MEAN = {
	PyVarObject_HEAD_INIT(&IntegrationType_Type, 1)
	{ MEAN }
};
static PyLongObject _IntegrationType_MIN = {
	PyVarObject_HEAD_INIT(&IntegrationType_Type, 1)
	{ MIN }
};
static PyLongObject _IntegrationType_MAX = {
	PyVarObject_HEAD_INIT(&IntegrationType_Type, 1)
	{ MAX }
};
static PyLongObject _IntegrationType_FIRST = {
	PyVarObject_HEAD_INIT(&IntegrationType_Type, 1)
	{ FIRST }
};
static PyLongObject _IntegrationType_LAST = {
	PyVarObject_HEAD_INIT(&IntegrationType_Type, 1)
	{ LAST }
};

#define BPy_IntegrationType_MEAN  ((PyObject *)&_IntegrationType_MEAN)
#define BPy_IntegrationType_MIN   ((PyObject *)&_IntegrationType_MIN)
#define BPy_IntegrationType_MAX   ((PyObject *)&_IntegrationType_MAX)
#define BPy_IntegrationType_FIRST ((PyObject *)&_IntegrationType_FIRST)
#define BPy_IntegrationType_LAST  ((PyObject *)&_IntegrationType_LAST)

//-------------------MODULE INITIALIZATION--------------------------------
int IntegrationType_Init(PyObject *module)
{
	PyObject *m, *d, *f;
	
	if (module == NULL)
		return -1;

	if (PyType_Ready(&IntegrationType_Type) < 0)
		return -1;
	Py_INCREF(&IntegrationType_Type);
	PyModule_AddObject(module, "IntegrationType", (PyObject *)&IntegrationType_Type);

	PyDict_SetItemString(IntegrationType_Type.tp_dict, "MEAN", BPy_IntegrationType_MEAN);
	PyDict_SetItemString(IntegrationType_Type.tp_dict, "MIN", BPy_IntegrationType_MIN);
	PyDict_SetItemString(IntegrationType_Type.tp_dict, "MAX", BPy_IntegrationType_MAX);
	PyDict_SetItemString(IntegrationType_Type.tp_dict, "FIRST", BPy_IntegrationType_FIRST);
	PyDict_SetItemString(IntegrationType_Type.tp_dict, "LAST", BPy_IntegrationType_LAST);
	
	m = PyModule_Create(&module_definition);
	if (m == NULL)
		return -1;
	Py_INCREF(m);
	PyModule_AddObject(module, "Integrator", m);

	// from Integrator import *
	d = PyModule_GetDict(m);
	for (PyMethodDef *p = module_functions; p->ml_name; p++) {
		f = PyDict_GetItemString(d, p->ml_name);
		Py_INCREF(f);
		PyModule_AddObject(module, p->ml_name, f);
	}

	return 0;
}

///////////////////////////////////////////////////////////////////////////////////////////

#ifdef __cplusplus
}
#endif
