BallanceBlenderHelper/ballance_blender_plugin/PROPS_virtools_material.py

import bpy
from . import UTILS_constants, UTILS_functions, UTILS_virtools_prop

class BALLANCE_OT_apply_virtools_material(bpy.types.Operator):
    """Apply Virtools Material to Blender Material."""
    bl_idname = "ballance.apply_virtools_material"
    bl_label = "Apply Virtools Material"
    bl_options = {'UNDO'}

    @classmethod
    def poll(cls, context):
        return context.material is not None

    def execute(self, context):
        mtl = context.material
        mtl_data = UTILS_virtools_prop.get_virtools_material_data(mtl)

        # check enable, [0] is enable_virtools_material
        if mtl_data[0]:
            UTILS_functions.create_material_nodes(mtl, mtl_data)
        else:
            UTILS_functions.show_message_box(("Virtools Material is not enabled.", ), "Apply Failed", 'ERROR')

        return {'FINISHED'}

class BALLANCE_OT_parse_virtools_material(bpy.types.Operator):
    """Apply Virtools Material to Blender Material."""
    bl_idname = "ballance.parse_virtools_material"
    bl_label = "Parse from Blender Principled BSDF"
    bl_options = {'UNDO'}

    @classmethod
    def poll(cls, context):
        return context.material is not None

    def execute(self, context):
        mtl = context.material
        mtl_data = UTILS_functions.parse_material_nodes(mtl)
        if mtl_data is None:
            UTILS_functions.show_message_box(("Fail to parse Principled BSDF.", ), "Parsing Failed", 'ERROR')
        else:
            UTILS_virtools_prop.set_virtools_material_data(mtl, mtl_data)

        return {'FINISHED'}

class BALLANCE_OT_preset_virtools_material(bpy.types.Operator):
    """Preset Virtools Material with Original Ballance Data."""
    bl_idname = "ballance.preset_virtools_material"
    bl_label = "Preset Virtools Material"
    bl_options = {'UNDO'}

    preset_type: bpy.props.EnumProperty(
        name="Preset",
        description="The preset which you want to apply.",
        items=tuple(
            (str(idx), item["human-readable"], "Suit for: " + ", ".join(item["member"]))
            for idx, item in enumerate(UTILS_constants.floor_materialStatistic)
        ),
    )

    @classmethod
    def poll(cls, context):
        return context.material is not None
    
    def invoke(self, context, event):
        wm = context.window_manager
        return wm.invoke_props_dialog(self)
    
    def draw(self, context):
        self.layout.prop(self, "preset_type")

    def execute(self, context):
        preset_idx = int(self.preset_type)
        preset_data = UTILS_constants.floor_materialStatistic[preset_idx]

        # get data self and only change core colors
        mtl = context.material
        vtmtl = UTILS_virtools_prop.get_virtools_material(mtl)

        vtmtl.ambient = preset_data['data']['ambient']
        vtmtl.diffuse = preset_data['data']['diffuse']
        vtmtl.specular = preset_data['data']['specular']
        vtmtl.emissive = preset_data['data']['emissive']
        vtmtl.specular_power = preset_data['data']['power']

        return {'FINISHED'}

class BALLANCE_PT_virtools_material(bpy.types.Panel):
    """Show Virtools Material Properties."""
    bl_label = "Virtools Material"
    bl_idname = "BALLANCE_PT_virtools_material"
    bl_space_type = 'PROPERTIES'
    bl_region_type = 'WINDOW'
    bl_context = "material"

    @classmethod
    def poll(cls, context):
        return context.material is not None

    def draw_header(self, context):
        # draw a checkbox in header
        target = UTILS_virtools_prop.get_virtools_material(context.material)
        self.layout.prop(target, "enable_virtools_material", text="")

    def draw(self, context):
        # get layout and target
        layout = self.layout
        target = UTILS_virtools_prop.get_virtools_material(context.material)

        # decide visible
        layout.enabled = target.enable_virtools_material

        # draw layout
        row = layout.row()
        row.label(text="Basic Parameters")
        row.operator(BALLANCE_OT_preset_virtools_material.bl_idname, text="", icon="PRESET")
        layout.prop(target, 'ambient')
        layout.prop(target, 'diffuse')
        layout.prop(target, 'specular')
        layout.prop(target, 'emissive')
        layout.prop(target, 'specular_power')
        layout.prop(target, 'texture', emboss=True)

        layout.separator()
        layout.label(text="Advanced Parameters")
        layout.prop(target, 'alpha_test')
        layout.prop(target, 'alpha_blend')
        layout.prop(target, 'z_buffer')
        layout.prop(target, 'two_sided')

        layout.separator()
        layout.label(text="Operations")
        layout.operator(BALLANCE_OT_apply_virtools_material.bl_idname, icon="NODETREE")
        layout.operator(BALLANCE_OT_parse_virtools_material.bl_idname, icon="HIDE_OFF")