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