582 lines
21 KiB
Python
582 lines
21 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
SUW Core - Material Manager Module
|
||
拆分自: suw_impl.py (Line 880-1200, 6470-6950)
|
||
用途: Blender材质管理、纹理处理、材质应用
|
||
版本: 1.0.0
|
||
作者: SUWood Team
|
||
"""
|
||
|
||
from .memory_manager import memory_manager
|
||
import time
|
||
import logging
|
||
from typing import Dict, Any, Optional
|
||
|
||
# 设置日志
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 检查Blender可用性
|
||
try:
|
||
import bpy
|
||
BLENDER_AVAILABLE = True
|
||
except ImportError:
|
||
BLENDER_AVAILABLE = False
|
||
|
||
# 导入内存管理器
|
||
|
||
# ==================== 材质管理器类 ====================
|
||
|
||
|
||
class MaterialManager:
|
||
"""材质管理器 - 负责所有材质相关操作"""
|
||
|
||
def __init__(self):
|
||
"""
|
||
初始化材质管理器 - 完全独立，不依赖suw_impl
|
||
"""
|
||
self.textures = {} # 材质缓存
|
||
self.material_cache = {} # 【修复】添加缺少的材质缓存字典
|
||
self.material_stats = {
|
||
"materials_created": 0,
|
||
"textures_loaded": 0,
|
||
"creation_errors": 0
|
||
}
|
||
|
||
# 材质类型配置
|
||
self.mat_type = "normal" # 当前材质类型
|
||
self.back_material = True # 是否应用背面材质
|
||
|
||
logger.info("MaterialManager 初始化完成")
|
||
|
||
def init_materials(self):
|
||
"""初始化材质 - 减少注册调用"""
|
||
try:
|
||
if not BLENDER_AVAILABLE:
|
||
return
|
||
|
||
logger.debug("初始化材质...")
|
||
|
||
# 创建基础材质
|
||
materials_to_create = [
|
||
("mat_default", (0.8, 0.8, 0.8, 1.0)),
|
||
("mat_select", (1.0, 0.5, 0.0, 1.0)),
|
||
("mat_normal", (0.7, 0.7, 0.7, 1.0)),
|
||
("mat_obverse", (0.9, 0.9, 0.9, 1.0)),
|
||
("mat_reverse", (0.6, 0.6, 0.6, 1.0)),
|
||
("mat_thin", (0.5, 0.5, 0.5, 1.0)),
|
||
]
|
||
|
||
for mat_name, color in materials_to_create:
|
||
if mat_name not in bpy.data.materials:
|
||
material = bpy.data.materials.new(name=mat_name)
|
||
material.use_nodes = True
|
||
|
||
# 设置基础颜色
|
||
if material.node_tree:
|
||
principled = material.node_tree.nodes.get(
|
||
"Principled BSDF")
|
||
if principled:
|
||
principled.inputs['Base Color'].default_value = color
|
||
|
||
# 只注册一次
|
||
memory_manager.register_object(material)
|
||
self.textures[mat_name] = material
|
||
else:
|
||
# 如果材质已存在，直接使用
|
||
self.textures[mat_name] = bpy.data.materials[mat_name]
|
||
|
||
logger.info("材质初始化完成")
|
||
|
||
except Exception as e:
|
||
logger.error(f"初始化材质失败: {e}")
|
||
|
||
def add_mat_rgb(self, mat_id: str, alpha: float, r: int, g: int, b: int):
|
||
"""添加RGB材质"""
|
||
try:
|
||
if not BLENDER_AVAILABLE:
|
||
return None
|
||
|
||
# 检查材质是否已存在
|
||
if mat_id in self.material_cache:
|
||
material_name = self.material_cache[mat_id]
|
||
if material_name in bpy.data.materials:
|
||
return bpy.data.materials[material_name]
|
||
|
||
# 创建新材质
|
||
material = bpy.data.materials.new(mat_id)
|
||
material.use_nodes = True
|
||
|
||
# 设置颜色
|
||
if material.node_tree:
|
||
principled = material.node_tree.nodes.get("Principled BSDF")
|
||
if principled:
|
||
color = (r/255.0, g/255.0, b/255.0, alpha)
|
||
principled.inputs[0].default_value = color
|
||
|
||
# 设置透明度
|
||
if alpha < 1.0:
|
||
material.blend_method = 'BLEND'
|
||
# Alpha input
|
||
principled.inputs[21].default_value = alpha
|
||
|
||
# 缓存材质
|
||
self.material_cache[mat_id] = material.name
|
||
self.textures[mat_id] = material
|
||
memory_manager.register_object(material)
|
||
|
||
logger.info(f"创建RGB材质: {mat_id}")
|
||
return material
|
||
|
||
except Exception as e:
|
||
logger.error(f"创建RGB材质失败: {e}")
|
||
return None
|
||
|
||
def get_texture(self, key: str):
|
||
"""获取纹理材质 - 修复版本，支持Default_前缀查找"""
|
||
if not BLENDER_AVAILABLE:
|
||
return None
|
||
|
||
try:
|
||
# 检查键是否有效
|
||
if not key:
|
||
return self.textures.get("mat_default")
|
||
|
||
# 【修复1】从缓存中获取
|
||
if key in self.textures:
|
||
material = self.textures[key]
|
||
# 验证材质是否仍然有效
|
||
if material and material.name in bpy.data.materials:
|
||
return material
|
||
else:
|
||
# 清理无效的缓存
|
||
del self.textures[key]
|
||
|
||
# 【修复2】在现有材质中查找 - 支持多种匹配方式
|
||
for material in bpy.data.materials:
|
||
material_name = material.name
|
||
|
||
# 精确匹配
|
||
if key == material_name:
|
||
self.textures[key] = material
|
||
logger.info(f"✅ 找到精确匹配材质: {key}")
|
||
return material
|
||
|
||
# 包含匹配（处理Default_前缀）
|
||
if key in material_name:
|
||
self.textures[key] = material
|
||
logger.info(f"✅ 找到包含匹配材质: {key} -> {material_name}")
|
||
return material
|
||
|
||
# Default_前缀匹配
|
||
if material_name.startswith(f"Default_{key}"):
|
||
self.textures[key] = material
|
||
logger.info(f"✅ 找到Default_前缀材质: {key} -> {material_name}")
|
||
return material
|
||
|
||
# 【修复3】如果没找到，尝试创建默认材质
|
||
logger.warning(f"未找到纹理: {key}，尝试创建默认材质")
|
||
try:
|
||
# 创建默认材质
|
||
default_material = bpy.data.materials.new(
|
||
name=f"Default_{key}")
|
||
default_material.use_nodes = True
|
||
|
||
# 设置基础颜色
|
||
if default_material.node_tree:
|
||
principled = default_material.node_tree.nodes.get(
|
||
"Principled BSDF")
|
||
if principled:
|
||
# 使用灰色作为默认颜色
|
||
principled.inputs['Base Color'].default_value = (
|
||
0.7, 0.7, 0.7, 1.0)
|
||
|
||
# 缓存材质
|
||
self.textures[key] = default_material
|
||
if memory_manager:
|
||
memory_manager.register_object(default_material)
|
||
|
||
logger.info(f"✅ 创建默认材质: Default_{key}")
|
||
return default_material
|
||
|
||
except Exception as create_error:
|
||
logger.error(f"创建默认材质失败: {create_error}")
|
||
|
||
# 【修复4】返回默认材质
|
||
default_material = self.textures.get("mat_default")
|
||
if default_material and default_material.name in bpy.data.materials:
|
||
logger.warning(f"使用系统默认材质: {key}")
|
||
return default_material
|
||
|
||
logger.warning(f"未找到纹理: {key}")
|
||
return None
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取纹理失败: {e}")
|
||
return None
|
||
|
||
def apply_material_to_face(self, face, material):
|
||
"""为面应用材质"""
|
||
try:
|
||
if not face or not material or not BLENDER_AVAILABLE:
|
||
return
|
||
|
||
if hasattr(face, 'data') and face.data:
|
||
if not face.data.materials:
|
||
face.data.materials.append(material)
|
||
else:
|
||
face.data.materials[0] = material
|
||
|
||
except Exception as e:
|
||
logger.error(f"为面应用材质失败: {e}")
|
||
|
||
def create_transparent_material(self):
|
||
"""创建透明材质"""
|
||
try:
|
||
if not BLENDER_AVAILABLE:
|
||
return None
|
||
|
||
# 检查是否已存在透明材质
|
||
transparent_mat_name = "mat_transparent"
|
||
if transparent_mat_name in self.textures:
|
||
return self.textures[transparent_mat_name]
|
||
|
||
# 创建透明材质
|
||
material = bpy.data.materials.new(name=transparent_mat_name)
|
||
material.use_nodes = True
|
||
material.blend_method = 'BLEND'
|
||
|
||
# 设置透明属性
|
||
if material.node_tree:
|
||
principled = material.node_tree.nodes.get("Principled BSDF")
|
||
if principled:
|
||
# 设置基础颜色为半透明白色
|
||
principled.inputs['Base Color'].default_value = (
|
||
1.0, 1.0, 1.0, 0.5)
|
||
# 设置Alpha
|
||
principled.inputs['Alpha'].default_value = 0.5
|
||
|
||
# 缓存材质
|
||
self.textures[transparent_mat_name] = material
|
||
memory_manager.register_object(material)
|
||
|
||
logger.info("创建透明材质完成")
|
||
return material
|
||
|
||
except Exception as e:
|
||
logger.error(f"创建透明材质失败: {e}")
|
||
return None
|
||
|
||
# ==================== 【修复】添加缺少的c02方法 ====================
|
||
|
||
def c02(self, data: Dict[str, Any]):
|
||
"""add_texture - 添加纹理"""
|
||
try:
|
||
logger.info(
|
||
f"🎨 MaterialManager.c02: 处理纹理 {data.get('ckey', 'unknown')}")
|
||
|
||
if not BLENDER_AVAILABLE:
|
||
logger.warning("Blender不可用，跳过纹理创建")
|
||
return None
|
||
|
||
ckey = data.get("ckey")
|
||
if not ckey:
|
||
logger.warning("纹理键为空，跳过创建")
|
||
return None
|
||
|
||
# 检查纹理是否已存在
|
||
if ckey in self.textures:
|
||
existing_material = self.textures[ckey]
|
||
if existing_material and existing_material.name in bpy.data.materials:
|
||
logger.info(f"✅ 纹理 {ckey} 已存在")
|
||
return existing_material
|
||
else:
|
||
# 清理无效缓存
|
||
del self.textures[ckey]
|
||
|
||
# 创建新材质
|
||
material = bpy.data.materials.new(name=ckey)
|
||
material.use_nodes = True
|
||
|
||
# 获取材质节点
|
||
nodes = material.node_tree.nodes
|
||
links = material.node_tree.links
|
||
|
||
# 清理默认节点
|
||
nodes.clear()
|
||
|
||
# 创建基础节点
|
||
principled = nodes.new(type='ShaderNodeBsdfPrincipled')
|
||
principled.location = (0, 0)
|
||
|
||
output = nodes.new(type='ShaderNodeOutputMaterial')
|
||
output.location = (300, 0)
|
||
|
||
# 连接基础节点
|
||
links.new(principled.outputs['BSDF'], output.inputs['Surface'])
|
||
|
||
# 设置纹理图像
|
||
src_path = data.get("src")
|
||
if src_path:
|
||
try:
|
||
import os
|
||
if os.path.exists(src_path):
|
||
# 加载图像
|
||
image_name = os.path.basename(src_path)
|
||
image = bpy.data.images.get(image_name)
|
||
|
||
if not image:
|
||
image = bpy.data.images.load(src_path)
|
||
if memory_manager:
|
||
memory_manager.register_image(image)
|
||
|
||
# 创建纹理节点
|
||
tex_coord = nodes.new(type='ShaderNodeTexCoord')
|
||
tex_coord.location = (-600, 0)
|
||
|
||
tex_image = nodes.new(type='ShaderNodeTexImage')
|
||
tex_image.image = image
|
||
tex_image.location = (-300, 0)
|
||
|
||
# 连接节点
|
||
links.new(
|
||
tex_coord.outputs['UV'], tex_image.inputs['Vector'])
|
||
links.new(
|
||
tex_image.outputs['Color'], principled.inputs['Base Color'])
|
||
|
||
# 透明度
|
||
alpha_value = data.get("alpha", 1.0)
|
||
if alpha_value < 1.0:
|
||
links.new(
|
||
tex_image.outputs['Alpha'], tex_image.inputs['Alpha'])
|
||
material.blend_method = 'BLEND'
|
||
else:
|
||
# 文件不存在，使用纯色
|
||
principled.inputs['Base Color'].default_value = (
|
||
0.5, 0.5, 0.5, 1.0)
|
||
logger.warning(f"纹理文件不存在: {src_path}")
|
||
|
||
except Exception as img_error:
|
||
logger.error(f"加载图像失败: {img_error}")
|
||
# 红色表示错误
|
||
principled.inputs['Base Color'].default_value = (
|
||
1.0, 0.0, 0.0, 1.0)
|
||
else:
|
||
# 没有图片路径，使用RGB数据
|
||
r = data.get("r", 128) / 255.0
|
||
g = data.get("g", 128) / 255.0
|
||
b = data.get("b", 128) / 255.0
|
||
principled.inputs['Base Color'].default_value = (r, g, b, 1.0)
|
||
|
||
# 设置透明度
|
||
alpha_value = data.get("alpha", 1.0)
|
||
principled.inputs['Alpha'].default_value = alpha_value
|
||
if alpha_value < 1.0:
|
||
material.blend_method = 'BLEND'
|
||
|
||
# 设置其他属性
|
||
if "reflection" in data:
|
||
metallic_value = data["reflection"]
|
||
principled.inputs['Metallic'].default_value = metallic_value
|
||
|
||
if "reflection_glossiness" in data:
|
||
roughness_value = 1.0 - data["reflection_glossiness"]
|
||
principled.inputs['Roughness'].default_value = roughness_value
|
||
|
||
# 缓存材质
|
||
self.textures[ckey] = material
|
||
if memory_manager:
|
||
memory_manager.register_object(material)
|
||
|
||
# 更新统计
|
||
if hasattr(self, 'material_stats'):
|
||
self.material_stats["materials_created"] += 1
|
||
|
||
logger.info(f"✅ 创建纹理材质成功: {ckey}")
|
||
return material
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ MaterialManager.c02 执行失败: {e}")
|
||
self.material_stats["creation_errors"] += 1
|
||
return None
|
||
|
||
# ==================== 其他方法继续 ====================
|
||
|
||
def textured_surf(self, face, back_material, color, saved_color=None, scale_a=None, angle_a=None):
|
||
"""为表面应用纹理 - 保持原始方法名和参数"""
|
||
try:
|
||
if not face or not BLENDER_AVAILABLE:
|
||
return
|
||
|
||
# 获取材质
|
||
material = None
|
||
if color:
|
||
material = self.get_texture(color)
|
||
|
||
if not material and saved_color:
|
||
material = self.get_texture(saved_color)
|
||
|
||
if not material:
|
||
material = self.get_texture("mat_default")
|
||
|
||
# 应用材质
|
||
if material:
|
||
self.apply_material_to_face(face, material)
|
||
|
||
# 应用纹理变换
|
||
if scale_a or angle_a:
|
||
self.apply_texture_transform(face, material, scale_a, angle_a)
|
||
|
||
except Exception as e:
|
||
logger.error(f"应用表面纹理失败: {e}")
|
||
|
||
def apply_texture_transform(self, face, material, scale=None, angle=None):
|
||
"""应用纹理变换 - 保持原始方法名和参数"""
|
||
try:
|
||
if not face or not material or not BLENDER_AVAILABLE:
|
||
return
|
||
|
||
if not hasattr(face, 'data') or not face.data:
|
||
return
|
||
|
||
mesh = face.data
|
||
|
||
# 确保有UV层
|
||
if not mesh.uv_layers:
|
||
mesh.uv_layers.new(name="UVMap")
|
||
|
||
uv_layer = mesh.uv_layers.active
|
||
|
||
if uv_layer:
|
||
self.apply_uv_transform(uv_layer, scale, angle)
|
||
|
||
except Exception as e:
|
||
logger.error(f"应用纹理变换失败: {e}")
|
||
|
||
def apply_uv_transform(self, uv_layer, scale, angle):
|
||
"""应用UV变换 - 保持原始方法名和参数"""
|
||
try:
|
||
if not uv_layer:
|
||
return
|
||
|
||
import math
|
||
|
||
# 应用缩放和旋转
|
||
if scale or angle:
|
||
for loop in uv_layer.data:
|
||
u, v = loop.uv
|
||
|
||
# 应用缩放
|
||
if scale:
|
||
u *= scale
|
||
v *= scale
|
||
|
||
# 应用旋转
|
||
if angle:
|
||
angle_rad = math.radians(angle)
|
||
cos_a = math.cos(angle_rad)
|
||
sin_a = math.sin(angle_rad)
|
||
|
||
# 绕中心点旋转
|
||
u_centered = u - 0.5
|
||
v_centered = v - 0.5
|
||
|
||
u_new = u_centered * cos_a - v_centered * sin_a + 0.5
|
||
v_new = u_centered * sin_a + v_centered * cos_a + 0.5
|
||
|
||
u, v = u_new, v_new
|
||
|
||
loop.uv = (u, v)
|
||
|
||
except Exception as e:
|
||
logger.error(f"应用UV变换失败: {e}")
|
||
|
||
def rotate_texture(self, face, scale, angle):
|
||
"""旋转纹理 - 保持原始方法名和参数"""
|
||
try:
|
||
if not face or not BLENDER_AVAILABLE:
|
||
return
|
||
|
||
if not hasattr(face, 'data') or not face.data:
|
||
return
|
||
|
||
mesh = face.data
|
||
if not mesh.uv_layers:
|
||
return
|
||
|
||
uv_layer = mesh.uv_layers.active
|
||
if uv_layer:
|
||
self.apply_uv_transform(uv_layer, scale, angle)
|
||
|
||
except Exception as e:
|
||
logger.error(f"旋转纹理失败: {e}")
|
||
|
||
def set_mat_type(self, mat_type: str):
|
||
"""设置材质类型"""
|
||
self.mat_type = mat_type
|
||
logger.info(f"设置材质类型: {mat_type}")
|
||
|
||
def get_mat_type(self) -> str:
|
||
"""获取当前材质类型"""
|
||
return self.mat_type
|
||
|
||
def clear_material_cache(self):
|
||
"""清理材质缓存"""
|
||
try:
|
||
if hasattr(self, 'material_cache'):
|
||
self.material_cache.clear()
|
||
# 保留基础材质，清理其他缓存
|
||
base_materials = ["mat_default", "mat_select",
|
||
"mat_normal", "mat_obverse", "mat_reverse", "mat_thin"]
|
||
filtered_textures = {
|
||
k: v for k, v in self.textures.items() if k in base_materials}
|
||
self.textures = filtered_textures
|
||
logger.info("材质缓存清理完成")
|
||
except Exception as e:
|
||
logger.error(f"清理材质缓存失败: {e}")
|
||
|
||
def get_material_stats(self) -> Dict[str, Any]:
|
||
"""获取材质管理器统计信息"""
|
||
try:
|
||
stats = {
|
||
"manager_type": "MaterialManager",
|
||
"cached_textures": len(self.textures),
|
||
"cached_materials": len(getattr(self, 'material_cache', {})),
|
||
"current_mat_type": self.mat_type,
|
||
"back_material": self.back_material,
|
||
"blender_available": BLENDER_AVAILABLE
|
||
}
|
||
|
||
if BLENDER_AVAILABLE:
|
||
stats["total_blender_materials"] = len(bpy.data.materials)
|
||
|
||
return stats
|
||
except Exception as e:
|
||
logger.error(f"获取材质统计失败: {e}")
|
||
return {"error": str(e)}
|
||
|
||
# ==================== 模块实例 ====================
|
||
|
||
|
||
# 全局实例，将由SUWImpl初始化时设置
|
||
material_manager = None
|
||
|
||
|
||
def init_material_manager():
|
||
"""初始化材质管理器 - 不再需要suw_impl参数"""
|
||
global material_manager
|
||
material_manager = MaterialManager()
|
||
return material_manager
|
||
|
||
|
||
def get_material_manager():
|
||
"""获取全局材质管理器实例"""
|
||
global material_manager
|
||
if material_manager is None:
|
||
material_manager = init_material_manager()
|
||
return material_manager
|
||
|
||
|
||
# 自动初始化
|
||
material_manager = init_material_manager()
|