bmesh py api:
[blender.git] / source / blender / python / bmesh / bmesh_py_types_meshdata.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2012 Blender Foundation.
19  * All rights reserved.
20  *
21  * Contributor(s): Campbell Barton
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file blender/python/bmesh/bmesh_py_types_meshdata.c
27  *  \ingroup pybmesh
28  *
29  * This file defines customdata types which can't be accessed as primitive
30  * python types such as MDeformVert, MLoopUV, MTexPoly
31  */
32
33 #include <Python.h>
34
35 #include "../mathutils/mathutils.h"
36
37 #include "DNA_meshdata_types.h"
38
39 #include "BLI_utildefines.h"
40 #include "BLI_math_vector.h"
41
42 /* Mesh Loop UV
43  * ************ */
44
45 typedef struct BPy_BMLoopUV {
46         PyObject_VAR_HEAD
47         MLoopUV *data;
48 } BPy_BMLoopUV;
49
50 static PyObject *bpy_bmloopuv_uv_get(BPy_BMLoopUV *self, void *UNUSED(closure))
51 {
52         return Vector_CreatePyObject(self->data->uv, 2, Py_WRAP, NULL);
53 }
54
55 static int bpy_bmloopuv_uv_set(BPy_BMLoopUV *self, PyObject *value, void *UNUSED(closure))
56 {
57         float tvec[2];
58         if (mathutils_array_parse(tvec, 2, 2, value, "BMLoop.uv") != -1) {
59                 copy_v2_v2(self->data->uv, tvec);
60                 return 0;
61         }
62         else {
63                 return -1;
64         }
65 }
66
67 static PyObject *bpy_bmloopuv_flag_get(BPy_BMLoopUV *self, void *flag_p)
68 {
69         const int flag = GET_INT_FROM_POINTER(flag_p);
70         return PyBool_FromLong(self->data->flag & flag);
71 }
72
73 static int bpy_bmloopuv_flag_set(BPy_BMLoopUV *self, PyObject *value, void *flag_p)
74 {
75         const int flag = GET_INT_FROM_POINTER(flag_p);
76
77         switch (PyLong_AsLong(value)) {
78                 case TRUE:
79                         self->data->flag |= flag;
80                         return 0;
81                 case FALSE:
82                         self->data->flag &= ~flag;
83                         return 0;
84                 default:
85                         PyErr_SetString(PyExc_TypeError,
86                                         "expected a boolean type 0/1");
87                         return -1;
88         }
89 }
90
91 static PyGetSetDef bpy_bmloopuv_getseters[] = {
92     /* attributes match rna_def_mloopuv  */
93     {(char *)"uv",          (getter)bpy_bmloopuv_uv_get,   (setter)bpy_bmloopuv_uv_set,   (char *)NULL, NULL},
94     {(char *)"pin_uv",      (getter)bpy_bmloopuv_flag_get, (setter)bpy_bmloopuv_flag_set, (char *)NULL, (void *)MLOOPUV_PINNED},
95     {(char *)"select",      (getter)bpy_bmloopuv_flag_get, (setter)bpy_bmloopuv_flag_set, (char *)NULL, (void *)MLOOPUV_VERTSEL},
96     {(char *)"select_edge", (getter)bpy_bmloopuv_flag_get, (setter)bpy_bmloopuv_flag_set, (char *)NULL, (void *)MLOOPUV_EDGESEL},
97
98     {NULL, NULL, NULL, NULL, NULL} /* Sentinel */
99 };
100
101 PyTypeObject BPy_BMLoopUV_Type = {{{0}}}; /* bm.loops.layers.uv.active */
102
103 static void bm_init_types_bmloopuv(void)
104 {
105         BPy_BMLoopUV_Type.tp_basicsize = sizeof(BPy_BMLoopUV);
106
107         BPy_BMLoopUV_Type.tp_name = "BMLoopUV";
108
109         BPy_BMLoopUV_Type.tp_doc = NULL; // todo
110
111         BPy_BMLoopUV_Type.tp_getset = bpy_bmloopuv_getseters;
112
113         BPy_BMLoopUV_Type.tp_flags = Py_TPFLAGS_DEFAULT;
114
115         PyType_Ready(&BPy_BMLoopUV_Type);
116 }
117
118 PyObject *BPy_BMLoopUV_CreatePyObject(struct MLoopUV *data)
119 {
120         BPy_BMLoopUV *self = PyObject_New(BPy_BMLoopUV, &BPy_BMLoopUV_Type);
121         self->data = data;
122         return (PyObject *)self;
123 }
124
125 /* --- End Mesh Loop UV --- */
126
127
128 /* call to init all types */
129 void BPy_BM_init_types_meshdata(void)
130 {
131         bm_init_types_bmloopuv();
132 }