missum/ArcGis_Py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初始化

Browse Source
This commit is contained in:
missum
2026-04-22 12:27:49 +08:00
commit 4857cb6e45
73 changed files with 20927 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

291
tools/ui/main_window.py Normal file
View File

@@ -0,0 +1,291 @@
import traceback
import json
import os
import sys
from PyQt6.QtWidgets import (
QMainWindow,
QWidget,
QVBoxLayout,
QTabWidget,
QTextEdit,
QLabel,
QPushButton,
QHBoxLayout,
)
from PyQt6.QtCore import pyqtSignal, QThreadPool
from PyQt6.QtGui import QTextCursor
from .tabs.export_layout_tab import ExportImageTab
from .tabs.raster_tab import RasterTab
from .tabs.export_map_tab import ExportMapTab
from .tabs.area_stat_tab import XlsxToJpgTab
from .tabs.acid_stat_tab import AcidStatsTab # 酸化专题图表格统计
from .tabs.soil_prop_stat_tab import SoilPropStatsTab
class MainWindow(QMainWindow):
log_signal = pyqtSignal(str) # 定义日志信号
"""主窗口"""
def __init__(self):
super().__init__()
self.setWindowTitle("ArcGIS工具集")
self.resize(600, 800)
self.settings_file = os.path.join(
os.path.dirname(__file__), "config", "settings.json"
)
print(self.settings_file)
self.ensure_settings_file()
self.settings = self.load_settings()
self.init_ui()
# 在创建UI后加载设置
self.load_ui_settings()
def init_ui(self):
central_widget = QWidget()
self.setCentralWidget(central_widget)
main_layout = QVBoxLayout(central_widget)
# 添加选项卡
tabs = QTabWidget()
self.export_map_tab = ExportMapTab(self)
self.export_map_tab.log_signal.connect(self.log_signal) # 连接日志信号
self.export_image_tab = ExportImageTab(self)
self.raster_tab = RasterTab(self)
self.area_stat_tab = XlsxToJpgTab(self)
self.acid_stat_tab = AcidStatsTab(self)
self.soil_prop_stat_tab = SoilPropStatsTab(self)
tabs.addTab(self.raster_tab, "栅格重分类")
tabs.addTab(self.area_stat_tab, "面积统计表制作")
tabs.addTab(self.export_map_tab, "导出工程文件")
tabs.addTab(self.export_image_tab, "导出成果图")
tabs.addTab(self.acid_stat_tab, "酸化专题图表格统计")
tabs.addTab(self.soil_prop_stat_tab, "土壤属性表制作")
main_layout.addWidget(tabs)
# 添加日志区域
self.log_area = QTextEdit()
self.log_area.setReadOnly(True)
self.log_area.setFixedHeight(120)
self.log_signal.connect(self.append_log) # 日志信号连接
main_layout.addWidget(QLabel("日志输出"))
main_layout.addWidget(self.log_area)
# 添加清除日志按钮
log_control_layout = QHBoxLayout()
clear_log_btn = QPushButton("清空日志")
clear_log_btn.clicked.connect(self.clear_log)
log_control_layout.addStretch()
log_control_layout.addWidget(clear_log_btn)
main_layout.addLayout(log_control_layout)
# 连接信号与槽
self.raster_tab.value_changed.connect(self.export_map_tab.update_config_file)
self.raster_tab.value_changed.connect(self.area_stat_tab.update_config_file)
self.raster_tab.value_changed.connect(self.soil_prop_stat_tab.update_config_file)
def ensure_settings_file(self):
"""确保配置文件存在，如不存在则创建"""
if not os.path.exists(self.settings_file):
default_settings = self.get_default_settings()
try:
os.makedirs(os.path.dirname(self.settings_file), exist_ok=True)
with open(self.settings_file, "w", encoding="utf-8") as f:
json.dump(default_settings, f, indent=4, ensure_ascii=False)
except Exception as e:
print(f"创建配置文件失败: {str(e)}")
def load_settings(self):
"""加载设置"""
try:
if os.path.exists(self.settings_file):
with open(self.settings_file, "r", encoding="utf-8") as f:
return json.load(f)
return self.get_default_settings()
except Exception as e:
print(f"加载设置失败: {str(e)}")
return self.get_default_settings()
def save_settings(self):
"""保存设置"""
try:
config_dir = os.path.dirname(self.settings_file)
if not os.path.exists(config_dir):
os.makedirs(config_dir, exist_ok=True)
with open(self.settings_file, "w", encoding="utf-8") as f:
json.dump(self.settings, f, indent=4, ensure_ascii=False)
print("配置已更新")
except Exception as e:
print(f"保存设置失败: {str(e)}")
def get_default_settings(self):
"""获取默认设置"""
return {
"export_map_settings": {
"template_aprx_file": "D:/工作/ArcGisPro/模板/工程模板.aprx",
"output_path": "D:/工作/ArcGisPro/输出文件夹",
"data_source_path": "D:/工作/ArcGisPro/数据源/土壤属性图矢量数据.gdb",
"config_file": "D:/arcpystudy/ArcGisPro/tools/ui/raster_test_config.json",
"county_name": "港南区",
"symbol_path": "D:/工作/ArcGisPro/模板/符号系统",
},
"export_image_settings": {
"input_aprx_path": "D:/工作/ArcGisPro/输出文件夹/港南区_工作空间",
"output_image_path": "D:/工作/ArcGisPro/输出文件夹",
"default_format": "PDF",
"resolution": 300,
'force_regenerate': True,
'use_multiprocessing': True,
'process_count': 3
},
"raster_settings": {
"input_folder": "D:/工作/三普成果编制/港南区土壤属性图成果最终",
"batch_output_folder": "D:/工作/ArcGisPro/输出文件夹/港南区_工作空间",
"config_file_path": "D:/arcpystudy/ArcGisPro/tools/ui/raster_test_config.json",
"simplify": True,
"min_area": 10000,
"area_unit": "平方米",
},
"area_stat_settings": {
"input_folder": "D:/工作/三普成果编制/港南区土壤属性图成果最终",
"batch_output_folder": "D:/工作/ArcGisPro/输出文件夹/港南区_工作空间",
"config_file_path": "D:/arcpystudy/ArcGisPro/tools/ui/raster_test_config.json",
"dltb_polygon": "D:/工作/三普成果编制/港南区土壤属性图成果最终/港南区_地类图_2021-09-01.shp",
"xzq_polygon": "D:/工作/三普成果编制/港南区土壤属性图成果最终/港南区_区县图_2021-09-01.shp",
},
}
def closeEvent(self, event):
"""窗口关闭处理"""
thread_pool = QThreadPool.globalInstance()
if thread_pool is not None:
thread_pool.waitForDone(2000)
self.update_settings()
self.save_settings()
super().closeEvent(event)
def update_settings(self):
"""更新设置"""
try:
# 更新导出工程设置
if hasattr(self.export_map_tab, "get_settings"):
export_map_settings = self.export_map_tab.get_settings()
if export_map_settings:
self.settings["export_map_settings"] = export_map_settings
# 更新导出成果图设置
if hasattr(self.export_image_tab, "get_layout_settings"):
raster_settings = self.export_image_tab.get_layout_settings()
if raster_settings:
self.settings["export_image_settings"]= raster_settings
# 更新栅格处理设置
if hasattr(self.raster_tab, "get_raster_settings"):
raster_settings = self.raster_tab.get_raster_settings()
if raster_settings:
self.settings["raster_settings"]= raster_settings
# 更新面积统计设置
if hasattr(self.area_stat_tab, "get_area_stat_settings"):
area_stat_settings = self.area_stat_tab.get_area_stat_settings()
if area_stat_settings:
self.settings["area_stat_settings"]= area_stat_settings
# 更新酸化统计设置
if hasattr(self.acid_stat_tab, "get_acid_stat_settings"):
acid_stat_settings = self.acid_stat_tab.get_acid_stat_settings()
if acid_stat_settings:
self.settings["acid_stat_settings"]= acid_stat_settings
# 更新土壤属性统计设置
if hasattr(self.soil_prop_stat_tab, "get_soil_prop_stat_settings"):
soil_prop_stat_settings = self.soil_prop_stat_tab.get_soil_prop_stat_settings()
if soil_prop_stat_settings:
self.settings["soil_prop_stat_settings"]= soil_prop_stat_settings
except Exception as e:
print(f"更新设置失败: {str(e)}")
# 回溯
traceback.print_exc()
def load_ui_settings(self):
"""加载UI设置到各个组件"""
try:
# 加载成果图导出标签页设置
if hasattr(self.export_image_tab, "load_settings"):
self.export_image_tab.load_settings()
# 加载栅格处理标签页设置
if hasattr(self.raster_tab, "load_settings"):
self.raster_tab.load_settings(self.settings["raster_settings"])
# 加载面积统计标签页设置
if hasattr(self.area_stat_tab, "load_settings"):
self.area_stat_tab.load_settings(self.settings["area_stat_settings"])
# 加载酸化专题统计标签页设置
if hasattr(self.acid_stat_tab, "load_settings"):
self.acid_stat_tab.load_settings(self.settings["acid_stat_settings"])
# 加载土壤属性统计标签页设置
if hasattr(self.soil_prop_stat_tab, "load_settings"):
self.soil_prop_stat_tab.load_settings(self.settings["soil_prop_stat_settings"])
except Exception as e:
print(f"加载UI设置失败: {str(e)}")
def append_log(self, message):
"""在日志区域追加信息"""
try:
from datetime import datetime
timestamp = datetime.now().strftime("%H:%M:%S")
# 如果传入字符串中已存在时间戳，则不添加时间戳
if "[" in message and "]" in message:
formatted_message = message
else:
formatted_message = f"[{timestamp}] {message}"
# 根据消息类型设置不同颜色
if "错误" in message or "失败" in message or "出错" in message:
formatted_message = f'<span style="color: #e74c3c;">{formatted_message}</span>'
elif "警告" in message:
formatted_message = f'<span style="color: #f39c12;">{formatted_message}</span>'
elif "成功" in message or "完成" in message:
formatted_message = f'<span style="color: #27ae60;">{formatted_message}</span>'
else:
formatted_message = f'<span style="color: #34495e;">{formatted_message}</span>'
self.log_area.append(formatted_message)
self.log_area.verticalScrollBar().setValue(
self.log_area.verticalScrollBar().maximum()
)
# 如果消息太多，清除旧消息
if self.log_area.document().lineCount() > 500:
cursor = self.log_area.textCursor()
cursor.movePosition(QTextCursor.MoveOperation.Start)
cursor.movePosition(
QTextCursor.MoveOperation.Down, QTextCursor.MoveMode.KeepAnchor, 100
)
cursor.removeSelectedText()
except Exception as e:
print(f"追加日志失败: {str(e)}")
def clear_log(self):
"""清空日志区域"""
self.log_area.clear()
# 如果直接运行该模块
if __name__ == "__main__":
from PyQt6.QtWidgets import QApplication
app = QApplication(sys.argv)
window = MainWindow()
window.show()
sys.exit(app.exec())