def paint(self, painter, option, widget=None):
brush = self.brush()
if self.__hover:
brush = QBrush(brush.color().darker(110))
painter.setBrush(brush)
painter.setPen(self.pen())
painter.drawEllipse(self.rect())
def paint(self, painter, option, widget=None):
brush = self.brush()
if self.__hover:
brush = QBrush(brush.color().darker(110))
painter.setBrush(brush)
painter.setPen(self.pen())
painter.drawEllipse(self.rect())
def brush_darker(brush, factor):
"""Return a copy of the brush darkened by factor.
"""
grad = brush.gradient()
if grad:
return QBrush(gradient_darker(grad, factor))
else:
brush = QBrush(brush)
brush.setColor(brush.color().darker(factor))
return brush
def brush_darker(brush, factor):
"""Return a copy of the brush darkened by factor.
"""
grad = brush.gradient()
if grad:
return QBrush(gradient_darker(grad, factor))
else:
brush = QBrush(brush)
brush.setColor(brush.color().darker(factor))
return brush
class FeatureHighlightItem(QgsMapCanvasItem):
# Code ported from QGIS QgsHighlight
def __init__(self, mapCanvas, feature, layer):
super(FeatureHighlightItem, self).__init__(mapCanvas)
self._mapCanvas = None # QgsMapCanvas
self._brush = QBrush()
self._pen = QPen()
self._feature = None # QgsFeature()
self._layer = None # QgsMapLayer()
self._buffer = 0.0
self._minWidth = 0.0
self._mapCanvas = mapCanvas
if (not layer or not feature or not isinstance(feature, QgsFeature)
or not feature.geometry() or feature.geometry().isEmpty()
or not feature.geometry().isGeosValid()):
return
self._feature = QgsFeature(feature) # Force deep copy
self._layer = layer
self.setLineColor(Application.highlightLineColor())
self.setFillColor(Application.highlightFillColor())
self._minWidth = Application.highlightMinimumWidth()
self._buffer = Application.highlightBuffer()
if self._mapCanvas.mapSettings().hasCrsTransformEnabled():
ct = self._mapCanvas.mapSettings().layerTransform(self._layer)
if ct:
self._feature.geometry().transform(ct)
self.updateRect()
self.update()
def remove(self):
self._mapCanvas.scene().removeItem(self)
def setLineWidth(self, width):
self._pen.setWidth(width)
def setLineColor(self, color):
self._pen.setColor(color)
def setFillColor(self, fillColor):
self._brush.setColor(fillColor)
self._brush.setStyle(Qt.SolidPattern)
def setBuffer(self, buff):
self._buffer = buff
def setMinWidth(self, width):
self._minWidth = width
def layer(self):
return self._layer
def updatePosition(self):
pass
# protected:
def paint(self, painter, option=None, widget=None): # Override
if not self._feature:
return
mapSettings = self._mapCanvas.mapSettings()
context = QgsRenderContext.fromMapSettings(mapSettings)
renderer = self._getRenderer(context, self._pen.color(),
self._brush.color())
if renderer:
context.setPainter(painter)
renderer.startRender(context, self._layer.fields())
renderer.renderFeature(self._feature, context)
renderer.stopRender(context)
def updateRect(self):
if self._feature and self._feature.constGeometry():
m2p = self._mapCanvas.mapSettings().mapToPixel()
topLeft = m2p.toMapPoint(0, 0)
res = m2p.mapUnitsPerPixel()
imageSize = self._mapCanvas.mapSettings().outputSize()
rect = QgsRectangle(topLeft.x(), topLeft.y(),
topLeft.x() + imageSize.width() * res,
topLeft.y() - imageSize.height() * res)
self.setRect(rect)
self.setVisible(True)
else:
self.setRect(QgsRectangle())
# private:
def _setSymbol(self, symbol, context, color, fillColor):
if not symbol:
return
for symbolLayer in reversed(symbol.symbolLayers()):
if symbolLayer:
if symbolLayer.subSymbol():
self._setSymbol(symbolLayer.subSymbol(), context, color,
fillColor)
else:
symbolLayer.setColor(color)
symbolLayer.setOutlineColor(color)
symbolLayer.setFillColor(fillColor)
if isinstance(symbolLayer, QgsSimpleMarkerSymbolLayerV2):
symbolLayer.setOutlineWidth(
self._getSymbolWidth(
context, symbolLayer.outlineWidth(),
symbolLayer.outlineWidthUnit()))
if symbolLayer.type() == QgsSymbolV2.Line:
symbolLayer.setWidth(
self._getSymbolWidth(context, symbolLayer.width(),
symbolLayer.widthUnit()))
if symbolLayer.type() == QgsSymbolV2.Fill:
symbolLayer.setBorderWidth(
self._getSymbolWidth(context,
symbolLayer.borderWidth(),
symbolLayer.outputUnit()))
symbolLayer.removeDataDefinedProperty('color')
symbolLayer.removeDataDefinedProperty('color_border')
def _getSymbolWidth(self, context, width, unit):
scale = 1.0
if unit == QgsSymbolV2.MapUnit:
scale = QgsSymbolLayerV2Utils.lineWidthScaleFactor(
context,
QgsSymbolV2.MM) / QgsSymbolLayerV2Utils.lineWidthScaleFactor(
context, QgsSymbolV2.MapUnit)
width = max(width + 2 * self._buffer * scale, self._minWidth * scale)
return width
def _getRenderer(self, context, color, fillColor):
renderer = None
if self._layer and self._layer.rendererV2():
renderer = self._layer.rendererV2().clone()
if renderer:
for symbol in renderer.symbols2(context):
self._setSymbol(symbol, context, color, fillColor)
return renderer
class FeatureHighlightItem(QgsMapCanvasItem):
# Code ported from QGIS QgsHighlight
def __init__(self, mapCanvas, feature, layer):
super(FeatureHighlightItem, self).__init__(mapCanvas)
self._mapCanvas = None # QgsMapCanvas
self._brush = QBrush()
self._pen = QPen()
self._feature = None # QgsFeature()
self._layer = None # QgsMapLayer()
self._buffer = 0.0
self._minWidth = 0.0
self._mapCanvas = mapCanvas
if (not layer
or not feature
or not isinstance(feature, QgsFeature)
or not feature.geometry()
or feature.geometry().isEmpty()
or not feature.geometry().isGeosValid()):
return
self._feature = QgsFeature(feature) # Force deep copy
self._layer = layer
self.setLineColor(Application.highlightLineColor())
self.setFillColor(Application.highlightFillColor())
self._minWidth = Application.highlightMinimumWidth()
self._buffer = Application.highlightBuffer()
if self._mapCanvas.mapSettings().hasCrsTransformEnabled():
ct = self._mapCanvas.mapSettings().layerTransform(self._layer)
if ct:
self._feature.geometry().transform(ct)
self.updateRect()
self.update()
def remove(self):
self._mapCanvas.scene().removeItem(self)
def setLineWidth(self, width):
self._pen.setWidth(width)
def setLineColor(self, color):
self._pen.setColor(color)
def setFillColor(self, fillColor):
self._brush.setColor(fillColor)
self._brush.setStyle(Qt.SolidPattern)
def setBuffer(self, buff):
self._buffer = buff
def setMinWidth(self, width):
self._minWidth = width
def layer(self):
return self._layer
def updatePosition(self):
pass
# protected:
def paint(self, painter, option=None, widget=None): # Override
if not self._feature:
return
mapSettings = self._mapCanvas.mapSettings()
context = QgsRenderContext.fromMapSettings(mapSettings)
renderer = self._getRenderer(context, self._pen.color(), self._brush.color())
if renderer:
context.setPainter(painter)
renderer.startRender(context, self._layer.fields())
renderer.renderFeature(self._feature, context)
renderer.stopRender(context)
def updateRect(self):
if self._feature and self._feature.constGeometry():
m2p = self._mapCanvas.mapSettings().mapToPixel()
topLeft = m2p.toMapPoint(0, 0)
res = m2p.mapUnitsPerPixel()
imageSize = self._mapCanvas.mapSettings().outputSize()
rect = QgsRectangle(topLeft.x(), topLeft.y(), topLeft.x() + imageSize.width()
* res, topLeft.y() - imageSize.height() * res)
self.setRect(rect)
self.setVisible(True)
else:
self.setRect(QgsRectangle())
# private:
def _setSymbol(self, symbol, context, color, fillColor):
if not symbol:
return
for symbolLayer in reversed(symbol.symbolLayers()):
if symbolLayer:
if symbolLayer.subSymbol():
self._setSymbol(symbolLayer.subSymbol(), context, color, fillColor)
else:
symbolLayer.setColor(color)
symbolLayer.setOutlineColor(color)
symbolLayer.setFillColor(fillColor)
if isinstance(symbolLayer, QgsSimpleMarkerSymbolLayerV2):
symbolLayer.setOutlineWidth(
self._getSymbolWidth(context, symbolLayer.outlineWidth(), symbolLayer.outlineWidthUnit()))
if symbolLayer.type() == QgsSymbolV2.Line:
symbolLayer.setWidth(
self._getSymbolWidth(context, symbolLayer.width(), symbolLayer.widthUnit()))
if symbolLayer.type() == QgsSymbolV2.Fill:
symbolLayer.setBorderWidth(
self._getSymbolWidth(context, symbolLayer.borderWidth(), symbolLayer.outputUnit()))
symbolLayer.removeDataDefinedProperty('color')
symbolLayer.removeDataDefinedProperty('color_border')
def _getSymbolWidth(self, context, width, unit):
scale = 1.0
if unit == QgsSymbolV2.MapUnit:
scale = QgsSymbolLayerV2Utils.lineWidthScaleFactor(
context, QgsSymbolV2.MM) / QgsSymbolLayerV2Utils.lineWidthScaleFactor(context, QgsSymbolV2.MapUnit)
width = max(width + 2 * self._buffer * scale, self._minWidth * scale)
return width
def _getRenderer(self, context, color, fillColor):
renderer = None
if self._layer and self._layer.rendererV2():
renderer = self._layer.rendererV2().clone()
if renderer:
for symbol in renderer.symbols2(context):
self._setSymbol(symbol, context, color, fillColor)
return renderer
