-
Notifications
You must be signed in to change notification settings - Fork 0
/
ThinGreyscale.py
154 lines (137 loc) · 6.26 KB
/
ThinGreyscale.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
# -*- coding: utf-8 -*-
"""
/***************************************************************************
ThinGreyscale
A QGIS plugin
Thin a greyscale image to a skeleton
-------------------
begin : 2014-12-22
git sha : $Format:%H$
copyright : (C) 2014 by Håvard Tveite, NMBU
email : havard.tveite@nmbu.no
***************************************************************************/
/***************************************************************************
* *
* 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. *
* *
***************************************************************************/
"""
import os.path
from qgis.PyQt.QtCore import QSettings, QTranslator, qVersion
from qgis.PyQt.QtCore import QCoreApplication
from qgis.PyQt.QtGui import QIcon
from qgis.PyQt.QtWidgets import QAction
from qgis.core import QgsMapLayer, QgsProject
#from qgis.core import QgsMessageLog
# Initialize Qt resources from file resources.py
from . resources import *
# Import the code for the dialog
from .ThinGreyscale_dialog import ThinGreyscaleDialog
class ThinGreyscale:
"""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
pluginPath = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(
pluginPath,
'i18n',
#'ThinGreyscale_{}.qm'.format(locale))
'{}.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 = ThinGreyscaleDialog(self.iface)
# Declare instance attributes
self.THINGRAYSCALE = self.tr(u'&Thin greyscale image to skeleton')
self.THINGRAYSCALEAMP = self.tr('&ThinGreyscale')
self.menu = self.THINGRAYSCALE
# TODO: We are going to let the user set this up in a future iteration
#self.toolbar = self.iface.addToolBar(u'ThinGreyscale')
#self.toolbar.setObjectName(u'ThinGreyscale')
# 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('ThinGreyscale', message)
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
#icon_path = ':/plugins/ThinGreyscale/icon.png'
icon_path = os.path.join(os.path.dirname(__file__), "icon.png")
self.action = QAction(
QIcon(icon_path),
self.menu, self.iface.mainWindow())
# connect the action to the run method
self.action.triggered.connect(self.run)
# Add menu item and Add toolbar icon
if hasattr(self.iface, 'addPluginToRasterMenu'):
self.iface.addPluginToRasterMenu(self.menu, self.action)
self.iface.addRasterToolBarIcon(self.action)
else:
self.iface.addPluginToMenu(self.menu, self.action)
self.iface.addToolBarIcon(self.action)
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
if hasattr(self.iface, 'removePluginRasterMenu'):
self.iface.removePluginRasterMenu(self.menu, self.action)
self.iface.removeRasterToolBarIcon(self.action)
else:
self.iface.removePluginMenu(self.menu, self.action)
self.iface.removeToolBarIcon(self.action)
def run(self):
"""Run method that performs all the real work"""
# show the dialog
#self.dlg.show()
self.dlg.progressBar.setValue(0.0)
# Populate the inputRaster comboBox
# Should also check for the type of raster (must be
# singleband grayscale)
self.dlg.inputRaster.clear()
layers = QgsProject.instance().mapLayers()
for id in layers.keys():
#gdalmetadata = alayer.metadata()
# Skip WMS layers
#WMSstring = 'Web Map Service'
#wmspos = gdalmetadata.find(WMSstring)
#if wmspos != -1:
# continue
#provstring = '<p>GDAL provider</p>\n'
#providerpos = gdalmetadata.find(provstring)
#if providerpos == -1:
# continue
#brpos = gdalmetadata.find('<br>', providerpos + len(provstring))
#aftprovpos = int(providerpos + len(provstring))
#gdalprovider = gdalmetadata[aftprovpos:int(brpos)]
if layers[id].type() == QgsMapLayer.RasterLayer:
self.dlg.inputRaster.addItem(layers[id].name(), id)
# show the dialog (needed for the messagebar cancel button)
self.dlg.show()
# Run the dialog event loop
#result = self.dlg.exec_()
# See if OK was pressed
#if result:
# pass
#QgsMessageLog.logMessage('Run method finished.',
# self.dlg.THINGREYSCALE,
# QgsMessageLog.INFO)