/*
 * $Id: CurNurb.c,v 1.2 2004/07/29 12:00:34 stiv Exp $
 * ***** BEGIN GPL/BL DUAL 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. The Blender
 * Foundation also sells licenses for use in proprietary software under
 * the Blender License.  See http://www.blender.org/BL/ for information
 * about this.
 *
 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
 * All rights reserved.
 *
 * This is a new part of Blender.
 *
 * Contributor(s): 
 *
 * ***** END GPL/BL DUAL LICENSE BLOCK *****
 */

#include "Python.h"
#include "DNA_curve_types.h"
#include "BKE_curve.h"
#include "MEM_guardedalloc.h"
#include "bpy_types.h"
#include "modules.h"
#include "gen_utils.h"
#include "CurNurb.h"


/*-------------------------------------------------------------

stuff in this section should be placed in bpy_types.h

-----------------------------------------------------------*/


/*
 * forward declarations go here
 */


extern PyMethodDef BPy_CurNurb_methods[];
PyObject *CurNurb_CreatePyObject (Nurb * blen_nurb);
static PyObject *CurNurb_setMatIndex (BPy_CurNurb * self, PyObject * args);
static PyObject *CurNurb_getMatIndex (BPy_CurNurb * self);
/* static PyObject* CurNurb_setXXX( BPy_CurNurb* self, PyObject* args ); */
PyObject *CurNurb_getPoint (BPy_CurNurb * self, int index);
static int CurNurb_length (PyInstanceObject * inst);
static PyObject *CurNurb_getIter (BPy_CurNurb * self);
static PyObject *CurNurb_iterNext (BPy_CurNurb * self);
PyObject *CurNurb_append (BPy_CurNurb * self, PyObject * args);
PyObject *CurNurb_pointAtIndex (Nurb * nurb, int index);
static PyObject *CurNurb_isNurb (BPy_CurNurb * self);

char M_CurNurb_doc[] = "CurNurb";


/*	
  CurNurb_Type callback function prototypes:                          
*/

static void CurNurb_dealloc (BPy_CurNurb * self);
static int CurNurb_compare (BPy_CurNurb * a, BPy_CurNurb * b);
static PyObject *CurNurb_getAttr (BPy_CurNurb * self, char *name);
static int CurNurb_setAttr (BPy_CurNurb * self, char *name, PyObject * v);
static PyObject *CurNurb_repr (BPy_CurNurb * self);




void CurNurb_dealloc (BPy_CurNurb * self)
{
	PyObject_DEL (self);
}



static PyObject *CurNurb_getAttr (BPy_CurNurb * self, char *name)
{
	PyObject *attr = Py_None;

	if(strcmp (name, "mat_index") == 0)
		attr = PyInt_FromLong (self->nurb->mat_nr);

	else if(strcmp (name, "points") == 0)
		attr = PyInt_FromLong (self->nurb->pntsu);

	if(!attr)
		return EXPP_ReturnPyObjError (PyExc_MemoryError,
					      "couldn't create PyObject");

	/* member attribute found, return it */
	if(attr != Py_None)
		return attr;

	/* not an attribute, search the methods table */
	return Py_FindMethod (BPy_CurNurb_methods, (PyObject *) self, name);
}


/*
  setattr
*/

static int CurNurb_setAttr (BPy_CurNurb * self, char *name, PyObject * value)
{
	PyObject *valtuple;
	PyObject *error = NULL;

	/* make a tuple to pass to our type methods */
	valtuple = Py_BuildValue ("(O)", value);

	if(!valtuple)
		return EXPP_ReturnIntError (PyExc_MemoryError,
					    "CurNurb.setAttr: cannot create pytuple");

	if(strcmp (name, "mat_index") == 0)
		error = CurNurb_setMatIndex (self, valtuple);

	else {			/* error - no match for name */
		Py_DECREF (valtuple);

		if((strcmp (name, "ZZZZ") == 0) ||	/* user tried to change a */
		   (strcmp (name, "ZZZZ") == 0))	/* constant dict type ... */
			return EXPP_ReturnIntError (PyExc_AttributeError,
						    "constant dictionary -- cannot be changed");
		else
			return EXPP_ReturnIntError (PyExc_KeyError,
						    "attribute not found");
	}


	Py_DECREF (valtuple);	/* since it is not being returned */
	if(error != Py_None)
		return -1;

	Py_DECREF (Py_None);
	return 0;		/* normal exit */
}

