import sys
from PyQt5.QtWidgets import QScrollArea, QApplicati
on, QMainWindow, QVBoxLayout, QHBoxLayout, QWidget, QPushButt
on, QLineEdit, QLabel, QFileDialog, QRadioButt
on, QComboBox, QCheckBox, QGroupBox, QListWidget, QProgressBar
from PyQt5.QtCore import QUrl, QRegExp, QTimer, Qt
from PyQt5.QtWebEngineWidgets import QWebEngineView, QWebEnginePage
from PyQt5.QtG
ui import QIc
on, QIntValidator, QRegExpValidator
from PyQt5.QtWebChannel import QWebChannel # 导入 QWebChannel
import c
onfigparser
import os
import subprocess
from PyQt5.QtWidgets import QMessageBox
import js
on
import time
import docker
import ctypes
import logging
import tkinter as tk
from tkinter import messagebox
import threading
import psutil
import shutil
import 文件服务2
from 服务校验 import validate_service
# 外部函数
sys.path.append
(os.path.dirname
(__file__
)) # 添加当前文件的目录到路径
# import 连接数据库添加历史倾斜摄影 # 直接
if getattr
(sys, 'frozen', False
):
# 如果是打包后的exe，使用sys.executable
base_dir = os.path.dirname
(sys.executable
)
else:
# 如果是普通脚本，使用__file__
base_dir = os.path.dirname
(os.path.abspath
(__file__
))
# 将'\'替换为'/'
base_dir = base_dir.replace
('\\', '/'
)
# 将第一个字母盘符写为大写
base_dir = base_dir[0].upper
() + base_dir[1:]
print
('文件路径',base_dir
)
class MainWindow
(QMainWindow
):
def __init__
(self
):
super
().__init__
()
self.task_started = True # 标志任务是否开始
self.current_img_index = 0 # 当前
处理的图片索引
self.setWindowTitle
("实时三维"
)
self.setGeometry
(100, 100, 1366, 768
)
self.setWindowIc
on(QIc
on('./ic
on.ico'
))
self.process = N
one # 用于存储子进程的引用
check_docker_running
() # 检查docker是否运行
self.node_process = N
one # 用于存储 node 进程的引用
self.start_node_process
() # 启动 node 进程
# 初始化变量存储路径和选择值
self.paths = {
"project": base_dir,
"sensor": "",
"1": "",
"texture": os.path.join
(base_dir, "ModelingScope.kml"
),
"localti
ons":''
}
self.
command_line = ""
self.data_source = "影像"
self.current_progress_key = N
one
# 主窗口布局
main_layout = QHBoxLayout
()
# 左侧
控件布局
left_layout = QVBoxLayout
()
# 示例
控件
self.start_butt
on = QPushButt
on("开始任务"
)
# 开始任务按钮点击事件绑定run_
command方法
self.start_butt
on.clicked.c
onnect
(lambda:self.run_
command(self.
command_line
))
left_layout.addWidget
(self.start_butt
on)
# 只准点击一次开始任务按钮，点击后状态显示停止任务
# stop_butt
on = QPushButt
on("停止任务"
)
# stop_butt
on.clicked.c
onnect
(lambda:self.stopTask
())
# left_layout.addWidget
(stop_butt
on)
# left_layout.addWidget
(QLabel
("空闲"
))
# 项目工程路径
# self.createPathInput
(left_layout, "项目工程路径:", "project"
)
stop_butt
on2 = QPushButt
on("文件传输"
)
stop_butt
on2.clicked.c
onnect
(lambda:self.file_transfer
())
left_layout.addWidget
(stop_butt
on2
)
# 平台上传
stop_butt
on3 = QPushButt
on("上传平台"
)
stop_butt
on3.clicked.c
onnect
(lambda:self.file_transfer_to_platform
())
left_layout.addWidget
(stop_butt
on3
)
# 任务队列路径
self.createPathInput
(left_layout, "任务队列路径:", "sensor"
)
# 图片文件路径
self.createPathInput
(left_layout, "图片文件路径:", "localti
ons"
)
# # 相机文件路径
# self.createPathInput_file
(left_layout, "相机文件路径
(.js
on):", "1", "js
on"
)
left_layout.addWidget
(QLabel
("数据源:"
))
radiobutt
ons_layout = QHBoxLayout
()
radio_image = QRadioButt
on("影像"
)
# radio_video = QRadioButt
on("视频"
)
radio_image.setChecked
(True
)
radio_image.toggled.c
onnect
(lambda: self.setDataSource
("影像", radio_image.isChecked
()))
# radio_video.toggled.c
onnect
(lambda: self.setDataSource
("视频", radio_video.isChecked
()))
radiobutt
ons_layout.addWidget
(radio_image
)
# radiobutt
ons_layout.addWidget
(radio_video
)
left_layout.addLayout
(radiobutt
ons_layout
)
# 建模范围
# self.createPathInput_file
(left_layout, "建模范围
(.kml
):", "texture", "kml"
)
# 经纬度输入部分
# left_layout.addWidget
(QLabel
("输入建模范围:"
))
# self.coordinates_layout = QVBoxLayout
()
# # 添加一个默认的坐标输入
# for _ in range
(4
):
# self.add_coordinate_input
()
# add_butt
on = QPushButt
on("+"
)
# add_butt
on.clicked.c
onnect
(self.add_coordinate_input
)
# self.coordinates_layout.addWidget
(add_butt
on)
# left_layout.addLayout
(self.coordinates_layout
)
# 添加经纬度输入框
# 初始化经纬度输入框布局
self.coordinates_layout = QVBoxLayout
()
self.input_widgets = []
# 添加四组默认的输入框
for _ in range
(4
):
self.add_coordinate_input
()
# 添加和删除按钮
butt
on_layout = QHBoxLayout
()
add_butt
on = QPushButt
on("+"
)
remove_butt
on = QPushButt
on("-"
)
add_butt
on.clicked.c
onnect
(self.add_coordinate_input
)
remove_butt
on.clicked.c
onnect
(self.remove_coordinate_input
)
butt
on_layout.addWidget
(add_butt
on)
butt
on_layout.addWidget
(remove_butt
on)
# 滚动区域
scroll_area = QScrollArea
()
scroll_c
ontent = QWidget
()
scroll_c
ontent.setLayout
(self.coordinates_layout
)
scroll_area.setWidget
(scroll_c
ontent
)
scroll_area.setWidgetResizable
(True
)
scroll_area.setFixedHeight
(180
)
left_layout.addWidget
(QLabel
("输入建模范围:"
))
left_layout.addWidget
(scroll_area
)
left_layout.addLayout
(butt
on_layout
)
# 相对航高（米）
# left_layout.addWidget
(QLabel
("相对航高（米）:"
))
# self.elevati
on_input = QLineEdit
(""
)
# left_layout.addWidget
(self.elevati
on_input
)
# 地面分辨率
# left_layout.addWidget
(QLabel
("像元大小
(um
):"
))
# self.minphot
onSize = QLineEdit
(""
)
# left_layout.addWidget
(self.minphot
onSize
)
# 建模结果文件名
left_layout.addWidget
(QLabel
("建模结果文件名:"
))
self.result_file_name = QLineEdit
(""
)
left_layout.addWidget
(self.result_file_name
)
self.result_file_name.textChanged.c
onnect
(self.
update_js
on)
# 建模精度
left_layout.addWidget
(QLabel
("建模精度:"
))
self.precisi
on_combo = QComboBox
()
self.precisi
on_combo.addItem
("快速"
)
self.precisi
on_combo.addItem
("普通"
)
self.precisi
on_combo.addItem
("精细"
)
left_layout.addWidget
(self.precisi
on_combo
)
# 高级设置
advanced_group = QGroupBox
("高级设置"
)
advanced_layout = QVBoxLayout
()
advanced_layout.addWidget
(QLabel
("图片集大小（张）:"
))
self.tile_size_input = QLineEdit
("20"
)
self.tile_size_input.setValidator
(QIntValidator
()) # 限制输入为整数
advanced_layout.addWidget
(self.tile_size_input
)
advanced_layout.addWidget
(QLabel
("照片获取等待时间（秒）:"
))
self.interval_input = QLineEdit
("10"
)
self.interval_input.setValidator
(QIntValidator
()) # 限制输入为整数
advanced_layout.addWidget
(self.interval_input
)
advanced_layout.addWidget
(QLabel
("边飞边建:"
))
self.checkbox1 = QRadioButt
on("正射"
)
self.checkbox2 = QRadioButt
on("倾斜"
)
self.checkbox3 = QRadioButt
on("全建"
)
self.checkbox1.setChecked
(False
)
self.checkbox2.setChecked
(True
)
self.checkbox3.setChecked
(False
)
# advanced_layout.addWidget
(checkbox1
)
# advanced_layout.addWidget
(checkbox2
)
advanced_layout.addWidget
(self.checkbox1
)
advanced_layout.addWidget
(self.checkbox2
)
advanced_layout.addWidget
(self.checkbox3
)
# 按钮
# Merge_models = QPushButt
on("合并模型"
)
# c
onnecti
on = 连接数据库添加历史倾斜摄影.create_c
onnecti
on("localhost", "root", "ysxx_0407H123", "product-backsage2"
)
# Merge_models.clicked.c
onnect
(lambda:连接数据库添加历史倾斜摄影.query_users
(c
onnecti
on))
# advanced_layout.addWidget
(Merge_models
)
advanced_group.setLayout
(advanced_layout
)
left_layout.addWidget
(advanced_group
)
# 实时状态显示
self.status_group = QGroupBox
("实时状态"
)
status_layout = QVBoxLayout
()
self.file_count_label = QLabel
("文件数量: 0"
)
self.file_names_list = QListWidget
()
self.docker_count_label = QLabel
("Docker容器数量: 0"
)
# 添加进度条和文字标签
self.progress_stage_label = QLabel
("进度阶段: 第一块进度"
)
self.progress_bar = QProgressBar
()
self.progress_bar.setAlignment
(Qt.AlignCenter
)
self.progress_bar.setRange
(0, 100
)
self.progress_bar.setValue
(0
)
status_layout.addWidget
(self.file_count_label
)
status_layout.addWidget
(self.file_names_list
)
status_layout.addWidget
(self.docker_count_label
)
status_layout.addWidget
(self.progress_stage_label
)
status_layout.addWidget
(self.progress_bar
)
self.status_group.setLayout
(status_layout
)
left_layout.addWidget
(self.status_group
)
# 设置左侧宽度
left_widget = QWidget
()
left_widget.setLayout
(left_layout
)
left_widget.setFixedWidth
(300
)
# 创建Web引擎视图
self.web_view = QWebEngineView
()
self.web_view.load
(QUrl
("http://localhost:3123"
))
web_channel = QWebChannel
(self.web_view.page
()) # 创建 WebChannel 实例
self.web_view.page
().setWebChannel
(web_channel
) # 关联 WebChannel
self.web_view.page
().runJavaScript
("setInterval
(functi
on(){locati
on.reload
()}, 1000
);"
)
# 将左侧
控件和Web视图添加到主布局
main_layout.addWidget
(left_widget
)
main_layout.addWidget
(self.web_view
)
# 设置主widget
central_widget = QWidget
()
central_widget.setLayout
(main_layout
)
self.setCentralWidget
(central_widget
)
# 定时
更新状态
self.timer = QTimer
()
self.timer.timeout.c
onnect
(self.
update_status
)
self.timer.start
(1000
) # 每秒
更新一次状态
# 启动进度条
更新循环
self.
update_progress_timer = QTimer
()
self.
update_progress_timer.timeout.c
onnect
(self.
update_progress_bar
)
self.
update_progress_timer.start
(1000
) # 每秒检查一次进度
def
update_status
(self
):
#
更新文件数量和文件名
folder_path = self.paths["sensor"]
if os.path.exists
(folder_path
):
files = os.listdir
(folder_path
)
self.file_count_label.setText
(f"文件数量: {len
(files
)}"
)
self.file_names_list.clear
()
for file_name in files:
self.file_names_list.addItem
(file_name
)
else:
self.file_count_label.setText
("文件数量: 0"
)
self.file_names_list.clear
()
#
更新Docker容器数量
try:
result = subprocess.run
(
["docker", "ps", "--filter", "ancestor=opendr
onemap/odm:gpu", "--format", "{{.ID}}"],
capture_output=True,
text=True,
shell=False,
creati
onflags=subprocess.CREATE_NO_WINDOW # 确保不弹出窗口
)
c
ontainer_ids = result.stdout.strip
().split
('\n'
)
if c
ontainer_ids and c
ontainer_ids[0] == '':
c
ontainer_ids = [] # 如果结果为空字符串，将其设为空列表
logging.debug
(f"C
ontainer IDs: {c
ontainer_ids}"
)
self.docker_count_label.setText
(f"Docker容器数量: {len
(c
ontainer_ids
)}"
)
except subprocess.CalledProcessError as e:
logging.error
(f"获取Docker容器信息时出错: {e}"
)
self.docker_count_label.setText
("Docker容器数量: 0"
)
except FileNotFoundError as e:
logging.error
(f"Docker命令未找到: {e}"
)
self.docker_count_label.setText
("Docker容器数量: 0"
)
def add_coordinate_input
(self
):
"""添加一个新的经纬度输入行"""
layout = QHBoxLayout
()
lng_input = QLineEdit
()
lat_input = QLineEdit
()
lng_input.setPlaceholderText
("经度"
)
lat_input.setPlaceholderText
("纬度"
)
# 使用正则表达式限制输入为正负加小数
validator = QRegExpValidator
(QRegExp
(r"^-?\d+
(\.\d+
)?$"
))
lng_input.setValidator
(validator
)
lat_input.setValidator
(validator
)
layout.addWidget
(lng_input
)
layout.addWidget
(lat_input
)
self.coordinates_layout.addLayout
(layout
)
self.input_widgets.append
((lng_input, lat_input
))
# 收集经纬度坐标
coordinates = []
for lng_input, lat_input in self.input_widgets:
lng = lng_input.text
()
lat = lat_input.text
()
print
(lng, lat
)
if lng and lat:
coordinates.append
(f"{lng},{lat}"
)
# 将所有经纬度坐标合并为一个字符串，用分号分隔
coordinates_str = ";".join
(coordinates
)
print
("经纬度",coordinates_str
)
# 将经纬度加入kml文件
def generate_kml
(self
):
"""生成KML内容"""
kml_c
ontent = f"""<?xml versi
on="1.0" encoding="utf-8"?>
<kml xmlns="http://www.opengis.net/kml/2.2">
<Document>
<Schema id="Dataset_203sup" name="Dataset_203sup">
<SimpleField name="SmUserID" type="int"/>
</Schema>
<Folder>
<Placemark>
<name/>
<ExtendedData>
<SchemaData schemaUrl="#Dataset_203sup">
<SimpleData name="SmUserID">0</SimpleData>
</SchemaData>
</ExtendedData>
<Polyg
on>
<outerBoundaryIs>
<LinearRing>
<coordinates>
"""
coordinates = []
for lng_input, lat_input in self.input_widgets:
lng = lng_input.text
()
lat = lat_input.text
()
if lng and lat:
coordinates.append
(f"{lng},{lat}"
)
# 在最后再添加第一个坐标点
if coordinates:
coordinates.append
(coordinates[0]
)
kml_c
ontent += "\n".join
(coordinates
) + "\n </coordinates>\n </LinearRing>\n </outerBoundaryIs>\n </Polyg
on>\n </Placemark>\n </Folder>\n </Document>\n</kml>"
# 写入KML文件
kml_path = self.paths["texture"]
with open
(kml_path, 'w', encoding='utf-8'
) as kml_file:
kml_file.write
(kml_c
ontent
)
print
("KML文件已写入:", kml_path
)
def remove_coordinate_input
(self
):
"""删除最下面的一个经纬度输入行"""
if len
(self.input_widgets
) > 4:
layout = self.coordinates_layout.takeAt
(len
(self.input_widgets
) - 1
)
for i in range
(layout.count
()):
widget = layout.itemAt
(i
).widget
()
if widget:
widget.deleteLater
()
self.input_widgets.pop
()
else:
QMessageBox.warning
(self, "提示", "至少需要保留四组坐标。"
)
def start_node_process
(self
):
"""启动 node 进程并保存其引用"""
self.node_process = subprocess.Popen
('node result_server_logs.js', shell=False, creati
onflags=subprocess.CREATE_NO_WINDOW
)
def
update_js
on(self
):
# 获取建模结果文件名
result_filename = self.result_file_name.text
()
# 创建要写入的 JS
ON 数据
data = {
"name": result_filename
}
# 写入 JS
ON 文件
with open
('./result/resultName.js
on', 'w', encoding='utf-8'
) as js
on_file:
js
on.dump
(data, js
on_file, ensure_ascii=False, indent=4
)
def create_c
onfig_file
(self,
):
"""生成配置文件 c
onfig.ini"""
c
onfig = c
onfigparser.C
onfigParser
()
# 裁剪相机文件目录，若裁剪失败，弹窗提示用户重新选择文件
if self.paths["1"] is not N
one and self.paths["project"] is not N
one:
project_path = self.paths["project"] + "/"
if project_path in self.paths["1"]:
self.paths["1"] = self.paths["1"].replace
(project_path, ""
)
# else:
# QMessageBox.warning
(self, "提示", "项目路径未包含在相机文件路径中。"
)
else:
# QMessageBox.warning
(self, "提示", "路径不能为空。"
)
print
("路径不能为空。"
)
self.paths["1"] = self.paths["1"].replace
(self.paths["project"]+"/", ""
)
c
onfig['settings'] = {
'kmlPath': self.paths["texture"], # 建模范围
# 'elevati
on': self.elevati
on_input.text
(), # 相对航高
'precisi
on': self.precisi
on_combo.currentText
(), # 建模精度
'interval': '2.0', # 照片获取等待时间
'imagesnum': self.tile_size_input.text
(), # 每多少张图片生成一个文件夹
'imagesFolderPath':self.paths["project"]+'/'+ self.paths["sensor"], # 图像文件夹路径
# 'minphot
onSize':self.minphot
onSize.text
(), # 地面分辨率
'projectPath':self.paths["project"], # 项目工程路径
'taskQueuePath':self.paths["sensor"], # 任务队列路径
'cameraFile':self.paths["1"], # 相机文件路径
'dataSource':self.data_source, # 数据源
'customTileSize':self.tile_size_input.text
(), # 自定义瓦片大小
'imagesWaitTime':self.interval_input.text
(), # 照片获取等待时间
'DEM':False, # 是否生成DEM
'DSM':False, # 是否生成DSM
'zhengshe':self.checkbox1.isChecked
(), # 是否生成正射图
'qingxie':self.checkbox2.isChecked
(), # 是否生成倾斜图
'quanjian':self.checkbox3.isChecked
(), # 是否生成全建图
'minImagesPerSquare':4 , #建模一块所需数量
'resultNameDir':'result/MODEL/'+self.result_file_name.text
(), # 建模结果文件名
"overlapRatio":0.7, #重叠率
"sideOverlapRatio":0.8, #侧重叠率
"localti
ons":self.paths["localti
ons"],
"centerpointl
ongitude":114.25,
"centerpointlatitude":30.58,
"imagesnum":0,
"ziplocalti
ons":self.paths["project"]+'/'+ "ziplocalti
ons"
}
with open
('c
onfig.ini', 'w', encoding='utf-8'
) as c
onfigfile:
c
onfig.write
(c
onfigfile
)
print
("配置文件 c
onfig.ini 已创建。"
)
def run_
command(self,
command):
for lng_input, lat_input in self.input_widgets:
lng = lng_input.text
()
lat = lat_input.text
()
if not lng or not lat:
QMessageBox.warning
(self, "提示", "请填写完整的建模经纬度范围 。"
)
return
# 检查输入框是否为空
if not self.paths["sensor"]:
QMessageBox.warning
(self, "提示", "任务队列路径不能为空。"
)
return
self.task_started = True # 设置任务开始标志为 True
# # 任务路径不能和之前的相同
# c
onfig = c
onfigparser.C
onfigParser
()
# try:
# with open
('c
onfig.ini', 'r', encoding='utf-8'
) as c
onfigfile:
# c
onfig.read_file
(c
onfigfile
)
# except Excepti
on as e:
# print
(f"读取配置文件时出错: {e}"
)
# # return
# folder_path = c
onfig.get
('settings', 'imagesFolderPath', fallback=N
one
)
# print
(folder_path
)
# if folder_path == self.paths["project"]+"/"+self.paths["sensor"]:
# QMessageBox.warning
(self, "提示", "任务队列路径不能重复。"
)
# return
# 清空目标文件夹，如果存在
# destinati
on_folder = self.paths["project"]+"/"+self.paths["sensor"]
# if os.path.exists
(destinati
on_folder
):
# shutil.rmtree
(destinati
on_folder
)
# if not self.paths["camera_file"]:
# QMessageBox.warning
(self, "提示", "相机文件路径不能为空。"
)
# return
if not self.paths["texture"]:
QMessageBox.warning
(self, "提示", "建模范围路径不能为空。"
)
return
if not self.paths["localti
ons"]:
QMessageBox.warning
(self, "提示", "图片文件路径不能为空。"
)
return
if not self.result_file_name.text
():
QMessageBox.warning
(self, "提示", "建模结果文件名不能为空。"
)
return
# if not self.elevati
on_input.text
():
# QMessageBox.warning
(self, "提示", "相对航高不能为空。"
)
# return
# if not self.minphot
onSize.text
():
# QMessageBox.warning
(self, "提示", "像元大小不能为空。"
)
# return
if not self.tile_size_input.text
():
QMessageBox.warning
(self, "提示", "自定义瓦片大小不能为空。"
)
return
if not self.interval_input.text
():
QMessageBox.warning
(self, "提示", "照片获取等待时间不能为空。"
)
return
self.create_c
onfig_file
() # 调用生成配置文件的方法
# 将按钮状态设为不可用
self.start_butt
on.setEnabled
(False
)
# self.start_butt
on.setText
("正在
处理..."
)
self.start_butt
on.repaint
()
self.start_butt
on.
update()
# 调用文件服务每多少张图生成一个文件夹
# 直接调用 main 函数
# 创建一个新的线程来运行文件服务
file_service_thread = threading.Thread
(target=文件服务2.main
)
file_service_thread.start
() # 启动线程
self.generate_kml
() # 生成KML文件
# 生成配置文件后，调用startB
uilding.exe
def is_exe_running
(exe_name
):
# 检查是否有指定的exe正在运行
for proc in psutil.process_iter
(['pid', 'name']
):
try:
if proc.info['name'] == exe_name:
return True
except
(psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess
):
pass
return False
# 先判断是否有startb
uilding.exe在运行，如果有则关闭所有docker容器，然后关闭startb
uilding.exe，再启动startb
uilding.exe
if is_exe_running
("startB
uilding.exe"
):
# 关闭所有基于"opendr
onemap/odm:gpu"的容器
self.stop_docker_c
ontainer
("opendr
onemap/odm:gpu"
)
# 关闭startB
uilding.exe
self.kill_exe
("startB
uilding.exe"
)
self.process =subprocess.Popen
("startB
uilding.exe", shell=False,creati
onflags=subprocess.CREATE_NO_WINDOW
)
time.sleep
(3
) # 等待1秒，等待startB
uilding.exe启动完成
#
刷新页面
self.web_view.reload
() # 使用 QWebEngineView 的 reload 方法
# print
(command)
# import subprocess
# subprocess.Popen
(command, shell=True
)
def file_transfer
(self
):
# 调用文件传输模块
self.process =subprocess.Popen
("地表建模-三维航测数据同步.exe", shell=False,creati
onflags=subprocess.CREATE_NO_WINDOW
)
# 平台上传
def file_transfer_to_platform
(self
):
# 调用文件传输模块
self.process =subprocess.Popen
("结果保存.exe", shell=False,creati
onflags=subprocess.CREATE_NO_WINDOW
)
def kill_exe
(self, exe_name
):
"""
杀死exe进程
:param exe_name:进程名字
:return:无
"""
os.system
('taskkill /f /t /im '+exe_name
)#MESMTPC.exe程序名字
print
("杀死进程{}".format
(exe_name
))
def stopTask
(self
):
"""停止任务并关闭占用端口3123的进程，同时关闭Docker容器"""
# 创建进度
对话框
# 创建一个新的窗口
# 显示关闭提示框
msg_box = QMessageBox
()
msg_box.setWindowTitle
("关闭提示"
)
msg_box.setText
("软件正在关闭中，请稍候..."
)
msg_box.setStandardButt
ons
(QMessageBox.NoButt
on)
msg_box.setModal
(True
) # 模态
对话框
msg_box.show
()
QApplicati
on.processEvents
() #
处理事件，确保
对话框显示
# 关闭 node 进程
if self.node_process: # 检查 node 进程是否存在
self.node_process.kill
() # 终止 node 进程
print
("已关闭 node 进程"
)
# 查找使用3123的进程
result = subprocess.run
("netstat -ano | findstr :3123", capture_output=True, text=True, shell=False, creati
onflags=subprocess.CREATE_NO_WINDOW
)
lines = result.stdout.splitlines
()
if lines:
for line in lines:
if line: # 确保不是空行
parts = line.split
()
pid = parts[-1] # 假设最后一部分是PID
if pid.isdigit
():
os.system
(f"taskkill /PID {pid} /F"
) # 强制关闭该进程
print
(f"已关闭占用端口3123的进程 {pid}"
)
# 停止 startB
uilding.exe
exe_name = 'startB
uilding.exe'
self.kill_exe
(exe_name
)
# 循环关闭Docker容器，直到3秒没有关闭为止
stop_attempts = 0
while stop_attempts < 5:
self.stop_docker_c
ontainer
("opendr
onemap/odm:gpu"
)
#
更新进度
对话框
# progress_dialog.setLabelText
("正在关闭Docker容器...请稍候..."
)
# 等待1秒
time.sleep
(1
)
# 检查容器是否还在运行
result = subprocess.run
(
["docker", "ps", "--filter", "ancestor=opendr
onemap/odm:gpu", "--format", "{{.ID}}"],
capture_output=True,
text=True
)
if not result.stdout.strip
(): # 如果没有运行的容器，退出循环
break
stop_attempts += 1
print
("任务停止完成。"
)
# 任务停止完成后关闭
对话框
msg_box.close
()
# 停止dockers
def stop_docker_c
ontainer
(self, image_name
):
# 获取所有基于image_name的容器ID
result = subprocess.run
(
["docker", "ps", "-f", f"ancestor={image_name}", "-q"],
capture_output=True,
text=True
)
c
ontainer_ids = result.stdout.strip
().split
('\n'
)
# 停止每个容器
for c
ontainer_id in c
ontainer_ids:
subprocess.run
(["docker", "stop", c
ontainer_id]
)
# 所有容器都停止后，返回True，否则返回False
return not c
ontainer_ids
# def createPathInput
(self, layout, label_text, path_key
):
# """创建一个路径输入组件，包括标签、输入框和选择按钮"""
# layout.addWidget
(QLabel
(label_text
))
# path_layout = QHBoxLayout
()
# path_edit = QLineEdit
(self.paths[path_key]
)
# path_butt
on = QPushButt
on("..."
)
# path_butt
on.setFixedWidth
(30
)
# path_butt
on.clicked.c
onnect
(lambda: self.selectPath
(path_edit, path_key
))
# path_edit.textChanged.c
onnect
(lambda text: self.savePath
(text, path_key
))
# path_layout.addWidget
(path_edit
)
# path_layout.addWidget
(path_butt
on)
# layout.addLayout
(path_layout
)
# # 选择文件路径
# def createPathInput_file
(self, layout, label_text, path_key,type
):
# """创建一个路径输入组件，包括标签、输入框和选择按钮"""
# layout.addWidget
(QLabel
(label_text
))
# path_layout = QHBoxLayout
()
# path_edit = QLineEdit
(self.paths[path_key]
)
# path_butt
on = QPushButt
on("..."
)
# path_butt
on.setFixedWidth
(30
)
# path_butt
on.clicked.c
onnect
(lambda: self.setPath
(path_edit, path_key,type
))
# path_edit.textChanged.c
onnect
(lambda text: self.savePath
(text, path_key
))
# path_layout.addWidget
(path_edit
)
# path_layout.addWidget
(path_butt
on)
# layout.addLayout
(path_layout
)
# def selectPath
(self, path_edit, path_key
):
# """打开文件
对话框，设置路径到输入框"""
# folder_path = QFileDialog.getExistingDirectory
(self, "选择文件夹"
)
# if folder_path:
# path_edit.setText
(folder_path
)
# self.savePath
(folder_path, path_key
)
# # 打开文件
对话框，选择文件路径，设置到输入框中，并保存到变量中
# def setPath
(self, path_edit, path_key,type
):
# # file_path, _ = QFileDialog.getOpenFileName
(self, "选择文件", "", "All Files
(*
)"
)
# # 只能选js
on文件
# if type=="js
on":
# file_path, _ = QFileDialog.getOpenFileName
(self, "选择文件", "", "js
on Files
(*.js
on)"
)
# elif type=="kml":
# file_path, _ = QFileDialog.getOpenFileName
(self, "选择文件", "", "kml Files
(*.kml
)"
)
# if file_path:
# path_edit.setText
(file_path
)
# self.savePath
(file_path, path_key
)
def createPathInput
(self, layout, label_text, path_key
):
"""创建一个路径输入组件，包括标签、输入框和选择按钮"""
layout.addWidget
(QLabel
(label_text
))
path_layout = QHBoxLayout
()
path_edit = QLineEdit
(self.paths[path_key]
)
path_edit.setRead
Only
(True
) # 设置为只读状态
path_butt
on = QPushButt
on("..."
)
path_butt
on.setFixedWidth
(30
)
path_butt
on.clicked.c
onnect
(lambda: self.selectPath
(path_edit, path_key
))
path_edit.textChanged.c
onnect
(lambda text: self.savePath
(text, path_key
))
path_layout.addWidget
(path_edit
)
path_layout.addWidget
(path_butt
on)
layout.addLayout
(path_layout
)
def createPathInput_file
(self, layout, label_text, path_key, type
):
"""创建一个路径输入组件，包括标签、输入框和选择按钮"""
layout.addWidget
(QLabel
(label_text
))
path_layout = QHBoxLayout
()
path_edit = QLineEdit
(self.paths[path_key]
)
path_butt
on = QPushButt
on("..."
)
path_butt
on.setFixedWidth
(30
)
path_butt
on.clicked.c
onnect
(lambda: self.setPath
(path_edit, path_key, type
))
path_edit.textChanged.c
onnect
(lambda text: self.savePath
(text, path_key
))
path_layout.addWidget
(path_edit
)
path_layout.addWidget
(path_butt
on)
layout.addLayout
(path_layout
)
def selectPath
(self, path_edit, path_key
):
"""打开文件夹选择
对话框，设置路径到输入框"""
folder_path = QFileDialog.getExistingDirectory
(self, "选择文件夹", base_dir
)
base_path = base_dir
if folder_path:
# 假如path_key是localti
ons，则不需要判断是否在C:/RealTimeModeling下
if path_key == "localti
ons":
path_edit.setText
(folder_path
)
self.savePath
(folder_path, path_key
)
else:
print
(folder_path,base_dir,folder_path.startswith
(base_dir
))
# 检查选择的目录是否在 C:/RealTimeModeling 下
if folder_path.startswith
(base_dir
):
path_edit.setText
(folder_path
)
self.savePath
(folder_path, path_key
)
else:
QMessageBox.warning
(self, "提示", f"只能选择123 {base_path} 目录下的文件夹。"
)
def setPath
(self, path_edit, path_key, type
):
"""打开文件
对话框，选择文件路径，设置到输入框中，并保存到变量中"""
base_path = base_dir
if type == "js
on":
file_path, _ = QFileDialog.getOpenFileName
(self, "选择文件", base_dir, "JS
ON Files
(*.js
on)"
)
elif type == "kml":
file_path, _ = QFileDialog.getOpenFileName
(self, "选择文件", base_dir, "KML Files
(*.kml
)"
)
# Check if the selected path starts with the allowed directory
if file_path and file_path.startswith
(base_dir
):
path_edit.setText
(file_path
)
self.savePath
(file_path, path_key
)
else:
QMessageBox.warning
(self, "提示", f"只能选择 {base_path} 目录下的文件夹。"
)
def savePath
(self, path, path_key
):
"""保存路径到变量"""
self.paths[path_key] = path
print
(self.paths
)
self.
updateCommand()
def
updateCommand(self
):
# 原始命令 docker run -ti --rm -v C:\3d\源文件\皂角林:/datasets --gpus all opendr
onemap/odm:gpu --project-path /datasets 皂角林 --3d-tiles --pc-rectify --pc-ept --pc-quality low --feature-quality medium --boundary "/datasets/皂角林/boundary.js
on" --orthophoto-png --orthophoto-cutline
line1=":/datasets --gpus all opendr
onemap/odm:gpu --project-path /datasets "
line2= " --3d-tiles --pc-rectify --orthophoto-png --orthophoto-cutline --pc-ept --pc-quality low --feature-quality medium --boundary /datasets"
# self.paths["sensor"]去掉包含 self.paths["project"] 的字符串
self.paths["sensor"] = self.paths["sensor"].replace
(self.paths["project"]+"/", ""
)
# self.paths["texture"]=self.paths["texture"].replace
(self.paths["project"], ""
)
"""
更新命令行"""
# self.
command_line = "docker run --rm -ti -v " + self.paths["project"] + ":/data/project -v " + self.paths["sensor"] + ":/data/sensor -v " + self.paths["camera_file"] + ":/data/camera_file -v " + self.paths["texture"] + ":/data/texture -p 3123:3123 zyf/3dmodel:latest " + self.data_source + " " + self.elevati
on_input.text
() + " " + self.precisi
on_combo.currentText
() + " " + self.tile_size_input.text
() + " " + self.interval_input.text
() + " " + " ".join
([str
(checkbox.isChecked
()) for checkbox in [checkbox1, checkbox2, checkbox3]]
)
print
("工程目录",
self.paths["project"] ,
"任务队列",
self.paths["sensor"],
"相机文件",
# self.paths["1"],
"建模范围",
self.paths["texture"],
"数据源",
self.data_source ,
# "相对航高",
# self.elevati
on_input.text
(),
"建模精度",
# self.precisi
on_combo.currentText
() ,
"自定义瓦片大小",
self.tile_size_input.text
(),
"照片获取等待时间",
self.interval_input.text
() ,
"是否生成正射",
self.checkbox3.isChecked
()
)
# " ".join
([str
(checkbox.isChecked
()) for checkbox in [self.checkbox1, self.checkbox2, self.checkbox3]]
))
def setDataSource
(self, source, checked
):
"""保存数据源选择"""
if checked:
self.data_source = source
print
(self.data_source
)
def closeEvent
(self, event
):
"""重写关闭事件以在退出时停止任务"""
self.stopTask
() # 调用停止任务的方法
event.accept
() # 允许窗口关闭
def
update_progress_bar
(self
):
"""根据Progress.ini文件
更新进度条，支持多个 imgs 阶段
处理"""
if not self.task_started:
return # 如果任务未开始，不
更新进度条
c
onfig = c
onfigparser.C
onfigParser
()
c
onfig.read
('d:\\实时建模\\实时建模V3\\Progress.ini'
)
# 获取当前要
处理的 imgsi
current_index = getattr
(self, 'current_img_index', 0
)
current_key = f'imgs{current_index + 1}'
# 检查是否有新的 imgsi=1 出现
if 'Progress' in c
onfig and current_key in c
onfig['Progress'] and c
onfig['Progress'][current_key] == '1':
print
(f"检测到 {current_key}=1，立即进入完成阶段"
)
self.current_img_index += 1
self.progress_stage_label.setText
(f"进度阶段: 第{self.current_img_index}块进度"
)
self._start_immediate_complete
()
return
# 如果还没开始自动增长，并且没有检测到任何 imgsi=1
if not hasattr
(self, '_auto_grow_started'
):
self._auto_grow_started = True
self._auto_grow_start_time = time.time
()
self._auto_grow_target = 80
self._auto_grow_durati
on = 120 # 120秒增长到80%
self._auto_grow_start_value = 0
self.progress_bar.setValue
(0
)
self.progress_stage_label.setText
(f"进度阶段: 第{self.current_img_index}块进度"
)
# 计算当前时间进度
elapsed = time.time
() - self._auto_grow_start_time
if elapsed >= self._auto_grow_durati
on:
self.progress_bar.setValue
(self._auto_grow_target
)
return
# 线性增长
progress = int
((elapsed / self._auto_grow_durati
on) * self._auto_grow_target
)
self.progress_bar.setValue
(min
(progress, self._auto_grow_target
))
# 每秒检查一次进度
QTimer.singleShot
(1000, self.
update_progress_bar
)
def _start_immediate_complete
(self
):
"""
处理 imgsi=1 的情况：5秒内增长到100%，5秒后归零"""
start_value = self.progress_bar.value
()
target_value = 100
durati
on = 5000 # 5秒
steps = 50 # 每次
更新间隔
delta =
(target_value - start_value
) /
(durati
on / steps
)
def
update():
current = self.progress_bar.value
()
if current >= target_value:
QTimer.singleShot
(5000, self._reset_progress
) # 5秒后归零
return
# 强制转换为整数
next_value = int
(min
(current + delta, target_value
))
self.progress_bar.setValue
(next_value
)
QTimer.singleShot
(steps,
update)
update()
def run_
commandss
(command):
logging.info
(f"运行命令: {
command}"
)
result = 0
result = subprocess.run
(command, shell=True
)
return result.returncode
def check_docker_running
():
"""检查 Docker 是否正在运行，如果没有运行则显示警告弹窗"""
# try:
# client = docker.from_env
() # 尝试获取 Docker 客户端
# client.ping
() # 发送 ping 请求确认 Docker 是否可用
# except docker.errors.DockerExcepti
on as e:
# # 如果发生异常，说明 Docker 没有运行
# show_docker_warning
() # 显示 Docker 未运行的警告
# sys.exit
() # 退出程序
# except Excepti
on as e:
# # 捕获其他类型的异常
# show_docker_warning
()
# sys.exit
()
try:
# 运行 docker info 命令
result = subprocess.run
(['docker', 'info'], capture_output=True, text=True, check=True
)
# 打印输出信息
print
(result.stdout
)
print
("Docker 正常启动"
)
except subprocess.CalledProcessError as e:
# 如果命令
执行失败，打印错误信息
print
("Docker 未正常启动"
)
show_docker_warning
() # 显示 Docker 未运行的警告
print
(e.stderr
)
sys.exit
()
except FileNotFoundError:
# 如果 docker 命令未找到，说明 Docker 未安装
print
("Docker 未安装或未正确配置"
)
show_docker_warning
() # 显示 Docker 未运行的警告
sys.exit
()
def show_docker_warning
():
"""显示 Docker 未运行的警告"""
QMessageBox.warning
(N
one, "警告", "请打开 Docker 并确保它正常运行！重新开始建模任务！"
)
def start_service
():
# 验证许可证
if not validate_service
():
return # 如果验证失败，退出服务
# 验证成功，继续运行服务
run_service
()
def run_service
():
# 您的服务主逻辑
# 打开软件时清空结果文件名
# 创建要写入的 JS
ON 数据
data = {
"name": ''
}
# 写入 JS
ON 文件
with open
('./result/resultName.js
on', 'w', encoding='utf-8'
) as js
on_file:
js
on.dump
(data, js
on_file, ensure_ascii=False, indent=4
)
ctypes.windll.user32.ShowWindow
(ctypes.windll.kernel32.GetC
onsoleWindow
(), 0
)
# 运行一个命令node result_server_logs.js
# subprocess.Popen
('node result_server_logs.js', shell=False,creati
onflags=subprocess.CREATE_NO_WINDOW
)
app = QApplicati
on(sys.argv
)
window = MainWindow
()
window.show
()
sys.exit
(app.exec_
())
if __name__ == "__main__":
start_service
()
本文围绕MFC展开，探讨了基于对话框程序中OnIdle无法工作的问题，指出MFC在模态对话框处理上的变化及WM_ENTERIDLE的局限性，介绍了WM_KICKIDLE作为替代方案。还提及使ON_COMMAND_UPDATE_UI处理函数在对话框中工作的方法，需处理WM_KICKIDLE并调用UpdateDialogControls。
扫一扫
6133
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
