def editOptions(self):
'''Show dialog for option editing'''
diag = OptionsDialog(self)
logger.info('Options dialog started')
if diag.exec_() == QDialog.Accepted:
self.options.update(diag.getOptions())
logger.info('Options changed')
self.websocketServer.updatePort(self.options['websocket-port'])
self.websocketServer.updateWhitelist(
self.options['websocket-whitelist'])
if self.options['websocket-enabled']:
self.startWebsocketServer()
else:
self.websocketServer.close()
class AutoPolygon:
"""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
self.canvas = self.iface.mapCanvas()
# 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',
'Auto_Polygon_{}.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)
#Configuration
self.config = ConfigParser.RawConfigParser()
# Create the dialog (after translation) and keep reference
self.dlg = AutoPolygonDialog()
self.dlg_options = OptionsDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'Automatic Polygon')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'AutoPolygon')
self.toolbar.setObjectName(u'AutoPolygon')
# 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('Auto_Polygon', 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."""
# Add actions
#Automatic creation of a polygon with a point
icon_path = ':/plugins/AutoPolygon/icons/rectangle_point.png'
self.rectdigitpoint = self.add_action(
icon_path,
text=self.tr(u'Automatic creation of a polygon with a point'),
callback=self.rectcreate_point,
add_to_menu=True,
add_to_toolbar=False,
parent=self.iface.mainWindow())
#Automatic creation of a polygon with a line
icon_path = ':/plugins/AutoPolygon/icons/rectangle_line.png'
self.rectdigitline = self.add_action(
icon_path,
text=self.tr(u'Automatic creation of a polygon with a line'),
callback=self.rectcreate_line,
add_to_menu=True,
add_to_toolbar=False,
parent=self.iface.mainWindow())
# Get the tool to digitize a rectangle
self.rectdigittool = RectDigitTool(self.canvas)
#Configuring the rectangle tool
self.RectMaj()
#Polygon creation tools
self.rectdigittools = QToolButton(self.toolbar)
self.rectdigittools.setPopupMode(QToolButton.MenuButtonPopup)
self.rectdigittools.addActions(
[self.rectdigitpoint, self.rectdigitline])
self.rectdigittools.setDefaultAction(self.rectdigitpoint)
self.toolbar.addWidget(self.rectdigittools)
#Creating the option button
icon_path = ':/plugins/AutoPolygon/icons/option.png'
self.options_conf = self.add_action(icon_path,
text=self.tr(u'Options'),
callback=self.options,
add_to_menu=True,
add_to_toolbar=True,
parent=self.iface.mainWindow())
#Connect to the event change layer
QObject.connect(self.iface,
SIGNAL("currentLayerChanged(QgsMapLayer*)"),
self.toggle)
# At the end of the feature creation, call the function createFeature
QObject.connect(self.rectdigittool,
SIGNAL("rbFinished(PyQt_PyObject)"),
self.createFeature)
# By default, put the enabled button to False
self.rectdigitpoint.setEnabled(False)
self.rectdigitline.setEnabled(False)
#If a layer is already selected, you are asked to perform toggle if the user requests editing
self.toggle()
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.actions:
self.iface.removePluginMenu(self.tr(u'&Automatic Polygon'), action)
self.iface.removeToolBarIcon(action)
#Disconnecting events
QObject.disconnect(self.rectdigittool,
SIGNAL("rbFinished(PyQt_PyObject)"),
self.createFeature)
QObject.disconnect(self.iface,
SIGNAL("currentLayerChanged(QgsMapLayer*)"),
self.toggle)
def run(self):
"""Run method that performs all the real work"""
# show the dialog
self.dlg.show()
# Run the dialog event loop
result = self.dlg.exec_()
# See if OK was pressed
if result:
# Do something useful here - delete the line containing pass and
# substitute with your code.
pass
#Function changing a layer
def toggle(self):
mc = self.canvas
layer = mc.currentLayer()
#Decide whether the plugin button/menu is enabled or disabled
if layer <> None:
if layer.type() == QgsMapLayer.VectorLayer:
#If the layer is editable and the geometry is a type polygon
if (layer.isEditable() and layer.geometryType() == 2):
#It makes clickable button
self.rectdigitpoint.setEnabled(True)
self.rectdigitline.setEnabled(True)
#Connect to the event editing stopped
QObject.connect(layer, SIGNAL("editingStopped()"),
self.toggle)
#Disconnect to the event editing started
QObject.disconnect(layer, SIGNAL("editingStarted()"),
self.toggle)
else:
#It makes unclickable button
self.rectdigitpoint.setEnabled(False)
self.rectdigitline.setEnabled(False)
if layer.geometryType() == 2:
#Connect to the event editing started
QObject.connect(layer, SIGNAL("editingStarted()"),
self.toggle)
else:
#Disconnect to the event editing started
QObject.disconnect(layer, SIGNAL("editingStarted()"),
self.toggle)
#Disconnect to the event editing stopped
QObject.disconnect(layer, SIGNAL("editingStopped()"),
self.toggle)
else:
#Disconnect to the event editing started
QObject.disconnect(layer, SIGNAL("editingStarted()"),
self.toggle)
def rectcreate_point(self):
self.canvas.setMapTool(self.rectdigittool)
self.rectdigittools.setDefaultAction(self.rectdigitpoint)
result = self.rectdigittool.SetPositionTypeSaisie("Point")
if result is not None:
self.iface.messageBar().pushMessage(self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3)
def rectcreate_line(self):
self.canvas.setMapTool(self.rectdigittool)
self.rectdigittools.setDefaultAction(self.rectdigitline)
result = self.rectdigittool.SetPositionTypeSaisie("Line")
if result is not None:
self.iface.messageBar().pushMessage(self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3)
#Function when create feature
def createFeature(self, geom):
settings = QSettings()
mc = self.canvas
#Get the current layer
layer = mc.currentLayer()
if layer <> None:
renderer = mc.mapRenderer()
#Get the layer SRID
layerCRSSrsid = layer.crs().srsid()
projectCRSSrsid = renderer.destinationCrs().srsid()
provider = layer.dataProvider()
f = QgsFeature()
#On the Fly reprojection.
if layerCRSSrsid != projectCRSSrsid:
geom.transform(
QgsCoordinateTransform(projectCRSSrsid, layerCRSSrsid))
f.setGeometry(geom)
# add attribute fields to feature
fields = layer.pendingFields()
# vector api change update
f.initAttributes(fields.count())
for i in range(fields.count()):
f.setAttribute(i, provider.defaultValue(i))
layer.beginEditCommand("Create a feature")
layer.addFeature(f)
layer.endEditCommand()
else:
QObject.disconnect(self.rectdigittool,
SIGNAL("rbFinished(PyQt_PyObject)"),
self.createFeature)
# Showing the option dialog box
def options(self):
"""Run method that performs all the real work"""
# show the dialog
self.dlg_options.show()
# Run the dialog event loop
result = self.dlg_options.exec_()
# See if OK was pressed
if result:
# Do something useful here - delete the line containing pass and
# substitute with your code.
#Configuring the rectangle tool
self.RectMaj()
pass
def RectMaj(self):
#Configuring the rectangle tool
try:
self.config.read(self.plugin_dir + '/AutoPolygon.cfg')
except:
return
try:
width = self.config.getfloat('RECTANGLE', 'width')
height = self.config.getfloat('RECTANGLE', 'height')
position = self.config.get('RECTANGLE', 'position')
except:
#If there were errors, the default values are put
width = 2
height = 1
position = self.tr(u'Bottom Left').encode('utf-8').strip()
self.rectdigittool.SetWidth(width)
self.rectdigittool.SetHeight(height)
result = self.rectdigittool.SetPositionPoint(position)
#If the position doesn't existSi la position n'existe pas
if result is not None:
self.iface.messageBar().pushMessage(self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3)
class AutoPolygon:
"""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
self.canvas = self.iface.mapCanvas()
# 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',
'Auto_Polygon_{}.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)
#Configuration
self.config = ConfigParser.RawConfigParser()
# Create the dialog (after translation) and keep reference
self.dlg = AutoPolygonDialog()
self.dlg_options = OptionsDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'Automatic Polygon')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'AutoPolygon')
self.toolbar.setObjectName(u'AutoPolygon')
# 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('Auto_Polygon', 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."""
# Add actions
#Automatic creation of a polygon with a point
icon_path = ':/plugins/AutoPolygon/icons/rectangle_point.png'
self.rectdigitpoint = self.add_action(
icon_path,
text=self.tr(u'Automatic creation of a polygon with a point'),
callback=self.rectcreate_point,
add_to_menu=True,
add_to_toolbar=False,
parent=self.iface.mainWindow())
#Automatic creation of a polygon with a line
icon_path = ':/plugins/AutoPolygon/icons/rectangle_line.png'
self.rectdigitline = self.add_action(
icon_path,
text=self.tr(u'Automatic creation of a polygon with a line'),
callback=self.rectcreate_line,
add_to_menu=True,
add_to_toolbar=False,
parent=self.iface.mainWindow())
# Get the tool to digitize a rectangle
self.rectdigittool = RectDigitTool( self.canvas )
#Configuring the rectangle tool
self.RectMaj()
#Polygon creation tools
self.rectdigittools = QToolButton(self.toolbar)
self.rectdigittools.setPopupMode(QToolButton.MenuButtonPopup)
self.rectdigittools.addActions( [ self.rectdigitpoint, self.rectdigitline ] )
self.rectdigittools.setDefaultAction(self.rectdigitpoint)
self.toolbar.addWidget(self.rectdigittools)
#Creating the option button
icon_path = ':/plugins/AutoPolygon/icons/option.png'
self.options_conf = self.add_action(
icon_path,
text=self.tr(u'Options'),
callback=self.options,
add_to_menu=True,
add_to_toolbar=True,
parent=self.iface.mainWindow())
#Connect to the event change layer
QObject.connect(self.iface, SIGNAL("currentLayerChanged(QgsMapLayer*)"), self.toggle)
# At the end of the feature creation, call the function createFeature
QObject.connect(self.rectdigittool, SIGNAL("rbFinished(PyQt_PyObject)"), self.createFeature)
# By default, put the enabled button to False
self.rectdigitpoint.setEnabled(False)
self.rectdigitline.setEnabled(False)
#If a layer is already selected, you are asked to perform toggle if the user requests editing
self.toggle()
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.actions:
self.iface.removePluginMenu(
self.tr(u'&Automatic Polygon'),
action)
self.iface.removeToolBarIcon(action)
#Disconnecting events
QObject.disconnect(self.rectdigittool, SIGNAL("rbFinished(PyQt_PyObject)"), self.createFeature)
QObject.disconnect(self.iface, SIGNAL("currentLayerChanged(QgsMapLayer*)"), self.toggle)
def run(self):
"""Run method that performs all the real work"""
# show the dialog
self.dlg.show()
# Run the dialog event loop
result = self.dlg.exec_()
# See if OK was pressed
if result:
# Do something useful here - delete the line containing pass and
# substitute with your code.
pass
#Function changing a layer
def toggle(self):
mc = self.canvas
layer = mc.currentLayer()
#Decide whether the plugin button/menu is enabled or disabled
if layer <> None:
if layer.type() == QgsMapLayer.VectorLayer:
#If the layer is editable and the geometry is a type polygon
if (layer.isEditable() and layer.geometryType() == 2):
#It makes clickable button
self.rectdigitpoint.setEnabled(True)
self.rectdigitline.setEnabled(True)
#Connect to the event editing stopped
QObject.connect(layer,SIGNAL("editingStopped()"),self.toggle)
#Disconnect to the event editing started
QObject.disconnect(layer,SIGNAL("editingStarted()"),self.toggle)
else:
#It makes unclickable button
self.rectdigitpoint.setEnabled(False)
self.rectdigitline.setEnabled(False)
if layer.geometryType() == 2:
#Connect to the event editing started
QObject.connect(layer,SIGNAL("editingStarted()"),self.toggle)
else:
#Disconnect to the event editing started
QObject.disconnect(layer,SIGNAL("editingStarted()"),self.toggle)
#Disconnect to the event editing stopped
QObject.disconnect(layer,SIGNAL("editingStopped()"),self.toggle)
else:
#Disconnect to the event editing started
QObject.disconnect(layer,SIGNAL("editingStarted()"),self.toggle)
def rectcreate_point(self):
self.canvas.setMapTool(self.rectdigittool)
self.rectdigittools.setDefaultAction(self.rectdigitpoint)
result = self.rectdigittool.SetPositionTypeSaisie("Point")
if result is not None :
self.iface.messageBar().pushMessage(
self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3
)
def rectcreate_line(self):
self.canvas.setMapTool(self.rectdigittool)
self.rectdigittools.setDefaultAction(self.rectdigitline)
result = self.rectdigittool.SetPositionTypeSaisie("Line")
if result is not None :
self.iface.messageBar().pushMessage(
self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3
)
#Function when create feature
def createFeature(self, geom):
settings = QSettings()
mc = self.canvas
#Get the current layer
layer = mc.currentLayer()
if layer <> None:
renderer = mc.mapRenderer()
#Get the layer SRID
layerCRSSrsid = layer.crs().srsid()
projectCRSSrsid = renderer.destinationCrs().srsid()
provider = layer.dataProvider()
f = QgsFeature()
#On the Fly reprojection.
if layerCRSSrsid != projectCRSSrsid:
geom.transform(QgsCoordinateTransform(projectCRSSrsid, layerCRSSrsid))
f.setGeometry(geom)
# add attribute fields to feature
fields = layer.pendingFields()
# vector api change update
f.initAttributes(fields.count())
for i in range(fields.count()):
f.setAttribute(i,provider.defaultValue(i))
layer.beginEditCommand("Create a feature")
layer.addFeature(f)
layer.endEditCommand()
else:
QObject.disconnect(self.rectdigittool, SIGNAL("rbFinished(PyQt_PyObject)"), self.createFeature)
# Showing the option dialog box
def options(self):
"""Run method that performs all the real work"""
# show the dialog
self.dlg_options.show()
# Run the dialog event loop
result = self.dlg_options.exec_()
# See if OK was pressed
if result:
# Do something useful here - delete the line containing pass and
# substitute with your code.
#Configuring the rectangle tool
self.RectMaj()
pass
def RectMaj(self):
#Configuring the rectangle tool
try: self.config.read( self.plugin_dir+'/AutoPolygon.cfg' )
except: return
try:
width=self.config.getfloat('RECTANGLE','width')
height=self.config.getfloat('RECTANGLE','height')
position=self.config.get('RECTANGLE','position')
except:
#If there were errors, the default values are put
width=2
height=1
position= self.tr(u'Bottom Left').encode('utf-8').strip()
self.rectdigittool.SetWidth (width)
self.rectdigittool.SetHeight (height)
result = self.rectdigittool.SetPositionPoint(position)
#If the position doesn't existSi la position n'existe pas
if result is not None :
self.iface.messageBar().pushMessage(
self.tr(u'Warning'),
self.tr(result),
level=QgsMessageBar.WARNING,
duration=3
)
