给QGIS开发一个卷帘工具

2024-04-01 01:08
文章标签 工具 开发 qgis 卷帘

本文主要是介绍给QGIS开发一个卷帘工具,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

版权声明:未经作者允许不得转载,此插件不得用于商业用途。

目录

开发环境

插件开发

__init__.py

map_swipe_plugin.py

map_swipe_tool.py

active

deactivate

canvasPressEvent

canvasReleaseEvent

canvasMoveEvent

swipe_map.py

实现结果


开发环境

  • QGIS 3.2
  • python(QGIS自带)

插件开发

关于QGIS中使用python开发插件的方法,自行查看官方文档

__init__.py

from .map_swipe_plugin import MapSwipePlugindef classFactory(iface):return MapSwipePlugin(iface)

map_swipe_plugin.py

import osfrom PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import QWidget, QHBoxLayout, QComboBox, QPushButton
from qgis.core import QgsProject
from qgis.gui import QgsMapToolPanfrom .map_swipe_tool import MapSwipeToolplugin_path = os.path.dirname(__file__)class MapSwipePlugin:def __init__(self, iface):self.iface = ifaceself.canvas = iface.mapCanvas()# 图层变化信号QgsProject.instance().layerTreeRoot().layerOrderChanged.connect(self.combobox_add_items)def initGui(self):self.menu = self.title = "卷帘工具"self._create_widget()self.tool = MapSwipeTool(plugin_path, self.combobox, self.iface)self.tool.deactivated.connect(self.tool_deactivated)self.widget_action = self.iface.addToolBarWidget(self.widget)def unload(self):self.canvas.setMapTool(QgsMapToolPan(self.iface.mapCanvas()))self.iface.removeToolBarIcon(self.widget_action)del self.widget_actiondef run(self):if self.canvas.mapTool() != self.tool:self.prevTool = self.canvas.mapTool()self.canvas.setMapTool(self.tool)else:self.canvas.setMapTool(self.prevTool)if self.pushbutton.isChecked() and self.combobox.isHidden():self.combobox.show()self.combobox_add_items()else:self.combobox.hide()def _create_widget(self):icon = QIcon(os.path.join(plugin_path, 'icon.png'))# 新建widgetself.widget = QWidget(self.iface.mainWindow())self.hlayout = QHBoxLayout(self.widget)self.hlayout.setContentsMargins(0, 0, 0, 0)self.pushbutton = QPushButton(icon, '', self.widget)self.pushbutton.setCheckable(True)self.pushbutton.setFlat(True)self.combobox = QComboBox(self.widget)self.hlayout.addWidget(self.pushbutton)self.hlayout.addWidget(self.combobox)self.combobox.hide()self.combobox_add_items()self.pushbutton.clicked.connect(self.run)def combobox_add_items(self):self.combobox.clear()layers = QgsProject.instance().layerTreeRoot().layerOrder()self.combobox.addItems([layer.name() for layer in layers])def tool_deactivated(self):'''tool非激活状态'''self.pushbutton.setChecked(False)self.combobox.hide()

map_swipe_tool.py

