import os
import shutil

# 输入文件夹路径：只需修改这三行
json_folder = r"G:\python_ws_g\data\manycircle"
tiff_folder = r"G:\python_ws_g\data\rgbd_tiff"
output_folder = r"G:\python_ws_g\data\tiff_json"

# 创建输出文件夹（如不存在）
os.makedirs(output_folder, exist_ok=True)

# 提取 json 文件中的标识前缀（用于匹配）
json_files = [f for f in os.listdir(json_folder) if f.endswith('_color.json')]
json_keys = {f.replace('_color.json', ''): f for f in json_files}

# 遍历 tiff 文件，尝试匹配 json
matched_count = 0
for tiff_name in os.listdir(tiff_folder):
    if not tiff_name.endswith('.tiff'):
        continue
    if '_rgbd.tiff' not in tiff_name:
        continue

    key = tiff_name.replace('_rgbd.tiff', '')
    if key in json_keys:
        # 匹配成功，复制两份文件到目标目录
        json_src = os.path.join(json_folder, json_keys[key])
        tiff_src = os.path.join(tiff_folder, tiff_name)

        shutil.copy(json_src, os.path.join(output_folder, json_keys[key]))
        shutil.copy(tiff_src, os.path.join(output_folder, tiff_name))
        print(f"匹配并复制：{json_keys[key]} <-> {tiff_name}")
        matched_count += 1

print(f"\n✅ 共复制 {matched_count} 对文件到：{output_folder}")
