1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# Copyright 2018 The glTF-Blender-IO authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import functools
import bpy
def cached(func):
"""
Decorate the cache gather functions results.
The gather function is only executed if its result isn't in the cache yet
:param func: the function to be decorated. It will have a static __cache member afterwards
:return:
"""
@functools.wraps(func)
def wrapper_cached(*args, **kwargs):
assert len(args) >= 2 and 0 <= len(kwargs) <= 1, "Wrong signature for cached function"
cache_key_args = args
# make a shallow copy of the keyword arguments so that 'export_settings' can be removed
cache_key_kwargs = dict(kwargs)
if kwargs.get("export_settings"):
export_settings = kwargs["export_settings"]
# 'export_settings' should not be cached
del cache_key_kwargs["export_settings"]
else:
export_settings = args[-1]
cache_key_args = args[:-1]
__by_name = [bpy.types.Object, bpy.types.Scene, bpy.types.Material, bpy.types.Action, bpy.types.Mesh]
# we make a tuple from the function arguments so that they can be used as a key to the cache
cache_key = ()
for i in cache_key_args:
if type(i) in __by_name:
cache_key += (i.name,)
else:
cache_key += (i,)
for i in cache_key_kwargs.values():
if type(i) in __by_name:
cache_key += (i.name,)
else:
cache_key += (i,)
# invalidate cache if export settings have changed
if not hasattr(func, "__export_settings") or export_settings != func.__export_settings:
func.__cache = {}
func.__export_settings = export_settings
# use or fill cache
if cache_key in func.__cache:
return func.__cache[cache_key]
else:
result = func(*args)
func.__cache[cache_key] = result
return result
return wrapper_cached
# TODO: replace "cached" with "unique" in all cases where the caching is functional and not only for performance reasons
call_or_fetch = cached
unique = cached
|