def testMoveToTopActionEmbeddedGroup(self):
"""Test move to top action on embeddedgroup layer"""
view = QgsLayerTreeView()
group = self.project.layerTreeRoot().addGroup("embeddedgroup")
group.addLayer(self.layer4)
group.addLayer(self.layer5)
groupname = group.name()
view.setModel(self.model)
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(self.nodeOrder(self.project.layerTreeRoot().children()), [
self.layer.name(),
self.layer2.name(),
self.layer3.name(),
groupname,
groupname + '-' + self.layer4.name(),
groupname + '-' + self.layer5.name(),
])
view.setCurrentLayer(self.layer5)
movetotop = actions.actionMoveToTop()
movetotop.trigger()
self.assertEqual(self.nodeOrder(self.project.layerTreeRoot().children()), [
self.layer.name(),
self.layer2.name(),
self.layer3.name(),
groupname,
groupname + '-' + self.layer5.name(),
groupname + '-' + self.layer4.name(),
])
def testMoveToTopActionLayer(self):
"""Test move to top action on layer"""
view = QgsLayerTreeView()
view.setModel(self.model)
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(self.project.layerTreeRoot().layerOrder(), [self.layer, self.layer2, self.layer3])
view.setCurrentLayer(self.layer3)
movetotop = actions.actionMoveToTop()
movetotop.trigger()
self.assertEqual(self.project.layerTreeRoot().layerOrder(), [self.layer3, self.layer, self.layer2])
def testSetModel(self):
view = QgsLayerTreeView()
# should not work
string_list_model = QStringListModel()
view.setModel(string_list_model)
self.assertFalse(view.model())
# should work
view.setModel(self.model)
self.assertEqual(view.model(), self.model)
def testProxyModelCurrentIndex(self):
"""Test a crash spotted out while developing the proxy model"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_tester = QAbstractItemModelTester(view.layerTreeModel())
view.setCurrentLayer(self.layer3)
self.layer3.setFlags(self.layer.Private)
def testSetModel(self):
view = QgsLayerTreeView()
# should not work
string_list_model = QStringListModel()
view.setModel(string_list_model)
self.assertFalse(view.model())
# should work
view.setModel(self.model)
self.assertEqual(view.model(), self.model)
def testMoveToTopActionLayer(self):
"""Test move to top action on layer"""
view = QgsLayerTreeView()
view.setModel(self.model)
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(self.project.layerTreeRoot().layerOrder(),
[self.layer, self.layer2, self.layer3])
view.setCurrentLayer(self.layer3)
movetotop = actions.actionMoveToTop()
movetotop.trigger()
self.assertEqual(self.project.layerTreeRoot().layerOrder(),
[self.layer3, self.layer, self.layer2])
def testDefaultActions(self):
view = QgsLayerTreeView()
view.setModel(self.model)
actions = QgsLayerTreeViewDefaultActions(view)
# show in overview action
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentNode().customProperty('overview', 0), False)
show_in_overview = actions.actionShowInOverview()
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0), True)
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0), False)
def testDefaultActions(self):
view = QgsLayerTreeView()
view.setModel(self.model)
actions = QgsLayerTreeViewDefaultActions(view)
# show in overview action
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentNode().customProperty('overview', 0), False)
show_in_overview = actions.actionShowInOverview()
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0), True)
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0), False)
def testSetCurrentLayer(self):
view = QgsLayerTreeView()
view.setModel(self.model)
current_layer_changed_spy = QSignalSpy(view.currentLayerChanged)
self.assertFalse(view.currentLayer())
view.setCurrentLayer(self.layer3)
self.assertEqual(view.currentLayer(), self.layer3)
self.assertEqual(len(current_layer_changed_spy), 1)
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentLayer(), self.layer)
self.assertEqual(len(current_layer_changed_spy), 2)
view.setCurrentLayer(None)
self.assertFalse(view.currentLayer())
self.assertEqual(len(current_layer_changed_spy), 3)
def testSetModel(self):
view = QgsLayerTreeView()
# should not work
string_list_model = QStringListModel()
view.setModel(string_list_model)
self.assertFalse(view.model())
# should work
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
self.assertEqual(view.layerTreeModel(), self.model)
def testSetCurrentLayer(self):
view = QgsLayerTreeView()
view.setModel(self.model)
current_layer_changed_spy = QSignalSpy(view.currentLayerChanged)
self.assertFalse(view.currentLayer())
view.setCurrentLayer(self.layer3)
self.assertEqual(view.currentLayer(), self.layer3)
self.assertEqual(len(current_layer_changed_spy), 1)
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentLayer(), self.layer)
self.assertEqual(len(current_layer_changed_spy), 2)
view.setCurrentLayer(None)
self.assertFalse(view.currentLayer())
self.assertEqual(len(current_layer_changed_spy), 3)
def testMoveToBottomActionLayer(self):
"""Test move to bottom action on layer"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(self.project.layerTreeRoot().layerOrder(),
[self.layer, self.layer2, self.layer3])
view.setCurrentLayer(self.layer)
movetobottom = actions.actionMoveToBottom()
movetobottom.trigger()
self.assertEqual(self.project.layerTreeRoot().layerOrder(),
[self.layer2, self.layer3, self.layer])
def testSetLayerVisible(self):
view = QgsLayerTreeView()
view.setModel(self.model)
self.project.layerTreeRoot().findLayer(self.layer).setItemVisibilityChecked(True)
self.project.layerTreeRoot().findLayer(self.layer2).setItemVisibilityChecked(True)
self.assertTrue(self.project.layerTreeRoot().findLayer(self.layer).itemVisibilityChecked())
self.assertTrue(self.project.layerTreeRoot().findLayer(self.layer2).itemVisibilityChecked())
view.setLayerVisible(None, True)
view.setLayerVisible(self.layer, True)
self.assertTrue(self.project.layerTreeRoot().findLayer(self.layer).itemVisibilityChecked())
view.setLayerVisible(self.layer2, False)
self.assertFalse(self.project.layerTreeRoot().findLayer(self.layer2).itemVisibilityChecked())
view.setLayerVisible(self.layer2, True)
self.assertTrue(self.project.layerTreeRoot().findLayer(self.layer2).itemVisibilityChecked())
def testAddGroupActionGroups(self):
"""Test add group action on several groups"""
view = QgsLayerTreeView()
group = self.project.layerTreeRoot().insertGroup(0, "embeddedgroup")
group.addLayer(self.layer4)
groupname = group.name()
group2 = self.project.layerTreeRoot().insertGroup(0, "embeddedgroup2")
group2.addLayer(self.layer5)
groupname2 = group2.name()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(
self.nodeOrder(self.project.layerTreeRoot().children()), [
groupname2,
groupname2 + '-' + self.layer5.name(),
groupname,
groupname + '-' + self.layer4.name(),
self.layer.name(),
self.layer2.name(),
self.layer3.name(),
])
selectionMode = view.selectionMode()
view.setSelectionMode(QgsLayerTreeView.MultiSelection)
nodeLayerIndex = view.node2index(group)
view.setCurrentIndex(nodeLayerIndex)
nodeLayerIndex2 = view.node2index(group2)
view.setCurrentIndex(nodeLayerIndex2)
view.setSelectionMode(selectionMode)
addgroup = actions.actionAddGroup()
addgroup.trigger()
self.assertEqual(
self.nodeOrder(self.project.layerTreeRoot().children()), [
self.groupname + '1', self.groupname + '1' + '-' + groupname,
self.groupname + '1' + '-' + groupname + '-' +
self.layer4.name(), self.groupname + '1' + '-' + groupname2,
self.groupname + '1' + '-' + groupname2 + '-' +
self.layer5.name(),
self.layer.name(),
self.layer2.name(),
self.layer3.name()
])
def testSetCurrentLayer(self):
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
current_layer_changed_spy = QSignalSpy(view.currentLayerChanged)
self.assertFalse(view.currentLayer())
view.setCurrentLayer(self.layer3)
self.assertEqual(view.currentLayer(), self.layer3)
self.assertEqual(len(current_layer_changed_spy), 1)
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentLayer(), self.layer)
self.assertEqual(len(current_layer_changed_spy), 2)
view.setCurrentLayer(None)
self.assertFalse(view.currentLayer())
self.assertEqual(len(current_layer_changed_spy), 3)
def testDefaultActions(self):
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
actions = QgsLayerTreeViewDefaultActions(view)
# show in overview action
view.setCurrentLayer(self.layer)
self.assertEqual(view.currentNode().customProperty('overview', 0),
False)
show_in_overview = actions.actionShowInOverview()
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0),
True)
show_in_overview.trigger()
self.assertEqual(view.currentNode().customProperty('overview', 0),
False)
def testNode2IndexMethods(self):
"""Test node2index and node2sourceIndex"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_model = view.layerTreeModel()
proxy_model = view.proxyModel()
proxy_index = proxy_model.index(1, 0)
node2 = view.index2node(proxy_index)
self.assertEqual(node2.name(), 'layer2')
proxy_layer2_index = view.node2index(node2)
self.assertEqual(proxy_layer2_index, view.node2index(node2))
source_index = tree_model.index(1, 0)
tree_layer2_index = view.node2sourceIndex(node2)
self.assertEqual(tree_layer2_index, view.node2sourceIndex(node2))
def testMoveToBottomActionLayerAndGroup(self):
"""Test move to top action for a group and it's layer simultaneously"""
view = QgsLayerTreeView()
group = self.project.layerTreeRoot().insertGroup(0, "embeddedgroup")
group.addLayer(self.layer4)
group.addLayer(self.layer5)
groupname = group.name()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
actions = QgsLayerTreeViewDefaultActions(view)
self.assertEqual(
self.nodeOrder(self.project.layerTreeRoot().children()), [
groupname,
groupname + '-' + self.layer4.name(),
groupname + '-' + self.layer5.name(),
self.layer.name(),
self.layer2.name(),
self.layer3.name(),
])
selectionMode = view.selectionMode()
view.setSelectionMode(QgsLayerTreeView.MultiSelection)
nodeLayerIndex = view.node2index(group)
view.setCurrentIndex(nodeLayerIndex)
view.setCurrentLayer(self.layer4)
view.setSelectionMode(selectionMode)
movetobottom = actions.actionMoveToBottom()
movetobottom.trigger()
self.assertEqual(
self.nodeOrder(self.project.layerTreeRoot().children()), [
self.layer.name(),
self.layer2.name(),
self.layer3.name(),
groupname,
groupname + '-' + self.layer5.name(),
groupname + '-' + self.layer4.name(),
])
class see_map(QDialog):
#app2 = QgsApplication([], True)
#app2.initQgis()
def __init__(self, file_name):
super().__init__()
# Creación del Lienzo / Canvas
self.canvas = QgsMapCanvas(self)
self.setWindowTitle("Map of Track / Mapa del Recorrido")
self.canvas.setCanvasColor(Qt.white)
self.initGui()
if not file_name == None:
vlayer = QgsVectorLayer(file_name, "Track Converted", "ogr")
#vlayer.updateExtents()
self.project.instance().addMapLayer(vlayer)
vlayer.renderer().symbol().setWidth(2)
vlayer.renderer().symbol().setColor(QColor.fromRgb(250, 0, 0))
self.view.refreshLayerSymbology(vlayer.id())
uri = 'crs=EPSG:4326&dpiMode=7&format=image/jpeg&layers=IGNBaseTodo&styles=default&tileMatrixSet=EPSG:4326&url=http://www.ign.es/wmts/ign-base'
map_spain = QgsRasterLayer(uri, "Spain", "wms")
self.root.insertLayer(1, map_spain)
self.project.instance().addMapLayer(map_spain)
if not map_spain.isValid():
print("Track failed to load! /n" + file_name)
else:
print("Track Load Correct! " + file_name)
else:
uri = 'crs=EPSG:4326&dpiMode=7&format=image/jpeg&layers=IGNBaseTodo&styles=default&tileMatrixSet=EPSG:4326&url=http://www.ign.es/wmts/ign-base'
map_spain = QgsRasterLayer(uri, "Spain", "wms")
self.project.instance().addMapLayer(map_spain)
if not map_spain.isValid():
print("Track failed to load! /n" + file_name)
else:
print("Track Load Correct! " + file_name)
def zoomIn(self):
self.canvas.setMapTool(self.toolZoomIn)
def zoomOut(self):
self.canvas.setMapTool(self.toolZoomOut)
def pan(self):
self.canvas.setMapTool(self.toolPan)
def file(self):
filename, _ = QFileDialog.getOpenFileName(self,
'Selecciona Fichero GPX',
'c:/', 'QGis (*.qgz)')
if filename:
self.project.read(filename)
def initGui(self):
icon = QIcon()
icon.addPixmap(
QPixmap(
"C:/Roberto/Visual_Studio_Code/GisBike/programa/IMG/Qgis.png"),
QIcon.Normal, QIcon.Off)
self.setWindowIcon(icon)
self.setWindowModality(Qt.WindowModal)
self.setWindowFlags(Qt.Window
| Qt.WindowSystemMenuHint
| Qt.WindowMinimizeButtonHint
| Qt.WindowMaximizeButtonHint
| Qt.WindowCloseButtonHint)
self.resize(1200, 600)
self.verticalLayout_2 = QVBoxLayout(self)
self.splitter = QSplitter(self)
self.splitter.setOrientation(Qt.Horizontal)
self.actionZoomIn = QPushButton("Zoom in", self.splitter)
self.actionZoomOut = QPushButton("Zoom out", self.splitter)
self.actionPan = QPushButton("Pan", self.splitter)
self.actionFile = QPushButton("File", self.splitter)
self.project = QgsProject.instance()
self.project.read('C:/Users/Fcc/Desktop/QGis/pruebas2.qgz')
self.root = QgsProject.instance().layerTreeRoot()
self.bridge = QgsLayerTreeMapCanvasBridge(self.root, self.canvas)
self.bridge.setCanvasLayers()
self.bridge.setAutoSetupOnFirstLayer(True)
#https://gis.stackexchange.com/questions/141516/adding-legend-to-canvas-in-standalone-pyqgis-application
self.model = QgsLayerTreeModel(self.root)
self.model.setFlag(QgsLayerTreeModel.AllowNodeReorder)
self.model.setFlag(QgsLayerTreeModel.AllowNodeRename)
self.model.setFlag(QgsLayerTreeModel.AllowNodeChangeVisibility)
self.model.setFlag(QgsLayerTreeModel.ShowLegend)
self.view = QgsLayerTreeView(self.splitter)
self.view.setModel(self.model)
self.view.setFixedWidth(150)
#self.view.resize(150,600)
self.widget = QWidget(self.splitter)
self.verticalLayout = QVBoxLayout(self.widget)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.addWidget(self.actionZoomIn)
self.horizontalLayout.addWidget(self.actionZoomOut)
self.horizontalLayout.addWidget(self.actionPan)
self.horizontalLayout.addWidget(self.actionFile)
self.verticalLayout.addLayout(self.horizontalLayout)
self.verticalLayout.addWidget(self.canvas)
self.verticalLayout_2.addWidget(self.splitter)
self.actionZoomIn.clicked.connect(self.zoomIn)
self.actionZoomOut.clicked.connect(self.zoomOut)
self.actionPan.clicked.connect(self.pan)
self.actionFile.clicked.connect(self.file)
# create the map tools
self.toolPan = QgsMapToolPan(self.canvas)
self.toolZoomIn = QgsMapToolZoom(self.canvas, False) # false = in
self.toolZoomOut = QgsMapToolZoom(self.canvas, True) # true = out
class Window(QMainWindow, Ui_MainWindow):
def __init__(self):
super(Window, self).__init__()
self.setupUi(self)
self.layers = []
# ---------初始化图层和画布----------
self.vl = QVBoxLayout(self.dockWidgetContents)
self.layer_tree_view = QgsLayerTreeView(self)
self.vl.addWidget(self.layer_tree_view)
self.hl = QHBoxLayout(self.frame)
self.mapCanvas = QgsMapCanvas(self.frame)
self.hl.addWidget(self.mapCanvas)
# ---------初始化图层和画布----------
self.action_open.triggered.connect(self.action_open_triggered)
# 建立桥梁
self.model = QgsLayerTreeModel(QgsProject.instance().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.layer_tree_view.setModel(self.model)
self.layer_tree_bridge = QgsLayerTreeMapCanvasBridge(
QgsProject.instance().layerTreeRoot(), self.mapCanvas)
# 设置识别工具
self.identifyTool = QgsMapToolIdentifyFeature(self.mapCanvas)
# 发送识别的要素
self.identifyTool.featureIdentified.connect(self.print_features)
self.mapCanvas.setMapTool(self.identifyTool)
def action_open_triggered(self):
data_file, ext = QFileDialog.getOpenFileName(self, '打开', '', '所有文件(*)')
if data_file:
if data_file.endswith('.shp'):
basename = os.path.splitext(os.path.basename(data_file))[0]
layer = QgsVectorLayer(data_file, basename, "ogr")
QgsProject.instance().addMapLayer(layer)
self.layers.append(layer)
self.mapCanvas.setExtent(layer.extent())
self.mapCanvas.setLayers(self.layers)
self.mapCanvas.refresh()
self.layer_tree_view.setCurrentLayer(layer)
# 设置需要识别的图层
self.identifyTool.setLayer(layer)
elif data_file.endswith('.qgz') or data_file.endswith('.qgs'):
QgsProject.instance().read(data_file)
else:
print('error')
def print_features(self, feature):
print(feature.attributes())
print(feature.fields())
print(feature.geometry())
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)
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}')
# coding: utf-8 from qgis.core import QgsProject, QgsLayerTreeModel from qgis.gui import QgsLayerTreeView root = QgsProject.instance().layerTreeRoot() model = QgsLayerTreeModel(root) view = QgsLayerTreeView() view.setModel(model) view.show()
def testProxyModel(self):
"""Test proxy model filtering and private layers"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_model = view.layerTreeModel()
proxy_model = view.proxyModel()
self.assertEqual(tree_model.rowCount(), 3)
self.assertEqual(proxy_model.rowCount(), 3)
items = []
for r in range(tree_model.rowCount()):
items.append(tree_model.data(tree_model.index(r, 0)))
self.assertEqual(items, ['layer1', 'layer2', 'layer3'])
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2', 'layer3'])
self.layer3.setFlags(self.layer.Private)
self.assertEqual(tree_model.rowCount(), 3)
self.assertEqual(proxy_model.rowCount(), 2)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2'])
view.setShowPrivateLayers(True)
self.assertEqual(proxy_model.rowCount(), 3)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2', 'layer3'])
view.setShowPrivateLayers(False)
self.assertEqual(proxy_model.rowCount(), 2)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2'])
# Test filters
proxy_model.setFilterText('layer2')
self.assertEqual(proxy_model.rowCount(), 1)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer2'])