import os
from PyQt5.QtCore import Qt, QPoint
from PyQt5.QtGui import QCursor, QPixmap
from qgis.gui import QgsMapTool
from qgis.core import QgsProject, QgsMapSettings, QgsMapRendererParallelJob
from .swipe_map import SwipeMapclass MapSwipeTool(QgsMapTool):def __init__(self, plugin_path, combobox, iface):super(MapSwipeTool, self).__init__(iface.mapCanvas())self.combobox = comboboxself.map_canvas = iface.mapCanvas()self.view = iface.layerTreeView()self.swipe = SwipeMap(self.map_canvas)self.hasSwipe = Noneself.start_point = QPoint()self.cursorSV = QCursor(QPixmap(os.path.join(plugin_path, 'images/split_v.png')))self.cursorSH = QCursor(QPixmap(os.path.join(plugin_path, 'images/split_h.png')))self.cursorUP = QCursor(QPixmap(os.path.join(plugin_path, 'images/up.png')))self.cursorDOWN = QCursor(QPixmap(os.path.join(plugin_path, 'images/down.png')))self.cursorLEFT = QCursor(QPixmap(os.path.join(plugin_path, 'images/left.png')))self.cursorRIGHT = QCursor(QPixmap(os.path.join(plugin_path, 'images/right.png')))def activate(self):self.map_canvas.setCursor(QCursor(Qt.CrossCursor))self._connect()self.hasSwipe = Falseself.setLayersSwipe()def canvasPressEvent(self, e):self.hasSwipe = Truedirection = Nonew, h = self.map_canvas.width(), self.map_canvas.height()if 0.25 * w < e.x() < 0.75 * w and e.y() < 0.5 * h:direction = 0  # '⬇'self.swipe.isVertical = Falseif 0.25 * w < e.x() < 0.75 * w and e.y() > 0.5 * h:direction = 1  # '⬆'self.swipe.isVertical = Falseif e.x() < 0.25 * w:direction = 2  # '➡'self.swipe.isVertical = Trueif e.x() > 0.75 * w:direction = 3  # '⬅'self.swipe.isVertical = Trueself.swipe.set_direction(direction)self.map_canvas.setCursor(self.cursorSH if self.swipe.isVertical else self.cursorSV)self.swipe.set_img_extent(e.x(), e.y())def canvasReleaseEvent(self, e):self.hasSwipe = Falseself.canvasMoveEvent(e)# 鼠标释放后,移除绘制的线self.swipe.set_img_extent(-9999, -9999)def canvasMoveEvent(self, e):if self.hasSwipe:self.swipe.set_img_extent(e.x(), e.y())else:# 设置当前cursorw, h = self.map_canvas.width(), self.map_canvas.height()if e.x() < 0.25 * w:self.canvas().setCursor(self.cursorRIGHT)if e.x() > 0.75 * w:self.canvas().setCursor(self.cursorLEFT)if 0.25 * w < e.x() < 0.75 * w and e.y() < 0.5 * h:self.canvas().setCursor(self.cursorDOWN)if 0.25 * w < e.x() < 0.75 * w and e.y() > 0.5 * h:self.canvas().setCursor(self.cursorUP)def _connect(self, isConnect=True):signal_slot = ({'signal': self.map_canvas.mapCanvasRefreshed, 'slot': self.setMap},{'signal': self.combobox.currentIndexChanged, 'slot': self.setLayersSwipe},{'signal': QgsProject.instance().removeAll, 'slot': self.disable})if isConnect:for item in signal_slot:item['signal'].connect(item['slot'])else:for item in signal_slot:item['signal'].disconnect(item['slot'])def setLayersSwipe(self, ):current_layer = QgsProject.instance().mapLayersByName(self.combobox.currentText())if len(current_layer) == 0:returnlayers = QgsProject.instance().layerTreeRoot().layerOrder()layer_list = []for layer in layers:if layer.id() == current_layer[0].id():continuelayer_list.append(layer)self.swipe.clear()self.swipe.setLayersId(layer_list)self.setMap()def disable(self):self.swipe.clear()self.hasSwipe = Falsedef deactivate(self):self.deactivated.emit()self.swipe.clear()self._connect(False)def setMap(self):def finished():self.swipe.setContent(job.renderedImage(), self.map_canvas.extent())if len(self.swipe.layers) == 0:returnsettings = QgsMapSettings(self.map_canvas.mapSettings())settings.setLayers(self.swipe.layers)job = QgsMapRendererParallelJob(settings)job.start()job.finished.connect(finished)job.waitForFinished()

MapSwipeTool继承了QgsMapTool类,重新实现了activate、deactivate、canvasPressEvent、canvasReleaseEvent、canvasMoveEvent方法,还用到了QgsMapRendererParallelJob类,QgsMapRendererParallelJob是QGIS提供的并行绘图类,不做详解,下面对重新实现的方法详细说明:

active

工具激活后执行此函数

deactivate

工具非激活状态执行此函数

canvasPressEvent

    def canvasPressEvent(self, e):self.hasSwipe = Truedirection = Nonew, h = self.map_canvas.width(), self.map_canvas.height()if 0.25 * w < e.x() < 0.75 * w and e.y() < 0.5 * h:direction = 0  # '⬇'self.swipe.isVertical = Falseif 0.25 * w < e.x() < 0.75 * w and e.y() > 0.5 * h:direction = 1  # '⬆'self.swipe.isVertical = Falseif e.x() < 0.25 * w:direction = 2  # '➡'self.swipe.isVertical = Trueif e.x() > 0.75 * w:direction = 3  # '⬅'self.swipe.isVertical = Trueself.swipe.set_direction(direction)self.map_canvas.setCursor(self.cursorSH if self.swipe.isVertical else self.cursorSV)self.swipe.set_img_extent(e.x(), e.y())

