Made the Freestyle Python API compatible with Python 3.
[blender.git] / source / blender / freestyle / intern / python / Interface1D / BPy_FEdge.cpp
1 #include "BPy_FEdge.h"
2
3 #include "../BPy_Convert.h"
4 #include "../BPy_Id.h"
5 #include "../Interface0D/BPy_SVertex.h"
6 #include "../Interface1D/BPy_ViewEdge.h"
7 #include "../BPy_Nature.h"
8
9 #ifdef __cplusplus
10 extern "C" {
11 #endif
12
13 ///////////////////////////////////////////////////////////////////////////////////////////
14
15 /*---------------  Python API function prototypes for FEdge instance  -----------*/
16 static int FEdge___init__(BPy_FEdge *self, PyObject *args, PyObject *kwds);
17 static PyObject * FEdge___copy__( BPy_FEdge *self );
18 static PyObject * FEdge_vertexA( BPy_FEdge *self );
19 static PyObject * FEdge_vertexB( BPy_FEdge *self );
20 static PyObject * FEdge___getitem__( BPy_FEdge *self, PyObject *args );
21 static PyObject * FEdge_nextEdge( BPy_FEdge *self );
22 static PyObject * FEdge_previousEdge( BPy_FEdge *self );
23 static PyObject * FEdge_viewedge( BPy_FEdge *self );
24 static PyObject * FEdge_isSmooth( BPy_FEdge *self );
25 static PyObject * FEdge_setVertexA( BPy_FEdge *self , PyObject *args);
26 static PyObject * FEdge_setVertexB( BPy_FEdge *self , PyObject *args);
27 static PyObject * FEdge_setId( BPy_FEdge *self , PyObject *args);
28 static PyObject * FEdge_setNextEdge( BPy_FEdge *self , PyObject *args);
29 static PyObject * FEdge_setPreviousEdge( BPy_FEdge *self , PyObject *args);
30 static PyObject * FEdge_setSmooth( BPy_FEdge *self , PyObject *args); 
31 static PyObject * FEdge_setNature( BPy_FEdge *self, PyObject *args );
32 static PyObject * FEdge_setViewEdge( BPy_FEdge *self, PyObject *args );
33 static PyObject * FEdge_verticesBegin( BPy_FEdge *self );
34 static PyObject * FEdge_verticesEnd( BPy_FEdge *self );
35 static PyObject * FEdge_pointsBegin( BPy_FEdge *self, PyObject *args );
36 static PyObject * FEdge_pointsEnd( BPy_FEdge *self, PyObject *args );
37
38 /*----------------------FEdge instance definitions ----------------------------*/
39 static PyMethodDef BPy_FEdge_methods[] = {      
40         {"__copy__", ( PyCFunction ) FEdge___copy__, METH_NOARGS, "() Cloning method."},
41         {"vertexA", ( PyCFunction ) FEdge_vertexA, METH_NOARGS, "() Returns the first SVertex."},
42         {"vertexB", ( PyCFunction ) FEdge_vertexB, METH_NOARGS, "() Returns the second SVertex."},
43         {"__getitem__", ( PyCFunction ) FEdge___getitem__, METH_VARARGS, "(int i) Returns the first SVertex if i=0, the seccond SVertex if i=1."},
44         {"nextEdge", ( PyCFunction ) FEdge_nextEdge, METH_NOARGS, "() Returns the FEdge following this one in the ViewEdge. If this FEdge is the last of the ViewEdge, 0 is returned."},
45         {"previousEdge", ( PyCFunction ) FEdge_previousEdge, METH_NOARGS, "Returns the Edge preceding this one in the ViewEdge. If this FEdge is the first one of the ViewEdge, 0 is returned."},
46         {"viewedge", ( PyCFunction ) FEdge_viewedge, METH_NOARGS, "Returns a pointer to the ViewEdge to which this FEdge belongs to."},
47         {"isSmooth", ( PyCFunction ) FEdge_isSmooth, METH_NOARGS, "() Returns true if this FEdge is a smooth FEdge."},
48         {"setVertexA", ( PyCFunction ) FEdge_setVertexA, METH_VARARGS, "(SVertex v) Sets the first SVertex. ."},
49         {"setVertexB", ( PyCFunction ) FEdge_setVertexB, METH_VARARGS, "(SVertex v) Sets the second SVertex. "},
50         {"setId", ( PyCFunction ) FEdge_setId, METH_VARARGS, "(Id id) Sets the FEdge Id ."},
51         {"setNextEdge", ( PyCFunction ) FEdge_setNextEdge, METH_VARARGS, "(FEdge e) Sets the pointer to the next FEdge. "},
52         {"setPreviousEdge", ( PyCFunction ) FEdge_setPreviousEdge, METH_VARARGS, "(FEdge e) Sets the pointer to the previous FEdge. "},
53         {"setSmooth", ( PyCFunction ) FEdge_setSmooth, METH_VARARGS, "(bool b) Sets the flag telling whether this FEdge is smooth or sharp. true for Smooth, false for Sharp. "},
54         {"setViewEdge", ( PyCFunction ) FEdge_setViewEdge, METH_VARARGS, "(ViewEdge ve) Sets the ViewEdge to which this FEdge belongs to."},
55         {"setNature", ( PyCFunction ) FEdge_setNature, METH_VARARGS, "(Nature n) Sets the nature of this FEdge. "},
56         {"verticesBegin", ( PyCFunction ) FEdge_verticesBegin, METH_NOARGS, "() Returns an iterator over the 2 (!) SVertex pointing to the first SVertex."},
57         {"verticesEnd", ( PyCFunction ) FEdge_verticesEnd, METH_NOARGS, "() Returns an iterator over the 2 (!) SVertex pointing after the last SVertex. "},
58         {"pointsBegin", ( PyCFunction ) FEdge_pointsBegin, METH_VARARGS, "(float t=0) Returns an iterator over the FEdge points, pointing to the first point. The difference with verticesBegin() is that here we can iterate over points of the FEdge at a any given sampling t. Indeed, for each iteration, a virtual point is created."},
59         {"pointsEnd", ( PyCFunction ) FEdge_pointsEnd, METH_VARARGS, "(float t=0) Returns an iterator over the FEdge points, pointing after the last point. The difference with verticesEnd() is that here we can iterate over points of the FEdge at a any given sampling t. Indeed, for each iteration, a virtual point is created."},
60         {NULL, NULL, 0, NULL}
61 };
62
63 /*-----------------------BPy_FEdge type definition ------------------------------*/
64
65 PyTypeObject FEdge_Type = {
66         PyObject_HEAD_INIT(NULL)
67         "FEdge",                        /* tp_name */
68         sizeof(BPy_FEdge),              /* tp_basicsize */
69         0,                              /* tp_itemsize */
70         0,                              /* tp_dealloc */
71         0,                              /* tp_print */
72         0,                              /* tp_getattr */
73         0,                              /* tp_setattr */
74         0,                              /* tp_reserved */
75         0,                              /* tp_repr */
76         0,                              /* tp_as_number */
77         0,                              /* tp_as_sequence */
78         0,                              /* tp_as_mapping */
79         0,                              /* tp_hash  */
80         0,                              /* tp_call */
81         0,                              /* tp_str */
82         0,                              /* tp_getattro */
83         0,                              /* tp_setattro */
84         0,                              /* tp_as_buffer */
85         Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
86         "FEdge objects",                /* tp_doc */
87         0,                              /* tp_traverse */
88         0,                              /* tp_clear */
89         0,                              /* tp_richcompare */
90         0,                              /* tp_weaklistoffset */
91         0,                              /* tp_iter */
92         0,                              /* tp_iternext */
93         BPy_FEdge_methods,              /* tp_methods */
94         0,                              /* tp_members */
95         0,                              /* tp_getset */
96         &Interface1D_Type,              /* tp_base */
97         0,                              /* tp_dict */
98         0,                              /* tp_descr_get */
99         0,                              /* tp_descr_set */
100         0,                              /* tp_dictoffset */
101         (initproc)FEdge___init__,       /* tp_init */
102         0,                              /* tp_alloc */
103         0,                              /* tp_new */
104 };
105
106 //------------------------INSTANCE METHODS ----------------------------------
107         
108 int FEdge___init__(BPy_FEdge *self, PyObject *args, PyObject *kwds)
109 {
110         PyObject *obj1 = 0, *obj2 = 0;
111
112     if (! PyArg_ParseTuple(args, "|OO", &obj1, &obj2) )
113         return -1;
114
115         if( !obj1 ){
116                 self->fe = new FEdge();
117
118         } else if( !obj2 && BPy_FEdge_Check(obj1) ) {
119                 self->fe = new FEdge(*( ((BPy_FEdge *) obj1)->fe ));
120                 
121         } else if( obj2 && BPy_SVertex_Check(obj1) && BPy_SVertex_Check(obj2) ) {
122                 self->fe = new FEdge( ((BPy_SVertex *) obj1)->sv, ((BPy_SVertex *) obj2)->sv );
123
124         } else {
125                 PyErr_SetString(PyExc_TypeError, "invalid argument(s)");
126                 return -1;
127         }
128
129         self->py_if1D.if1D = self->fe;
130         self->py_if1D.borrowed = 0;
131
132         return 0;
133 }
134
135
136 PyObject * FEdge___copy__( BPy_FEdge *self ) {
137         BPy_FEdge *py_fe;
138         
139         py_fe = (BPy_FEdge *) FEdge_Type.tp_new( &FEdge_Type, 0, 0 );
140         
141         py_fe->fe = new FEdge( *(self->fe) );
142         py_fe->py_if1D.if1D = py_fe->fe;
143         py_fe->py_if1D.borrowed = 0;
144
145         return (PyObject *) py_fe;
146 }
147
148 PyObject * FEdge_vertexA( BPy_FEdge *self ) {   
149         SVertex *A = self->fe->vertexA();
150         if( A ){
151                 return BPy_SVertex_from_SVertex( *A );
152         }
153                 
154         Py_RETURN_NONE;
155 }
156
157 PyObject * FEdge_vertexB( BPy_FEdge *self ) {
158         SVertex *B = self->fe->vertexB();
159         if( B ){
160                 return BPy_SVertex_from_SVertex( *B );
161         }
162                 
163         Py_RETURN_NONE;
164 }
165
166
167 PyObject * FEdge___getitem__( BPy_FEdge *self, PyObject *args ) {
168         int i;
169
170         if(!( PyArg_ParseTuple(args, "i", &i) ))
171                 return NULL;
172         if(!(i == 0 || i == 1)) {
173                 PyErr_SetString(PyExc_IndexError, "index must be either 0 or 1");
174                 return NULL;
175         }
176         
177         SVertex *v = self->fe->operator[](i);
178         if( v )
179                 return BPy_SVertex_from_SVertex( *v );
180
181         Py_RETURN_NONE;
182 }
183
184 PyObject * FEdge_nextEdge( BPy_FEdge *self ) {
185         FEdge *fe = self->fe->nextEdge();
186         if( fe )
187                 return Any_BPy_FEdge_from_FEdge( *fe );
188
189         Py_RETURN_NONE;
190 }
191
192 PyObject * FEdge_previousEdge( BPy_FEdge *self ) {
193         FEdge *fe = self->fe->previousEdge();
194         if( fe )
195                 return Any_BPy_FEdge_from_FEdge( *fe );
196
197         Py_RETURN_NONE;
198 }
199
200 PyObject * FEdge_viewedge( BPy_FEdge *self ) {
201         ViewEdge *ve = self->fe->viewedge();
202         if( ve )
203                 return BPy_ViewEdge_from_ViewEdge( *ve );
204
205         Py_RETURN_NONE;
206 }
207
208 PyObject * FEdge_isSmooth( BPy_FEdge *self ) {
209         return PyBool_from_bool( self->fe->isSmooth() );
210 }
211         
212 PyObject *FEdge_setVertexA( BPy_FEdge *self , PyObject *args) {
213         PyObject *py_sv;
214
215         if(!( PyArg_ParseTuple(args, "O!", &SVertex_Type, &py_sv) ))
216                 return NULL;
217
218         self->fe->setVertexA( ((BPy_SVertex *) py_sv)->sv );
219
220         Py_RETURN_NONE;
221 }
222
223 PyObject *FEdge_setVertexB( BPy_FEdge *self , PyObject *args) {
224         PyObject *py_sv;
225
226         if(!( PyArg_ParseTuple(args, "O", &py_sv) && BPy_SVertex_Check(py_sv) )) {
227                 cout << "ERROR: FEdge_setVertexB" << endl;
228                 Py_RETURN_NONE;
229         }
230
231         self->fe->setVertexB( ((BPy_SVertex *) py_sv)->sv );
232
233         Py_RETURN_NONE;
234 }
235
236 PyObject *FEdge_setId( BPy_FEdge *self , PyObject *args) {
237         PyObject *py_id;
238
239         if(!( PyArg_ParseTuple(args, "O!", &Id_Type, &py_id) ))
240                 return NULL;
241
242         self->fe->setId(*( ((BPy_Id *) py_id)->id ));
243
244         Py_RETURN_NONE;
245 }
246
247
248 PyObject *FEdge_setNextEdge( BPy_FEdge *self , PyObject *args) {
249         PyObject *py_fe;
250
251         if(!( PyArg_ParseTuple(args, "O!", &FEdge_Type, &py_fe) ))
252                 return NULL;
253
254         self->fe->setNextEdge( ((BPy_FEdge *) py_fe)->fe );
255
256         Py_RETURN_NONE;
257 }
258
259 PyObject *FEdge_setPreviousEdge( BPy_FEdge *self , PyObject *args) {
260         PyObject *py_fe;
261
262         if(!( PyArg_ParseTuple(args, "O!", &FEdge_Type, &py_fe) ))
263                 return NULL;
264
265         self->fe->setPreviousEdge( ((BPy_FEdge *) py_fe)->fe );
266
267         Py_RETURN_NONE;
268 }
269
270 PyObject * FEdge_setNature( BPy_FEdge *self, PyObject *args ) {
271         PyObject *py_n;
272
273         if(!( PyArg_ParseTuple(args, "O!", &Nature_Type, &py_n) ))
274                 return NULL;
275         
276         PyObject *i = (PyObject *) &( ((BPy_Nature *) py_n)->i );
277         self->fe->setNature( PyLong_AsLong(i) );
278
279         Py_RETURN_NONE;
280 }
281
282
283 PyObject * FEdge_setViewEdge( BPy_FEdge *self, PyObject *args ) {
284         PyObject *py_ve;
285
286         if(!( PyArg_ParseTuple(args, "O!", &ViewEdge_Type, &py_ve) ))
287                 return NULL;
288
289         ViewEdge *ve = ((BPy_ViewEdge *) py_ve)->ve;
290         self->fe->setViewEdge( ve );
291
292         Py_RETURN_NONE;
293 }
294
295
296
297 PyObject *FEdge_setSmooth( BPy_FEdge *self , PyObject *args) {
298         PyObject *py_b;
299
300         if(!( PyArg_ParseTuple(args, "O", &py_b) ))
301                 return NULL;
302
303         self->fe->setSmooth( bool_from_PyBool(py_b) );
304
305         Py_RETURN_NONE;
306 }
307
308
309 PyObject * FEdge_verticesBegin( BPy_FEdge *self ) {
310         Interface0DIterator if0D_it( self->fe->verticesBegin() );
311         return BPy_Interface0DIterator_from_Interface0DIterator( if0D_it, 0 );
312 }
313
314 PyObject * FEdge_verticesEnd( BPy_FEdge *self ) {
315         Interface0DIterator if0D_it( self->fe->verticesEnd() );
316         return BPy_Interface0DIterator_from_Interface0DIterator( if0D_it, 1 );
317 }
318
319
320 PyObject * FEdge_pointsBegin( BPy_FEdge *self, PyObject *args ) {
321         float f = 0;
322
323         if(!( PyArg_ParseTuple(args, "|f", &f)  ))
324                 return NULL;
325         
326         Interface0DIterator if0D_it( self->fe->pointsBegin(f) );
327         return BPy_Interface0DIterator_from_Interface0DIterator( if0D_it, 0 );
328 }
329
330 PyObject * FEdge_pointsEnd( BPy_FEdge *self, PyObject *args ) {
331         float f = 0;
332
333         if(!( PyArg_ParseTuple(args, "|f", &f)  ))
334                 return NULL;
335         
336         Interface0DIterator if0D_it( self->fe->pointsEnd(f) );
337         return BPy_Interface0DIterator_from_Interface0DIterator( if0D_it, 1 );
338 }
339 ///////////////////////////////////////////////////////////////////////////////////////////
340
341 #ifdef __cplusplus
342 }
343 #endif