/*

 * raytracermodule.cc: Python module

 *

 * This file is part of Pyrit Ray Tracer.

 *

 * Copyright 2006, 2007, 2008  Radek Brich

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to deal

 * in the Software without restriction, including without limitation the rights

 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

 * copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice shall be included in

 * all copies or substantial portions of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

 * THE SOFTWARE.

 */

#include <Python.h>

#include <vector>

#include "raytracer.h"

#include "scene.h"

#include "octree.h"

//=========================== Light Source Object ===========================

typedef struct {

	PyObject_HEAD

	Light *light;

} LightObject;

static PyObject *Light_Constructor(PyObject* self, PyObject* args, PyObject *kwd);

static void Light_Destructor(PyObject* self);

static PyObject *Light_Getattr(PyObject *self, char *name);

static PyObject *Light_castshadows(PyObject* self, PyObject* args);

static PyTypeObject LightType = {

	PyObject_HEAD_INIT(NULL)

	0,				/*ob_size*/

	"Light",			/*tp_name*/

	sizeof(LightObject),		/*tp_basicsize*/

	0,				/*tp_itemsize*/

	/* methods */

	Light_Destructor,		/*tp_dealloc*/

	0,				/*tp_print*/

	Light_Getattr,			/*tp_getattr*/

	0,				/*tp_setattr*/

	0,				/*tp_compare*/

	0,				/*tp_repr*/

	0,				/*tp_as_number*/

	0,				/*tp_as_sequence*/

	0,				/*tp_as_mapping*/

	0,				/*tp_hash */

};

static PyMethodDef LightMethods[] = {

	{"castshadows", (PyCFunction)Light_castshadows, METH_VARARGS, "Enable or disable shadows from this light."},

	{NULL, NULL}

};

static PyObject* Light_Constructor(PyObject* self, PyObject* args, PyObject *kwd)

{

	LightObject *v;

	static char *kwdlist[] = {"position", "colour", NULL};

	PyObject *TPos, *TCol = NULL;

	Float px, py, pz;

	Float cr = 1.0, cg = 1.0, cb = 1.0;

	if (!PyArg_ParseTupleAndKeywords(args, kwd, "O!|O!", kwdlist,

		&PyTuple_Type, &TPos, &PyTuple_Type, &TCol))

		return NULL;

	if (!PyArg_ParseTuple(TPos, "fff", &px, &py, &pz))

		return NULL;

	if (TCol && !PyArg_ParseTuple(TCol, "fff", &cr, &cg, &cb))

		return NULL;

	v = PyObject_New(LightObject, &LightType);

	v->light = new Light(Vector3(px, py, pz), Colour(cr, cg, cb));

	return (PyObject*)v;

}

static void Light_Destructor(PyObject* self)

{

	delete ((LightObject *)self)->light;

	PyObject_Del(self);

}

static PyObject *Light_Getattr(PyObject *self, char *name)

{

	return Py_FindMethod(LightMethods, self, name);

}

static PyObject *Light_castshadows(PyObject* self, PyObject* args)

{

	int shadows = 1;

	if (!PyArg_ParseTuple(args, "i", &shadows))

		return NULL;

	((LightObject *)self)->light->castShadows(shadows);

	Py_INCREF(Py_None);

	return Py_None;

}

//=========================== Camera Object ===========================

typedef struct {

	PyObject_HEAD

	Camera *camera;

} CameraObject;

static PyObject *Camera_Constructor(PyObject* self, PyObject* args, PyObject *kwd);

static void Camera_Destructor(PyObject* self);

static PyObject *Camera_Getattr(PyObject *self, char *name);

static PyObject *Camera_setEye(PyObject* self, PyObject* args);

static PyObject *Camera_rotate(PyObject* self, PyObject* args);

static PyTypeObject CameraType = {

	PyObject_HEAD_INIT(NULL)

	0,				/*ob_size*/

	"Camera",			/*tp_name*/

	sizeof(CameraObject),		/*tp_basicsize*/

	0,				/*tp_itemsize*/

	/* methods */

	Camera_Destructor,		/*tp_dealloc*/

	0,				/*tp_print*/

	Camera_Getattr,			/*tp_getattr*/

	0,				/*tp_setattr*/

	0,				/*tp_compare*/

	0,				/*tp_repr*/

	0,				/*tp_as_number*/

	0,				/*tp_as_sequence*/

	0,				/*tp_as_mapping*/

	0,				/*tp_hash */

};

