Bugfix #24837
[blender.git] / source / blender / makesrna / intern / rna_meta.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * Contributor(s): Blender Foundation (2008), Juho Vepsalainen, Jiri Hnidek
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include <stdlib.h>
26
27 #include "RNA_define.h"
28 #include "RNA_enum_types.h"
29
30 #include "rna_internal.h"
31
32 #include "DNA_mesh_types.h"
33 #include "DNA_meta_types.h"
34
35 #ifdef RNA_RUNTIME
36
37 #include "BLI_math.h"
38
39 #include "MEM_guardedalloc.h"
40
41 #include "DNA_scene_types.h"
42 #include "DNA_object_types.h"
43
44 #include "BKE_mball.h"
45 #include "BKE_depsgraph.h"
46 #include "BKE_main.h"
47
48 #include "WM_types.h"
49 #include "WM_api.h"
50
51 static int rna_Meta_texspace_editable(PointerRNA *ptr)
52 {
53         MetaBall *mb= (MetaBall*)ptr->data;
54         return (mb->texflag & MB_AUTOSPACE)? 0: PROP_EDITABLE;
55 }
56
57 static void rna_Meta_texspace_loc_get(PointerRNA *ptr, float *values)
58 {
59         MetaBall *mb= (MetaBall*)ptr->data;
60         
61         /* tex_space_mball() needs object.. ugh */
62         
63         copy_v3_v3(values, mb->loc);
64 }
65
66 static void rna_Meta_texspace_loc_set(PointerRNA *ptr, const float *values)
67 {
68         MetaBall *mb= (MetaBall*)ptr->data;
69         
70         copy_v3_v3(mb->loc, values);
71 }
72
73 static void rna_Meta_texspace_size_get(PointerRNA *ptr, float *values)
74 {
75         MetaBall *mb= (MetaBall*)ptr->data;
76         
77         /* tex_space_mball() needs object.. ugh */
78         
79         copy_v3_v3(values, mb->size);
80 }
81
82 static void rna_Meta_texspace_size_set(PointerRNA *ptr, const float *values)
83 {
84         MetaBall *mb= (MetaBall*)ptr->data;
85         
86         copy_v3_v3(mb->size, values);
87 }
88
89
90 static void rna_MetaBall_update_data(Main *bmain, Scene *scene, PointerRNA *ptr)
91 {
92         MetaBall *mb= ptr->id.data;
93         Object *ob;
94
95         /* cheating way for importers to avoid slow updates */
96         if(mb->id.us > 0) {
97                 for(ob=bmain->object.first; ob; ob= ob->id.next)
98                         if(ob->data == mb)
99                                 copy_mball_properties(scene, ob);
100         
101                 DAG_id_flush_update(&mb->id, OB_RECALC_DATA);
102                 WM_main_add_notifier(NC_GEOM|ND_DATA, mb);
103         }
104 }
105
106 static MetaElem *rna_MetaBall_elements_new(MetaBall *mb, int type)
107 {
108         MetaElem *ml= add_metaball_element(mb, type);
109
110         /* cheating way for importers to avoid slow updates */
111         if(mb->id.us > 0) {
112                 DAG_id_flush_update(&mb->id, OB_RECALC_DATA);
113                 WM_main_add_notifier(NC_GEOM|ND_DATA, &mb->id);
114         }
115
116         return ml;
117 }
118
119 static void rna_MetaBall_elements_remove(MetaBall *mb, ReportList *reports, MetaElem *ml)
120 {
121         int found= 0;
122
123         found= BLI_remlink_safe(&mb->elems, ml);
124
125         if(!found) {
126                 BKE_reportf(reports, RPT_ERROR, "MetaBall \"%s\" does not contain spline given", mb->id.name+2);
127                 return;
128         }
129
130         MEM_freeN(ml);
131         /* invalidate pointer!, no can do */
132
133         /* cheating way for importers to avoid slow updates */
134         if(mb->id.us > 0) {
135                 DAG_id_flush_update(&mb->id, OB_RECALC_DATA);
136                 WM_main_add_notifier(NC_GEOM|ND_DATA, &mb->id);
137         }
138 }
139
140 #else
141
142 static void rna_def_metaelement(BlenderRNA *brna)
143 {
144         StructRNA *srna;
145         PropertyRNA *prop;
146
147         srna= RNA_def_struct(brna, "MetaElement", NULL);
148         RNA_def_struct_sdna(srna, "MetaElem");
149         RNA_def_struct_ui_text(srna, "Meta Element", "Blobby element in a MetaBall datablock");
150         RNA_def_struct_ui_icon(srna, ICON_OUTLINER_DATA_META);
151
152         /* enums */
153         prop= RNA_def_property(srna, "type", PROP_ENUM, PROP_NONE);
154         RNA_def_property_enum_items(prop, metaelem_type_items);
155         RNA_def_property_ui_text(prop, "Type", "Metaball types");
156         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
157         
158         /* number values */
159         prop= RNA_def_property(srna, "co", PROP_FLOAT, PROP_TRANSLATION);
160         RNA_def_property_float_sdna(prop, NULL, "x");
161         RNA_def_property_array(prop, 3);
162         RNA_def_property_ui_text(prop, "Location", "");
163         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
164
165         prop= RNA_def_property(srna, "rotation", PROP_FLOAT, PROP_QUATERNION);
166         RNA_def_property_float_sdna(prop, NULL, "quat");
167         RNA_def_property_ui_text(prop, "Rotation", "");
168         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
169
170         prop= RNA_def_property(srna, "radius", PROP_FLOAT, PROP_UNSIGNED|PROP_UNIT_LENGTH);
171         RNA_def_property_float_sdna(prop, NULL, "rad");
172         RNA_def_property_ui_text(prop, "Radius", "");
173         RNA_def_property_range(prop, 0.0f, FLT_MAX);
174         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
175
176         prop= RNA_def_property(srna, "size_x", PROP_FLOAT, PROP_DISTANCE);
177         RNA_def_property_float_sdna(prop, NULL, "expx");
178         RNA_def_property_range(prop, 0.0f, 20.0f);
179         RNA_def_property_ui_text(prop, "Size X", "Size of element, use of components depends on element type");
180         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
181
182         prop= RNA_def_property(srna, "size_y", PROP_FLOAT, PROP_DISTANCE);
183         RNA_def_property_float_sdna(prop, NULL, "expy");
184         RNA_def_property_range(prop, 0.0f, 20.0f);
185         RNA_def_property_ui_text(prop, "Size Y", "Size of element, use of components depends on element type");
186         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
187
188         prop= RNA_def_property(srna, "size_z", PROP_FLOAT, PROP_DISTANCE);
189         RNA_def_property_float_sdna(prop, NULL, "expz");
190         RNA_def_property_range(prop, 0.0f, 20.0f);
191         RNA_def_property_ui_text(prop, "Size Z", "Size of element, use of components depends on element type");
192         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
193         
194         prop= RNA_def_property(srna, "stiffness", PROP_FLOAT, PROP_NONE);
195         RNA_def_property_float_sdna(prop, NULL, "s");
196         RNA_def_property_range(prop, 0.0f, 10.0f);
197         RNA_def_property_ui_text(prop, "Stiffness", "Stiffness defines how much of the element to fill");
198         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
199         
200         /* flags */
201         prop= RNA_def_property(srna, "use_negative", PROP_BOOLEAN, PROP_NONE);
202         RNA_def_property_boolean_sdna(prop, NULL, "flag", MB_NEGATIVE);
203         RNA_def_property_ui_text(prop, "Negative", "Set metaball as negative one");
204         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
205         
206         prop= RNA_def_property(srna, "hide", PROP_BOOLEAN, PROP_NONE);
207         RNA_def_property_boolean_sdna(prop, NULL, "flag", MB_HIDE);
208         RNA_def_property_ui_text(prop, "Hide", "Hide element");
209         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
210 }
211
212 /* mball.elements */
213 static void rna_def_metaball_elements(BlenderRNA *brna, PropertyRNA *cprop)
214 {
215         StructRNA *srna;
216         PropertyRNA *prop;
217
218         FunctionRNA *func;
219         PropertyRNA *parm;
220
221         RNA_def_property_srna(cprop, "MetaBallElements");
222         srna= RNA_def_struct(brna, "MetaBallElements", NULL);
223         RNA_def_struct_sdna(srna, "MetaBall");
224         RNA_def_struct_ui_text(srna, "Meta Elements", "Collection of metaball elements");
225
226         func= RNA_def_function(srna, "new", "rna_MetaBall_elements_new");
227         RNA_def_function_ui_description(func, "Add a new spline to the curve.");
228         parm= RNA_def_enum(func, "type", metaelem_type_items, MB_BALL, "", "type for the new meta element.");
229         parm= RNA_def_pointer(func, "element", "MetaElement", "", "The newly created metaelement.");
230         RNA_def_function_return(func, parm);
231
232         func= RNA_def_function(srna, "remove", "rna_MetaBall_elements_remove");
233         RNA_def_function_ui_description(func, "Remove a spline from a curve.");
234         RNA_def_function_flag(func, FUNC_USE_REPORTS);
235         parm= RNA_def_pointer(func, "element", "MetaElement", "", "The element to remove.");
236         RNA_def_property_flag(parm, PROP_REQUIRED|PROP_NEVER_NULL);
237
238         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
239         RNA_def_property_pointer_sdna(prop, NULL, "lastelem");
240         RNA_def_property_ui_text(prop, "Active Element", "Last selected element");
241 }
242
243 static void rna_def_metaball(BlenderRNA *brna)
244 {
245         StructRNA *srna;
246         PropertyRNA *prop;
247         static EnumPropertyItem prop_update_items[] = {
248                 {MB_UPDATE_ALWAYS, "UPDATE_ALWAYS", 0, "Always", "While editing, update metaball always"},
249                 {MB_UPDATE_HALFRES, "HALFRES", 0, "Half", "While editing, update metaball in half resolution"},
250                 {MB_UPDATE_FAST, "FAST", 0, "Fast", "While editing, update metaball without polygonization"},
251                 {MB_UPDATE_NEVER, "NEVER", 0, "Never", "While editing, don't update metaball at all"},
252                 {0, NULL, 0, NULL, NULL}};
253         
254         srna= RNA_def_struct(brna, "MetaBall", "ID");
255         RNA_def_struct_ui_text(srna, "MetaBall", "Metaball datablock to defined blobby surfaces");
256         RNA_def_struct_ui_icon(srna, ICON_META_DATA);
257
258         prop= RNA_def_property(srna, "elements", PROP_COLLECTION, PROP_NONE);
259         RNA_def_property_collection_sdna(prop, NULL, "elems", NULL);
260         RNA_def_property_struct_type(prop, "MetaElement");
261         RNA_def_property_ui_text(prop, "Elements", "Meta elements");
262         rna_def_metaball_elements(brna, prop);
263
264         /* enums */
265         prop= RNA_def_property(srna, "update_method", PROP_ENUM, PROP_NONE);
266         RNA_def_property_enum_sdna(prop, NULL, "flag");
267         RNA_def_property_enum_items(prop, prop_update_items);
268         RNA_def_property_ui_text(prop, "Update", "Metaball edit update behavior");
269         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
270         
271         /* number values */
272         prop= RNA_def_property(srna, "resolution", PROP_FLOAT, PROP_DISTANCE);
273         RNA_def_property_float_sdna(prop, NULL, "wiresize");
274         RNA_def_property_range(prop, 0.050f, 1.0f);
275         RNA_def_property_ui_text(prop, "Wire Size", "Polygonization resolution in the 3D viewport");
276         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
277         
278         prop= RNA_def_property(srna, "render_resolution", PROP_FLOAT, PROP_DISTANCE);
279         RNA_def_property_float_sdna(prop, NULL, "rendersize");
280         RNA_def_property_range(prop, 0.050f, 1.0f);
281         RNA_def_property_ui_text(prop, "Render Size", "Polygonization resolution in rendering");
282         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
283         
284         prop= RNA_def_property(srna, "threshold", PROP_FLOAT, PROP_NONE);
285         RNA_def_property_float_sdna(prop, NULL, "thresh");
286         RNA_def_property_range(prop, 0.0f, 5.0f);
287         RNA_def_property_ui_text(prop, "Threshold", "Influence of meta elements");
288         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
289
290         /* texture space */
291         prop= RNA_def_property(srna, "use_auto_texspace", PROP_BOOLEAN, PROP_NONE);
292         RNA_def_property_boolean_sdna(prop, NULL, "texflag", MB_AUTOSPACE);
293         RNA_def_property_ui_text(prop, "Auto Texture Space", "Adjusts active object's texture space automatically when transforming object");
294         
295         prop= RNA_def_property(srna, "texspace_location", PROP_FLOAT, PROP_TRANSLATION);
296         RNA_def_property_array(prop, 3);
297         RNA_def_property_ui_text(prop, "Texture Space Location", "Texture space location");
298         RNA_def_property_editable_func(prop, "rna_Meta_texspace_editable");
299         RNA_def_property_float_funcs(prop, "rna_Meta_texspace_loc_get", "rna_Meta_texspace_loc_set", NULL);     
300         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
301         
302         prop= RNA_def_property(srna, "texspace_size", PROP_FLOAT, PROP_XYZ);
303         RNA_def_property_array(prop, 3);
304         RNA_def_property_ui_text(prop, "Texture Space Size", "Texture space size");
305         RNA_def_property_editable_func(prop, "rna_Meta_texspace_editable");
306         RNA_def_property_float_funcs(prop, "rna_Meta_texspace_size_get", "rna_Meta_texspace_size_set", NULL);
307         RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");
308         
309         /* not supported yet
310          prop= RNA_def_property(srna, "texspace_rot", PROP_FLOAT, PROP_EULER);
311          RNA_def_property_float(prop, NULL, "rot");
312          RNA_def_property_ui_text(prop, "Texture Space Rotation", "Texture space rotation");
313          RNA_def_property_editable_func(prop, "rna_Meta_texspace_editable");
314          RNA_def_property_update(prop, 0, "rna_MetaBall_update_data");*/
315         
316         /* materials */
317         prop= RNA_def_property(srna, "materials", PROP_COLLECTION, PROP_NONE);
318         RNA_def_property_collection_sdna(prop, NULL, "mat", "totcol");
319         RNA_def_property_struct_type(prop, "Material");
320         RNA_def_property_ui_text(prop, "Materials", "");
321         RNA_def_property_srna(prop, "IDMaterials"); /* see rna_ID.c */  
322         
323         /* anim */
324         rna_def_animdata_common(srna);
325 }
326
327 void RNA_def_meta(BlenderRNA *brna)
328 {
329         rna_def_metaelement(brna);
330         rna_def_metaball(brna);
331 }
332
333 #endif