QGISプラグイン開発の勉強(2)

概要

前回（https://anorith.hatenablog.com/entry/2023/10/28/234120）の続き
プラグインのアイコン・ショートカット設定
画像を保存する処理の追加

参考

ChatGPT
- GPT-4を使用しました。
- 記事の最後に少しコメントを記載しています。

アイコン設定とショートカット設定

前回のプラグインフォルダをそのまま使用します。フォルダにアイコン用の画像を格納します。

アイコンには以下の画像を使用しました。DALL·E3に白背景のアイコンを描いてもらい、ノイズ除去のためvectorizer.aiで一度ベクター化＆pngで再保存して、白背景部分をPythonで透過に処理してアイコンにしました（AIに頼りすぎでは・・・？）。

まずインポート部分にアイコン設定用のモジュールを追加します。

from PyQt5.QtWidgets import QAction
from PyQt5.QtGui import QIcon

次に__init__.pyのinitGuiの箇所を以下のように書き換えます。

    def initGui(self):
        self.action = QAction('MyPluginAction', self.iface.mainWindow())

        icon_path = os.path.join(os.path.dirname(__file__), "icon.png")        
        self.action.setIcon(QIcon(icon_path))
        self.action.setShortcut("Ctrl+T")

        self.action.triggered.connect(self.run)
        self.iface.addToolBarIcon(self.action)

以下のようにアイコンが反映されました。オリジナルのアイコンだと一気にプラグイン作ってる感じが出ますね。

またショートカットCtrl+Tも使用してrunの処理が行えるようになっています。

画像保存プラグインの作成

インポート部分に必要なモジュールを追加します。

import os
from PyQt5.QtWidgets import QAction
from PyQt5.QtGui import QIcon
from qgis.core import QgsProject, QgsMapRendererParallelJob

runの中身を以下の内容をに書き換えます（ChatGPTの出力そのままです）。

    def run(self):
        # 現在のプロジェクトのファイルパスを取得
        project_path = QgsProject.instance().fileName()
        if not project_path:
            # プロジェクトが保存されていない場合は、処理を中止
            return
        
        # プロジェクトのディレクトリパスを取得
        project_directory = os.path.dirname(project_path)
        image_path = os.path.join(project_directory, "map_image.png")

        # 現在の地図の設定を取得
        map_settings = self.iface.mapCanvas().mapSettings()

        # レンダリングジョブを作成して地図を画像として保存
        job = QgsMapRendererParallelJob(map_settings)
        job.start()
        job.waitForFinished()
        image = job.renderedImage()

        image.save(image_path)