31 lines
939 B
Python
31 lines
939 B
Python
|
import bpy
|
||
|
|
import bpy.utils.previews
|
||
|
|
import os
|
||
|
|
from . import UTILS_constants
|
||
|
|
|
||
|
|
# ImagePreviewCollection ccreated by Blender
|
||
|
|
floor_icons = None
|
||
|
|
# a map. key is block name, value is loaded icon id
|
||
|
|
floor_icons_map: dict = {}
|
||
|
|
|
||
|
|
def register_icons():
|
||
|
|
global floor_icons, floor_icons_map
|
||
|
|
|
||
|
|
icon_path = os.path.join(os.path.dirname(__file__), "icons")
|
||
|
|
floor_icons = bpy.utils.previews.new()
|
||
|
|
for key, value in UTILS_constants.floor_blockDict.items():
|
||
|
|
blockIconName = "Ballance_FloorIcon_" + key
|
||
|
|
floor_icons.load(blockIconName, os.path.join(icon_path, "floor", value["BindingDisplayTexture"]), 'IMAGE')
|
||
|
|
floor_icons_map[key] = floor_icons[blockIconName].icon_id
|
||
|
|
|
||
|
|
def unregister_icons():
|
||
|
|
global floor_icons, floor_icons_map
|
||
|
|
|
||
|
|
bpy.utils.previews.remove(floor_icons)
|
||
|
|
floor_icons_map.clear()
|
||
|
|
|
||
|
|
def get_floor_icon(floor_blk_name: str):
|
||
|
|
global floor_icons_map
|
||
|
|
|
||
|
|
return floor_icons_map[floor_blk_name]
|