/*
  compare
  in this case, we consider two CurNurbs equal, if they point to the same
  blender data.
*/

static int CurNurb_compare (BPy_CurNurb * a, BPy_CurNurb * b)
{
	Nurb *pa = a->nurb;
	Nurb *pb = b->nurb;

	return (pa == pb) ? 0 : -1;
}


/*
  factory method to create a BPy_CurNurb from a Blender Nurb
*/

PyObject *CurNurb_CreatePyObject (Nurb * blen_nurb)
{
	BPy_CurNurb *pyNurb;

	pyNurb = (BPy_CurNurb *) PyObject_NEW (BPy_CurNurb, &CurNurb_Type);

	if(!pyNurb)
		return EXPP_ReturnPyObjError (PyExc_MemoryError,
					      "could not create BPy_CurNurb PyObject");

	pyNurb->nurb = blen_nurb;
	return (PyObject *) pyNurb;
}


/*
 *  CurNurb_repr
 */
static PyObject *CurNurb_repr (BPy_CurNurb * self)
{				/* used by 'repr' */

	return PyString_FromFormat ("[CurNurb \"%d\"]", self->nurb->type);
}


static PyObject *M_CurNurb_New (PyObject * self, PyObject * args)
{
	return (PyObject *) 0;

}



/*
 * CurNurb_append( point )
 * append a new point to a nurb curve.
 * arg is BezTriple or list of xyzw floats 
 */

PyObject *CurNurb_append (BPy_CurNurb * self, PyObject * args)
{
	Nurb *nurb = self->nurb;

	return CurNurb_appendPointToNurb (nurb, args);
}



/*
 * CurNurb_appendPointToNurb
 * this is a non-bpy utility func to add a point to a given nurb
 */

PyObject *CurNurb_appendPointToNurb (Nurb * nurb, PyObject * args)
{

	int i;
	int size;
	PyObject *pyOb;
	int npoints;

	/*
	   do we have a list of four floats or a BezTriple?
	 */
	PyArg_ParseTuple (args, "O", &pyOb);

	if(BezTriple_CheckPyObject (pyOb)) {
		BezTriple *tmp;
		npoints = nurb->pntsu;

/*		printf("\ndbg: got a BezTriple\n"); */
		tmp = nurb->bezt;	/* save old points */
		nurb->bezt =
			(BezTriple *) MEM_mallocN (sizeof (BezTriple) *
						   (npoints + 1),
						   "CurNurb_append2");

		if(!nurb->bezt)
			return (EXPP_ReturnPyObjError
				(PyExc_MemoryError, "allocation failed"));

		/* copy old points to new */
		memmove (nurb->bezt, tmp, sizeof (BezTriple) * npoints);
		if( tmp )
			MEM_freeN (tmp);
		nurb->pntsu++;
		/* add new point to end of list */
		memcpy (nurb->bezt + npoints,
			BezTriple_FromPyObject (pyOb), sizeof (BezTriple));

	} else if(PySequence_Check (pyOb)) {
		size = PySequence_Size (pyOb);
/*		printf("\ndbg: got a sequence of size %d\n", size );  */
		if(size == 4) {
			BPoint *tmp;
			npoints = nurb->pntsu;

			tmp = nurb->bp;	/* save old pts */

			nurb->bp =
				(BPoint *) MEM_mallocN (sizeof (BPoint) *
							(npoints + 1),
							"CurNurb_append1");
			if(!nurb->bp)
				return (EXPP_ReturnPyObjError
					(PyExc_MemoryError,
					 "allocation failed"));

			memmove (nurb->bp, tmp, sizeof (BPoint) * npoints);
			if( tmp) 
				MEM_freeN (tmp);

			++nurb->pntsu;
			/* initialize new BPoint from old */
			memcpy (nurb->bp + npoints, nurb->bp, sizeof (BPoint));

			for(i = 0; i < 4; ++i) {
				float tmpx =
					(float) PyFloat_AsDouble
					(PySequence_GetItem (pyOb, i));
				nurb->bp[npoints].vec[i] = tmpx;

			}

			makeknots (nurb, 1, nurb->flagu >> 1);

		} else if(size == 3) {	/* 3 xyz coords */
			printf ("\nNot Yet Implemented!\n");

		}

	} else {
		/* bail with error */
		return (EXPP_ReturnPyObjError
			(PyExc_AttributeError, "expected better stuff"));

	}

	return (EXPP_incr_ret (Py_None));
}


