forked from bartvdbraak/blender
263 lines
7.4 KiB
Python
263 lines
7.4 KiB
Python
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
import bpy
|
|
|
|
narrowui = 180
|
|
|
|
|
|
from properties_physics_common import point_cache_ui
|
|
from properties_physics_common import effector_weights_ui
|
|
|
|
|
|
class PhysicButtonsPanel(bpy.types.Panel):
|
|
bl_space_type = 'PROPERTIES'
|
|
bl_region_type = 'WINDOW'
|
|
bl_context = "physics"
|
|
|
|
def poll(self, context):
|
|
ob = context.object
|
|
rd = context.scene.render
|
|
return (ob and ob.type == 'MESH') and (not rd.use_game_engine)
|
|
|
|
|
|
class PHYSICS_PT_smoke(PhysicButtonsPanel):
|
|
bl_label = "Smoke"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
md = context.smoke
|
|
ob = context.object
|
|
wide_ui = context.region.width > narrowui
|
|
|
|
split = layout.split()
|
|
|
|
if md:
|
|
# remove modifier + settings
|
|
split.set_context_pointer("modifier", md)
|
|
split.operator("object.modifier_remove", text="Remove")
|
|
|
|
row = split.row(align=True)
|
|
row.prop(md, "render", text="")
|
|
row.prop(md, "realtime", text="")
|
|
|
|
else:
|
|
# add modifier
|
|
split.operator("object.modifier_add", text="Add").type = 'SMOKE'
|
|
if wide_ui:
|
|
split.label()
|
|
|
|
if md:
|
|
if wide_ui:
|
|
layout.prop(md, "smoke_type", expand=True)
|
|
else:
|
|
layout.prop(md, "smoke_type", text="")
|
|
|
|
if md.smoke_type == 'TYPE_DOMAIN':
|
|
|
|
domain = md.domain_settings
|
|
|
|
split = layout.split()
|
|
|
|
col = split.column()
|
|
col.label(text="Resolution:")
|
|
col.prop(domain, "maxres", text="Divisions")
|
|
col.label(text="Particle:")
|
|
col.prop(domain, "initial_velocity", text="Initial Velocity")
|
|
|
|
if wide_ui:
|
|
col = split.column()
|
|
col.label(text="Behavior:")
|
|
col.prop(domain, "alpha")
|
|
col.prop(domain, "beta")
|
|
col.prop(domain, "dissolve_smoke", text="Dissolve")
|
|
sub = col.column()
|
|
sub.active = domain.dissolve_smoke
|
|
sub.prop(domain, "dissolve_speed", text="Time")
|
|
sub.prop(domain, "dissolve_smoke_log", text="Slow")
|
|
|
|
elif md.smoke_type == 'TYPE_FLOW':
|
|
|
|
flow = md.flow_settings
|
|
|
|
split = layout.split()
|
|
|
|
col = split.column()
|
|
col.prop(flow, "outflow")
|
|
col.label(text="Particle System:")
|
|
col.prop_object(flow, "psys", ob, "particle_systems", text="")
|
|
|
|
if md.flow_settings.outflow:
|
|
if wide_ui:
|
|
col = split.column()
|
|
else:
|
|
if wide_ui:
|
|
col = split.column()
|
|
col.label(text="Behavior:")
|
|
col.prop(flow, "temperature")
|
|
col.prop(flow, "density")
|
|
|
|
#elif md.smoke_type == 'TYPE_COLL':
|
|
# layout.separator()
|
|
|
|
|
|
class PHYSICS_PT_smoke_groups(PhysicButtonsPanel):
|
|
bl_label = "Smoke Groups"
|
|
bl_default_closed = True
|
|
|
|
def poll(self, context):
|
|
md = context.smoke
|
|
return md and (md.smoke_type == 'TYPE_DOMAIN')
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
group = context.smoke.domain_settings
|
|
wide_ui = context.region.width > narrowui
|
|
|
|
split = layout.split()
|
|
|
|
col = split.column()
|
|
col.label(text="Flow Group:")
|
|
col.prop(group, "fluid_group", text="")
|
|
|
|
#col.label(text="Effector Group:")
|
|
#col.prop(group, "eff_group", text="")
|
|
|
|
if wide_ui:
|
|
col = split.column()
|
|
col.label(text="Collision Group:")
|
|
col.prop(group, "coll_group", text="")
|
|
|
|
|
|
class PHYSICS_PT_smoke_cache(PhysicButtonsPanel):
|
|
bl_label = "Smoke Cache"
|
|
bl_default_closed = True
|
|
|
|
def poll(self, context):
|
|
md = context.smoke
|
|
return md and (md.smoke_type == 'TYPE_DOMAIN')
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
domain = context.smoke.domain_settings
|
|
|
|
layout.label(text="Compression:")
|
|
layout.prop(domain, "smoke_cache_comp", expand=True)
|
|
|
|
md = context.smoke.domain_settings
|
|
cache = md.point_cache_low
|
|
|
|
point_cache_ui(self, context, cache, (cache.baked is False), 0, 1)
|
|
|
|
|
|
class PHYSICS_PT_smoke_highres(PhysicButtonsPanel):
|
|
bl_label = "Smoke High Resolution"
|
|
bl_default_closed = True
|
|
|
|
def poll(self, context):
|
|
md = context.smoke
|
|
return md and (md.smoke_type == 'TYPE_DOMAIN')
|
|
|
|
def draw_header(self, context):
|
|
high = context.smoke.domain_settings
|
|
|
|
self.layout.prop(high, "highres", text="")
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
md = context.smoke.domain_settings
|
|
wide_ui = context.region.width > narrowui
|
|
|
|
split = layout.split()
|
|
|
|
col = split.column()
|
|
col.label(text="Resolution:")
|
|
col.prop(md, "amplify", text="Divisions")
|
|
col.prop(md, "viewhighres")
|
|
|
|
if wide_ui:
|
|
col = split.column()
|
|
col.label(text="Noise Method:")
|
|
col.row().prop(md, "noise_type", text="")
|
|
col.prop(md, "strength")
|
|
|
|
|
|
class PHYSICS_PT_smoke_cache_highres(PhysicButtonsPanel):
|
|
bl_label = "Smoke High Resolution Cache"
|
|
bl_default_closed = True
|
|
|
|
def poll(self, context):
|
|
md = context.smoke
|
|
return md and (md.smoke_type == 'TYPE_DOMAIN') and md.domain_settings.highres
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
domain = context.smoke.domain_settings
|
|
|
|
layout.label(text="Compression:")
|
|
layout.prop(domain, "smoke_cache_high_comp", expand=True)
|
|
|
|
|
|
md = context.smoke.domain_settings
|
|
cache = md.point_cache_high
|
|
|
|
point_cache_ui(self, context, cache, (cache.baked is False), 0, 1)
|
|
|
|
|
|
class PHYSICS_PT_smoke_field_weights(PhysicButtonsPanel):
|
|
bl_label = "Smoke Field Weights"
|
|
bl_default_closed = True
|
|
|
|
def poll(self, context):
|
|
smoke = context.smoke
|
|
return (smoke and smoke.smoke_type == 'TYPE_DOMAIN')
|
|
|
|
def draw(self, context):
|
|
domain = context.smoke.domain_settings
|
|
effector_weights_ui(self, context, domain.effector_weights)
|
|
|
|
|
|
classes = [
|
|
PHYSICS_PT_smoke,
|
|
PHYSICS_PT_smoke_field_weights,
|
|
PHYSICS_PT_smoke_cache,
|
|
PHYSICS_PT_smoke_highres,
|
|
PHYSICS_PT_smoke_groups,
|
|
PHYSICS_PT_smoke_cache_highres]
|
|
|
|
|
|
def register():
|
|
register = bpy.types.register
|
|
for cls in classes:
|
|
register(cls)
|
|
|
|
|
|
def unregister():
|
|
unregister = bpy.types.unregister
|
|
for cls in classes:
|
|
unregister(cls)
|
|
|
|
if __name__ == "__main__":
|
|
register()
|