Onan Chew
a08d888acb
- operator strings are referenced via the class property rather than a hard coded string - class names have been cleaned to have _OT_ or _PT_ depending on the kind of class it was. - Class names should mostly be CamelCase now.
89 lines
4.0 KiB
Python
89 lines
4.0 KiB
Python
from bpy.types import UIList, Panel, UILayout, Object, Context,Material, Operator
|
|
import bpy
|
|
from ..core.register import register_wrap
|
|
from .panel import AvatarToolKit_PT_AvatarToolkitPanel, CATEGORY_NAME
|
|
from ..core.common import SceneMatClass, MaterialListBool, get_selected_armature
|
|
from ..functions.atlas_materials import AvatarToolKit_OT_AtlasMaterials
|
|
from ..functions.translations import t
|
|
|
|
@register_wrap
|
|
class AvatarToolKit_OT_ExpandSectionMaterials(Operator):
|
|
bl_idname = 'avatar_toolkit.expand_section_materials'
|
|
bl_label = ""
|
|
bl_description = ""
|
|
|
|
@classmethod
|
|
def poll(cls, context: Context) -> bool:
|
|
return True
|
|
|
|
def execute(self, context: Context) -> set:
|
|
if not context.scene.texture_atlas_Has_Mat_List_Shown:
|
|
context.scene.materials.clear()
|
|
newlist: list[Material] = []
|
|
for obj in bpy.context.scene.objects:
|
|
if len(obj.material_slots)>0:
|
|
for mat_slot in obj.material_slots:
|
|
if mat_slot.material:
|
|
if mat_slot.material not in newlist:
|
|
newlist.append(mat_slot.material)
|
|
newitem: SceneMatClass = context.scene.materials.add()
|
|
newitem.mat = mat_slot.material
|
|
MaterialListBool.old_list[context.scene.name] = newlist
|
|
else:
|
|
context.scene.texture_atlas_Has_Mat_List_Shown = False
|
|
return {'FINISHED'}
|
|
|
|
@register_wrap
|
|
class AvatarToolKit_UL_MaterialTextureAtlasProperties(UIList):
|
|
bl_label = t("TextureAtlas.material_list_label")
|
|
bl_idname = "Material_UL_avatar_toolkit_texture_atlas_mat_list_mat"
|
|
bl_space_type = 'VIEW_3D'
|
|
bl_region_type = 'UI'
|
|
|
|
def draw_item(self , context: Context, layout: UILayout, data: bpy.types.Object, item:SceneMatClass, icon, active_data, active_propname, index):
|
|
if context.scene.texture_atlas_Has_Mat_List_Shown:
|
|
box = layout.box()
|
|
row = box.row()
|
|
row.label(text=item.mat.name, icon = "MATERIAL")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_albedo")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_normal")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_emission")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_ambient_occlusion")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_height")
|
|
col = box.row()
|
|
col.prop(item.mat, "texture_atlas_roughness")
|
|
|
|
@register_wrap
|
|
class AvatarToolKit_PT_TextureAtlasPanel(Panel):
|
|
bl_label = t("TextureAtlas.label")
|
|
bl_idname = "OBJECT_PT_avatar_toolkit_texture_atlas"
|
|
bl_space_type = 'VIEW_3D'
|
|
bl_region_type = 'UI'
|
|
bl_category = CATEGORY_NAME
|
|
bl_parent_id = AvatarToolKit_PT_AvatarToolkitPanel.bl_idname
|
|
bl_order = 4
|
|
|
|
def draw(self, context: Context):
|
|
layout = self.layout
|
|
armature = get_selected_armature(context)
|
|
|
|
if armature:
|
|
row = layout.row()
|
|
boxoutter = row.box()
|
|
direction_icon = 'RIGHTARROW' if not context.scene.texture_atlas_Has_Mat_List_Shown else 'DOWNARROW_HLT'
|
|
row = boxoutter.row()
|
|
row.operator(AvatarToolKit_OT_ExpandSectionMaterials.bl_idname, text=(t("TextureAtlas.reload_list") if not context.scene.texture_atlas_Has_Mat_List_Shown else t("TextureAtlas.loaded_list")), icon=direction_icon)
|
|
if context.scene.texture_atlas_Has_Mat_List_Shown:
|
|
row = boxoutter.row()
|
|
row.template_list(AvatarToolKit_UL_MaterialTextureAtlasProperties.bl_idname, 'material_list', context.scene, 'materials',
|
|
context.scene, 'texture_atlas_material_index', rows=12, type='DEFAULT')
|
|
row = layout.row()
|
|
row.operator(AvatarToolKit_OT_AtlasMaterials.bl_idname, text=t("TextureAtlas.atlas_materials"))
|
|
else:
|
|
layout.label(text=t("Tools.select_armature"), icon='ERROR')
|