mirror of https://github.com/H-uru/korman.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.5 KiB
75 lines
2.5 KiB
# 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 math |
|
|
|
class GoodNeighbor: |
|
"""Leave Things the Way You Found Them! (TM)""" |
|
|
|
def __enter__(self): |
|
self._tracking = {} |
|
return self |
|
|
|
def track(self, cls, attr, value): |
|
self._tracking[(cls, attr)] = getattr(cls, attr) |
|
setattr(cls, attr, value) |
|
|
|
def __exit__(self, type, value, traceback): |
|
for (cls, attr), value in self._tracking.items(): |
|
setattr(cls, attr, value) |
|
|
|
|
|
class TemporaryObject: |
|
def __init__(self, obj, remove_func): |
|
self._obj = obj |
|
self._remove_func = remove_func |
|
|
|
def __enter__(self): |
|
return self |
|
|
|
def __exit__(self, type, value, traceback): |
|
self._remove_func(self._obj) |
|
|
|
def __getattr__(self, attr): |
|
return getattr(self._obj, attr) |
|
|
|
|
|
def ensure_object_can_bake(bo, toggle): |
|
"""Ensures that we can use Blender's baking operators on this object. Side effect: also ensures |
|
that the object will enter edit mode when requested.""" |
|
scene = bpy.context.scene |
|
# we don't toggle.track this because it actually updates some kind of Blender internals... |
|
# therefore the old and new value are equal. the export operator will take care of this for us |
|
scene.layers = (True,) * len(scene.layers) |
|
|
|
toggle.track(bo, "hide", False) |
|
toggle.track(bo, "hide_render", False) |
|
toggle.track(bo, "hide_select", False) |
|
|
|
# Ensure only this object is selected |
|
for i in bpy.data.objects: |
|
i.select = i == bo |
|
|
|
def ensure_power_of_two(value): |
|
return pow(2, math.floor(math.log(value, 2))) |
|
|
|
def find_modifier(boname, modid): |
|
"""Given a Blender Object name, finds a given modifier and returns it or None""" |
|
bo = bpy.data.objects.get(boname, None) |
|
if bo is not None: |
|
# if they give us the wrong modid, it is a bug and an AttributeError |
|
return getattr(bo.plasma_modifiers, modid) |
|
return None
|
|
|