/*
 *  CurNurb_setMatIndex
 *
 *  set index into material list
 */

static PyObject *CurNurb_setMatIndex (BPy_CurNurb * self, PyObject * args)
{
	int index;

	if(!PyArg_ParseTuple (args, "i", &(index)))
		return (EXPP_ReturnPyObjError
			(PyExc_AttributeError, "expected integer argument"));

	/* fixme:  some range checking would be nice! */
	self->nurb->mat_nr = index;

	Py_INCREF (Py_None);
	return Py_None;
}

/*
 * CurNurb_getMatIndex
 *
 * returns index into material list
 */

static PyObject *CurNurb_getMatIndex (BPy_CurNurb * self)
{
	PyObject *index = PyInt_FromLong ((long) self->nurb->mat_nr);

	if(index)
		return index;

	return (EXPP_ReturnPyObjError (PyExc_RuntimeError,
				       "could not get material index"));
}


/*
 * CurNurb_getIter
 *
 * create an iterator for our CurNurb.
 * this iterator returns the points for this CurNurb.
 */

static PyObject *CurNurb_getIter (BPy_CurNurb * self)
{
	self->bp = self->nurb->bp;
	self->bezt = self->nurb->bezt;
	self->atEnd = 0;
	self->nextPoint = 0;

	/* set exhausted flag if both bp and bezt are zero */
	if((!self->bp) && (!self->bezt))
		self->atEnd = 1;

	Py_INCREF (self);
	return (PyObject *) self;
}



static PyObject *CurNurb_iterNext (BPy_CurNurb * self)
{
	PyObject *po;		/* return value */
	Nurb *pnurb = self->nurb;
	int npoints = pnurb->pntsu;

	/* are we at end already? */
	if(self->atEnd)
		return (EXPP_ReturnPyObjError (PyExc_StopIteration,
					       "iterator at end"));

	if(self->nextPoint < npoints) {

		po = CurNurb_pointAtIndex (self->nurb, self->nextPoint);
		self->nextPoint++;

		return po;

	} else {
		self->atEnd = 1;	/* set flag true */
	}

	return (EXPP_ReturnPyObjError (PyExc_StopIteration,
				       "iterator at end"));
}



/*
 * CurNurb_isNurb()
 * test whether spline nurb or bezier
 */

static PyObject *CurNurb_isNurb (BPy_CurNurb * self)
{
	/* NOTE: a Nurb has bp and bezt pointers
	 * depending on type.
	 * It is possible both are NULL if no points exist.
	 * in that case, we return False
	 */

	if(self->nurb->bp) {
		Py_INCREF (Py_True);
		return Py_True;
	} else {
		Py_INCREF (Py_False);
		return (Py_False);
	}
}


/*
   table of module methods
   these are the equivalent of class or static methods.
   you do not need an object instance to call one.
  
*/

static PyMethodDef M_CurNurb_methods[] = {
/*   name, method, flags, doc_string                */
	{"New", (PyCFunction) M_CurNurb_New, METH_VARARGS | METH_KEYWORDS,
	 " () - doc string"},
/*  {"Get", (PyCFunction) M_CurNurb_method, METH_NOARGS, " () - doc string"}, */
/*   {"method", (PyCFunction) M_CurNurb_method, METH_NOARGS, " () - doc string"}, */

	{NULL, NULL, 0, NULL}
};



/*
 * method table
 * table of instance methods
 * these methods are invoked on an instance of the type.
*/

