diff --git a/Tools/ModelTranslator/bl_bake.py b/Tools/ModelTranslator/bl_bake.py new file mode 100644 index 00000000..88e1f50e --- /dev/null +++ b/Tools/ModelTranslator/bl_bake.py @@ -0,0 +1,40 @@ +"""Blender 内运行:高低模 Cycles 烘焙,低模 UV 上出 normal/ao/color/metallic/roughness。 +调用:blender -b --factory-startup --python bl_bake.py -- \ + +""" +import math +import os +import sys + +sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) + +DEFAULT_RAY_PCT = 0.02 # 自动射线距离 = 低模包围盒对角线 * 2% +BBOX_TOL = 0.10 # 高低模包围盒尺寸相对差警告阈值 + +# (输出名, bake type, EMIT 来源的 Principled 输入, 目标图颜色空间) +BAKE_PASSES = [ + ("normal", 'NORMAL', None, 'Non-Color'), + ("ao", 'AO', None, 'Non-Color'), + ("color", 'EMIT', 'Base Color', 'sRGB'), + ("metallic", 'EMIT', 'Metallic', 'Non-Color'), + ("roughness", 'EMIT', 'Roughness', 'Non-Color'), +] + + +def estimate_ray_distance(bbox_dims, pct=DEFAULT_RAY_PCT): + """低模包围盒尺寸 (dx,dy,dz) -> 射线距离 = 对角线长 * pct。""" + return math.sqrt(sum(d * d for d in bbox_dims)) * pct + + +def output_stem(low_name): + """低模名去掉 _low 后缀作为输出前缀(对齐工具3纯网格贴图命名约定)。""" + return low_name[:-4] if low_name.endswith("_low") else low_name + + +def bbox_mismatch(high_dims, low_dims, tol=BBOX_TOL): + """任一轴尺寸相对差超 tol 返回 True;接近 0 的轴忽略(平面模型)。""" + for h, l in zip(high_dims, low_dims): + m = max(abs(h), abs(l)) + if m > 1e-9 and abs(h - l) / m > tol: + return True + return False diff --git a/Tools/ModelTranslator/tests/test_bl_bake.py b/Tools/ModelTranslator/tests/test_bl_bake.py new file mode 100644 index 00000000..7351b529 --- /dev/null +++ b/Tools/ModelTranslator/tests/test_bl_bake.py @@ -0,0 +1,47 @@ +import math +import os +import sys +import unittest + +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +import bl_bake as bb + + +class TestEstimateRayDistance(unittest.TestCase): + def test_two_percent_of_bbox_diagonal(self): + # 3-4-12 直角箱对角线 = 13 + self.assertAlmostEqual(bb.estimate_ray_distance((3.0, 4.0, 12.0)), 0.26) + + def test_custom_pct(self): + self.assertAlmostEqual( + bb.estimate_ray_distance((1.0, 0.0, 0.0), pct=0.5), 0.5) + + +class TestOutputStem(unittest.TestCase): + def test_strips_low_suffix(self): + self.assertEqual(bb.output_stem("well1500_low"), "well1500") + + def test_keeps_name_without_suffix(self): + self.assertEqual(bb.output_stem("well1500"), "well1500") + + def test_only_strips_trailing_suffix(self): + self.assertEqual(bb.output_stem("low_poly_low"), "low_poly") + + +class TestBboxMismatch(unittest.TestCase): + def test_identical_ok(self): + self.assertFalse(bb.bbox_mismatch((1.0, 2.0, 3.0), (1.0, 2.0, 3.0))) + + def test_within_10_percent_ok(self): + self.assertFalse(bb.bbox_mismatch((1.0, 2.0, 3.0), (1.05, 1.9, 3.2))) + + def test_one_axis_exceeds(self): + self.assertTrue(bb.bbox_mismatch((1.0, 2.0, 3.0), (1.0, 2.0, 3.5))) + + def test_zero_axis_ignored(self): + # 平面模型某轴为 0,不应误报 + self.assertFalse(bb.bbox_mismatch((1.0, 0.0, 3.0), (1.0, 0.0, 3.0))) + + +if __name__ == "__main__": + unittest.main()