114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
import bpy, bpy.utils.previews
|
|
import os, enum, typing
|
|
|
|
class BlenderPresetIcons(enum.Enum):
|
|
Info = 'INFO'
|
|
Warning = 'ERROR'
|
|
Error = 'CANCEL'
|
|
|
|
#region Custom Icons Helper
|
|
|
|
_g_SupportedImageExts: set[str] = set((
|
|
'.png',
|
|
))
|
|
|
|
_g_IconsManager: bpy.utils.previews.ImagePreviewCollection | None = None
|
|
|
|
_g_EmptyIcon: int = 0
|
|
_g_IconPrefix: str = "BlcBldPlg_"
|
|
|
|
_g_BmeIconsMap: dict[str, int] = {}
|
|
_g_BmeIconPrefix: str = _g_IconPrefix + 'Bme_'
|
|
_g_ComponentIconsMap: dict[str, int] = {}
|
|
_g_ComponentIconPrefix: str = _g_IconPrefix + 'Component_'
|
|
_g_GroupIconsMap: dict[str, int] = {}
|
|
_g_GroupIconPrefix: str = _g_IconPrefix + 'Group_'
|
|
|
|
def _iterate_folder_images(folder: str) -> typing.Iterator[tuple[str, str]]:
|
|
for name in os.listdir(folder):
|
|
# check whether it is file
|
|
filepath: str = os.path.join(folder, name)
|
|
if os.path.isfile(filepath):
|
|
# check file exts
|
|
(root, ext) = os.path.splitext(name)
|
|
if ext.lower() in _g_SupportedImageExts:
|
|
yield (filepath, root)
|
|
|
|
def _load_image_folder(
|
|
folder: str,
|
|
loader: bpy.utils.previews.ImagePreviewCollection,
|
|
container: dict[str, int],
|
|
name_prefix: str) -> None:
|
|
# iterate folder
|
|
for (filepath, filename_no_ext) in _iterate_folder_images(folder):
|
|
# generate name for unique
|
|
icon_name: str = name_prefix + filename_no_ext
|
|
# load it
|
|
loader.load(icon_name, filepath, 'IMAGE')
|
|
# add into list. use plain name (not the unique name)
|
|
container[filename_no_ext] = loader[icon_name].icon_id
|
|
|
|
#endregion
|
|
|
|
#region Custom Icons Visitors
|
|
|
|
def get_empty_icon() -> int:
|
|
return _g_EmptyIcon
|
|
|
|
def get_bme_icon(name: str) -> int | None:
|
|
return _g_BmeIconsMap.get(name, None)
|
|
|
|
def get_component_icon(name: str) -> int | None:
|
|
return _g_ComponentIconsMap.get(name, None)
|
|
|
|
def get_group_icon(name: str) -> int | None:
|
|
return _g_GroupIconsMap.get(name, None)
|
|
|
|
#endregion
|
|
|
|
def register():
|
|
global _g_IconsManager
|
|
global _g_EmptyIcon
|
|
global _g_BmeIconsMap, _g_ComponentIconsMap, _g_GroupIconsMap
|
|
|
|
# create preview collection and get icon folder
|
|
icons_folder: str = os.path.join(os.path.dirname(__file__), "icons")
|
|
_g_IconsManager = bpy.utils.previews.new()
|
|
|
|
# load empty icon as default fallback
|
|
empty_icon_name: str = _g_IconPrefix + 'EmptyIcon'
|
|
_g_IconsManager.load(empty_icon_name, os.path.join(icons_folder, "Empty.png"), 'IMAGE')
|
|
_g_EmptyIcon = _g_IconsManager[empty_icon_name].icon_id
|
|
|
|
# load bme, component, group icon
|
|
_load_image_folder(
|
|
os.path.join(icons_folder, 'bme'),
|
|
_g_IconsManager,
|
|
_g_BmeIconsMap,
|
|
_g_BmeIconPrefix
|
|
)
|
|
_load_image_folder(
|
|
os.path.join(icons_folder, 'component'),
|
|
_g_IconsManager,
|
|
_g_ComponentIconsMap,
|
|
_g_ComponentIconPrefix
|
|
)
|
|
_load_image_folder(
|
|
os.path.join(icons_folder, 'group'),
|
|
_g_IconsManager,
|
|
_g_GroupIconsMap,
|
|
_g_GroupIconPrefix
|
|
)
|
|
|
|
def unregister():
|
|
global _g_IconsManager
|
|
global _g_EmptyIcon
|
|
global _g_BmeIconsMap, _g_ComponentIconsMap, _g_GroupIconsMap
|
|
|
|
bpy.utils.previews.remove(_g_IconsManager)
|
|
_g_IconsManager = None
|
|
|
|
_g_BmeIconsMap.clear()
|
|
_g_ComponentIconsMap.clear()
|
|
_g_GroupIconsMap.clear()
|