static PyMethodDef BPy_CurNurb_methods[] = {
/*   name,     method,                    flags,         doc               */
/*  {"method", (PyCFunction) CurNurb_method, METH_NOARGS, " () - doc string"} */
	{"setMatIndex", (PyCFunction) CurNurb_setMatIndex, METH_VARARGS,
	 "( index ) - set index into materials list"},
	{"getMatIndex", (PyCFunction) CurNurb_getMatIndex, METH_NOARGS,
	 "( ) - get current material index"},
	{"append", (PyCFunction) CurNurb_append, METH_VARARGS,
	 "( point ) - add a new point.  arg is BezTriple or list of x,y,z,w floats"},
	{"isNurb", (PyCFunction) CurNurb_isNurb, METH_NOARGS,
	 "( ) - boolean function tests if this spline is type nurb or bezier"},
	{NULL, NULL, 0, NULL}
};


/* 
 *   methods for CurNurb as sequece
 */

static PySequenceMethods CurNurb_as_sequence = {
	(inquiry) CurNurb_length,	/* sq_length   */
	(binaryfunc) 0,		/* sq_concat */
	(intargfunc) 0,		/* sq_repeat */
	(intargfunc) CurNurb_getPoint,	/* sq_item */
	(intintargfunc) 0,	/* sq_slice */
	0,			/* sq_ass_item */
	0,			/* sq_ass_slice */
	(objobjproc) 0,		/* sq_contains */
	0,
	0
};



/*
  Object Type definition
  full blown 2.3 struct
*/

PyTypeObject CurNurb_Type = {
	PyObject_HEAD_INIT (NULL) /* required py macro */ 0,	/* ob_size */
	/*  For printing, in format "<module>.<name>" */
	"CurNurb",		/* char *tp_name; */
	sizeof (CurNurb_Type),	/* int tp_basicsize, */
	0,			/* tp_itemsize;  For allocation */

	/* Methods to implement standard operations */

	(destructor) CurNurb_dealloc,	/*    destructor tp_dealloc; */
	0,			/*    printfunc tp_print; */
	(getattrfunc) CurNurb_getAttr,	/*    getattrfunc tp_getattr; */
	(setattrfunc) CurNurb_setAttr,	/*    setattrfunc tp_setattr; */
	(cmpfunc) CurNurb_compare,	/*    cmpfunc tp_compare; */
	(reprfunc) CurNurb_repr,	/*    reprfunc tp_repr; */

	/* Method suites for standard classes */

	0,			/*    PyNumberMethods *tp_as_number; */
	&CurNurb_as_sequence,	/*    PySequenceMethods *tp_as_sequence; */
	0,			/*    PyMappingMethods *tp_as_mapping; */

	/* More standard operations (here for binary compatibility) */

	0,			/*    hashfunc tp_hash; */
	0,			/*    ternaryfunc tp_call; */
	0,			/*    reprfunc tp_str; */
	0,			/*    getattrofunc tp_getattro; */
	0,			/*    setattrofunc tp_setattro; */

	/* Functions to access object as input/output buffer */
	0,			/*    PyBufferProcs *tp_as_buffer; */

  /*** Flags to define presence of optional/expanded features ***/
	Py_TPFLAGS_DEFAULT,	/*    long tp_flags; */

	0,			/*  char *tp_doc;  Documentation string */
  /*** Assigned meaning in release 2.0 ***/
	/* call function for all accessible objects */
	0,			/*    traverseproc tp_traverse; */

	/* delete references to contained objects */
	0,			/*    inquiry tp_clear; */

  /***  Assigned meaning in release 2.1 ***/
  /*** rich comparisons ***/
	0,			/*  richcmpfunc tp_richcompare; */

  /***  weak reference enabler ***/
	0,			/* long tp_weaklistoffset; */

  /*** Added in release 2.2 ***/
	/*   Iterators */
	(getiterfunc) CurNurb_getIter,	/*    getiterfunc tp_iter; */
	(iternextfunc) CurNurb_iterNext,	/*    iternextfunc tp_iternext; */

  /*** Attribute descriptor and subclassing stuff ***/
	BPy_CurNurb_methods,	/*    struct PyMethodDef *tp_methods; */
	0,			/*    struct PyMemberDef *tp_members; */
	0,			/*    struct PyGetSetDef *tp_getset; */
	0,			/*    struct _typeobject *tp_base; */
	0,			/*    PyObject *tp_dict; */
	0,			/*    descrgetfunc tp_descr_get; */
	0,			/*    descrsetfunc tp_descr_set; */
	0,			/*    long tp_dictoffset; */
	0,			/*    initproc tp_init; */
	0,			/*    allocfunc tp_alloc; */
	0,			/*    newfunc tp_new; */
	/*  Low-level free-memory routine */
	0,			/*    freefunc tp_free;  */
	/* For PyObject_IS_GC */
	0,			/*    inquiry tp_is_gc;  */
	0,			/*    PyObject *tp_bases; */
	/* method resolution order */
	0,			/*    PyObject *tp_mro;  */
	0,			/*    PyObject *tp_cache; */
	0,			/*    PyObject *tp_subclasses; */
	0,			/*    PyObject *tp_weaklist; */
	0
};