画布的鼠标按压事件,将画布分为四个部分,获得卷帘的方向,然后设置鼠标光标,具体实现方式自行脑补,原理如下图:

canvasReleaseEvent

    def canvasReleaseEvent(self, e):self.hasSwipe = Falseself.canvasMoveEvent(e)# 鼠标释放后,移除绘制的线self.swipe.set_img_extent(-9999, -9999)

画布的鼠标释放事件,鼠标释放后要把绘制的线移出画布

self.swipe.set_img_extent(-9999, -9999)

canvasMoveEvent

    def canvasMoveEvent(self, e):if self.hasSwipe:self.swipe.set_img_extent(e.x(), e.y())else:# 设置当前cursorw, h = self.map_canvas.width(), self.map_canvas.height()if e.x() < 0.25 * w:self.canvas().setCursor(self.cursorRIGHT)if e.x() > 0.75 * w:self.canvas().setCursor(self.cursorLEFT)if 0.25 * w < e.x() < 0.75 * w and e.y() < 0.5 * h:self.canvas().setCursor(self.cursorDOWN)if 0.25 * w < e.x() < 0.75 * w and e.y() > 0.5 * h:self.canvas().setCursor(self.cursorUP)

画布的鼠标移动事件,当使用卷帘时,获得鼠标的位置(e.x(),e.y()),当不使用卷帘时,根据鼠标的位置设置鼠标的光标形状,原理与canvasPressEvent相同

swipe_map.py

from PyQt5.QtCore import QRectF, QPointF, Qt
from PyQt5.QtGui import QPen, QColor
from qgis.gui import QgsMapCanvasItemclass SwipeMap(QgsMapCanvasItem):def __init__(self, canvas):super(SwipeMap, self).__init__(canvas)self.length = 0self.isVertical = Trueself.layers = []self.is_paint = Falsedef setContent(self, image, rect):self.copyimage = imageself.setRect(rect)def clear(self):del self.layers[:]self.is_paint = Falsedef setLayersId(self, layers):del self.layers[:]for item in layers:self.layers.append(item)def set_direction(self, direction):# 0:'⬇', 1:'⬆', 2:'➡', 3:'⬅'if direction == 0:self.direction = 0elif direction == 1:self.direction = 1elif direction == 2:self.direction = 2else:self.direction = 3self.startx, self.starty, self.endx, self.endy = 0, 0, self.boundingRect().width(), self.boundingRect().height()def set_img_extent(self, x, y):self.x = xself.y = yif self.direction == 0:  # 0:'⬇'self.endy = yelif self.direction == 1:  # 1:'⬆'self.starty = yelif self.direction == 2:  # 2:'➡'self.endx = xelse:  # 3:'⬅'self.startx = xself.is_paint = Trueself.update()def paint(self, painter, *args):if len(self.layers) == 0 or self.is_paint == False:returnw = self.boundingRect().width()h = self.boundingRect().height()pen = QPen(Qt.DashDotDotLine)pen.setColor(QColor(18, 150, 219))pen.setWidth(4)if self.isVertical:painter.setPen(pen)painter.drawLine(QPointF(self.x, 0), QPointF(self.x, h))else:painter.setPen(pen)painter.drawLine(QPointF(0, self.y), QPointF(w, self.y))image = self.copyimage.copy(self.startx, self.starty, self.endx, self.endy)painter.drawImage(QRectF(self.startx, self.starty, self.endx, self.endy), image)

此模块为绘制地图模块,首先获得绘制地图的范围

self.startx, self.starty, self.endx, self.endy = 0, 0, self.boundingRect().width(), self.boundingRect().height()

然后重写绘制函数paint

    def paint(self, painter, *args):if len(self.layers) == 0 or self.is_paint == False:returnw = self.boundingRect().width()h = self.boundingRect().height()pen = QPen(Qt.DashDotDotLine)pen.setColor(QColor(18, 150, 219))pen.setWidth(4)if self.isVertical:painter.setPen(pen)painter.drawLine(QPointF(self.x, 0), QPointF(self.x, h))else:painter.setPen(pen)painter.drawLine(QPointF(0, self.y), QPointF(w, self.y))image = self.copyimage.copy(self.startx, self.starty, self.endx, self.endy)painter.drawImage(QRectF(self.startx, self.starty, self.endx, self.endy), image)

