UAV/post_pro/obj_post_pro.py

import os
import logging
import subprocess
from typing import Tuple, Dict


class ObjPostProcessor:
    def __init__(self, output_dir: str):
        self.output_dir = output_dir
        self.logger = logging.getLogger('UAV_Preprocess.ObjPostProcessor')

    def create_metadata_xml(self, osgb_dir: str, lon: float, lat: float, bounding_box):
        """创建metadata.xml文件，包含地理参考信息
        Args:
            osgb_dir: osgb输出目录
            lon: 中心点经度
            lat: 中心点纬度
        """
        try:
            metadata_content = f'''<?xml version="1.0" encoding="utf-8"?>
    <ModelMetadata version="1">
        <!-- Spatial Reference System -->
        <SRS>EPSG:4326</SRS>

        <!-- Center point in Spatial Reference System (in Longitude, Latitude, Height) -->
        <SRSOrigin>{lon},{lat},0.000000</SRSOrigin>

        <!-- Bounding Box with Two Points -->
        <BoundingBox>
            <!-- Left-Bottom Corner (Longitude, Latitude) -->
            <LB_lon>{bounding_box[0]}</LB_lon>
            <LB_lat>{bounding_box[1]}</LB_lat>

            <!-- Right-Top Corner (Longitude, Latitude) -->
            <RU_lon>{bounding_box[2]}</RU_lon>
            <RU_lat>{bounding_box[3]}</RU_lat>
        </BoundingBox>

        <Texture>
            <ColorSource>Visible</ColorSource>
        </Texture>
    </ModelMetadata>
    '''
            
            # metadata.xml 放在根目录
            metadata_path = os.path.join(osgb_dir, 'metadata.xml')
            with open(metadata_path, 'w', encoding='utf-8') as f:
                f.write(metadata_content)
                
            self.logger.info(f"已创建metadata.xml: {metadata_path}")
            
        except Exception as e:
            self.logger.error(f"创建metadata.xml时发生错误: {str(e)}")
            raise

    def convert_to_osgb(self, center_lon, center_lat, bounding_box):
        """将obj转换为osgb，并创建metadata.xml
        Args:
            center_coords: (longitude, latitude)中心点经纬度坐标
        """
        try:
            # 获取合并后的obj文件路径
            obj_dir = os.path.join(self.output_dir, 'texturing')
            obj_file = os.path.join(obj_dir, 'textured_model.obj')
            if not os.path.exists(obj_file):
                raise Exception(f"未找到obj文件: {obj_file}")

            # 创建osgb目录结构
            osgb_dir = os.path.join(self.output_dir, 'osgb')
            osgb_data_dir = os.path.join(osgb_dir, 'Data', 'textured_model')
            os.makedirs(osgb_data_dir, exist_ok=True)

            # 输出文件路径
            output_osgb = os.path.join(osgb_data_dir, 'textured_model.osgb')

            # 构建osgconv命令
            cmd = [
                'osgconv',
                '--compressed',
                '--smooth',
                '--fix-transparency',
                '-o', '0,1,0-0,0,-1',
                obj_file,
                output_osgb
            ]

            # 执行命令
            self.logger.info(f"执行osgconv命令：{' '.join(cmd)}")
            result = subprocess.run(cmd, capture_output=True, text=True)

            if result.returncode != 0:
                raise Exception(f"osgb格式转换失败: {result.stderr}")
            self.logger.info(f"转换完成: {output_osgb}")

            # 创建metadata.xml
            self.create_metadata_xml(osgb_dir, center_lon, center_lat, bounding_box)

            return True

        except Exception as e:
            self.logger.error(f"转换osgb时发生错误: {str(e)}")
            return False