static PyMethodDef CameraMethods[] = {

	{"setEye", (PyCFunction)Camera_setEye, METH_VARARGS, "Set eye of the camera."},

	{"rotate", (PyCFunction)Camera_rotate, METH_VARARGS, "Rotate camera with a quaternion."},

	{NULL, NULL}

};

static PyObject* Camera_Constructor(PyObject* self, PyObject* args, PyObject *kwd)

{

	CameraObject *v;

	static char *kwdlist[] = {"eye", "p", "u", "v", NULL};

	PyObject *TEye = NULL, *Tp = NULL, *Tu = NULL, *Tv = NULL;

	Float ex=0.0,  ey=0.0, ez=10.0;

	Float px=0.0,  py=0.0, pz=-1.0;

	Float ux=-1.0, uy=0.0, uz=0.0;

	Float vx=0.0,  vy=1.0, vz=0.0;

	if (!PyArg_ParseTupleAndKeywords(args, kwd, "|O!O!O!O!", kwdlist,

		&PyTuple_Type, &TEye, &PyTuple_Type, &Tp, &PyTuple_Type, &Tu, &PyTuple_Type, &Tv))

		return NULL;

	if (TEye)

		if (!PyArg_ParseTuple(TEye, "fff", &ex, &ey, &ez))

			return NULL;

	if (Tp)

		if (!PyArg_ParseTuple(Tp, "fff", &px, &py, &pz))

			return NULL;

	if (Tu)

		if (!PyArg_ParseTuple(Tu, "fff", &ux, &uy, &uz))

			return NULL;

	if (Tv)

		if (!PyArg_ParseTuple(Tv, "fff", &vx, &vy, &vz))

			return NULL;

	v = PyObject_New(CameraObject, &CameraType);

	v->camera = new Camera(Vector3(ex, ey, ez),

		Vector3(px, py, pz), Vector3(ux, uy, uz), Vector3(vx, vy, vz));

	return (PyObject*)v;

}

static void Camera_Destructor(PyObject* self)

{

	delete ((CameraObject *)self)->camera;

	PyObject_Del(self);

}

static PyObject *Camera_Getattr(PyObject *self, char *name)

{

	return Py_FindMethod(CameraMethods, self, name);

}

static PyObject *Camera_setEye(PyObject* self, PyObject* args)

{

	PyObject *TEye = NULL;

	Float ex=0.0,  ey=0.0, ez=10.0;

	if (!PyArg_ParseTuple(args, "O!", &PyTuple_Type, &TEye))

		return NULL;

	if (TEye)

		if (!PyArg_ParseTuple(TEye, "fff", &ex, &ey, &ez))

			return NULL;

	((CameraObject *)self)->camera->setEye(Vector3(ex, ey, ez));

	Py_INCREF(Py_None);

	return Py_None;

}

static PyObject *Camera_rotate(PyObject* self, PyObject* args)

{

	PyObject *Tq = NULL;

	Float qa, qb, qc, qd;

	if (!PyArg_ParseTuple(args, "O!", &PyTuple_Type, &Tq))

		return NULL;

	if (!PyArg_ParseTuple(Tq, "ffff", &qa, &qb, &qc, &qd))

		return NULL;

	((CameraObject *)self)->camera->rotate(Quaternion(qa, qb, qc, qd).normalize());

	Py_INCREF(Py_None);

	return Py_None;

}

//=========================== Material Object ===========================

typedef struct {

	PyObject_HEAD

	Material *material;

} MaterialObject;

static PyObject *Material_Constructor(PyObject* self, PyObject* args, PyObject *kwd);

static void Material_Destructor(PyObject* self);

static PyObject *Material_Getattr(PyObject *self, char *name);

static PyObject *Material_setPhong(PyObject* self, PyObject* args);

static PyObject *Material_setReflectivity(PyObject* self, PyObject* args);

static PyObject *Material_setTransmissivity(PyObject* self, PyObject* args);

static PyObject *Material_setSmooth(PyObject* self, PyObject* args);

