82911ebb3be97368ddc285533f8808458eddfe3b
[blender.git] / source / blender / makesrna / intern / rna_fluidsim.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).
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 /** \file blender/makesrna/intern/rna_fluidsim.c
26  *  \ingroup RNA
27  */
28
29
30 #include <stdlib.h>
31
32 #include "RNA_define.h"
33
34 #include "rna_internal.h"
35
36 #include "DNA_object_fluidsim.h"
37
38 #include "WM_api.h"
39 #include "WM_types.h"
40
41 #ifdef RNA_RUNTIME
42
43 #include "MEM_guardedalloc.h"
44
45 #include "DNA_scene_types.h"
46 #include "DNA_particle_types.h"
47
48 #include "BKE_depsgraph.h"
49 #include "BKE_fluidsim.h"
50 #include "BKE_global.h"
51 #include "BKE_main.h"
52 #include "BKE_modifier.h"
53 #include "BKE_particle.h"
54 #include "BKE_pointcache.h"
55
56 static StructRNA* rna_FluidSettings_refine(struct PointerRNA *ptr)
57 {
58         FluidsimSettings *fss= (FluidsimSettings*)ptr->data;
59
60         switch(fss->type) {
61                 case OB_FLUIDSIM_DOMAIN:
62                         return &RNA_DomainFluidSettings;
63                 case OB_FLUIDSIM_FLUID:
64                         return &RNA_FluidFluidSettings;
65                 case OB_FLUIDSIM_OBSTACLE:
66                         return &RNA_ObstacleFluidSettings;
67                 case OB_FLUIDSIM_INFLOW:
68                         return &RNA_InflowFluidSettings;
69                 case OB_FLUIDSIM_OUTFLOW:
70                         return &RNA_OutflowFluidSettings;
71                 case OB_FLUIDSIM_PARTICLE:
72                         return &RNA_ParticleFluidSettings;
73                 case OB_FLUIDSIM_CONTROL:
74                         return &RNA_ControlFluidSettings;
75                 default:
76                         return &RNA_FluidSettings;
77         }
78 }
79
80 static void rna_fluid_update(Main *bmain, Scene *scene, PointerRNA *ptr)
81 {
82         Object *ob= ptr->id.data;
83
84         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
85         WM_main_add_notifier(NC_OBJECT|ND_MODIFIER, ob);
86 }
87
88 static int fluidsim_find_lastframe(FluidsimSettings *fss)
89 {
90         char targetDir[FILE_MAXFILE+FILE_MAXDIR], targetFile[FILE_MAXFILE+FILE_MAXDIR];
91         int curFrame = 1;
92
93         BLI_snprintf(targetDir, sizeof(targetDir), "%sfluidsurface_final_####.bobj.gz", fss->surfdataPath);
94         BLI_path_abs(targetDir, G.main->name);
95
96         do {
97                 BLI_strncpy(targetFile, targetDir, sizeof(targetFile));
98                 BLI_path_frame(targetFile, curFrame++, 0);
99         } while(BLI_exist(targetFile));
100
101         return curFrame - 1;
102 }
103
104 static void rna_fluid_find_enframe(Main *bmain, Scene *scene, PointerRNA *ptr)
105 {
106         Object *ob= ptr->id.data;
107         FluidsimModifierData *fluidmd= (FluidsimModifierData*)modifiers_findByType(ob, eModifierType_Fluidsim);
108
109         if(fluidmd->fss->flag & OB_FLUIDSIM_REVERSE) {
110                 fluidmd->fss->lastgoodframe = fluidsim_find_lastframe(fluidmd->fss);
111         }
112         else {
113                 fluidmd->fss->lastgoodframe = -1;
114         }
115         rna_fluid_update(bmain, scene, ptr);
116 }
117
118 static void rna_FluidSettings_update_type(Main *bmain, Scene *scene, PointerRNA *ptr)
119 {
120         Object *ob= (Object*)ptr->id.data;
121         FluidsimModifierData *fluidmd;
122         ParticleSystemModifierData *psmd;
123         ParticleSystem *psys;
124         ParticleSettings *part;
125         
126         fluidmd= (FluidsimModifierData*)modifiers_findByType(ob, eModifierType_Fluidsim);
127         fluidmd->fss->flag &= ~OB_FLUIDSIM_REVERSE; // clear flag
128
129         /* remove fluidsim particle system */
130         if(fluidmd->fss->type & OB_FLUIDSIM_PARTICLE) {
131                 for(psys=ob->particlesystem.first; psys; psys=psys->next)
132                         if(psys->part->type == PART_FLUID)
133                                 break;
134
135                 if(ob->type == OB_MESH && !psys) {
136                         /* add particle system */
137                         part= psys_new_settings("ParticleSettings", bmain);
138                         psys= MEM_callocN(sizeof(ParticleSystem), "particle_system");
139
140                         part->type= PART_FLUID;
141                         psys->part= part;
142                         psys->pointcache= BKE_ptcache_add(&psys->ptcaches);
143                         psys->flag |= PSYS_ENABLED;
144                         sprintf(psys->name, "FluidParticles");
145                         BLI_addtail(&ob->particlesystem,psys);
146
147                         /* add modifier */
148                         psmd= (ParticleSystemModifierData*)modifier_new(eModifierType_ParticleSystem);
149                         sprintf(psmd->modifier.name, "FluidParticleSystem" );
150                         psmd->psys= psys;
151                         BLI_addtail(&ob->modifiers, psmd);
152                         modifier_unique_name(&ob->modifiers, (ModifierData *)psmd);
153                 }
154         }
155         else {
156                 for(psys=ob->particlesystem.first; psys; psys=psys->next) {
157                         if(psys->part->type == PART_FLUID) {
158                                 /* clear modifier */
159                                 psmd= psys_get_modifier(ob, psys);
160                                 BLI_remlink(&ob->modifiers, psmd);
161                                 modifier_free((ModifierData *)psmd);
162
163                                 /* clear particle system */
164                                 BLI_remlink(&ob->particlesystem, psys);
165                                 psys_free(ob, psys);
166                         }
167                 }
168         }
169
170         rna_fluid_update(bmain, scene, ptr);
171 }
172
173 static void rna_DomainFluidSettings_memory_estimate_get(PointerRNA *ptr, char *value)
174 {
175 #ifdef DISABLE_ELBEEM
176         value[0]= '\0';
177 #else
178         Object *ob= (Object*)ptr->id.data;
179         FluidsimSettings *fss= (FluidsimSettings*)ptr->data;
180
181         fluid_estimate_memory(ob, fss, value);
182 #endif
183 }
184
185 static int rna_DomainFluidSettings_memory_estimate_length(PointerRNA *ptr)
186 {
187 #ifdef DISABLE_ELBEEM
188         return 0;
189 #else
190         return 31;
191 #endif
192 }
193
194 static char *rna_FluidSettings_path(PointerRNA *ptr)
195 {
196         FluidsimSettings *fss = (FluidsimSettings*)ptr->data;
197         ModifierData *md= (ModifierData *)fss->fmd;
198
199         return BLI_sprintfN("modifiers[\"%s\"].settings", md->name);
200 }
201
202 static void rna_FluidMeshVertex_data_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
203 {
204         FluidsimSettings *fss = (FluidsimSettings*)ptr->data;
205         rna_iterator_array_begin(iter, fss->meshVelocities, sizeof(float)*3, fss->totvert, 0, NULL);
206 }
207
208 static int rna_FluidMeshVertex_data_length(PointerRNA *ptr)
209 {
210         FluidsimSettings *fss = (FluidsimSettings*)ptr->data;
211         return fss->totvert;
212 }
213
214 #else
215
216 static void rna_def_fluidsim_slip(StructRNA *srna)
217 {
218         PropertyRNA *prop;
219
220         static EnumPropertyItem slip_items[] = {
221                 {OB_FSBND_NOSLIP, "NOSLIP", 0, "No Slip", "Obstacle causes zero normal and tangential velocity (=sticky). Default for all. Only option for moving objects"},
222                 {OB_FSBND_PARTSLIP, "PARTIALSLIP", 0, "Partial Slip", "Mix between no-slip and free-slip. Non moving objects only!"},
223                 {OB_FSBND_FREESLIP, "FREESLIP", 0, "Free Slip", "Obstacle only causes zero normal velocity (=not sticky). Non moving objects only!"},
224                 {0, NULL, 0, NULL, NULL}};
225
226         prop= RNA_def_property(srna, "slip_type", PROP_ENUM, PROP_NONE);
227         RNA_def_property_enum_bitflag_sdna(prop, NULL, "typeFlags");
228         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
229         RNA_def_property_enum_items(prop, slip_items);
230         RNA_def_property_ui_text(prop, "Slip Type", "");
231
232         prop= RNA_def_property(srna, "partial_slip_factor", PROP_FLOAT, PROP_NONE);
233         RNA_def_property_float_sdna(prop, NULL, "partSlipValue");
234         RNA_def_property_range(prop, 0.0f, 1.0f);
235         RNA_def_property_ui_text(prop, "Partial Slip Amount", "Amount of mixing between no- and free-slip, 0 is no slip and 1 is free slip");
236 }
237
238 static void rna_def_fluid_mesh_vertices(BlenderRNA *brna)
239 {
240         StructRNA *srna;
241         PropertyRNA *prop;
242         
243         srna= RNA_def_struct(brna, "FluidMeshVertex", NULL);
244         RNA_def_struct_sdna(srna, "FluidVertexVelocity");
245         RNA_def_struct_ui_text(srna, "Fluid Mesh Vertex", "Vertex of a simulated fluid mesh");
246         RNA_def_struct_ui_icon(srna, ICON_VERTEXSEL);
247
248         prop= RNA_def_property(srna, "velocity", PROP_FLOAT, PROP_VELOCITY);
249         RNA_def_property_array(prop, 3);
250         RNA_def_property_float_sdna(prop, NULL, "vel");
251         RNA_def_property_ui_text(prop, "Velocity", "");
252         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
253 }
254
255
256 static void rna_def_fluidsim_domain(BlenderRNA *brna)
257 {
258         StructRNA *srna;
259         PropertyRNA *prop;
260
261         static EnumPropertyItem quality_items[] = {
262                 {OB_FSDOM_GEOM, "GEOMETRY", 0, "Geometry", "Display geometry"},
263                 {OB_FSDOM_PREVIEW, "PREVIEW", 0, "Preview", "Display preview quality results"},
264                 {OB_FSDOM_FINAL, "FINAL", 0, "Final", "Display final quality results"},
265                 {0, NULL, 0, NULL, NULL}};
266
267         static EnumPropertyItem viscosity_items[] = {
268                 {1, "MANUAL", 0, "Manual", "Manual viscosity settings"},
269                 {2, "WATER", 0, "Water", "Viscosity of 1.0 * 10^-6"},
270                 {3, "OIL", 0, "Oil", "Viscosity of 5.0 * 10^-5"},
271                 {4, "HONEY", 0, "Honey", "Viscosity of 2.0 * 10^-3"},
272                 {0, NULL, 0, NULL, NULL}};
273
274         srna= RNA_def_struct(brna, "DomainFluidSettings", "FluidSettings");
275         RNA_def_struct_sdna(srna, "FluidsimSettings");
276         RNA_def_struct_ui_text(srna, "Domain Fluid Simulation Settings", "Fluid simulation settings for the domain of a fluid simulation");
277
278         /* standard settings */
279
280         prop= RNA_def_property(srna, "resolution", PROP_INT, PROP_NONE);
281         RNA_def_property_int_sdna(prop, NULL, "resolutionxyz");
282         RNA_def_property_range(prop, 1, 1024);
283         RNA_def_property_ui_text(prop, "Resolution", "Domain resolution in X,Y and Z direction");
284
285         prop= RNA_def_property(srna, "preview_resolution", PROP_INT, PROP_NONE);
286         RNA_def_property_int_sdna(prop, NULL, "previewresxyz");
287         RNA_def_property_range(prop, 1, 100);
288         RNA_def_property_ui_text(prop, "Preview Resolution", "Preview resolution in X,Y and Z direction");
289
290         prop= RNA_def_property(srna, "viewport_display_mode", PROP_ENUM, PROP_NONE);
291         RNA_def_property_enum_sdna(prop, NULL, "guiDisplayMode");
292         RNA_def_property_enum_items(prop, quality_items);
293         RNA_def_property_ui_text(prop, "Viewport Display Mode", "How to display the mesh in the viewport");
294         RNA_def_property_update(prop, 0, "rna_fluid_update");
295
296         prop= RNA_def_property(srna, "render_display_mode", PROP_ENUM, PROP_NONE);
297         RNA_def_property_enum_sdna(prop, NULL, "renderDisplayMode");
298         RNA_def_property_enum_items(prop, quality_items);
299         RNA_def_property_ui_text(prop, "Render Display Mode", "How to display the mesh for rendering");
300
301         prop= RNA_def_property(srna, "use_reverse_frames", PROP_BOOLEAN, PROP_NONE);
302         RNA_def_property_boolean_sdna(prop, NULL, "flag", OB_FLUIDSIM_REVERSE);
303         RNA_def_property_ui_text(prop, "Reverse Frames", "Reverse fluid frames");
304         RNA_def_property_update(prop, 0, "rna_fluid_find_enframe");
305
306         prop= RNA_def_property(srna, "filepath", PROP_STRING, PROP_FILEPATH);
307         RNA_def_property_string_maxlength(prop, 240);
308         RNA_def_property_string_sdna(prop, NULL, "surfdataPath");
309         RNA_def_property_ui_text(prop, "Path", "Directory (and/or filename prefix) to store baked fluid simulation files in");
310         RNA_def_property_update(prop, 0, "rna_fluid_update");
311
312         prop= RNA_def_property(srna, "memory_estimate", PROP_STRING, PROP_NONE);
313         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
314         RNA_def_property_string_funcs(prop, "rna_DomainFluidSettings_memory_estimate_get", "rna_DomainFluidSettings_memory_estimate_length", NULL);
315         RNA_def_property_ui_text(prop, "Memory Estimate", "Estimated amount of memory needed for baking the domain");
316
317         /* advanced settings */
318         prop= RNA_def_property(srna, "gravity", PROP_FLOAT, PROP_ACCELERATION);
319         RNA_def_property_float_sdna(prop, NULL, "grav");
320         RNA_def_property_array(prop, 3);
321         RNA_def_property_range(prop, -1000.1, 1000.1);
322         RNA_def_property_ui_text(prop, "Gravity", "Gravity in X, Y and Z direction");
323         
324         prop= RNA_def_property(srna, "use_time_override", PROP_BOOLEAN, PROP_NONE);
325         RNA_def_property_boolean_sdna(prop, NULL, "flag", OB_FLUIDSIM_OVERRIDE_TIME);
326         RNA_def_property_ui_text(prop, "Override Time", "Use a custom start and end time (in seconds) instead of the scene's timeline");
327         
328         prop= RNA_def_property(srna, "start_time", PROP_FLOAT, PROP_TIME);
329         RNA_def_property_float_sdna(prop, NULL, "animStart");
330         RNA_def_property_range(prop, 0, 100);
331         RNA_def_property_ui_text(prop, "Start Time", "Simulation time of the first blender frame (in seconds)");
332         
333         prop= RNA_def_property(srna, "end_time", PROP_FLOAT, PROP_TIME);
334         RNA_def_property_float_sdna(prop, NULL, "animEnd");
335         RNA_def_property_range(prop, 0, 100);
336         RNA_def_property_ui_text(prop, "End Time", "Simulation time of the last blender frame (in seconds)");
337         
338         prop= RNA_def_property(srna, "simulation_scale", PROP_FLOAT, PROP_NONE);
339         RNA_def_property_float_sdna(prop, NULL, "realsize");
340         RNA_def_property_range(prop, 0.001, 10);
341         RNA_def_property_ui_text(prop, "Real World Size", "Size of the simulation domain in metres");
342         
343         prop= RNA_def_property(srna, "viscosity_preset", PROP_ENUM, PROP_NONE);
344         RNA_def_property_enum_sdna(prop, NULL, "viscosityMode");
345         RNA_def_property_enum_items(prop, viscosity_items);
346         RNA_def_property_ui_text(prop, "Viscosity Preset", "Set viscosity of the fluid to a preset value, or use manual input");
347
348         prop= RNA_def_property(srna, "viscosity_base", PROP_FLOAT, PROP_NONE);
349         RNA_def_property_float_sdna(prop, NULL, "viscosityValue");
350         RNA_def_property_range(prop, 0, 10);
351         RNA_def_property_ui_text(prop, "Viscosity Base", "Viscosity setting: value that is multiplied by 10 to the power of (exponent*-1)");
352
353         prop= RNA_def_property(srna, "viscosity_exponent", PROP_INT, PROP_NONE);
354         RNA_def_property_int_sdna(prop, NULL, "viscosityExponent");
355         RNA_def_property_range(prop, 0, 10);
356         RNA_def_property_ui_text(prop, "Viscosity Exponent", "Negative exponent for the viscosity value (to simplify entering small values e.g. 5*10^-6.)");
357
358         prop= RNA_def_property(srna, "grid_levels", PROP_INT, PROP_NONE);
359         RNA_def_property_int_sdna(prop, NULL, "maxRefine");
360         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
361         RNA_def_property_range(prop, -1, 4);
362         RNA_def_property_ui_text(prop, "Grid Levels", "Number of coarsened grids to use (-1 for automatic)");
363
364         prop= RNA_def_property(srna, "compressibility", PROP_FLOAT, PROP_NONE);
365         RNA_def_property_float_sdna(prop, NULL, "gstar");
366         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
367         RNA_def_property_range(prop, 0.001, 0.1);
368         RNA_def_property_ui_text(prop, "Compressibility", "Allowed compressibility due to gravitational force for standing fluid. (directly affects simulation step size)");
369
370         /* domain boundary settings */
371
372         rna_def_fluidsim_slip(srna);
373
374         prop= RNA_def_property(srna, "surface_smooth", PROP_FLOAT, PROP_NONE);
375         RNA_def_property_float_sdna(prop, NULL, "surfaceSmoothing");
376         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
377         RNA_def_property_range(prop, 0.0, 5.0);
378         RNA_def_property_ui_text(prop, "Surface Smoothing", "Amount of surface smoothing. A value of 0 is off, 1 is normal smoothing and more than 1 is extra smoothing");
379
380         prop= RNA_def_property(srna, "surface_subdivisions", PROP_INT, PROP_NONE);
381         RNA_def_property_int_sdna(prop, NULL, "surfaceSubdivs");
382         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
383         RNA_def_property_range(prop, 0, 5);
384         RNA_def_property_ui_text(prop, "Surface Subdivisions", "Number of isosurface subdivisions. This is necessary for the inclusion of particles into the surface generation. Warning - can lead to longer computation times!");
385
386         prop= RNA_def_property(srna, "use_speed_vectors", PROP_BOOLEAN, PROP_NONE);
387         RNA_def_property_boolean_negative_sdna(prop, NULL, "domainNovecgen", 0);
388         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
389         RNA_def_property_ui_text(prop, "Generate Speed Vectors", "Generate speed vectors for vector blur");
390
391         /* no collision object surface */
392         prop= RNA_def_property(srna, "surface_noobs", PROP_BOOLEAN, PROP_NONE);
393         RNA_def_property_boolean_sdna(prop, NULL, "typeFlags", OB_FSSG_NOOBS);
394         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
395         RNA_def_property_ui_text(prop, "Hide fluid surface", "");
396
397         /* particles */
398
399         prop= RNA_def_property(srna, "tracer_particles", PROP_INT, PROP_NONE);
400         RNA_def_property_int_sdna(prop, NULL, "generateTracers");
401         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
402         RNA_def_property_range(prop, 0, 10000);
403         RNA_def_property_ui_text(prop, "Tracer Particles", "Number of tracer particles to generate");
404
405         prop= RNA_def_property(srna, "generate_particles", PROP_FLOAT, PROP_NONE);
406         RNA_def_property_float_sdna(prop, NULL, "generateParticles");
407         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
408         RNA_def_property_range(prop, 0.0, 10.0);
409         RNA_def_property_ui_text(prop, "Generate Particles", "Amount of particles to generate (0=off, 1=normal, >1=more)");
410         
411         /* simulated fluid mesh data */
412         prop= RNA_def_property(srna, "fluid_mesh_vertices", PROP_COLLECTION, PROP_NONE);
413         RNA_def_property_struct_type(prop, "FluidMeshVertex");
414         RNA_def_property_ui_text(prop, "Fluid Mesh Vertices", "Vertices of the fluid mesh generated by simulation");
415         RNA_def_property_collection_funcs(prop, "rna_FluidMeshVertex_data_begin", "rna_iterator_array_next", "rna_iterator_array_end", "rna_iterator_array_get", "rna_FluidMeshVertex_data_length", NULL, NULL, NULL);
416         rna_def_fluid_mesh_vertices(brna);
417 }
418
419 static void rna_def_fluidsim_volume(StructRNA *srna)
420 {
421         PropertyRNA *prop;
422
423         static EnumPropertyItem volume_type_items[] = {
424                 {1, "VOLUME", 0, "Volume", "Use only the inner volume of the mesh"},
425                 {2, "SHELL", 0, "Shell", "Use only the outer shell of the mesh"},
426                 {3, "BOTH", 0, "Both", "Use both the inner volume and the outer shell of the mesh"},
427                 {0, NULL, 0, NULL, NULL}};
428
429         prop= RNA_def_property(srna, "volume_initialization", PROP_ENUM, PROP_NONE);
430         RNA_def_property_enum_bitflag_sdna(prop, NULL, "volumeInitType");
431         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
432         RNA_def_property_enum_items(prop, volume_type_items);
433         RNA_def_property_ui_text(prop, "Volume Initialization", "Volume initialization type");
434
435         prop= RNA_def_property(srna, "use_animated_mesh", PROP_BOOLEAN, PROP_NONE);
436         RNA_def_property_boolean_sdna(prop, NULL, "domainNovecgen", 0);
437         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
438         RNA_def_property_ui_text(prop, "Export Animated Mesh", "Export this mesh as an animated one. Slower, only use if really necessary (e.g. armatures or parented objects), animated pos/rot/scale IPOs do not require it");
439 }
440
441 static void rna_def_fluidsim_active(StructRNA *srna)
442 {
443         PropertyRNA *prop;
444         
445         prop= RNA_def_property(srna, "use", PROP_BOOLEAN, PROP_NONE);
446         RNA_def_property_boolean_sdna(prop, NULL, "flag", OB_FLUIDSIM_ACTIVE);
447         RNA_def_property_ui_text(prop, "Enabled", "Object contributes to the fluid simulation");
448 }
449
450 static void rna_def_fluidsim_fluid(BlenderRNA *brna)
451 {
452         StructRNA *srna;
453         PropertyRNA *prop;
454
455         srna= RNA_def_struct(brna, "FluidFluidSettings", "FluidSettings");
456         RNA_def_struct_sdna(srna, "FluidsimSettings");
457         RNA_def_struct_ui_text(srna, "Fluid Fluid Simulation Settings", "Fluid simulation settings for the fluid in the simulation");
458
459         rna_def_fluidsim_active(srna);
460         rna_def_fluidsim_volume(srna);
461         
462         prop= RNA_def_property(srna, "initial_velocity", PROP_FLOAT, PROP_VELOCITY);
463         RNA_def_property_float_sdna(prop, NULL, "iniVelx");
464         RNA_def_property_array(prop, 3);
465         RNA_def_property_range(prop, -1000.1, 1000.1);
466         RNA_def_property_ui_text(prop, "Initial Velocity", "Initial velocity of fluid");
467 }
468
469 static void rna_def_fluidsim_obstacle(BlenderRNA *brna)
470 {
471         StructRNA *srna;
472         PropertyRNA *prop;
473
474         srna= RNA_def_struct(brna, "ObstacleFluidSettings", "FluidSettings");
475         RNA_def_struct_sdna(srna, "FluidsimSettings");
476         RNA_def_struct_ui_text(srna, "Obstacle Fluid Simulation Settings", "Fluid simulation settings for obstacles in the simulation");
477
478         rna_def_fluidsim_active(srna);
479         rna_def_fluidsim_volume(srna);
480         rna_def_fluidsim_slip(srna);
481
482         prop= RNA_def_property(srna, "impact_factor", PROP_FLOAT, PROP_NONE);
483         RNA_def_property_float_sdna(prop, NULL, "surfaceSmoothing");
484         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
485         RNA_def_property_range(prop, -2.0, 10.0);
486         RNA_def_property_ui_text(prop, "Impact Factor", "This is an unphysical value for moving objects - it controls the impact an obstacle has on the fluid, =0 behaves a bit like outflow (deleting fluid), =1 is default, while >1 results in high forces. Can be used to tweak total mass");
487 }
488
489 static void rna_def_fluidsim_inflow(BlenderRNA *brna)
490 {
491         StructRNA *srna;
492         PropertyRNA *prop;
493
494         srna= RNA_def_struct(brna, "InflowFluidSettings", "FluidSettings");
495         RNA_def_struct_sdna(srna, "FluidsimSettings");
496         RNA_def_struct_ui_text(srna, "Inflow Fluid Simulation Settings", "Fluid simulation settings for objects adding fluids in the simulation");
497
498         rna_def_fluidsim_active(srna);
499         rna_def_fluidsim_volume(srna);
500
501         prop= RNA_def_property(srna, "inflow_velocity", PROP_FLOAT, PROP_VELOCITY);
502         RNA_def_property_float_sdna(prop, NULL, "iniVelx");
503         RNA_def_property_array(prop, 3);
504         RNA_def_property_range(prop, -1000.1, 1000.1);
505         RNA_def_property_ui_text(prop, "Inflow Velocity", "Initial velocity of fluid");
506
507         prop= RNA_def_property(srna, "use_local_coords", PROP_BOOLEAN, PROP_NONE);
508         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
509         RNA_def_property_boolean_sdna(prop, NULL, "typeFlags", OB_FSINFLOW_LOCALCOORD);
510         RNA_def_property_ui_text(prop, "Local Coordinates", "Use local coordinates for inflow. (e.g. for rotating objects)");
511 }
512
513 static void rna_def_fluidsim_outflow(BlenderRNA *brna)
514 {
515         StructRNA *srna;
516
517         srna= RNA_def_struct(brna, "OutflowFluidSettings", "FluidSettings");
518         RNA_def_struct_sdna(srna, "FluidsimSettings");
519         RNA_def_struct_ui_text(srna, "Outflow Fluid Simulation Settings", "Fluid simulation settings for objects removing fluids from the simulation");
520
521         rna_def_fluidsim_active(srna);
522         rna_def_fluidsim_volume(srna);
523 }
524
525 static void rna_def_fluidsim_particle(BlenderRNA *brna)
526 {
527         StructRNA *srna;
528         PropertyRNA *prop;
529
530         srna= RNA_def_struct(brna, "ParticleFluidSettings", "FluidSettings");
531         RNA_def_struct_sdna(srna, "FluidsimSettings");
532         RNA_def_struct_ui_text(srna, "Particle Fluid Simulation Settings", "Fluid simulation settings for objects storing fluid particles generated by the simulation");
533
534         prop= RNA_def_property(srna, "use_drops", PROP_BOOLEAN, PROP_NONE);
535         RNA_def_property_boolean_sdna(prop, NULL, "typeFlags", OB_FSPART_DROP);
536         RNA_def_property_ui_text(prop, "Drops", "Show drop particles");
537
538         prop= RNA_def_property(srna, "use_floats", PROP_BOOLEAN, PROP_NONE);
539         RNA_def_property_boolean_sdna(prop, NULL, "typeFlags", OB_FSPART_FLOAT);
540         RNA_def_property_ui_text(prop, "Floats", "Show floating foam particles");
541
542         prop= RNA_def_property(srna, "show_tracer", PROP_BOOLEAN, PROP_NONE);
543         RNA_def_property_boolean_sdna(prop, NULL, "typeFlags", OB_FSPART_TRACER);
544         RNA_def_property_ui_text(prop, "Tracer", "Show tracer particles");
545
546         prop= RNA_def_property(srna, "particle_influence", PROP_FLOAT, PROP_NONE);
547         RNA_def_property_float_sdna(prop, NULL, "particleInfSize");
548         RNA_def_property_range(prop, 0.0, 2.0);
549         RNA_def_property_ui_text(prop, "Particle Influence", "Amount of particle size scaling: 0=off (all same size), 1=full (range 0.2-2.0), >1=stronger");
550
551         prop= RNA_def_property(srna, "alpha_influence", PROP_FLOAT, PROP_NONE);
552         RNA_def_property_float_sdna(prop, NULL, "particleInfAlpha");
553         RNA_def_property_range(prop, 0.0, 2.0);
554         RNA_def_property_ui_text(prop, "Alpha Influence", "Amount of particle alpha change, inverse of size influence: 0=off (all same alpha), 1=full. (large particles get lower alphas, smaller ones higher values)");
555
556         prop= RNA_def_property(srna, "filepath", PROP_STRING, PROP_FILEPATH);
557         RNA_def_property_string_maxlength(prop, 240);
558         RNA_def_property_string_sdna(prop, NULL, "surfdataPath");
559         RNA_def_property_ui_text(prop, "Path", "Directory (and/or filename prefix) to store and load particles from");
560         RNA_def_property_update(prop, 0, "rna_fluid_update");
561 }
562
563 static void rna_def_fluidsim_control(BlenderRNA *brna)
564 {
565         StructRNA *srna;
566         PropertyRNA *prop;
567
568         srna= RNA_def_struct(brna, "ControlFluidSettings", "FluidSettings");
569         RNA_def_struct_sdna(srna, "FluidsimSettings");
570         RNA_def_struct_ui_text(srna, "Control Fluid Simulation Settings", "Fluid simulation settings for objects controlling the motion of fluid in the simulation");
571
572         rna_def_fluidsim_active(srna);
573         
574         prop= RNA_def_property(srna, "start_time", PROP_FLOAT, PROP_TIME);
575         RNA_def_property_float_sdna(prop, NULL, "cpsTimeStart");
576         RNA_def_property_range(prop, 0.0, 100.0);
577         RNA_def_property_ui_text(prop, "Start Time", "Specifies time when the control particles are activated");
578         
579         prop= RNA_def_property(srna, "end_time", PROP_FLOAT, PROP_TIME);
580         RNA_def_property_float_sdna(prop, NULL, "cpsTimeEnd");
581         RNA_def_property_range(prop, 0.0, 100.0);
582         RNA_def_property_ui_text(prop, "End Time", "Specifies time when the control particles are deactivated");
583
584         prop= RNA_def_property(srna, "attraction_strength", PROP_FLOAT, PROP_NONE);
585         RNA_def_property_float_sdna(prop, NULL, "attractforceStrength");
586         RNA_def_property_range(prop, -10.0, 10.0);
587         RNA_def_property_ui_text(prop, "Attraction Strength", "Force strength for directional attraction towards the control object");
588
589         prop= RNA_def_property(srna, "attraction_radius", PROP_FLOAT, PROP_NONE);
590         RNA_def_property_float_sdna(prop, NULL, "attractforceRadius");
591         RNA_def_property_range(prop, 0.0, 10.0);
592         RNA_def_property_ui_text(prop, "Attraction Radius", "Specifies the force field radius around the control object");
593         
594         prop= RNA_def_property(srna, "velocity_strength", PROP_FLOAT, PROP_NONE);
595         RNA_def_property_float_sdna(prop, NULL, "velocityforceStrength");
596         RNA_def_property_range(prop, 0.0, 10.0);
597         RNA_def_property_ui_text(prop, "Velocity Strength", "Force strength of how much of the control object's velocity is influencing the fluid velocity");
598
599         prop= RNA_def_property(srna, "velocity_radius", PROP_FLOAT, PROP_NONE);
600         RNA_def_property_float_sdna(prop, NULL, "velocityforceRadius");
601         RNA_def_property_range(prop, 0.0, 10.0);
602         RNA_def_property_ui_text(prop, "Velocity Radius", "Specifies the force field radius around the control object");
603
604         prop= RNA_def_property(srna, "quality", PROP_FLOAT, PROP_NONE);
605         RNA_def_property_float_sdna(prop, NULL, "cpsQuality");
606         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
607         RNA_def_property_range(prop, 5.0, 100.0);
608         RNA_def_property_ui_text(prop, "Quality", "Specifies the quality which is used for object sampling. (higher = better but slower)");
609
610         prop= RNA_def_property(srna, "use_reverse_frames", PROP_BOOLEAN, PROP_NONE);
611         RNA_def_property_boolean_sdna(prop, NULL, "flag", OB_FLUIDSIM_REVERSE);
612         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
613         RNA_def_property_ui_text(prop, "Reverse Frames", "Reverse control object movement");
614         RNA_def_property_update(prop, 0, "rna_fluid_find_enframe");
615 }
616
617 void RNA_def_fluidsim(BlenderRNA *brna)
618 {
619         StructRNA *srna;
620         PropertyRNA *prop;
621
622         static EnumPropertyItem prop_fluid_type_items[] = {
623                 {OB_FLUIDSIM_ENABLE, "NONE", 0, "None", ""},
624                 {OB_FLUIDSIM_DOMAIN, "DOMAIN", 0, "Domain", "Bounding box of this object represents the computational domain of the fluid simulation"},
625                 {OB_FLUIDSIM_FLUID, "FLUID", 0, "Fluid", "Object represents a volume of fluid in the simulation"},
626                 {OB_FLUIDSIM_OBSTACLE, "OBSTACLE", 0, "Obstacle", "Object is a fixed obstacle"},
627                 {OB_FLUIDSIM_INFLOW, "INFLOW", 0, "Inflow", "Object adds fluid to the simulation"},
628                 {OB_FLUIDSIM_OUTFLOW, "OUTFLOW", 0, "Outflow", "Object removes fluid from the simulation"},
629                 {OB_FLUIDSIM_PARTICLE, "PARTICLE", 0, "Particle", "Object is made a particle system to display particles generated by a fluidsim domain object"},
630                 {OB_FLUIDSIM_CONTROL, "CONTROL", 0, "Control", "Object is made a fluid control mesh, which influences the fluid"},
631                 {0, NULL, 0, NULL, NULL}};
632
633
634         srna= RNA_def_struct(brna, "FluidSettings", NULL);
635         RNA_def_struct_sdna(srna, "FluidsimSettings");
636         RNA_def_struct_refine_func(srna, "rna_FluidSettings_refine");
637         RNA_def_struct_path_func(srna, "rna_FluidSettings_path");
638         RNA_def_struct_ui_text(srna, "Fluid Simulation Settings", "Fluid simulation settings for an object taking part in the simulation");
639
640         prop= RNA_def_property(srna, "type", PROP_ENUM, PROP_NONE);
641         RNA_def_property_enum_bitflag_sdna(prop, NULL, "type");
642         RNA_def_property_enum_items(prop, prop_fluid_type_items);
643         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
644         RNA_def_property_ui_text(prop, "Type", "Type of participation in the fluid simulation");
645         RNA_def_property_update(prop, 0, "rna_FluidSettings_update_type");
646
647         //prop= RNA_def_property(srna, "ipo", PROP_POINTER, PROP_NONE);
648         //RNA_def_property_ui_text(prop, "IPO Curves", "IPO curves used by fluid simulation settings");
649
650         /* types */
651
652         rna_def_fluidsim_domain(brna);
653         rna_def_fluidsim_fluid(brna);
654         rna_def_fluidsim_obstacle(brna);
655         rna_def_fluidsim_inflow(brna);
656         rna_def_fluidsim_outflow(brna);
657         rna_def_fluidsim_particle(brna);
658         rna_def_fluidsim_control(brna);
659 }
660
661
662 #endif