2da35b55bd51e65e2917832620cb2d23e5b1861e
[blender.git] / source / blender / makesrna / intern / rna_mesh_utils.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /** \file
18  * \ingroup RNA
19  */
20
21 #ifndef __RNA_MESH_UTILS_H__
22 #define __RNA_MESH_UTILS_H__
23
24 /* Macros to help reduce code clutter in rna_mesh.c */
25
26 /* Define the accessors for a basic CustomDataLayer collection */
27 #define DEFINE_CUSTOMDATA_LAYER_COLLECTION(collection_name, customdata_type, layer_type) \
28   /* check */ \
29   static int rna_##collection_name##_check(CollectionPropertyIterator *UNUSED(iter), void *data) \
30   { \
31     CustomDataLayer *layer = (CustomDataLayer *)data; \
32     return (layer->type != layer_type); \
33   } \
34   /* begin */ \
35   static void rna_Mesh_##collection_name##s_begin(CollectionPropertyIterator *iter, \
36                                                   PointerRNA *ptr) \
37   { \
38     CustomData *data = rna_mesh_##customdata_type(ptr); \
39     if (data) { \
40       rna_iterator_array_begin(iter, \
41                                (void *)data->layers, \
42                                sizeof(CustomDataLayer), \
43                                data->totlayer, \
44                                0, \
45                                rna_##collection_name##_check); \
46     } \
47     else { \
48       rna_iterator_array_begin(iter, NULL, 0, 0, 0, NULL); \
49     } \
50   } \
51   /* length */ \
52   static int rna_Mesh_##collection_name##s_length(PointerRNA *ptr) \
53   { \
54     CustomData *data = rna_mesh_##customdata_type(ptr); \
55     return data ? CustomData_number_of_layers(data, layer_type) : 0; \
56   } \
57   /* index range */ \
58   static void rna_Mesh_##collection_name##_index_range( \
59       PointerRNA *ptr, int *min, int *max, int *UNUSED(softmin), int *UNUSED(softmax)) \
60   { \
61     CustomData *data = rna_mesh_##customdata_type(ptr); \
62     *min = 0; \
63     *max = data ? CustomData_number_of_layers(data, layer_type) - 1 : 0; \
64     *max = MAX2(0, *max); \
65   }
66
67 /* Define the accessors for special CustomDataLayers in the collection
68  * (active, render, clone, stencil, etc) */
69 #define DEFINE_CUSTOMDATA_LAYER_COLLECTION_ACTIVEITEM( \
70     collection_name, customdata_type, layer_type, active_type, layer_rna_type) \
71 \
72   static PointerRNA rna_Mesh_##collection_name##_##active_type##_get(PointerRNA *ptr) \
73   { \
74     CustomData *data = rna_mesh_##customdata_type(ptr); \
75     CustomDataLayer *layer; \
76     if (data) { \
77       int index = CustomData_get_##active_type##_layer_index(data, layer_type); \
78       layer = (index == -1) ? NULL : &data->layers[index]; \
79     } \
80     else { \
81       layer = NULL; \
82     } \
83     return rna_pointer_inherit_refine(ptr, &RNA_##layer_rna_type, layer); \
84   } \
85 \
86   static void rna_Mesh_##collection_name##_##active_type##_set( \
87       struct ReportList *UNUSED(reports), PointerRNA *ptr, PointerRNA value) \
88   { \
89     Mesh *me = rna_mesh(ptr); \
90     CustomData *data = rna_mesh_##customdata_type(ptr); \
91     int a; \
92     if (data) { \
93       CustomDataLayer *layer; \
94       int layer_index = CustomData_get_layer_index(data, layer_type); \
95       for (layer = data->layers + layer_index, a = 0; layer_index + a < data->totlayer; \
96            layer++, a++) { \
97         if (value.data == layer) { \
98           CustomData_set_layer_##active_type(data, layer_type, a); \
99           BKE_mesh_update_customdata_pointers(me, true); \
100           return; \
101         } \
102       } \
103     } \
104   } \
105 \
106   static int rna_Mesh_##collection_name##_##active_type##_index_get(PointerRNA *ptr) \
107   { \
108     CustomData *data = rna_mesh_##customdata_type(ptr); \
109     if (data) { \
110       return CustomData_get_##active_type##_layer(data, layer_type); \
111     } \
112     else { \
113       return 0; \
114     } \
115   } \
116 \
117   static void rna_Mesh_##collection_name##_##active_type##_index_set(PointerRNA *ptr, int value) \
118   { \
119     Mesh *me = rna_mesh(ptr); \
120     CustomData *data = rna_mesh_##customdata_type(ptr); \
121     if (data) { \
122       CustomData_set_layer_##active_type(data, layer_type, value); \
123       BKE_mesh_update_customdata_pointers(me, true); \
124     } \
125   }
126
127 #endif /* __RNA_MESH_UTILS_H__ */