mirror of https://github.com/H-uru/korman.git
11 changed files with 408 additions and 229 deletions
@ -0,0 +1,167 @@ |
|||||||
|
# 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/>. |
||||||
|
|
||||||
|
from __future__ import annotations |
||||||
|
|
||||||
|
from bpy.props import * |
||||||
|
|
||||||
|
from typing import * |
||||||
|
import warnings |
||||||
|
|
||||||
|
# Workaround for Blender memory management limitation, |
||||||
|
# don't change this to a literal in the code! |
||||||
|
_ENTIRE_ANIMATION = "(Entire Animation)" |
||||||
|
|
||||||
|
def _get_object_animation_names(self, object_attr: str) -> Sequence[Tuple[str, str, str]]: |
||||||
|
target_object = getattr(self, object_attr) |
||||||
|
if target_object is not None: |
||||||
|
items = [(anim.animation_name, anim.animation_name, "") |
||||||
|
for anim in target_object.plasma_modifiers.animation.subanimations] |
||||||
|
else: |
||||||
|
items = [(_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")] |
||||||
|
|
||||||
|
# We always want "(Entire Animation)", if it exists, to be the first item. |
||||||
|
entire = items.index((_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")) |
||||||
|
if entire not in (-1, 0): |
||||||
|
items.pop(entire) |
||||||
|
items.insert(0, (_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")) |
||||||
|
|
||||||
|
return items |
||||||
|
|
||||||
|
def animation(object_attr: str, **kwargs) -> str: |
||||||
|
def get_items(self, context): |
||||||
|
return _get_object_animation_names(self, object_attr) |
||||||
|
|
||||||
|
return EnumProperty(items=get_items, **kwargs) |
||||||
|
|
||||||
|
# These are the kinds of physical bounds Plasma can work with. |
||||||
|
# This sequence is acceptable in any EnumProperty |
||||||
|
_bounds_types = ( |
||||||
|
("box", "Bounding Box", "Use a perfect bounding box"), |
||||||
|
("sphere", "Bounding Sphere", "Use a perfect bounding sphere"), |
||||||
|
("hull", "Convex Hull", "Use a convex set encompasing all vertices"), |
||||||
|
("trimesh", "Triangle Mesh", "Use the exact triangle mesh (SLOW!)") |
||||||
|
) |
||||||
|
|
||||||
|
def _bounds_type_index(key: str) -> int: |
||||||
|
return list(zip(*_bounds_types))[0].index(key) |
||||||
|
|
||||||
|
def _bounds_type_str(idx: int) -> str: |
||||||
|
return _bounds_types[idx][0] |
||||||
|
|
||||||
|
def _get_bounds(physics_attr: Optional[str]) -> Callable[[Any], int]: |
||||||
|
def getter(self) -> int: |
||||||
|
physics_object = getattr(self, physics_attr) if physics_attr is not None else self.id_data |
||||||
|
if physics_object is not None: |
||||||
|
return _bounds_type_index(physics_object.plasma_modifiers.collision.bounds) |
||||||
|
return _bounds_type_index("hull") |
||||||
|
return getter |
||||||
|
|
||||||
|
def _set_bounds(physics_attr: Optional[str]) -> Callable[[Any, int], None]: |
||||||
|
def setter(self, value: int): |
||||||
|
physics_object = getattr(self, physics_attr) if physics_attr is not None else self.id_data |
||||||
|
if physics_object is not None: |
||||||
|
physics_object.plasma_modifiers.collision.bounds = _bounds_type_str(value) |
||||||
|
return setter |
||||||
|
|
||||||
|
def bounds(physics_attr: Optional[str] = None, store_on_collider: bool = True, **kwargs) -> str: |
||||||
|
assert not {"items", "get", "set"} & kwargs.keys(), "You cannot use the `items`, `get`, or `set` keyword arguments" |
||||||
|
if store_on_collider: |
||||||
|
kwargs["get"] = _get_bounds(physics_attr) |
||||||
|
kwargs["set"] = _set_bounds(physics_attr) |
||||||
|
else: |
||||||
|
warnings.warn("Storing bounds properties outside of the collision modifier is deprecated.", category=DeprecationWarning) |
||||||
|
if "default" not in kwargs: |
||||||
|
kwargs["default"] = "hull" |
||||||
|
return EnumProperty( |
||||||
|
items=_bounds_types, |
||||||
|
**kwargs |
||||||
|
) |
||||||
|
|
||||||
|
def upgrade_bounds(bl, bounds_attr: str) -> None: |
||||||
|
# Only perform this process if the property has a value. Otherwise, we'll |
||||||
|
# wind up blowing away the collision modifier's settings with nonsense. |
||||||
|
if not bl.is_property_set(bounds_attr): |
||||||
|
return |
||||||
|
|
||||||
|
# Before we unregister anything, grab a copy of what the collision modifier currently thinks. |
||||||
|
bounds_value_curr = getattr(bl, bounds_attr) |
||||||
|
|
||||||
|
# So, here's the deal. If someone has been playing with nodes and changed the bounds type, |
||||||
|
# Blender will think the property has been set, even if they wound up with the property |
||||||
|
# at the default value. I don't know that we can really trust the default in the property |
||||||
|
# definition to be the same as the old default (they shouldn't be different, but let's be safe). |
||||||
|
# So, let's apply rough justice. If the destination property thinks it's a triangle mesh, we |
||||||
|
# don't need to blow that away - it's a very specific non default setting. |
||||||
|
if bounds_value_curr == "trimesh": |
||||||
|
return |
||||||
|
|
||||||
|
# Unregister the new/correct proxy bounds property (with getter/setter) and re-register |
||||||
|
# the property without the proxy functions to get the old value. Reregister the new property |
||||||
|
# again and set it. |
||||||
|
cls = bl.__class__ |
||||||
|
prop_func, prop_def = getattr(cls, bounds_attr) |
||||||
|
RemoveProperty(cls, attr=bounds_attr) |
||||||
|
del prop_def["attr"] |
||||||
|
|
||||||
|
# Remove the things we don't want in a copy to prevent hosing the new property. |
||||||
|
old_prop_def = dict(prop_def) |
||||||
|
del old_prop_def["get"] |
||||||
|
del old_prop_def["set"] |
||||||
|
setattr(cls, bounds_attr, prop_func(**old_prop_def)) |
||||||
|
bounds_value_new = getattr(bl, bounds_attr) |
||||||
|
|
||||||
|
# Re-register new property. |
||||||
|
RemoveProperty(cls, attr=bounds_attr) |
||||||
|
setattr(cls, bounds_attr, prop_func(**prop_def)) |
||||||
|
|
||||||
|
# Only set the property if the value different to avoid thrashing and log spam. |
||||||
|
if bounds_value_curr != bounds_value_new: |
||||||
|
print(f"Stashing bounds property: [{bl.name}] ({cls.__name__}) {bounds_value_curr} -> {bounds_value_new}") # TEMP |
||||||
|
setattr(bl, bounds_attr, bounds_value_new) |
||||||
|
|
||||||
|
def _get_texture_animation_names(self, object_attr: str, material_attr: str, texture_attr: str) -> Sequence[Tuple[str, str, str]]: |
||||||
|
target_object = getattr(self, object_attr) |
||||||
|
material = getattr(self, material_attr) |
||||||
|
texture = getattr(self, texture_attr) |
||||||
|
|
||||||
|
if texture is not None: |
||||||
|
items = [(anim.animation_name, anim.animation_name, "") |
||||||
|
for anim in texture.plasma_layer.subanimations] |
||||||
|
elif material is not None or target_object is not None: |
||||||
|
if material is None: |
||||||
|
materials = (i.material for i in target_object.material_slots if i and i.material) |
||||||
|
else: |
||||||
|
materials = (material,) |
||||||
|
layer_props = (i.texture.plasma_layer for mat in materials for i in mat.texture_slots if i and i.texture) |
||||||
|
all_anims = frozenset((anim.animation_name for i in layer_props for anim in i.subanimations)) |
||||||
|
items = [(i, i, "") for i in all_anims] |
||||||
|
else: |
||||||
|
items = [(_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")] |
||||||
|
|
||||||
|
# We always want "(Entire Animation)", if it exists, to be the first item. |
||||||
|
entire = items.index((_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")) |
||||||
|
if entire not in (-1, 0): |
||||||
|
items.pop(entire) |
||||||
|
items.insert(0, (_ENTIRE_ANIMATION, _ENTIRE_ANIMATION, "")) |
||||||
|
|
||||||
|
return items |
||||||
|
|
||||||
|
def triprop_animation(object_attr: str, material_attr: str, texture_attr: str, **kwargs) -> str: |
||||||
|
def get_items(self, context): |
||||||
|
return _get_texture_animation_names(self, object_attr, material_attr, texture_attr) |
||||||
|
|
||||||
|
assert not {"items", "get", "set"} & kwargs.keys(), "You cannot use the `items`, `get`, or `set` keyword arguments" |
||||||
|
return EnumProperty(items=get_items, **kwargs) |
Loading…
Reference in new issue