source/blender/makesrna/intern/rna_mesh_utils.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

/*
 * 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.
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

/** \file
 * \ingroup RNA
 */

#ifndef __RNA_MESH_UTILS_H__
#define __RNA_MESH_UTILS_H__

/* Macros to help reduce code clutter in rna_mesh.c */

/* Define the accessors for a basic CustomDataLayer collection */
#define DEFINE_CUSTOMDATA_LAYER_COLLECTION(collection_name, customdata_type, layer_type)        \
	/* check */                                                                                 \
	static int rna_##collection_name##_check(                                                   \
	        CollectionPropertyIterator *UNUSED(iter), void *data)                               \
	{                                                                                           \
		CustomDataLayer *layer = (CustomDataLayer *)data;                                       \
		return (layer->type != layer_type);                                                     \
	}                                                                                           \
	/* begin */                                                                                 \
	static void rna_Mesh_##collection_name##s_begin(CollectionPropertyIterator *iter,           \
	                                                PointerRNA *ptr)                            \
	{                                                                                           \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		if (data) {                                                                             \
			rna_iterator_array_begin(iter,                                                      \
			                         (void *)data->layers, sizeof(CustomDataLayer),             \
			                         data->totlayer, 0,                                         \
			                         rna_##collection_name##_check);                            \
		}                                                                                       \
		else {                                                                                  \
			rna_iterator_array_begin(iter, NULL, 0, 0, 0, NULL);                                \
		}                                                                                       \
	}                                                                                           \
	/* length */                                                                                \
	static int rna_Mesh_##collection_name##s_length(PointerRNA *ptr)                            \
	{                                                                                           \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		return data ? CustomData_number_of_layers(data, layer_type) : 0;                        \
	}                                                                                           \
	/* index range */                                                                           \
	static void rna_Mesh_##collection_name##_index_range(                                       \
	        PointerRNA *ptr, int *min, int *max,                                                \
	        int *UNUSED(softmin), int *UNUSED(softmax))                                         \
	{                                                                                           \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		*min = 0;                                                                               \
		*max = data ? CustomData_number_of_layers(data, layer_type) - 1 : 0;                    \
		*max = MAX2(0, *max);                                                                   \
	}

/* Define the accessors for special CustomDataLayers in the collection
 * (active, render, clone, stencil, etc) */
#define DEFINE_CUSTOMDATA_LAYER_COLLECTION_ACTIVEITEM(collection_name, customdata_type,         \
	                                                  layer_type, active_type, layer_rna_type)  \
	                                                                                            \
	static PointerRNA rna_Mesh_##collection_name##_##active_type##_get(PointerRNA *ptr)         \
	{                                                                                           \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		CustomDataLayer *layer;                                                                 \
		if (data) {                                                                             \
			int index = CustomData_get_##active_type##_layer_index(data, layer_type);           \
			layer = (index == -1) ? NULL : &data->layers[index];                                \
		}                                                                                       \
		else {                                                                                  \
			layer = NULL;                                                                       \
		}                                                                                       \
		return rna_pointer_inherit_refine(ptr, &RNA_##layer_rna_type, layer);                   \
	}                                                                                           \
	                                                                                            \
	static void rna_Mesh_##collection_name##_##active_type##_set(PointerRNA *ptr,               \
	                                                             PointerRNA value)              \
	{                                                                                           \
		Mesh *me = rna_mesh(ptr);                                                               \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		int a;                                                                                  \
		if (data) {                                                                             \
			CustomDataLayer *layer;                                                             \
			int layer_index = CustomData_get_layer_index(data, layer_type);                     \
			for (layer = data->layers + layer_index, a = 0; layer_index + a < data->totlayer; layer++, a++) { \
				if (value.data == layer) {                                                      \
					CustomData_set_layer_##active_type(data, layer_type, a);                    \
					BKE_mesh_update_customdata_pointers(me, true);                              \
					return;                                                                     \
				}                                                                               \
			}                                                                                   \
		}                                                                                       \
	}                                                                                           \
	                                                                                            \
	static int rna_Mesh_##collection_name##_##active_type##_index_get(PointerRNA *ptr)          \
	{                                                                                           \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		if (data) {                                                                             \
			return CustomData_get_##active_type##_layer(data, layer_type);                      \
		}                                                                                       \
		else {                                                                                  \
			return 0;                                                                           \
		}                                                                                       \
	}                                                                                           \
	                                                                                            \
	static void rna_Mesh_##collection_name##_##active_type##_index_set(PointerRNA *ptr, int value) \
	{                                                                                           \
		Mesh *me = rna_mesh(ptr);                                                               \
		CustomData *data = rna_mesh_##customdata_type(ptr);                                     \
		if (data) {                                                                             \
			CustomData_set_layer_##active_type(data, layer_type, value);                        \
			BKE_mesh_update_customdata_pointers(me, true);                                      \
		}                                                                                       \
	}

#endif /* __RNA_MESH_UTILS_H__ */