used py error checking tools to fix some bugs & make pep8 corrections
[blender.git] / release / scripts / 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21
22 def point_cache_ui(self, cache, enabled, particles, smoke):
23     layout = self.layout
24     layout.set_context_pointer("PointCache", cache)
25
26     row = layout.row()
27     row.template_list(cache, "point_cache_list", cache, "active_point_cache_index", rows=2)
28     col = row.column(align=True)
29     col.itemO("ptcache.add_new", icon='ICON_ZOOMIN', text="")
30     col.itemO("ptcache.remove", icon='ICON_ZOOMOUT', text="")
31
32     row = layout.row()
33     row.itemL(text="File Name:")
34     if particles:
35         row.itemR(cache, "external")
36
37     if cache.external:
38         split = layout.split(percentage=0.80)
39         split.itemR(cache, "name", text="")
40         split.itemR(cache, "index", text="")
41
42         layout.itemL(text="File Path:")
43         layout.itemR(cache, "filepath", text="")
44
45         layout.itemL(text=cache.info)
46     else:
47         layout.itemR(cache, "name", text="")
48
49         if not particles:
50             row = layout.row()
51             row.enabled = enabled
52             row.itemR(cache, "start_frame")
53             row.itemR(cache, "end_frame")
54
55         row = layout.row()
56
57         if cache.baked == True:
58             row.itemO("ptcache.free_bake", text="Free Bake")
59         else:
60             row.item_booleanO("ptcache.bake", "bake", True, text="Bake")
61
62         sub = row.row()
63         sub.enabled = (cache.frames_skipped or cache.outdated) and enabled
64         sub.itemO("ptcache.bake", "bake", False, text="Calculate to Current Frame")
65
66         row = layout.row()
67         row.enabled = enabled
68         row.itemO("ptcache.bake_from_cache", text="Current Cache to Bake")
69         if not smoke:
70             row.itemR(cache, "step")
71
72         if not smoke:
73             row = layout.row()
74             sub = row.row()
75             sub.enabled = enabled
76             sub.itemR(cache, "quick_cache")
77             row.itemR(cache, "disk_cache")
78
79         layout.itemL(text=cache.info)
80
81         layout.itemS()
82
83         row = layout.row()
84         row.item_booleanO("ptcache.bake_all", "bake", True, text="Bake All Dynamics")
85         row.itemO("ptcache.free_bake_all", text="Free All Bakes")
86         layout.itemO("ptcache.bake_all", "bake", False, text="Update All Dynamics to current frame")
87
88
89 def effector_weights_ui(self, weights):
90     layout = self.layout
91
92     layout.itemR(weights, "group")
93
94     split = layout.split()
95     split.itemR(weights, "gravity", slider=True)
96     split.itemR(weights, "all", slider=True)
97
98     layout.itemS()
99
100     flow = layout.column_flow()
101     flow.itemR(weights, "force", slider=True)
102     flow.itemR(weights, "vortex", slider=True)
103     flow.itemR(weights, "magnetic", slider=True)
104     flow.itemR(weights, "wind", slider=True)
105     flow.itemR(weights, "curveguide", slider=True)
106     flow.itemR(weights, "texture", slider=True)
107     flow.itemR(weights, "harmonic", slider=True)
108     flow.itemR(weights, "charge", slider=True)
109     flow.itemR(weights, "lennardjones", slider=True)
110     flow.itemR(weights, "turbulence", slider=True)
111     flow.itemR(weights, "drag", slider=True)
112     flow.itemR(weights, "boid", slider=True)
113
114
115 def basic_force_field_settings_ui(self, field):
116     layout = self.layout
117     split = layout.split()
118
119     if not field or field.type == 'NONE':
120         return
121
122     col = split.column()
123
124     if field.type == 'DRAG':
125         col.itemR(field, "linear_drag", text="Linear")
126     else:
127         col.itemR(field, "strength")
128
129     if field.type == 'TURBULENCE':
130         col.itemR(field, "size")
131         col.itemR(field, "flow")
132     elif field.type == 'HARMONIC':
133         col.itemR(field, "harmonic_damping", text="Damping")
134     elif field.type == 'VORTEX' and field.shape != 'POINT':
135         col.itemR(field, "inflow")
136     elif field.type == 'DRAG':
137         col.itemR(field, "quadratic_drag", text="Quadratic")
138     else:
139         col.itemR(field, "flow")
140
141     col = split.column()
142     col.itemR(field, "noise")
143     col.itemR(field, "seed")
144     if field.type == 'TURBULENCE':
145         col.itemR(field, "global_coordinates", text="Global")
146
147     split = layout.split()
148
149     col = split.column()
150     col.itemL(text="Effect point:")
151     col.itemR(field, "do_location")
152     col.itemR(field, "do_rotation")
153
154     sub = split.column()
155     sub.itemL(text="Collision:")
156     sub.itemR(field, "do_absorption")
157
158
159 def basic_force_field_falloff_ui(self, field):
160     layout = self.layout
161     split = layout.split(percentage=0.35)
162
163     if not field or field.type == 'NONE':
164         return
165
166     col = split.column()
167     col.itemR(field, "z_direction", text="")
168     col.itemR(field, "use_min_distance", text="Use Minimum")
169     col.itemR(field, "use_max_distance", text="Use Maximum")
170
171     col = split.column()
172     col.itemR(field, "falloff_power", text="Power")
173
174     sub = col.column()
175     sub.active = field.use_min_distance
176     sub.itemR(field, "minimum_distance", text="Distance")
177
178     sub = col.column()
179     sub.active = field.use_max_distance
180     sub.itemR(field, "maximum_distance", text="Distance")