class IESProfileManager(DebugObject):
""" Manager which handles the different IES profiles, and also handles
loading .IES files """
def __init__(self, pipeline):
DebugObject.__init__(self)
self._pipeline = pipeline
self._entries = []
self._loader = IESProfileLoader()
self._max_entries = 32
self._create_storage()
def _create_storage(self):
""" Internal method to create the storage for the profile dataset textures """
self._storage_tex = Texture("IESDatasets")
self._storage_tex.setup_3d_texture(
512, 512, self._max_entries, Texture.T_float, Texture.F_r16)
self._storage_tex.set_minfilter(SamplerState.FT_linear)
self._storage_tex.set_magfilter(SamplerState.FT_linear)
self._storage_tex.set_wrap_u(SamplerState.WM_clamp)
self._storage_tex.set_wrap_v(SamplerState.WM_repeat)
self._storage_tex.set_wrap_w(SamplerState.WM_clamp)
self._pipeline.stage_mgr.add_input("IESDatasetTex", self._storage_tex)
self._pipeline.stage_mgr.define("MAX_IES_PROFILES", self._max_entries)
def load(self, filename):
""" Loads a profile from a given filename """
# Make filename unique
fname = Filename.from_os_specific(filename)
if not VirtualFileSystem.get_global_ptr().resolve_filename(fname, get_model_path().get_value(), "ies"):
self.error("Could not resolve", filename)
return -1
fname = fname.get_fullpath()
# Check for cache entries
if fname in self._entries:
return self._entries.index(fname)
# Check for out of bounds
if len(self._entries) >= self._max_entries:
# TODO: Could remove unused profiles here or regenerate texture
self.warn("Cannot load IES Profile, too many loaded! (Maximum: 32)")
# Try loading the dataset, and see what happes
try:
dataset = self._loader.load(fname)
except IESLoaderException as msg:
self.warn("Failed to load profile from", filename, ":", msg)
return -1
if not dataset:
return -1
# Dataset was loaded successfully, now copy it
dataset.generate_dataset_texture_into(self._storage_tex, len(self._entries))
self._entries.append(fname)
return len(self._entries) - 1