实现结果

项目地址:MapSwipeTool

这篇关于给QGIS开发一个卷帘工具的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/865850

相关文章

MySql match against工具详细用法

《MySqlmatchagainst工具详细用法》在MySQL中,MATCH……AGAINST是全文索引(Full-Textindex)的查询语法,它允许你对文本进行高效的全文搜素,支持自然语言搜... 目录一、全文索引的基本概念二、创建全文索引三、自然语言搜索四、布尔搜索五、相关性排序六、全文索引的限制七

基于Java实现回调监听工具类

《基于Java实现回调监听工具类》这篇文章主要为大家详细介绍了如何基于Java实现一个回调监听工具类,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录监听接口类 Listenable实际用法打印结果首先,会用到 函数式接口 Consumer, 通过这个可以解耦回调方法,下面先写一个

使用Python构建一个Hexo博客发布工具

《使用Python构建一个Hexo博客发布工具》虽然Hexo的命令行工具非常强大,但对于日常的博客撰写和发布过程,我总觉得缺少一个直观的图形界面来简化操作,下面我们就来看看如何使用Python构建一个... 目录引言Hexo博客系统简介设计需求技术选择代码实现主框架界面设计核心功能实现1. 发布文章2. 加

JS+HTML实现在线图片水印添加工具

《JS+HTML实现在线图片水印添加工具》在社交媒体和内容创作日益频繁的今天,如何保护原创内容、展示品牌身份成了一个不得不面对的问题,本文将实现一个完全基于HTML+CSS构建的现代化图片水印在线工具... 目录概述功能亮点使用方法技术解析延伸思考运行效果项目源码下载总结概述在社交媒体和内容创作日益频繁的

使用Python开发一个带EPUB转换功能的Markdown编辑器

《使用Python开发一个带EPUB转换功能的Markdown编辑器》Markdown因其简单易用和强大的格式支持,成为了写作者、开发者及内容创作者的首选格式,本文将通过Python开发一个Markd... 目录应用概览代码结构与核心组件1. 初始化与布局 (__init__)2. 工具栏 (setup_t

基于Python打造一个全能文本处理工具

《基于Python打造一个全能文本处理工具》:本文主要介绍一个基于Python+Tkinter开发的全功能本地化文本处理工具,它不仅具备基础的格式转换功能,更集成了中文特色处理等实用功能,有需要的... 目录1. 概述:当文本处理遇上python图形界面2. 功能全景图:六大核心模块解析3.运行效果4. 相

springboot项目中常用的工具类和api详解

《springboot项目中常用的工具类和api详解》在SpringBoot项目中,开发者通常会依赖一些工具类和API来简化开发、提高效率,以下是一些常用的工具类及其典型应用场景,涵盖Spring原生... 目录1. Spring Framework 自带工具类(1) StringUtils(2) Coll

Spring Shell 命令行实现交互式Shell应用开发

《SpringShell命令行实现交互式Shell应用开发》本文主要介绍了SpringShell命令行实现交互式Shell应用开发,能够帮助开发者快速构建功能丰富的命令行应用程序,具有一定的参考价... 目录引言一、Spring Shell概述二、创建命令类三、命令参数处理四、命令分组与帮助系统五、自定义S

基于Python实现高效PPT转图片工具

《基于Python实现高效PPT转图片工具》在日常工作中,PPT是我们常用的演示工具,但有时候我们需要将PPT的内容提取为图片格式以便于展示或保存,所以本文将用Python实现PPT转PNG工具,希望... 目录1. 概述2. 功能使用2.1 安装依赖2.2 使用步骤2.3 代码实现2.4 GUI界面3.效

基于Python和MoviePy实现照片管理和视频合成工具

《基于Python和MoviePy实现照片管理和视频合成工具》在这篇博客中,我们将详细剖析一个基于Python的图形界面应用程序,该程序使用wxPython构建用户界面,并结合MoviePy、Pill... 目录引言项目概述代码结构分析1. 导入和依赖2. 主类:PhotoManager初始化方法:__in