# merge_osgb_batch.py

import os
import subprocess
import glob
import re


# ================== 配置参数 ==================
IN_ROOT = r"D:\zlc\earthwork\before"
OUT_DIR = r"D:\zlc\earthwork\single2"
OSGCONV = r"C:\Soft\OpenSceneGraph-3.6.5\bin\osgconv.exe"
OSG_PLUGIN_PATH = r"C:\Soft\OpenSceneGraph-3.6.5\bin\osgPlugins-3.6.5"

BATCH_SIZE = 10
TEMP_PREFIX = "temp_merge_"
FINAL_OUTPUT = "merged.obj"
# ==============================================

def find_leaf_osgb_files(root_dir):
    """查找所有包含几何数据的 .osgb 文件（L18~L22）"""
    pattern = os.path.join(root_dir, "**", "*.osgb")
    all_files = glob.glob(pattern, recursive=True)
    
    valid_files = []
    for f in all_files:
        fname = os.path.basename(f)
        
        # 排除 root.osgb
        if 'root.osgb' in fname.lower():
            continue
            
        # 排除顶层结构文件：Tile_+000_+005.osgb
        if re.fullmatch(r'tile_\+\d+_\+\d+\.osgb', fname, re.IGNORECASE):
            continue
            
        # 排除低层级的 _0 文件：只排除 L17 及以下的 _0
        match = re.match(r'tile_\+\d+_\+\d+_l(\d+)_(0+)\.osgb', fname, re.IGNORECASE)
        if match:
            level = int(match.group(1))
            zeros = match.group(2)
            # 如果是 L17 及以下，且是 _0，排除
            if level <= 17 and zeros == '0':
                continue
            # 如果是 L18 及以上，即使是 _0 也保留（可能是有效瓦片）
            # 或者是 _00002300 这种非全零，也保留
            if level >= 18:
                pass  # 保留
            else:
                continue
        
        # ✅ 保留所有 L18 及以上层级的文件
        level_match = re.search(r'_l(\d+)_', fname, re.IGNORECASE)
        if level_match:
            level = int(level_match.group(1))
            if level >= 18:
                valid_files.append(f)
                continue
        
        # 可选：如果没匹配到层级，也保留（以防漏掉）
        # valid_files.append(f)
    
    valid_files.sort(key=lambda x: x.lower())
    return valid_files

def run_osgconv(args, env, cwd):
    """执行 osgconv 命令"""
    try:
        print(f"执行命令: {' '.join(args)}")
        result = subprocess.run(
            args,
            env=env,
            cwd=cwd,
            check=True,
            capture_output=True,
            text=True,
            encoding='utf-8'
        )
        print("stdout:", result.stdout.strip())
        return True
    except subprocess.CalledProcessError as e:
        print(f"❌ 命令执行失败：{e.cmd}")
        print(f"错误输出：{e.stderr}")
        return False

def main():
    print("� 正在检查环境...")

    if not os.path.exists(OSGCONV):
        print(f"❌ 错误：未找到 osgconv.exe！\n路径：{OSGCONV}")
        return

    if not os.path.exists(OSG_PLUGIN_PATH):
        print(f"❌ 错误：未找到 OSG 插件目录！\n路径：{OSG_PLUGIN_PATH}")
        return

    # ✅ 创建环境变量副本
    env = os.environ.copy()
    env['OSG_PLUGIN_PATH'] = OSG_PLUGIN_PATH
    osgconv_dir = os.path.dirname(OSGCONV)
    env['PATH'] = osgconv_dir + ';' + env.get('PATH', '')

    # 创建输出目录
    os.makedirs(OUT_DIR, exist_ok=True)
    print(f"� 已创建输出目录：{OUT_DIR}")

    # 查找所有叶子节点 .osgb 文件
    print("� 正在扫描 .osgb 文件...")
    leaf_files = find_leaf_osgb_files(IN_ROOT)

    if not leaf_files:
        print("❌ 错误：未找到任何符合条件的 .osgb 文件！")
        return

    print(f"✅ 找到 {len(leaf_files)} 个 .osgb 文件，开始分批合并...")

    intermediate_objs = []

    # ========== 第一轮：分批合并为中间 OBJ 文件 ==========
    for i in range(0, len(leaf_files), BATCH_SIZE):
        start_idx = i + 1
        end_idx = min(i + BATCH_SIZE, len(leaf_files))
        temp_obj = f"{TEMP_PREFIX}{start_idx:04d}.obj"

        print(f"� 合并第 {start_idx} ~ {end_idx} 个文件 → {temp_obj}")

        # ✅ 构建命令：不要手动加引号！
        cmd = [
            OSGCONV,
            "-O", "OutputTextureFiles"
        ]
        cmd.extend(leaf_files[i:end_idx])  # ✅ 让 subprocess 自动处理路径
        cmd.append(temp_obj)

        success = run_osgconv(cmd, env=env, cwd=OUT_DIR)
        if not success:
            print(f"❌ 合并失败：{temp_obj}")
            return
        else:
            print(f"✅ 成功生成：{temp_obj}")
            intermediate_objs.append(temp_obj)

    # ========== 第二轮：合并所有中间 OBJ 文件为最终 OBJ ==========
    if len(intermediate_objs) == 1:
        final_path = os.path.join(OUT_DIR, FINAL_OUTPUT)
        os.replace(os.path.join(OUT_DIR, intermediate_objs[0]), final_path)
        print(f"� 单批次完成，已重命名为：{FINAL_OUTPUT}")
    else:
        print(f"� 正在合并所有中间文件为 {FINAL_OUTPUT}...")
        final_cmd = [
            OSGCONV,
            "-O", "OutputTextureFiles"
        ]
        final_cmd.extend(intermediate_objs)  # ✅ 同样不要加引号
        final_cmd.append(FINAL_OUTPUT)

        success = run_osgconv(final_cmd, env=env, cwd=OUT_DIR)
        if not success:
            print(f"❌ 最终合并失败！")
            return

        print(f"� === 成功生成 {FINAL_OUTPUT} === �")

if __name__ == "__main__":
    main()