7ca97c204223d9e4ad4afce75f2a65f59521c64a
[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 from blf import gettext as _
24
25
26 class PhysicButtonsPanel():
27     bl_space_type = 'PROPERTIES'
28     bl_region_type = 'WINDOW'
29     bl_context = "physics"
30
31     @classmethod
32     def poll(cls, context):
33         rd = context.scene.render
34         return (context.object) and (not rd.use_game_engine)
35
36
37 def physics_add(self, layout, md, name, type, typeicon, toggles):
38     sub = layout.row(align=True)
39     if md:
40         sub.context_pointer_set("modifier", md)
41         sub.operator("object.modifier_remove", text=name, icon='X')
42         if(toggles):
43             sub.prop(md, "show_render", text="")
44             sub.prop(md, "show_viewport", text="")
45     else:
46         sub.operator("object.modifier_add", text=name, icon=typeicon).type = type
47
48
49 class PHYSICS_PT_add(PhysicButtonsPanel, Panel):
50     bl_label = ""
51     bl_options = {'HIDE_HEADER'}
52
53     def draw(self, context):
54         ob = context.object
55
56         layout = self.layout
57         layout.label(_("Enable physics for:"))
58         split = layout.split()
59         col = split.column()
60
61         if(context.object.field.type == 'NONE'):
62             col.operator("object.forcefield_toggle", text=_("Force Field"), icon='FORCE_FORCE')
63         else:
64             col.operator("object.forcefield_toggle", text=_("Force Field"), icon='X')
65
66         if(ob.type == 'MESH'):
67             physics_add(self, col, context.collision, _("Collision"), 'COLLISION', 'MOD_PHYSICS', False)
68             physics_add(self, col, context.cloth, _("Cloth"), 'CLOTH', 'MOD_CLOTH', 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 #cachetype 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 cache.use_external:
98         split = layout.split(percentage=0.80)
99         split.prop(cache, "name", text=_("File Name"))
100         split.prop(cache, "index", text="")
101
102         row = layout.row()
103         row.label(text=_("File Path:"))
104         row.prop(cache, "use_library_path", "Use Lib Path")
105
106         layout.prop(cache, "filepath", text="")
107
108         layout.label(text=cache.info)
109     else:
110         if cachetype == 'SMOKE':
111             if not bpy.data.is_saved:
112                 layout.label(text=_("Cache is disabled until the file is saved"))
113                 layout.enabled = False
114
115         if cache.use_disk_cache:
116             layout.prop(cache, "name", text=_("File Name"))
117         else:
118             layout.prop(cache, "name", text=_("Cache Name"))
119
120         row = layout.row(align=True)
121
122         if cachetype != 'PSYS':
123             row.enabled = enabled
124             row.prop(cache, "frame_start")
125             row.prop(cache, "frame_end")
126         if cachetype not in {'SMOKE', 'CLOTH'}:
127             row.prop(cache, "frame_step")
128             row.prop(cache, "use_quick_cache")
129         if cachetype != 'SMOKE':
130             layout.label(text=cache.info)
131
132         if cachetype != 'SMOKE':
133             split = layout.split()
134             split.enabled = enabled and bpy.data.is_saved
135
136             col = split.column()
137             col.prop(cache, "use_disk_cache")
138
139             col = split.column()
140             col.active = cache.use_disk_cache
141             col.prop(cache, "use_library_path", "Use Lib Path")
142
143             row = layout.row()
144             row.enabled = enabled and bpy.data.is_saved
145             row.active = cache.use_disk_cache
146             row.label(text=_("Compression:"))
147             row.prop(cache, "compression", expand=True)
148
149         layout.separator()
150
151         split = layout.split()
152
153         col = split.column()
154
155         if cache.is_baked == True:
156             col.operator("ptcache.free_bake", text=_("Free Bake"))
157         else:
158             col.operator("ptcache.bake", text=_("Bake")).bake = True
159
160         sub = col.row()
161         sub.enabled = (cache.frames_skipped or cache.is_outdated) and enabled
162         sub.operator("ptcache.bake", text=_("Calculate To Frame")).bake = False
163
164         sub = col.column()
165         sub.enabled = enabled
166         sub.operator("ptcache.bake_from_cache", text=_("Current Cache to Bake"))
167
168         col = split.column()
169         col.operator("ptcache.bake_all", text=_("Bake All Dynamics")).bake = True
170         col.operator("ptcache.free_bake_all", text=_("Free All Bakes"))
171         col.operator("ptcache.bake_all", text=_("Update All To Frame")).bake = False
172
173
174 def effector_weights_ui(self, context, weights):
175     layout = self.layout
176
177     layout.prop(weights, "group")
178
179     split = layout.split()
180
181     col = split.column()
182     col.prop(weights, "gravity", slider=True)
183
184     col = split.column()
185     col.prop(weights, "all", slider=True)
186
187     layout.separator()
188
189     split = layout.split()
190
191     col = split.column()
192     col.prop(weights, "force", slider=True)
193     col.prop(weights, "vortex", slider=True)
194     col.prop(weights, "magnetic", slider=True)
195     col.prop(weights, "wind", slider=True)
196     col.prop(weights, "curve_guide", slider=True)
197     col.prop(weights, "texture", slider=True)
198
199     col = split.column()
200     col.prop(weights, "harmonic", slider=True)
201     col.prop(weights, "charge", slider=True)
202     col.prop(weights, "lennardjones", slider=True)
203     col.prop(weights, "turbulence", slider=True)
204     col.prop(weights, "drag", slider=True)
205     col.prop(weights, "boid", slider=True)
206
207
208 def basic_force_field_settings_ui(self, context, field):
209     layout = self.layout
210
211     split = layout.split()
212
213     if not field or field.type == 'NONE':
214         return
215
216     col = split.column()
217
218     if field.type == 'DRAG':
219         col.prop(field, "linear_drag", text=_("Linear"))
220     else:
221         col.prop(field, "strength")
222
223     if field.type == 'TURBULENCE':
224         col.prop(field, "size")
225         col.prop(field, "flow")
226     elif field.type == 'HARMONIC':
227         col.prop(field, "harmonic_damping", text=_("Damping"))
228         col.prop(field, "rest_length")
229     elif field.type == 'VORTEX' and field.shape != 'POINT':
230         col.prop(field, "inflow")
231     elif field.type == 'DRAG':
232         col.prop(field, "quadratic_drag", text=_("Quadratic"))
233     else:
234         col.prop(field, "flow")
235
236     col = split.column()
237     sub = col.column(align=True)
238     sub.prop(field, "noise")
239     sub.prop(field, "seed")
240     if field.type == 'TURBULENCE':
241         col.prop(field, "use_global_coords", text=_("Global"))
242     elif field.type == 'HARMONIC':
243         col.prop(field, "use_multiple_springs")
244
245     split = layout.split()
246
247     col = split.column()
248     col.label(text=_("Effect point:"))
249     col.prop(field, "apply_to_location")
250     col.prop(field, "apply_to_rotation")
251
252     col = split.column()
253     col.label(text=_("Collision:"))
254     col.prop(field, "use_absorption")
255
256
257 def basic_force_field_falloff_ui(self, context, field):
258     layout = self.layout
259
260     split = layout.split(percentage=0.35)
261
262     if not field or field.type == 'NONE':
263         return
264
265     col = split.column()
266     col.prop(field, "z_direction", text="")
267
268     col = split.column()
269     col.prop(field, "falloff_power", text=_("Power"))
270
271     split = layout.split()
272     col = split.column()
273     row = col.row(align=True)
274     row.prop(field, "use_min_distance", text="")
275     sub = row.row()
276     sub.active = field.use_min_distance
277     sub.prop(field, "distance_min", text=_("Minimum"))
278
279     col = split.column()
280     row = col.row(align=True)
281     row.prop(field, "use_max_distance", text="")
282     sub = row.row()
283     sub.active = field.use_max_distance
284     sub.prop(field, "distance_max", text=_("Maximum"))
285
286 if __name__ == "__main__":  # only for live edit.
287     bpy.utils.register_module(__name__)