z轴平移

post_pro/conv_obj.py

@@ -34,20 +34,25 @@ class ConvertOBJ:
             first_grid_dir, "odm_orthophoto", "odm_orthophoto_log.txt")
         self.ref_east, self.ref_north = self.read_utm_offset(log_file)
 
+        # 以第一个grid计算高度的偏移量
+        min_z = self.get_min_z_from_obj(os.path.join(first_grid_dir, 'odm_texturing', 'odm_textured_model_geo.obj'))
+        min_25d_z = self.get_min_z_from_obj(os.path.join(first_grid_dir, 'odm_texturing_25d', 'odm_textured_model_geo.obj'))
+        high = min_25d_z - min_z
+
         for grid_id in grid_points.keys():
             try:
                 self._convert_single_grid(grid_id, grid_points)
             except Exception as e:
                 self.logger.error(f"网格 {grid_id} 转换失败: {str(e)}")
 
-        self._create_merged_metadata()
+        self._create_merged_metadata(high)
 
     def _convert_single_grid(self, grid_id, grid_points):
         """转换单个网格的OBJ文件"""
         # 1. 构建相关路径
         grid_name = f"grid_{grid_id[0]}_{grid_id[1]}"
         project_dir = os.path.join(self.output_dir, grid_name, "project")
-        texturing_dir = os.path.join(project_dir, "odm_texturing_25d")
+        texturing_dir = os.path.join(project_dir, "odm_texturing")
         texturing_dst_dir = os.path.join(project_dir, "odm_texturing_dst")
         opensfm_dir = os.path.join(project_dir, "opensfm")
         log_file = os.path.join(
@@ -83,7 +88,7 @@ class ConvertOBJ:
             tile_dir, f"Tile_{grid_id[0]}_{grid_id[1]}.osgb")
         shutil.copy2(output_osgb, target_osgb)
 
-    def _create_merged_metadata(self):
+    def _create_merged_metadata(self, high):
         """创建合并后的metadata.xml文件"""
         # 转换为WGS84经纬度
         center_lon, center_lat = self.transformer.transform(
@@ -91,7 +96,7 @@ class ConvertOBJ:
         metadata_content = f"""<?xml version="1.0" encoding="utf-8"?>
 <ModelMetadata version="1">
     <SRS>EPSG:4326</SRS>
-    <SRSOrigin>{center_lon},{center_lat},0.000000</SRSOrigin>
+    <SRSOrigin>{center_lon},{center_lat},{high}</SRSOrigin>
         <Texture>
             <ColorSource>Visible</ColorSource>
         </Texture>
@@ -211,3 +216,18 @@ class ConvertOBJ:
                         img = cv2.imread(dst_path, cv2.IMREAD_UNCHANGED)
                         self.logger.info(
                             f"下采样后文件大小: {file_size / (1024 * 1024):.2f} MB")
+
+    def get_min_z_from_obj(self, file_path):
+        min_z = float('inf')  # 初始值设为无穷大
+        with open(file_path, 'r') as obj_file:
+            for line in obj_file:
+                # 检查每一行是否是顶点定义（以 'v ' 开头）
+                if line.startswith('v '):
+                    # 获取顶点坐标
+                    parts = line.split()
+                    # 将z值转换为浮动数字
+                    z = float(parts[3])
+                    # 更新最小z值
+                    if z < min_z:
+                        min_z = z
+        return min_z