"""Blender 头图渲染脚本。"""

# pyright: reportMissingImports=false

import bpy
import numpy as np
from pathlib import Path
import os
import argparse
import sys


bpy.ops.preferences.addon_enable(module="photogrammetry_importer")

def load_obj_vertices(obj_file_path):
    """
    使用bpy加载OBJ文件并获取顶点数据。
    返回网格object和顶点数组。
    """
    # 清除当前场景中的所有网格对象
    bpy.ops.object.select_all(action='DESELECT')
    bpy.ops.object.select_by_type(type='MESH')
    bpy.ops.object.delete(use_global=False, confirm=False)

    # 判断文件是否存在
    if not os.path.exists(obj_file_path):
        print(f"{obj_file_path} not exist")
        return None, None

    # 导入OBJ
    bpy.ops.wm.obj_import(filepath=obj_file_path, forward_axis='Y', up_axis='Z')
    obj_original = [obj for obj in bpy.data.objects if obj.type == 'MESH'][0]
    bpy.context.view_layer.objects.active = obj_original
    obj_original.select_set(True)
    mesh = obj_original.data
    vertices = np.array([v.co for v in mesh.vertices])
    return obj_original, vertices

def render_and_save(camera, output_path, cam_name):
    """
    设置渲染参数并执行渲染操作，最后保存为图片
    :param camera: 相机对象
    :param output_path: 输出图片的路径
    :param cam_name: 相机名称
    """
    # 设置渲染引擎为 Workbench
    bpy.context.scene.render.engine = 'BLENDER_WORKBENCH'
    # bpy.context.scene.render.engine = 'WORKBENCH'
    # 设置 Workbench 渲染的着色模式为 Flat 光和 Texture 颜色
    bpy.context.scene.display.shading.light = 'FLAT'
    # bpy.context.scene.display.shading.color_type = 'ATTRIBUTE'
    bpy.context.scene.display.shading.color_type = 'TEXTURE'
    # 设置 Viewtransform 为 standard
    bpy.context.scene.view_settings.view_transform = 'Standard'

    # 设置背景颜色为白色
    bpy.context.scene.world.color = (1, 1, 1)

    # 设置输出路径
    bpy.context.scene.render.filepath = os.path.join(output_path, f"123_2.jpg")
    # 设置图像格式为 PNG
    bpy.context.scene.render.image_settings.file_format = 'JPEG'
    # 设置渲染分辨率为 100%
    bpy.context.scene.render.resolution_percentage = 100

    # 设置当前相机为活动相机
    bpy.context.scene.camera = camera
    # 执行渲染操作
    bpy.ops.render.render(write_still=True)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--sparse_file", type=str, required=True)
    parser.add_argument("--mesh_file", type=str, required=True)
    parser.add_argument("--output_path", type=str, required=True)
    args = parser.parse_args(sys.argv[sys.argv.index("--") + 1:])
    # --------------------- 示例使用 ---------------------
    sparse_file = args.sparse_file
    mesh_file = args.mesh_file
    output_path = args.output_path


    # 调用操作符
    bpy.ops.import_scene.colmap_model(    
        directory               = sparse_file,   # 指向模型文件夹
        import_points           = False,        # 导入稀疏点云
        import_cameras          = True,        # 导入相机
        add_background_images   = False,        # 把原始照片挂到相机上
        adjust_render_settings  = True,        # 自动把渲染分辨率改成相机分辨率
        adjust_clipping_distance= False         # 自动把裁剪面设置得更合适
    )
    print("导入成功！")

    mesh_object, vertices = load_obj_vertices(mesh_file)
    # 场景中查找相机名称为74_8_cam的相机
    camera = None
    for obj in bpy.data.objects:
        if obj.type == 'CAMERA':
            camera = obj
            if camera.name == "123_2_cam":
                render_and_save(camera, str(output_path), camera.name)
            # break
    # 加载 mesh，得到 mesh_object 和 顶点数组

    # if camera:
    #     print("找到名称为74_8_cam的相机。")
    # else:
    #     print("未找到名称为74_8_cam的相机。")

    """
    blender --background --python /data/code/pose_toolkit/render_img.py -- \
    --sparse_file "/data/datasets_20t/Downloads_google/rander_obj_image/2124/render_pose/" \
    --mesh_file "/data/datasets_20t/Downloads_google/rander_obj_image/2124/2124.obj" \
    --output_path "/data/datasets_20t/Downloads_google/rander_obj_image/2124/output"
    
    
    blender --background --python /data/code/pose_toolkit/render_img.py -- \
    --sparse_file "/data/datasets_20t/Downloads_google/rander_obj_image/1030/render_pose/" \
    --mesh_file "/data/datasets_20t/Downloads_google/rander_obj_image/1030/obj/1030.obj" \
    --output_path "/data/datasets_20t/Downloads_google/rander_obj_image/1030/output"
    
    
    blender --background --python /data/code/pose_toolkit/render_img.py -- \
    --sparse_file "/data/datasets_20t/Downloads_google/rander_obj_image/1026/render_pose/" \
    --mesh_file "/data/datasets_20t/Downloads_google/rander_obj_image/1026/obj/1026.obj" \
    --output_path "/data/datasets_20t/Downloads_google/rander_obj_image/1026/output"
    
    blender --background --python /data/code/pose_toolkit/render_img.py -- \
    --sparse_file "/data/datasets_20t/Downloads_google/rander_obj_image/959/render_pose/" \
    --mesh_file "/data/datasets_20t/Downloads_google/rander_obj_image/959/obj/959.obj" \
    --output_path "/data/datasets_20t/Downloads_google/rander_obj_image/959/output"
    
    """