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