yyc12345
ddf6b7befe
- correct all icon usage. use 'CANCEL' instead of 'ERROR' to indicate error correctly. and move these icon const into UTILS_icons_manager.py - auto rename generated rail section to match the default name standard. - give more clear signal about import and export bm file by using Operator.report() function.
136 lines
4.8 KiB
Python
136 lines
4.8 KiB
Python
import bpy
|
|
from . import UTILS_constants, UTILS_functions, UTILS_virtools_prop, UTILS_icons_manager
|
|
|
|
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", UTILS_icons_manager.blender_error_icon)
|
|
|
|
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", UTILS_icons_manager.blender_error_icon)
|
|
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")
|
|
|