class MainWindow(QMainWindow, Ui_MainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.setupUi(self)
self.first_flag = True
self.setWindowTitle('PyQGIS')
# 调整窗口大小
self.resize(800, 600)
# 初始化图层树
vl = QVBoxLayout(self.dockWidgetContents)
self.layerTreeView = QgsLayerTreeView(self)
vl.addWidget(self.layerTreeView)
# 初始化地图画布
self.mapCanvas = QgsMapCanvas(self)
hl = QHBoxLayout(self.frame)
hl.setContentsMargins(0, 0, 0, 0)
hl.addWidget(self.mapCanvas)
# 建立桥梁
self.model = QgsLayerTreeModel(PROJECT.layerTreeRoot(), self)
self.model.setFlag(QgsLayerTreeModel.AllowNodeRename)
self.model.setFlag(QgsLayerTreeModel.AllowNodeReorder)
self.model.setFlag(QgsLayerTreeModel.AllowNodeChangeVisibility)
self.model.setFlag(QgsLayerTreeModel.ShowLegendAsTree)
self.model.setAutoCollapseLegendNodes(10)
self.layerTreeView.setModel(self.model)
self.layerTreeBridge = QgsLayerTreeMapCanvasBridge(
PROJECT.layerTreeRoot(), self.mapCanvas, self)
# 显示经纬度
self.mapCanvas.xyCoordinates.connect(self.showLngLat)
# 打开工程
self.actionOpen.triggered.connect(self.actionOpenTriggered)
# 退出程序
self.actionQuit.triggered.connect(self.close)
# 地图工具
# TODO:放大、缩小没有图标
self.actionPanTriggered()
self.actionPan.triggered.connect(self.actionPanTriggered)
self.actionZoomin.triggered.connect(self.actionZoomInTriggered)
self.actionZoomout.triggered.connect(self.actionZoomOutTriggered)
self.actionIdentity.triggered.connect(self.actionIdentifyTriggered)
# 图层
self.actionShapefile.triggered.connect(self.actionShapefileTriggered)
self.actionCsv.triggered.connect(self.actionCsvTriggered)
self.actionPostGIS.triggered.connect(self.actionPostGISTriggered)
self.actionWFS.triggered.connect(self.actionWFSTriggered)
self.actionGeotiff.triggered.connect(self.actionGeotiffTriggered)
self.actionXYZ.triggered.connect(self.actionXYZTriggered)
# 绘图工具
self.actionPoint.triggered.connect(self.actionPointTriggered)
self.actionLine.triggered.connect(self.actionLineTriggered)
self.actionRectangle.triggered.connect(self.actionRectangleTriggered)
self.actionPolygon.triggered.connect(self.actionPolygonTriggered)
# 关于Qt
self.actionAboutQt.triggered.connect(
lambda: QMessageBox.aboutQt(self, '关于Qt'))
self.actionAbout.triggered.connect(
lambda: QMessageBox.about(self, '关于', 'PyQGIS二次开发'))
# self.actionPan.triggered.connect(self.actionPanTriggered)
# self.actionIdentify.triggered.connect(self.actionIdentifyTriggered)
# 图层右键菜单
self.customMenuProvider = CustomMenuProvider(self.layerTreeView,
self.mapCanvas)
self.layerTreeView.setMenuProvider(self.customMenuProvider)
# self.layerTreeRegistryBridge = QgsLayerTreeRegistryBridge(PROJECT.layerTreeRoot(), PROJECT, self)
def actionOpenTriggered(self):
"""打开工程"""
data_file, ext = QFileDialog.getOpenFileName(self, '打开', '',
'工程文件(*.qgs , *.qgz)')
if data_file:
PROJECT.read(data_file)
def actionPanTriggered(self):
self.mapTool = QgsMapToolPan(self.mapCanvas)
self.mapCanvas.setMapTool(self.mapTool)
def actionZoomInTriggered(self):
self.mapTool = QgsMapToolZoom(self.mapCanvas, False)
self.mapCanvas.setMapTool(self.mapTool)
def actionZoomOutTriggered(self):
self.mapTool = QgsMapToolZoom(self.mapCanvas, True)
self.mapCanvas.setMapTool(self.mapTool)
def actionIdentifyTriggered(self):
# 设置识别工具
self.identifyTool = QgsMapToolIdentifyFeature(self.mapCanvas)
self.identifyTool.featureIdentified.connect(self.showFeatures)
self.mapCanvas.setMapTool(self.identifyTool)
# 设置需要识别的图层
layers = self.mapCanvas.layers()
if layers:
# 识别画布中第一个图层
self.identifyTool.setLayer(layers[0])
def showFeatures(self, feature):
print(type(feature))
QMessageBox.information(self, '信息', ''.join(feature.attributes()))
def actionAddGroupTriggered(self):
PROJECT.layerTreeRoot().addGroup('group1')
def actionShapefileTriggered(self):
"""打开shp"""
data_file, ext = QFileDialog.getOpenFileName(self, '打开', '', '*.shp')
if data_file:
layer = QgsVectorLayer(
data_file,
os.path.splitext(os.path.basename(data_file))[0], "ogr")
self.addLayer(layer)
def actionCsvTriggered(self):
"""加载csv数据"""
data_file, ext = QFileDialog.getOpenFileName(self, '打开', '', '*.csv')
if data_file:
# 去掉盘符，否则图层无效
data_file = os.path.splitdrive(data_file)[1]
uri = f"file://{data_file}?delimiter=,&xField=x&yField=y"
print(uri)
layer = QgsVectorLayer(uri, "point", "delimitedtext")
self.addLayer(layer)
def actionPostGISTriggered(self):
"""加载postgis图层"""
dialog = PostGISDialog(self)
if dialog.exec_():
uri = QgsDataSourceUri()
uri.setConnection(dialog.lineEditHost.text(),
dialog.lineEditPort.text(),
dialog.lineEditDatabase.text(),
dialog.lineEditUsername.text(),
dialog.lineEditPassword.text())
# lineEditGeometryColumn：根据实际情况，可能为：wkb_geometry、geometry、the_geom...
uri.setDataSource("public", dialog.lineEditLayer.text(),
dialog.lineEditGeometryColumn.text())
layer = QgsVectorLayer(uri.uri(False), dialog.lineEditLayer.text(),
"postgres")
self.addLayer(layer)
def actionWFSTriggered(self):
"""加载天地图WFS图层"""
uri = 'http://gisserver.tianditu.gov.cn/TDTService/wfs?' \
'srsname=EPSG:4326&typename=TDTService:RESA&version=auto&request=GetFeature&service=WFS'
layer = QgsVectorLayer(uri, "RESA", "WFS")
self.addLayer(layer)
def actionGeotiffTriggered(self):
"""加载geotiff"""
data_file, ext = QFileDialog.getOpenFileName(self, '打开', '', '*.tif')
if data_file:
layer = QgsRasterLayer(data_file, os.path.basename(data_file))
self.addLayer(layer)
def actionXYZTriggered(self):
uri = 'type=xyz&' \
'url=https://www.google.cn/maps/vt?lyrs=s@804%26gl=cn%26x={x}%26y={y}%26z={z}&' \
'zmax=19&' \
'zmin=0&' \
'crs=EPSG3857'
layer = QgsRasterLayer(uri, 'google', 'wms')
self.addLayer(layer)
def addLayer(self, layer):
if layer.isValid():
if self.first_flag:
self.mapCanvas.setDestinationCrs(layer.crs())
self.mapCanvas.setExtent(layer.extent())
self.first_flag = False
PROJECT.addMapLayer(layer)
layers = [layer
] + [PROJECT.mapLayer(i) for i in PROJECT.mapLayers()]
self.mapCanvas.setLayers(layers)
self.mapCanvas.refresh()
else:
print('图层无效.')
def actionPointTriggered(self):
self.pointTool = PointMapTool(self.mapCanvas)
self.mapCanvas.setMapTool(self.pointTool)
def actionLineTriggered(self):
self.lineTool = LineMapTool(self.mapCanvas)
self.mapCanvas.setMapTool(self.lineTool)
def actionRectangleTriggered(self):
self.rectangleTool = RectangleMapTool(self.mapCanvas)
self.mapCanvas.setMapTool(self.rectangleTool)
def actionPolygonTriggered(self):
self.polygonTool = PolygonMapTool(self.mapCanvas)
self.mapCanvas.setMapTool(self.polygonTool)
def showLngLat(self, point):
x = point.x()
y = point.y()
self.statusbar.showMessage(f'经度:{x}, 纬度:{y}')
class MapExplorer(QMainWindow, Ui_MainWindow):
def __init__(self):
super(MapExplorer, self).__init__()
self.project = QgsProject()
self.band_synthesis_window = Band_Synthesis_Component()
self.kmeans_window = KMeans_Component()
self.meanshift_window = MeanShift_Component()
self.dbscan_window = DBScan_Component()
self.setupUi(self)
self.init_mapcanvas()
self.slot_connect()
self.init_layerTree()
self.style_stand()
def init_layerTree(self):
self.layer_tree_view = QgsLayerTreeView(self)
layout = QVBoxLayout(self.layerTreeWidget)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(self.layer_tree_view)
model = QgsLayerTreeModel(QgsProject.instance().layerTreeRoot(), self)
model.setFlag(QgsLayerTreeModel.AllowNodeRename)
model.setFlag(QgsLayerTreeModel.AllowNodeReorder)
model.setFlag(QgsLayerTreeModel.AllowNodeChangeVisibility)
model.setFlag(QgsLayerTreeModel.ShowLegendAsTree)
model.setAutoCollapseLegendNodes(10)
self.layer_tree_view.setModel(model)
provider = MenuProvider(self.layer_tree_view, self.mapCanvas,
self.project.instance())
self.layer_tree_view.setMenuProvider(provider)
# 注意self.layer_tree_bridge必须有
self.layer_tree_bridge = QgsLayerTreeMapCanvasBridge(
self.project.instance().layerTreeRoot(), self.mapCanvas)
def slot_connect(self):
self.actionopen_file.triggered.connect(self.action_open_triggered)
self.actionzoom_in.triggered.connect(self.action_zoomin_triggered)
self.actionzoom_out.triggered.connect(self.action_zoomout_triggered)
self.actionpan.triggered.connect(self.action_pan_triggered)
self.actionfull_extent.triggered.connect(
self.action_fullextent_triggered)
self.action_select.triggered.connect(self.action_select_triggered)
self.actionsave.triggered.connect(self.action_save_triggered)
self.mapdecoration.triggered.connect(self.mapDecoration_show)
self.actionAqua.triggered.connect(self.style_Aqua)
self.actionstand.triggered.connect(self.style_stand)
self.actionConsoleStule.triggered.connect(self.style_ConsoleStyle)
self.actionElegantDark.triggered.connect(self.style_ElegantDark)
self.actionManjaroMix.triggered.connect(self.style_ManjaroMix)
self.actionMaterialDark.triggered.connect(self.style_MaterialDark)
self.actionUbuntu.triggered.connect(self.style_Ubuntu)
self.action_KMeans.triggered.connect(self.action_KMeans_triggered)
self.action_DBScan.triggered.connect(self.action_DBScan_triggered)
self.action_MeanShift.triggered.connect(
self.action_Meanshift_triggered)
self.action_band_synthesis.triggered.connect(
self.action_band_synthesis_triggered)
self.kmeans_window.add_layer_signal.connect(
self.action_add_kmeans_result)
self.dbscan_window.add_layer_signal.connect(
self.action_add_dbscan_result)
self.meanshift_window.add_layer_signal.connect(
self.action_add_meanshift_result)
self.band_synthesis_window.add_layer_signal.connect(
self.action_add_band_synthesis_result)
def init_mapcanvas(self):
# 实例化地图画布
self.mapCanvas = QgsMapCanvas()
self.mapCanvas.xyCoordinates.connect(self.show_lonlat)
self.mapCanvas.setCanvasColor(QColor(60, 63, 65, 100))
# self.mapCanvas.show()
layout = QVBoxLayout(self.mapWidget)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(self.mapCanvas)
def loadMap(self, fullpath, classArray=None):
print(fullpath[-3:-1] + fullpath[-1])
imgClass = fullpath[-3:-1] + fullpath[-1]
basename = os.path.splitext(os.path.basename(fullpath))[0]
if imgClass == 'shp':
self.layer = QgsVectorLayer(fullpath, basename, "ogr")
else:
self.layer = QgsRasterLayer(fullpath, basename)
# 添加图例
if classArray:
self.node = QgsLayerTreeLayer(self.layer)
# self.classArray = [{'name':'class1','color':QColor(0,0,0)},{'name':'class1','color':QColor(0,255,0)}]
self.legend = CustomViewerLegend(classArray)
self.legend.createLayerTreeModelLegendNodes(self.node)
self.layer.setLegend(self.legend)
# 注册图层
self.project.instance().addMapLayer(self.layer)
self.mapCanvas.setLayers([self.layer])
# 设置图层范围
self.mapCanvas.setExtent(self.layer.extent())
self.mapCanvas.refresh()
self.layer_tree_view.setCurrentLayer(self.layer)
def action_open_triggered(self):
fullpath, format = QFileDialog.getOpenFileName(
self, '打开数据', '', '*.shp;;*.jpg;;*.png;;*.tif')
if os.path.exists(fullpath):
self.loadMap(fullpath)
def action_save_triggered(self):
fullpath, format = QFileDialog.getSaveFileName(self, '保存数据', '',
'*.tif')
if os.path.exists(fullpath):
self.mapCanvas.saveAsImage(fullpath)
def action_zoomin_triggered(self):
self.maptool = QgsMapToolZoom(self.mapCanvas, False)
self.mapCanvas.setMapTool(self.maptool)
def action_zoomout_triggered(self):
self.maptool = QgsMapToolZoom(self.mapCanvas, True)
self.mapCanvas.setMapTool(self.maptool)
def action_pan_triggered(self):
self.maptool = QgsMapToolPan(self.mapCanvas)
self.mapCanvas.setMapTool(self.maptool)
def action_fullextent_triggered(self):
self.mapCanvas.setExtent(self.layer.extent())
self.mapCanvas.refresh()
def action_select_triggered(self):
self.maptool = PolygonMapTool(self)
self.mapCanvas.setMapTool(self.maptool, True)
self.mapCanvas.refresh()
def mapDecoration_show(self):
self.mapdecoration_dia = layerOut(QgsProject.instance(),
self.qssStyleDia)
self.mapdecoration_dia.show()
# 显示鼠标点的经纬度信息
def show_lonlat(self, point):
x = point.x()
y = point.y()
self.statusbar.showMessage(f'经度:{x},纬度:{y}')
def style_stand(self):
styleFile = './QSS-master/style.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS('./QSS-master/ElegantDark.qss')
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_Aqua(self):
styleFile = './QSS-master/Aqua.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_ConsoleStyle(self):
styleFile = './QSS-master/ConsoleStyle.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_ElegantDark(self):
styleFile = './QSS-master/ElegantDark.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_ManjaroMix(self):
styleFile = './QSS-master/ManjaroMix.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_MaterialDark(self):
styleFile = './QSS-master/MaterialDark.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def style_Ubuntu(self):
styleFile = './QSS-master/Ubuntu.qss'
self.qssStyle = CommonHelper.readQSS(styleFile)
self.qssStyleDia = CommonHelper.readQSS(styleFile)
QMainWindow.setStyleSheet(self, self.qssStyle)
def action_band_synthesis_triggered(self):
self.band_synthesis_window.show()
def action_KMeans_triggered(self):
self.kmeans_window.show()
def action_DBScan_triggered(self):
self.dbscan_window.show()
def action_Meanshift_triggered(self):
self.meanshift_window.show()
def action_add_kmeans_result(self, kmeans_img_path, layer_legends):
self.loadMap(kmeans_img_path, layer_legends)
def action_add_dbscan_result(self, dbscan_img_path, layer_legends):
self.loadMap(dbscan_img_path, layer_legends)
def action_add_meanshift_result(self, meanshift_img_path, layer_legends):
self.loadMap(meanshift_img_path, layer_legends)
def action_add_band_synthesis_result(self, multiband_img_path):
self.loadMap(multiband_img_path)
