class PolygonComplexityDialogTest(unittest.TestCase): """Test dialog works.""" def setUp(self): """Runs before each test.""" self.dialog = PolygonComplexityDialog(None) def tearDown(self): """Runs after each test.""" self.dialog = None def test_dialog_ok(self): """Test we can click OK.""" button = self.dialog.button_box.button(QDialogButtonBox.Ok) button.click() result = self.dialog.result() self.assertEqual(result, QDialog.Accepted) def test_dialog_cancel(self): """Test we can click cancel.""" button = self.dialog.button_box.button(QDialogButtonBox.Cancel) button.click() result = self.dialog.result() self.assertEqual(result, QDialog.Rejected)
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', 'PolygonComplexity_{}.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) # Create the dialog (after translation) and keep reference self.dlg = PolygonComplexityDialog() # Declare instance attributes self.actions = [] self.menu = self.tr(u'&PolygonComplexity') # TODO: We are going to let the user set this up in a future iteration self.toolbar = self.iface.addToolBar(u'PolygonComplexity') self.toolbar.setObjectName(u'PolygonComplexity')
def setUp(self): """Runs before each test.""" self.dialog = PolygonComplexityDialog(None)
class PolygonComplexity: """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', 'PolygonComplexity_{}.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) # Create the dialog (after translation) and keep reference self.dlg = PolygonComplexityDialog() # Declare instance attributes self.actions = [] self.menu = self.tr(u'&PolygonComplexity') # TODO: We are going to let the user set this up in a future iteration self.toolbar = self.iface.addToolBar(u'PolygonComplexity') self.toolbar.setObjectName(u'PolygonComplexity') # 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('PolygonComplexity', 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/PolygonComplexity/logo.png' self.add_action( icon_path, text=self.tr(u'PolygonComplexity'), callback=self.run, parent=self.iface.mainWindow()) def unload(self): """Removes the plugin menu item and icon from QGIS GUI.""" for action in self.actions: self.iface.removePluginMenu( self.tr(u'&PolygonComplexity'), action) self.iface.removeToolBarIcon(action) def run(self): """Run method that performs all the real work""" # Populate the combo boxes self.dlg.populateLayers() # show the dialog self.dlg.show() # Run the dialog event loop result = self.dlg.exec_() # See if OK was pressed if result == 1: if not self.dlg.ui.inputLayer.currentText(): QMessageBox.information( self.iface.mainWindow(),"Info", "Please select a polygon layer") else: layer = ftu.getMapLayerByName(self.dlg.ui.inputLayer.currentText()) nFeatures = layer.featureCount() pr = layer.dataProvider() layer.startEditing() if self.dlg.ui.cnessCheckbox.isChecked() and layer.fieldNameIndex("CS") == -1 : pr.addAttributes( [ QgsField("CS", QVariant.Double)] ) layer.commitChanges() if self.dlg.ui.complCheckbox.isChecked() and layer.fieldNameIndex("CP") == -1 : pr.addAttributes( [ QgsField("CP", QVariant.Double)] ) layer.commitChanges() if self.dlg.ui.convCheckbox.isChecked() and layer.fieldNameIndex("CV") == -1 : pr.addAttributes( [ QgsField("CV", QVariant.Double)] ) layer.commitChanges() if self.dlg.ui.amplCheckbox.isChecked() and layer.fieldNameIndex("AP") == -1 : pr.addAttributes( [ QgsField("AP", QVariant.Double)] ) layer.commitChanges() if self.dlg.ui.freqCheckbox.isChecked() and layer.fieldNameIndex("FQ") == -1 : pr.addAttributes( [ QgsField("FQ", QVariant.Double)] ) layer.commitChanges() if self.dlg.ui.vertCheckbox.isChecked() and layer.fieldNameIndex("vert") == -1 : pr.addAttributes( [ QgsField("vert", QVariant.Int)] ) layer.commitChanges() # Loop over the features for i, feature in enumerate(layer.dataProvider().getFeatures()): self.iface.mainWindow().statusBar().showMessage( "Processing feature {} of {}".format(i + 1, nFeatures)) geom = feature.geometry() feature_compactness = find_feature_compactness(geom) feature_convexity = find_feature_convexity(geom) feature_amplitude = find_feature_amplitude(geom) feature_vertices = find_feature_vertices(geom) feature_notches = find_feature_notches(geom, self) feature_notches_normalized = float(feature_notches) / (feature_vertices - 3) feature_frequency = (16*((feature_notches_normalized - 0.5)**4)) - (8*(feature_notches_normalized - 0.5)**2) + 1 feature_complexity = find_feature_complexity(feature_convexity, feature_amplitude, feature_frequency) if self.dlg.ui.cnessCheckbox.isChecked(): layer.startEditing() feature['CS'] = feature_compactness layer.updateFeature(feature) if self.dlg.ui.complCheckbox.isChecked(): layer.startEditing() feature['CP'] = feature_complexity layer.updateFeature(feature) if self.dlg.ui.convCheckbox.isChecked(): layer.startEditing() feature['CV'] = feature_convexity layer.updateFeature(feature) if self.dlg.ui.amplCheckbox.isChecked(): layer.startEditing() feature['AP'] = feature_amplitude layer.updateFeature(feature) if self.dlg.ui.freqCheckbox.isChecked(): layer.startEditing() feature['FQ'] = feature_frequency layer.updateFeature(feature) if self.dlg.ui.vertCheckbox.isChecked(): layer.startEditing() feature['vert'] = feature_vertices - 1 layer.updateFeature(feature) layer.commitChanges()