static PyTypeObject MaterialType = {

	PyObject_HEAD_INIT(NULL)

	0,				/*ob_size*/

	"Material",			/*tp_name*/

	sizeof(MaterialObject),		/*tp_basicsize*/

	0,				/*tp_itemsize*/

	/* methods */

	Material_Destructor,		/*tp_dealloc*/

	0,				/*tp_print*/

	Material_Getattr,			/*tp_getattr*/

	0,				/*tp_setattr*/

	0,				/*tp_compare*/

	0,				/*tp_repr*/

	0,				/*tp_as_number*/

	0,				/*tp_as_sequence*/

	0,				/*tp_as_mapping*/

	0,				/*tp_hash */

};

static PyMethodDef MaterialMethods[] = {

	{"setPhong", (PyCFunction)Material_setPhong, METH_VARARGS, "Set ambient, diffuse, specular and shininess Phong model constants."},

	{"setReflectivity", (PyCFunction)Material_setReflectivity, METH_VARARGS, "Set reflectivity."},

	{"setTransmissivity", (PyCFunction)Material_setTransmissivity, METH_VARARGS, "Set transmissivity and refraction index."},

	{"setSmooth", (PyCFunction)Material_setSmooth, METH_VARARGS, "Set triangle smoothing."},

	{NULL, NULL}

};

static PyObject* Material_Constructor(PyObject* self, PyObject* args, PyObject *kwd)

{

	MaterialObject *v;

	static char *kwdlist[] = {"colour", NULL};

	PyObject *TCol = NULL;

	Float cr=1.0, cg=1.0, cb=1.0;

	if (!PyArg_ParseTupleAndKeywords(args, kwd, "|O!", kwdlist,

		&PyTuple_Type, &TCol))

		return NULL;

	if (TCol)

		if (!PyArg_ParseTuple(TCol, "fff", &cr, &cg, &cb))

			return NULL;

	v = PyObject_New(MaterialObject, &MaterialType);

	v->material = new Material(Colour(cr, cg, cb));

	return (PyObject*)v;

}

static void Material_Destructor(PyObject* self)

{

	delete ((MaterialObject *)self)->material;

	PyObject_Del(self);

}

static PyObject *Material_Getattr(PyObject *self, char *name)

{

	return Py_FindMethod(MaterialMethods, self, name);

}

static PyObject *Material_setPhong(PyObject* self, PyObject* args)

{

	Float amb, dif, spec, shin = 0.5;

	if (!PyArg_ParseTuple(args, "fff|f", &amb, &dif, &spec, &shin))

		return NULL;

	((MaterialObject *)self)->material->setPhong(amb, dif, spec, shin);

	Py_INCREF(Py_None);

	return Py_None;

}

static PyObject *Material_setReflectivity(PyObject* self, PyObject* args)

{

	Float refl;

	if (!PyArg_ParseTuple(args, "f", &refl))

		return NULL;

	((MaterialObject *)self)->material->setReflectivity(refl);

	Py_INCREF(Py_None);

	return Py_None;

}

static PyObject *Material_setTransmissivity(PyObject* self, PyObject* args)

{

	Float trans, rindex = 1.3;

	if (!PyArg_ParseTuple(args, "f|f", &trans, &rindex))

		return NULL;

	((MaterialObject *)self)->material->setTransmissivity(trans, rindex);

	Py_INCREF(Py_None);

	return Py_None;

}

static PyObject* Material_setSmooth(PyObject* self, PyObject* args)

{

	int smooth;

	if (!PyArg_ParseTuple(args, "i", &smooth))

		return NULL;

	((MaterialObject *)self)->material->setSmooth(smooth);

	Py_INCREF(Py_None);

	return Py_None;

}

//=========================== NormalVertex Object ===========================

typedef struct {

	PyObject_HEAD

	NormalVertex *nvertex;

} NormalVertexObject;

static PyObject *NormalVertex_Constructor(PyObject* self, PyObject* args, PyObject *kwd);

static void NormalVertex_Destructor(PyObject* self);

static PyObject *NormalVertex_Getattr(PyObject *self, char *name);

static PyObject *NormalVertex_setNormal(PyObject* self, PyObject* args);

static PyTypeObject NormalVertexType = {

	PyObject_HEAD_INIT(NULL)

	0,				/*ob_size*/

	"NormalVertex",			/*tp_name*/

	sizeof(NormalVertexObject),		/*tp_basicsize*/

	0,				/*tp_itemsize*/

	/* methods */

	NormalVertex_Destructor,		/*tp_dealloc*/

	0,				/*tp_print*/

	NormalVertex_Getattr,			/*tp_getattr*/

	0,				/*tp_setattr*/

	0,				/*tp_compare*/

	0,				/*tp_repr*/

	0,				/*tp_as_number*/

	0,				/*tp_as_sequence*/

	0,				/*tp_as_mapping*/

	0,				/*tp_hash */

};

