Merged changes in the trunk up to revision 54802.
[blender.git] / release / scripts / startup / bl_ui / properties_physics_common.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21 import bpy
22 from bpy.types import Panel
23
24 i18n_default_ctxt = bpy.app.translations.contexts.default
25
26
27 class PhysicButtonsPanel():
28     bl_space_type = 'PROPERTIES'
29     bl_region_type = 'WINDOW'
30     bl_context = "physics"
31
32     @classmethod
33     def poll(cls, context):
34         rd = context.scene.render
35         return (context.object) and (not rd.use_game_engine)
36
37
38 def physics_add(self, layout, md, name, type, typeicon, toggles):
39     sub = layout.row(align=True)
40     if md:
41         sub.context_pointer_set("modifier", md)
42         sub.operator("object.modifier_remove", text=name, text_ctxt=i18n_default_ctxt, icon='X')
43         if(toggles):
44             sub.prop(md, "show_render", text="")
45             sub.prop(md, "show_viewport", text="")
46     else:
47         sub.operator("object.modifier_add", text=name, text_ctxt=i18n_default_ctxt, icon=typeicon).type = type
48
49
50 def physics_add_special(self, layout, data, name, addop, removeop, typeicon):
51     sub = layout.row(align=True)
52     if data:
53         sub.operator(removeop, text=name, text_ctxt=i18n_default_ctxt, icon='X')
54     else:
55         sub.operator(addop, text=name, text_ctxt=i18n_default_ctxt, icon=typeicon)
56
57
58 class PHYSICS_PT_add(PhysicButtonsPanel, Panel):
59     bl_label = ""
60     bl_options = {'HIDE_HEADER'}
61
62     def draw(self, context):
63         ob = context.object
64
65         layout = self.layout
66         layout.label("Enable physics for:")
67         split = layout.split()
68         col = split.column()
69
70         if(context.object.field.type == 'NONE'):
71             col.operator("object.forcefield_toggle", text="Force Field", icon='FORCE_FORCE')
72         else:
73             col.operator("object.forcefield_toggle", text="Force Field", icon='X')
74
75         if(ob.type == 'MESH'):
76             physics_add(self, col, context.collision, "Collision", 'COLLISION', 'MOD_PHYSICS', False)
77             physics_add(self, col, context.cloth, "Cloth", 'CLOTH', 'MOD_CLOTH', True)
78             physics_add(self, col, context.dynamic_paint, "Dynamic Paint", 'DYNAMIC_PAINT', 'MOD_DYNAMICPAINT', True)
79
80         col = split.column()
81
82         if(ob.type == 'MESH' or ob.type == 'LATTICE'or ob.type == 'CURVE'):
83             physics_add(self, col, context.soft_body, "Soft Body", 'SOFT_BODY', 'MOD_SOFT', True)
84
85         if(ob.type == 'MESH'):
86             physics_add(self, col, context.fluid, "Fluid", 'FLUID_SIMULATION', 'MOD_FLUIDSIM', True)
87             physics_add(self, col, context.smoke, "Smoke", 'SMOKE', 'MOD_SMOKE', True)
88
89         if(ob.type == 'MESH'):
90             physics_add_special(self, col, ob.rigid_body, "Rigid Body",
91                                 "rigidbody.object_add",
92                                 "rigidbody.object_remove",
93                                 'MESH_ICOSPHERE')  # XXX: need dedicated icon
94
95         # all types of objects can have rigid body constraint
96         physics_add_special(self, col, ob.rigid_body_constraint, "Rigid Body Constraint",
97                             "rigidbody.constraint_add",
98                             "rigidbody.constraint_remove",
99                             'CONSTRAINT')  # RB_TODO needs better icon
100
101
102 # cache-type can be 'PSYS' 'HAIR' 'SMOKE' etc
103
104 def point_cache_ui(self, context, cache, enabled, cachetype):
105     layout = self.layout
106
107     layout.context_pointer_set("point_cache", cache)
108
109     if not cachetype == 'RIGID_BODY':
110         row = layout.row()
111         row.template_list("UI_UL_list", "point_caches", cache, "point_caches",
112                           cache.point_caches, "active_index", rows=2)
113         col = row.column(align=True)
114         col.operator("ptcache.add", icon='ZOOMIN', text="")
115         col.operator("ptcache.remove", icon='ZOOMOUT', text="")
116
117     row = layout.row()
118     if cachetype in {'PSYS', 'HAIR', 'SMOKE'}:
119         row.prop(cache, "use_external")
120
121         if cachetype == 'SMOKE':
122             row.prop(cache, "use_library_path", "Use Lib Path")
123
124     if cache.use_external:
125         split = layout.split(percentage=0.35)
126         col = split.column()
127         col.label(text="File Name:")
128         if cache.use_external:
129             col.label(text="File Path:")
130
131         col = split.column()
132         sub = col.split(percentage=0.70, align=True)
133         sub.prop(cache, "name", text="")
134         sub.prop(cache, "index", text="")
135         col.prop(cache, "filepath", text="")
136
137         cache_info = cache.info
138         if cache_info:
139             layout.label(text=cache_info)
140     else:
141         if cachetype in {'SMOKE', 'DYNAMIC_PAINT'}:
142             if not bpy.data.is_saved:
143                 layout.label(text="Cache is disabled until the file is saved")
144                 layout.enabled = False
145
146         if cache.use_disk_cache:
147             layout.prop(cache, "name", text="File Name")
148         else:
149             layout.prop(cache, "name", text="Cache Name")
150
151     if not cache.use_external or cachetype == 'SMOKE':
152         row = layout.row(align=True)
153
154         if cachetype not in {'PSYS', 'DYNAMIC_PAINT'}:
155             row.enabled = enabled
156             row.prop(cache, "frame_start")
157             row.prop(cache, "frame_end")
158         if cachetype not in {'SMOKE', 'CLOTH', 'DYNAMIC_PAINT', 'RIGID_BODY'}:
159             row.prop(cache, "frame_step")
160
161         if cachetype != 'SMOKE':
162             layout.label(text=cache.info)
163
164         if cachetype not in {'SMOKE', 'DYNAMIC_PAINT', 'RIGID_BODY'}:
165             split = layout.split()
166             split.enabled = enabled and bpy.data.is_saved
167
168             col = split.column()
169             col.prop(cache, "use_disk_cache")
170
171             col = split.column()
172             col.active = cache.use_disk_cache
173             col.prop(cache, "use_library_path", "Use Lib Path")
174
175             row = layout.row()
176             row.enabled = enabled and bpy.data.is_saved
177             row.active = cache.use_disk_cache
178             row.label(text="Compression:")
179             row.prop(cache, "compression", expand=True)
180
181         layout.separator()
182
183         split = layout.split()
184
185         col = split.column()
186
187         if cache.is_baked is True:
188             col.operator("ptcache.free_bake", text="Free Bake")
189         else:
190             col.operator("ptcache.bake", text="Bake").bake = True
191
192         sub = col.row()
193         sub.enabled = (cache.frames_skipped or cache.is_outdated) and enabled
194         sub.operator("ptcache.bake", text="Calculate To Frame").bake = False
195
196         sub = col.column()
197         sub.enabled = enabled
198         sub.operator("ptcache.bake_from_cache", text="Current Cache to Bake")
199
200         col = split.column()
201         col.operator("ptcache.bake_all", text="Bake All Dynamics").bake = True
202         col.operator("ptcache.free_bake_all", text="Free All Bakes")
203         col.operator("ptcache.bake_all", text="Update All To Frame").bake = False
204
205
206 def effector_weights_ui(self, context, weights, weight_type):
207     layout = self.layout
208
209     layout.prop(weights, "group")
210
211     split = layout.split()
212
213     split.prop(weights, "gravity", slider=True)
214     split.prop(weights, "all", slider=True)
215
216     layout.separator()
217
218     split = layout.split()
219
220     col = split.column()
221     col.prop(weights, "force", slider=True)
222     col.prop(weights, "vortex", slider=True)
223     col.prop(weights, "magnetic", slider=True)
224     col.prop(weights, "wind", slider=True)
225     col.prop(weights, "curve_guide", slider=True)
226     col.prop(weights, "texture", slider=True)
227     if weight_type != 'SMOKE':
228         col.prop(weights, "smokeflow", slider=True)
229
230     col = split.column()
231     col.prop(weights, "harmonic", slider=True)
232     col.prop(weights, "charge", slider=True)
233     col.prop(weights, "lennardjones", slider=True)
234     col.prop(weights, "turbulence", slider=True)
235     col.prop(weights, "drag", slider=True)
236     col.prop(weights, "boid", slider=True)
237
238
239 def basic_force_field_settings_ui(self, context, field):
240     layout = self.layout
241
242     split = layout.split()
243
244     if not field or field.type == 'NONE':
245         return
246
247     col = split.column()
248
249     if field.type == 'DRAG':
250         col.prop(field, "linear_drag", text="Linear")
251     else:
252         col.prop(field, "strength")
253
254     if field.type == 'TURBULENCE':
255         col.prop(field, "size")
256         col.prop(field, "flow")
257     elif field.type == 'HARMONIC':
258         col.prop(field, "harmonic_damping", text="Damping")
259         col.prop(field, "rest_length")
260     elif field.type == 'VORTEX' and field.shape != 'POINT':
261         col.prop(field, "inflow")
262     elif field.type == 'DRAG':
263         col.prop(field, "quadratic_drag", text="Quadratic")
264     else:
265         col.prop(field, "flow")
266
267     col = split.column()
268     sub = col.column(align=True)
269     sub.prop(field, "noise")
270     sub.prop(field, "seed")
271     if field.type == 'TURBULENCE':
272         col.prop(field, "use_global_coords", text="Global")
273     elif field.type == 'HARMONIC':
274         col.prop(field, "use_multiple_springs")
275
276     split = layout.split()
277
278     col = split.column()
279     col.label(text="Effect point:")
280     col.prop(field, "apply_to_location")
281     col.prop(field, "apply_to_rotation")
282
283     col = split.column()
284     col.label(text="Collision:")
285     col.prop(field, "use_absorption")
286
287
288 def basic_force_field_falloff_ui(self, context, field):
289     layout = self.layout
290
291     split = layout.split(percentage=0.35)
292
293     if not field or field.type == 'NONE':
294         return
295
296     col = split.column()
297     col.prop(field, "z_direction", text="")
298
299     col = split.column()
300     col.prop(field, "falloff_power", text="Power")
301
302     split = layout.split()
303     col = split.column()
304     row = col.row(align=True)
305     row.prop(field, "use_min_distance", text="")
306     sub = row.row()
307     sub.active = field.use_min_distance
308     sub.prop(field, "distance_min", text="Minimum")
309
310     col = split.column()
311     row = col.row(align=True)
312     row.prop(field, "use_max_distance", text="")
313     sub = row.row()
314     sub.active = field.use_max_distance
315     sub.prop(field, "distance_max", text="Maximum")
316
317 if __name__ == "__main__":  # only for live edit.
318     bpy.utils.register_module(__name__)