/*
 * CurNurb_length
 * returns the number of points in a Nurb
 * this is a tp_as_sequence method, not a regular instance method.
 */

static int CurNurb_length (PyInstanceObject * inst)
{
	Nurb *nurb;
	int len;

	if(CurNurb_CheckPyObject ((PyObject *) inst)) {
		nurb = ((BPy_CurNurb *) inst)->nurb;
		len = nurb->pntsu;
		return len;
	}

	return EXPP_ReturnIntError (PyExc_RuntimeError,
				    "arg is not a BPy_CurNurb");
}


/*
 * CurNurb_getPoint
 * returns the Nth point in a Nurb
 * this is one of the tp_as_sequence methods, hence the int N argument.
 * it is called via the [] operator, not as a usual instance method.
 */

PyObject *CurNurb_getPoint (BPy_CurNurb * self, int index)
{
	PyObject *pyo;
	Nurb *myNurb;

	int npoints;

	/* for convenince */
	myNurb = self->nurb;
	npoints = myNurb->pntsu;

	/* DELETED: bail if index < 0 */
	/* actually, this check is not needed since python treats */
	/* negative indices as starting from the right end of a sequence */

	/* bail if no Nurbs in Curve */
	if(npoints == 0)
		return (EXPP_ReturnPyObjError (PyExc_IndexError,
					       "no points in this CurNurb"));

	if(index >= npoints)	/* out of range!  */
		return (EXPP_ReturnPyObjError (PyExc_IndexError,
					       "index out of range"));

	pyo = CurNurb_pointAtIndex (myNurb, index);

	return (PyObject *) pyo;
}


/* 
 * this is an internal routine.  not callable directly from python
 */

PyObject *CurNurb_pointAtIndex (Nurb * nurb, int index)
{
	PyObject *pyo;

	if(nurb->bp) {		/* we have a nurb curve */
		int i;
		pyo = PyList_New (4);

		for(i = 0; i < 4; i++) {
			PyList_SetItem (pyo, i,
					PyFloat_FromDouble (nurb->bp[index].vec[i]));

		}

	} else if(nurb->bezt) {	/* we have a bezier */
		/* if an error occurs, we just pass it on */
		pyo = BezTriple_CreatePyObject (&(nurb->bezt[index]));

	} else			/* something is horribly wrong */
		/* neither bp or bezt is set && pntsu != 0 */
		return (EXPP_ReturnPyObjError (PyExc_SystemError,
					       "inconsistant structure found"));

	return (pyo);
}


int CurNurb_CheckPyObject (PyObject * py_obj)
{
	return (py_obj->ob_type == &CurNurb_Type);
}


PyObject *CurNurb_Init (void)
{
	PyObject *submodule;

	Curve_Type.ob_type = &PyType_Type;

	submodule =
		Py_InitModule3 ("Blender.CurNurb", M_CurNurb_methods,
				M_CurNurb_doc);
	return (submodule);
}