static PyMethodDef NormalVertexMethods[] = {

	{"setNormal", (PyCFunction)NormalVertex_setNormal, METH_VARARGS, "Set normal of this vertex."},

	{NULL, NULL}

};

static PyObject* NormalVertex_Constructor(PyObject* self, PyObject* args, PyObject *kwd)

{

	NormalVertexObject *v;

	static char *kwdlist[] = {"vector", "normal", NULL};

	PyObject *TVer = NULL;

	PyObject *TNor = NULL;

	Float vx, vy, vz, nx=0, ny=0, nz=0;

	if (!PyArg_ParseTupleAndKeywords(args, kwd, "O|O!", kwdlist,

		&TVer, &PyTuple_Type, &TNor))

		return NULL;

	if (!TNor && TVer->ob_type == &NormalVertexType)

	{

		v = PyObject_New(NormalVertexObject, &NormalVertexType);

		v->nvertex = new NormalVertex(((NormalVertexObject*)TVer)->nvertex);

	}

	else

	{

		if (!PyArg_ParseTuple(TVer, "fff", &vx, &vy, &vz))

			return NULL;

		if (TNor)

			if (!PyArg_ParseTuple(TNor, "fff", &nx, &ny, &nz))

				return NULL;

		v = PyObject_New(NormalVertexObject, &NormalVertexType);

		v->nvertex = new NormalVertex(Vector3(vx, vy, vz), Vector3(nx, ny, nz));

	}

	return (PyObject*)v;

}

static void NormalVertex_Destructor(PyObject* self)

{

	delete ((NormalVertexObject *)self)->nvertex;

	PyObject_Del(self);

}

static PyObject *NormalVertex_Getattr(PyObject *self, char *name)

{

	return Py_FindMethod(NormalVertexMethods, self, name);

}

static PyObject *NormalVertex_setNormal(PyObject* self, PyObject* args)

{

	PyObject *TNor = NULL;

	Float nx, ny, nz;

	if (!PyArg_ParseTuple(args, "O!", &PyTuple_Type, &TNor))

		return NULL;

	if (!PyArg_ParseTuple(TNor, "fff", &nx, &ny, &nz))

		return NULL;

	((NormalVertexObject *)self)->nvertex->setNormal(Vector3(nx,ny,nz).normalize());

	Py_INCREF(Py_None);

	return Py_None;

}

//=========================== Sphere Object ===========================

typedef struct {

	PyObject_HEAD

	Sphere *shape;

} SphereObject;

static PyObject *Sphere_Constructor(PyObject* self, PyObject* args, PyObject *kwd);

static void Sphere_Destructor(PyObject* self);

static PyObject *Sphere_Getattr(PyObject *self, char *name);

static PyTypeObject SphereType = {

	PyObject_HEAD_INIT(NULL)

	0,				/*ob_size*/

	"Sphere",			/*tp_name*/

	sizeof(SphereObject),		/*tp_basicsize*/

	0,				/*tp_itemsize*/

	/* methods */

	Sphere_Destructor,		/*tp_dealloc*/

	0,				/*tp_print*/

	Sphere_Getattr,			/*tp_getattr*/

	0,				/*tp_setattr*/

	0,				/*tp_compare*/

	0,				/*tp_repr*/

	0,				/*tp_as_number*/

	0,				/*tp_as_sequence*/

	0,				/*tp_as_mapping*/

	0,				/*tp_hash */

};

static PyMethodDef SphereMethods[] = {

	{NULL, NULL}

};

static PyObject* Sphere_Constructor(PyObject* self, PyObject* args, PyObject *kwd)

{

	SphereObject *v;

	MaterialObject *material;

	static char *kwdlist[] = {"centre", "radius", "material", NULL};

	PyObject *TCentre = NULL;

	Float cx, cy, cz, radius;

	if (!PyArg_ParseTupleAndKeywords(args, kwd, "O!fO!", kwdlist,

		&PyTuple_Type, &TCentre, &radius, &MaterialType, &material))

		return NULL;