# -*- coding: utf-8 -*- """ /*************************************************************************** SaveView A QGIS plugin Plugin loops over features and stores view into file. Generated by Plugin Builder: http://g-sherman.github.io/Qgis-Plugin-Builder/ ------------------- begin : 2019-03-13 git sha : $Format:%H$ copyright : (C) 2019 by Martin Landa email : martin.landa@fsv.cvut.cz ***************************************************************************/ /*************************************************************************** * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * ***************************************************************************/ """ from PyQt5.QtCore import QSettings, QTranslator, qVersion, QCoreApplication, Qt from PyQt5.QtGui import QIcon from PyQt5.QtWidgets import QAction # Initialize Qt resources from file resources.py from .resources import * from qgis.core import QgsProject, QgsMapLayer # from qgis.core import * # Import the code for the DockWidget from .save_views_dockwidget import SaveViewsDockWidget import os.path class SaveViews: """QGIS Plugin Implementation.""" def __init__(self, iface): """Constructor. :param iface: An interface instance that will be passed to this class which provides the hook by which you can manipulate the QGIS application at run time. :type iface: QgsInterface """ # Save reference to the QGIS interface self.iface = iface # initialize plugin directory self.plugin_dir = os.path.dirname(__file__) # initialize locale locale = QSettings().value('locale/userLocale')[0:2] locale_path = os.path.join( self.plugin_dir, 'i18n', 'SaveView_{}.qm'.format(locale)) if os.path.exists(locale_path): self.translator = QTranslator() self.translator.load(locale_path) if qVersion() > '4.3.3': QCoreApplication.installTranslator(self.translator) # Declare instance attributes self.actions = [] self.menu = self.tr(u'&Save Views') # TODO: We are going to let the user set this up in a future iteration self.toolbar = self.iface.addToolBar(u'SaveView') self.toolbar.setObjectName(u'SaveView') #print "** INITIALIZING SaveView" self.pluginIsActive = False self.dockwidget = None # noinspection PyMethodMayBeStatic def tr(self, message): """Get the translation for a string using Qt translation API. We implement this ourselves since we do not inherit QObject. :param message: String for translation. :type message: str, QString :returns: Translated version of message. :rtype: QString """ # noinspection PyTypeChecker,PyArgumentList,PyCallByClass return QCoreApplication.translate('SaveView', message) def add_action( self, icon_path, text, callback, enabled_flag=True, add_to_menu=True, add_to_toolbar=True, status_tip=None, whats_this=None, parent=None): """Add a toolbar icon to the toolbar. :param icon_path: Path to the icon for this action. Can be a resource path (e.g. ':/plugins/foo/bar.png') or a normal file system path. :type icon_path: str :param text: Text that should be shown in menu items for this action. :type text: str :param callback: Function to be called when the action is triggered. :type callback: function :param enabled_flag: A flag indicating if the action should be enabled by default. Defaults to True. :type enabled_flag: bool :param add_to_menu: Flag indicating whether the action should also be added to the menu. Defaults to True. :type add_to_menu: bool :param add_to_toolbar: Flag indicating whether the action should also be added to the toolbar. Defaults to True. :type add_to_toolbar: bool :param status_tip: Optional text to show in a popup when mouse pointer hovers over the action. :type status_tip: str :param parent: Parent widget for the new action. Defaults None. :type parent: QWidget :param whats_this: Optional text to show in the status bar when the mouse pointer hovers over the action. :returns: The action that was created. Note that the action is also added to self.actions list. :rtype: QAction """ icon = QIcon(icon_path) action = QAction(icon, text, parent) action.triggered.connect(callback) action.setEnabled(enabled_flag) if status_tip is not None: action.setStatusTip(status_tip) if whats_this is not None: action.setWhatsThis(whats_this) if add_to_toolbar: self.toolbar.addAction(action) if add_to_menu: self.iface.addPluginToMenu( self.menu, action) self.actions.append(action) return action def initGui(self): """Create the menu entries and toolbar icons inside the QGIS GUI.""" icon_path = ':/plugins/save_views/icon.png' self.add_action( icon_path, text=self.tr(u'Save views'), callback=self.run, parent=self.iface.mainWindow()) #-------------------------------------------------------------------------- def onClosePlugin(self): """Cleanup necessary items here when plugin dockwidget is closed""" #print "** CLOSING SaveView" # disconnects self.dockwidget.closingPlugin.disconnect(self.onClosePlugin) # remove this statement if dockwidget is to remain # for reuse if plugin is reopened # Commented next statement since it causes QGIS crashe # when closing the docked window: # self.dockwidget = None self.pluginIsActive = False def unload(self): """Removes the plugin menu item and icon from QGIS GUI.""" #print "** UNLOAD SaveView" for action in self.actions: self.iface.removePluginMenu( self.tr(u'&Save Views'), action) self.iface.removeToolBarIcon(action) # remove the toolbar del self.toolbar #-------------------------------------------------------------------------- def run(self): """Run method that loads and starts the plugin""" if not self.pluginIsActive: self.pluginIsActive = True #print "** STARTING SaveView" # dockwidget may not exist if: # first run of plugin # removed on close (see self.onClosePlugin method) if self.dockwidget == None: # Create the dockwidget (after translation) and keep reference self.dockwidget = SaveViewsDockWidget() # connect to provide cleanup on closing of dockwidget self.dockwidget.closingPlugin.connect(self.onClosePlugin) # show the dockwidget # TODO: fix to allow choice of dock location self.iface.addDockWidget(Qt.LeftDockWidgetArea, self.dockwidget) self.dockwidget.show() # connect the select_output_file() method to the clicked signal of the tool button widget self.dockwidget.toolButton.clicked.connect(self.select_output_dir) # connect the save_views() method to the clicked signal of the push button widget self.dockwidget.pushButton.clicked.connect(self.save_views) # populate the Combo Box with the vector layers loaded in QGIS from qgis.core import QgsProject, QgsMapLayer self.dockwidget.comboBox.clear() for layer in QgsProject.instance().mapLayers().values(): if layer.type() != QgsMapLayer.VectorLayer: continue self.dockwidget.comboBox.addItem(layer.name()) # clear the previously loaded text (if any) in the line edit widget self.dockwidget.lineEdit.clear() # open directory browser and populate the line edit widget def select_output_dir(self): from PyQt5.QtWidgets import QFileDialog self.dirname = QFileDialog.getExistingDirectory( self.dockwidget, "Select directory ", os.path.expanduser("~") ) self.dockwidget.lineEdit.setText(self.dirname) def save_views(self): from PyQt5.QtGui import QColor, QPixmap from qgis.core import QgsProject from qgis.utils import iface # save graphical output for every row in attribute table layer_name = self.dockwidget.comboBox.currentText() layer = QgsProject.instance().mapLayersByName(layer_name)[0] for feature in layer.getFeatures(): layer.selectByIds([feature.id()]) self.iface.mapCanvas().setSelectionColor(QColor("transparent")); box = layer.boundingBoxOfSelected() self.iface.mapCanvas().setExtent(box) pixmap = QPixmap(self.iface.mapCanvas().mapSettings().outputSize().width(), self.iface.mapCanvas().mapSettings().outputSize().height() ) mapfile = os.path.join(self.dirname, '{0}_{1:03d}.png'.format(layer_name, feature.id())) self.iface.mapCanvas().saveAsImage(mapfile, pixmap) layer.removeSelection() # save also full extend of vector layer canvas = self.iface.mapCanvas() canvas.setExtent(layer.extent()) pixmap = QPixmap(self.iface.mapCanvas().mapSettings().outputSize().width(), self.iface.mapCanvas().mapSettings().outputSize().height() ) mapfile = os.path.join(self.dirname, '{}_full.png'.format(layer_name)) self.iface.mapCanvas().saveAsImage(mapfile, pixmap)