mirror of https://github.com/H-uru/korman.git
11 changed files with 656 additions and 38 deletions
@ -0,0 +1,185 @@ |
|||||||
|
# This file is part of Korman. |
||||||
|
# |
||||||
|
# Korman 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 3 of the License, or |
||||||
|
# (at your option) any later version. |
||||||
|
# |
||||||
|
# Korman 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 Korman. If not, see <http://www.gnu.org/licenses/>. |
||||||
|
|
||||||
|
import bpy |
||||||
|
from collections import defaultdict |
||||||
|
import itertools |
||||||
|
from PyHSPlasma import * |
||||||
|
import weakref |
||||||
|
|
||||||
|
from ..exporter.explosions import ExportError |
||||||
|
|
||||||
|
def _get_puddle_class(exporter, name, vs): |
||||||
|
if vs: |
||||||
|
# sigh... thou shalt not... |
||||||
|
exporter.report.warn("'{}': Cannot use 'Water Ripple (Shallow) on a waveset--forcing to 'Water Ripple (Deep)", name) |
||||||
|
return plDynaRippleVSMgr |
||||||
|
return plDynaPuddleMgr |
||||||
|
|
||||||
|
def _get_footprint_class(exporter, name, vs): |
||||||
|
if vs: |
||||||
|
raise ExportError("'{}': Footprints cannot be attached to wavesets", name) |
||||||
|
return plDynaFootMgr |
||||||
|
|
||||||
|
class DecalConverter: |
||||||
|
_decal_lookup = { |
||||||
|
"footprint_dry": _get_footprint_class, |
||||||
|
"footprint_wet": _get_footprint_class, |
||||||
|
"puddle": _get_puddle_class, |
||||||
|
"ripple": lambda e, name, vs: plDynaRippleVSMgr if vs else plDynaRippleMgr, |
||||||
|
} |
||||||
|
|
||||||
|
def __init__(self, exporter): |
||||||
|
self._decal_managers = defaultdict(list) |
||||||
|
self._exporter = weakref.ref(exporter) |
||||||
|
self._notifies = defaultdict(set) |
||||||
|
|
||||||
|
def add_dynamic_decal_receiver(self, so, decal_name): |
||||||
|
# One decal manager in Blender can map to many Plasma decal managers. |
||||||
|
# The case we care about: a single water decal exporting to multiple DynaDecalMgrs |
||||||
|
# eg two wavesets (two mgrs) and two water planes (one mgr) |
||||||
|
# We don't care about: DynaDecalMgrs in another page. |
||||||
|
decal_mgrs, so_key = self._decal_managers.get(decal_name), so.key |
||||||
|
if decal_mgrs is None: |
||||||
|
raise ExportError("'{}': Invalid decal manager '{}'", so_key.name, decal_name) |
||||||
|
|
||||||
|
# If we are waveset water, then we can only have one target... |
||||||
|
waveset_id = plFactory.ClassIndex("plWaveSet7") |
||||||
|
waveset = next((i for i in so.modifiers if i.type == waveset_id), None) |
||||||
|
|
||||||
|
so_loc = so_key.location |
||||||
|
for key, decal_mgr in ((i, i.object) for i in decal_mgrs): |
||||||
|
if key.location == so_loc and getattr(decal_mgr, "waveSet", None) == waveset: |
||||||
|
decal_mgr.addTarget(so_key) |
||||||
|
|
||||||
|
# HACKAGE: Add the wet/dirty notifes now that we know about all the decal managers. |
||||||
|
notify_names = self._notifies[decal_name] |
||||||
|
notify_keys = itertools.chain.from_iterable((self._decal_managers[i] for i in notify_names)) |
||||||
|
for notify_key in notify_keys: |
||||||
|
for i in (i.object for i in decal_mgrs): |
||||||
|
i.addNotify(notify_key) |
||||||
|
# Don't need to do that again. |
||||||
|
del self._notifies[decal_name] |
||||||
|
|
||||||
|
def export_active_print_shape(self, print_shape, decal_name): |
||||||
|
decal_mgrs = self._decal_managers.get(decal_name) |
||||||
|
if decal_mgrs is None: |
||||||
|
raise ExportError("'{}': Invalid decal manager '{}'", print_shape.key.name, decal_name) |
||||||
|
for i in decal_mgrs: |
||||||
|
print_shape.addDecalMgr(i) |
||||||
|
|
||||||
|
def export_static_decal(self, bo): |
||||||
|
mat_mgr = self._exporter().mesh.material |
||||||
|
mat_keys = mat_mgr.get_materials(bo) |
||||||
|
if not mat_keys: |
||||||
|
raise ExportError("'{}': Cannot print decal onto object with no materials", bo.name) |
||||||
|
|
||||||
|
zFlags = hsGMatState.kZIncLayer | hsGMatState.kZNoZWrite |
||||||
|
for material in (i.object for i in mat_keys): |
||||||
|
# Only useful in a debugging context |
||||||
|
material.compFlags |= hsGMaterial.kCompDecal |
||||||
|
|
||||||
|
# zFlags should only be applied to the material's base layer |
||||||
|
# note: changing blend flags is unsafe here -- so don't even think about it! |
||||||
|
layer = mat_mgr.get_base_layer(material) |
||||||
|
layer.state.ZFlags |= zFlags |
||||||
|
|
||||||
|
def generate_dynamic_decal(self, bo, decal_name): |
||||||
|
decal = next((i for i in bpy.context.scene.plasma_scene.decal_managers if i.name == decal_name), None) |
||||||
|
if decal is None: |
||||||
|
raise ExportError("'{}': Invalid decal manager '{}'", bo.name, decal_name) |
||||||
|
|
||||||
|
exporter = self._exporter() |
||||||
|
decal_type = decal.decal_type |
||||||
|
is_waveset = bo.plasma_modifiers.water_basic.enabled |
||||||
|
pClass = self._decal_lookup[decal_type](exporter, decal_name, is_waveset) |
||||||
|
|
||||||
|
# DynaDecal Managers generate geometry at runtime, so we need to share them as much as |
||||||
|
# possible. However, it is best to keep things page local. Furthermore, wavesets cannot |
||||||
|
# share decal managers due to vertex shaders being used. |
||||||
|
name = "{}_{}".format(decal_name, bo.name) if is_waveset else decal_name |
||||||
|
decal_mgr = exporter.mgr.find_object(pClass, bl=bo, name=name) |
||||||
|
if decal_mgr is None: |
||||||
|
self._report.msg("Exporing decal manager '{}' to '{}'", decal_name, name, indent=2) |
||||||
|
|
||||||
|
decal_mgr = exporter.mgr.add_object(pClass, bl=bo, name=name) |
||||||
|
self._decal_managers[decal_name].append(decal_mgr.key) |
||||||
|
|
||||||
|
# Certain decals are required to be squares |
||||||
|
if decal_type in {"footprint_dry", "footprint_wet", "wake"}: |
||||||
|
length, width = decal.length / 100.0, decal.width / 100.0 |
||||||
|
else: |
||||||
|
length = max(decal.length, decal.width) / 100.0 |
||||||
|
width = max(decal.length, decal.width) / 100.0 |
||||||
|
|
||||||
|
image = decal.image |
||||||
|
if image is None: |
||||||
|
raise ExportError("'{}': decal manager '{}' has no image set", bo.name, decal_name) |
||||||
|
|
||||||
|
blend = getattr(hsGMatState, decal.blend) |
||||||
|
mats = exporter.mesh.material.export_print_materials(bo, image, name, blend) |
||||||
|
decal_mgr.matPreShade, decal_mgr.matRTShade = mats |
||||||
|
|
||||||
|
# Hardwired values from PlasmaMAX |
||||||
|
decal_mgr.maxNumVerts = 1000 |
||||||
|
decal_mgr.maxNumIdx = 1000 |
||||||
|
decal_mgr.intensity = decal.intensity / 100.0 |
||||||
|
decal_mgr.gridSizeU = 2.5 |
||||||
|
decal_mgr.gridSizeV = 2.5 |
||||||
|
decal_mgr.scale = hsVector3(length, width, 1.0) |
||||||
|
|
||||||
|
# Hardwired calculations from PlasmaMAX |
||||||
|
if decal_type in {"footprint_dry", "footprint_wet", "bullet"}: |
||||||
|
decal_mgr.rampEnd = 0.1 |
||||||
|
decal_mgr.decayStart = decal.life_span - (decal.life_span * 0.25) |
||||||
|
decal_mgr.lifeSpan = decal.life_span |
||||||
|
elif decal_type in {"puddle", "ripple", "torpedo", "wake"}: |
||||||
|
decal_mgr.rampEnd = 0.25 |
||||||
|
life_span = decal.life_span if decal_type == "torpedo" else length / 2.0 |
||||||
|
decal_mgr.decayStart = life_span * 0.8 |
||||||
|
decal_mgr.lifeSpan = life_span |
||||||
|
else: |
||||||
|
raise RuntimeError() |
||||||
|
|
||||||
|
# While any decal manager can be wet/dry, it really makes the most sense to only |
||||||
|
# expose wet footprints. In the future, we could expose the plDynaDecalEnableMsg |
||||||
|
# to nodes for advanced hacking. |
||||||
|
decal_mgr.waitOnEnable = decal_type == "footprint_wet" |
||||||
|
if decal_type in {"puddle", "ripple"}: |
||||||
|
decal_mgr.wetLength = decal.wet_time |
||||||
|
self._notifies[decal_name].update((i.name for i in decal.wet_managers |
||||||
|
if i.enabled and i.name != decal_name)) |
||||||
|
|
||||||
|
# UV Animations are hardcoded in PlasmaMAX. Any reason why we should expose this? |
||||||
|
# I can't think of any presently... Note testing the final instance instead of the |
||||||
|
# artist setting in case that gets overridden (puddle -> ripple) |
||||||
|
if isinstance(decal_mgr, plDynaPuddleMgr): |
||||||
|
decal_mgr.initUVW = hsVector3(5.0, 5.0, 5.0) |
||||||
|
decal_mgr.finalUVW = hsVector3(1.0, 1.0, 1.0) |
||||||
|
elif isinstance(decal_mgr, plDynaRippleMgr): |
||||||
|
# wakes, torpedos, and ripples... |
||||||
|
decal_mgr.initUVW = hsVector3(3.0, 3.0, 3.0) |
||||||
|
decal_mgr.finalUVW = hsVector3(1.0, 1.0, 1.0) |
||||||
|
|
||||||
|
if isinstance(decal_mgr, (plDynaRippleVSMgr, plDynaTorpedoVSMgr)): |
||||||
|
decal_mgr.waveSet = exporter.mgr.find_create_key(plWaveSet7, bl=bo) |
||||||
|
|
||||||
|
@property |
||||||
|
def _mgr(self): |
||||||
|
return self._exporter().mgr |
||||||
|
|
||||||
|
@property |
||||||
|
def _report(self): |
||||||
|
return self._exporter().report |
@ -0,0 +1,100 @@ |
|||||||
|
# This file is part of Korman. |
||||||
|
# |
||||||
|
# Korman 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 3 of the License, or |
||||||
|
# (at your option) any later version. |
||||||
|
# |
||||||
|
# Korman 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 Korman. If not, see <http://www.gnu.org/licenses/>. |
||||||
|
|
||||||
|
import bpy |
||||||
|
import functools |
||||||
|
from . import ui_list |
||||||
|
|
||||||
|
class SceneButtonsPanel: |
||||||
|
bl_space_type = "PROPERTIES" |
||||||
|
bl_region_type = "WINDOW" |
||||||
|
bl_context = "scene" |
||||||
|
|
||||||
|
@classmethod |
||||||
|
def poll(cls, context): |
||||||
|
return context.scene.render.engine == "PLASMA_GAME" |
||||||
|
|
||||||
|
|
||||||
|
class DecalManagerListUI(bpy.types.UIList): |
||||||
|
def draw_item(self, context, layout, data, item, icon, active_data, active_property, index=0, flt_flag=0): |
||||||
|
layout.prop(item, "display_name", emboss=False, text="", icon="BRUSH_DATA") |
||||||
|
|
||||||
|
|
||||||
|
class WetManagerListUI(bpy.types.UIList): |
||||||
|
def draw_item(self, context, layout, data, item, icon, active_data, active_property, index=0, flt_flag=0): |
||||||
|
if item.name: |
||||||
|
layout.label(item.name, icon="BRUSH_DATA") |
||||||
|
layout.prop(item, "enabled", text="") |
||||||
|
else: |
||||||
|
layout.label("[Empty]") |
||||||
|
|
||||||
|
|
||||||
|
class PlasmaDecalManagersPanel(SceneButtonsPanel, bpy.types.Panel): |
||||||
|
bl_label = "Plasma Decal Managers" |
||||||
|
|
||||||
|
def draw(self, context): |
||||||
|
layout, scene = self.layout, context.scene.plasma_scene |
||||||
|
|
||||||
|
ui_list.draw_list(layout, "DecalManagerListUI", "scene", scene, "decal_managers", |
||||||
|
"active_decal_index", name_prefix="Decal", name_prop="display_name", |
||||||
|
rows=3) |
||||||
|
|
||||||
|
try: |
||||||
|
decal_mgr = scene.decal_managers[scene.active_decal_index] |
||||||
|
except: |
||||||
|
pass |
||||||
|
else: |
||||||
|
box = layout.box().column() |
||||||
|
|
||||||
|
box.prop(decal_mgr, "decal_type") |
||||||
|
box.alert = decal_mgr.image is None |
||||||
|
box.prop(decal_mgr, "image") |
||||||
|
box.alert = False |
||||||
|
box.prop(decal_mgr, "blend") |
||||||
|
box.separator() |
||||||
|
|
||||||
|
split = box.split() |
||||||
|
col = split.column(align=True) |
||||||
|
col.label("Scale:") |
||||||
|
col.alert = decal_mgr.decal_type in {"ripple", "puddle", "bullet", "torpedo"} \ |
||||||
|
and decal_mgr.length != decal_mgr.width |
||||||
|
col.prop(decal_mgr, "length") |
||||||
|
col.prop(decal_mgr, "width") |
||||||
|
|
||||||
|
col = split.column() |
||||||
|
col.label("Draw Settings:") |
||||||
|
col.prop(decal_mgr, "intensity") |
||||||
|
sub = col.row() |
||||||
|
sub.active = decal_mgr.decal_type in {"footprint_dry", "footprint_wet", "bullet", "torpedo"} |
||||||
|
sub.prop(decal_mgr, "life_span") |
||||||
|
sub = col.row() |
||||||
|
sub.active = decal_mgr.decal_type in {"puddle", "ripple"} |
||||||
|
sub.prop(decal_mgr, "wet_time") |
||||||
|
|
||||||
|
if decal_mgr.decal_type in {"puddle", "ripple"}: |
||||||
|
box.separator() |
||||||
|
box.label("Wet Footprints:") |
||||||
|
ui_list.draw_list(box, "WetManagerListUI", "scene", decal_mgr, "wet_managers", |
||||||
|
"active_wet_index", rows=2, maxrows=3) |
||||||
|
try: |
||||||
|
wet_ref = decal_mgr.wet_managers[decal_mgr.active_wet_index] |
||||||
|
except: |
||||||
|
pass |
||||||
|
else: |
||||||
|
wet_mgr = next((i for i in scene.decal_managers if i.name == wet_ref.name), None) |
||||||
|
box.alert = getattr(wet_mgr, "decal_type", None) == "footprint_wet" |
||||||
|
box.prop_search(wet_ref, "name", scene, "decal_managers", icon="NONE") |
||||||
|
if wet_ref.name == decal_mgr.name: |
||||||
|
box.label(text="Circular reference", icon="ERROR") |
Loading…
Reference in new issue