ODM_pro/utils/command_runner.py

import os
import logging
import subprocess
import time
from typing import Dict
import pandas as pd
from preprocess.odm_monitor import ODMProcessMonitor


class CommandRunner:
    """执行网格处理命令的类"""

    def __init__(self, output_dir: str, max_retries: int = 3, mode: str = "快拼模式"):
        """
        初始化命令执行器
i
        Args:
            output_dir: 输出目录路径
            max_retries: 最大重试次数
        """
        self.output_dir = output_dir
        self.max_retries = max_retries
        self.logger = logging.getLogger('UAV_Preprocess.CommandRunner')
        self.monitor = ODMProcessMonitor(max_retries=max_retries, mode=mode)
        self.mode = mode

    def _run_command(self, grid_idx: int):
        """
        执行单个网格的命令

        Args:
            grid_idx: 网格索引

        Raises:
            Exception: 当命令执行失败时抛出异常
        """
        try:
            grid_dir = os.path.join(self.output_dir, f'grid_{grid_idx + 1}')
            grid_dir = grid_dir[0].lower() + grid_dir[1:].replace('\\', '/')
            if self.mode == "快拼模式":
                command = f"docker run -ti --rm -v {grid_dir}:/datasets opendronemap/odm --project-path /datasets project --feature-quality lowest --force-gps --fast-orthophoto --skip-3dmodel"
            else:
                command = f"docker run -ti --rm -v {grid_dir}:/datasets opendronemap/odm --project-path /datasets project --feature-quality lowest --force-gps"

            self.logger.info(f"开始执行命令: {command}")
            success, error_msg = self.monitor.run_odm_with_monitor(
                command, grid_dir, grid_idx)

            if not success:
                raise Exception(error_msg)

        except Exception as e:
            self.logger.error(f"网格 {grid_idx + 1} 处理失败: {str(e)}")
            raise

    def run_grid_commands(self, grid_points: Dict[int, pd.DataFrame], enable_grid_division: bool = True):
        """
        为每个网格顺序运行指定命令

        Args:
            grid_points: 网格点数据字典，键为网格索引，值为该网格的点数据
            enable_grid_division: 是否启用网格划分
        """
        if not enable_grid_division:
            self._run_command(0)
            return

        self.logger.info("开始执行网格处理命令")

        for grid_idx in grid_points.keys():
            try:
                self._run_command(grid_idx)
            except Exception as e:
                self.logger.error(f"网格 {grid_idx + 1} 处理失败，停止后续执行: {str(e)}")
                raise