MultLayerSelection.py

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 Multi Layer Selection
                                 A QGIS plugin
                              -------------------
        begin                : 2014-10-07
        git sha              : $Format:%H$
        copyright            : (C) 2014 by Felipe Ferrari
        email                : ferrari@gmail.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 PyQt4.QtCore import *
from PyQt4.QtGui import *
from qgis.gui import *
from qgis.core import *
from MultiLayerSelection import MultiLayerSelection
from MultiLayerRectangleSelection import MultiLayerRectangleSelection
# Initialize Qt resources from file resources.py
import resources

import os.path

class MultLayerSelection:
    """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',
            'MultLayerSelection_{}.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)
                
        self.toolbar = self.iface.addToolBar(u'MultipleSelection')
        self.toolbar.setObjectName(u'MultipleSelection')

    def createToolButton(self, parent, text):
        button = QToolButton(parent)
        button.setObjectName(text)
        button.setToolButtonStyle(Qt.ToolButtonIconOnly)
        button.setPopupMode(QToolButton.MenuButtonPopup)
        parent.addWidget(button)
        return button
    
    def createAction(self, icon_path, text, callback):
        action = QAction(
            QIcon(icon_path),
            text,
            self.iface.mainWindow())
        # connect the action to the run method
        action.setCheckable(True)
        action.toggled.connect(callback)
        return action
    
    def createClearAction(self, icon_path, text):
        action = QAction(
            QIcon(icon_path),
            text,
            self.iface.mainWindow())
        # connect the action to the run method
        action.setCheckable(False)
        action.triggered.connect(self.clear)
        return action
 
    def initGui(self):
        # Create action that will start plugin configuration
        self.actionCriar = self.createAction(":/plugins/MultipleLayerSelection/icon.png",
                                            u"Multi Selection by point",
                                            self.run)

        # Create action that will start plugin configuration
        self.actionCriarRectangle = self.createAction(":/plugins/MultipleLayerSelection/icon_rectangle.png",
                                                      u"Multiple Selection by Rectangle",
                                                      self.runRectangle)

        # Create action that will start plugin configuration
        self.actionClear = self.createClearAction(":/plugins/MultipleLayerSelection/icon_clear.png",
                                                      u"Clear selections")
                
        self.tool = MultiLayerSelection(self.iface.mapCanvas(), self.actionCriar) 
        self.toolRectangle = MultiLayerRectangleSelection(self.iface.mapCanvas(), self.actionCriarRectangle) 
        
        #QToolButtons
        self.selectionButton = self.createToolButton(self.toolbar, u'MultipleSelectionButton')
        self.selectionButton.addAction(self.actionCriar)     
        self.selectionButton.addAction(self.actionCriarRectangle)
        self.selectionButton.addAction(self.actionClear)
        self.selectionButton.setDefaultAction(self.actionCriar)    

    def unload(self):
        # Remove the plugin menu item and icon
        self.iface.mainWindow().removeToolBar(self.toolbar)
        
    def clear(self):
        self.actionCriar.setChecked(False)
        self.actionCriarRectangle.setChecked(False)
        self.selectionButton.setDefaultAction(self.selectionButton.sender())
        layers = self.iface.mapCanvas().layers()
        for layer in layers:
            if layer.type() == QgsMapLayer.RasterLayer:
                continue
            layer.removeSelection()

    def run(self, b):
        self.actionCriarRectangle.setChecked(False)
        self.selectionButton.setDefaultAction(self.selectionButton.sender())
        if b:
            self.iface.mapCanvas().setMapTool(self.tool)
        else:
            self.iface.mapCanvas().unsetMapTool(self.tool)

    def runRectangle(self, b):
        self.actionCriar.setChecked(False)
        self.selectionButton.setDefaultAction(self.selectionButton.sender())
        if b:
            self.iface.mapCanvas().setMapTool(self.toolRectangle)
        else:
            self.iface.mapCanvas().unsetMapTool(self.toolRectangle)