def subRenderer(self, color):
lineSymbol = QgsLineSymbolV2()
arrowSymbol = QgsArrowSymbolLayer().create()
arrowSymbol.setArrowWidth(0.0)
arrowSymbol.setArrowStartWidth(0.0)
arrowSymbol.setColor(color)
arrowSymbol.setIsCurved(False)
arrowSymbol.setIsRepeated(False)
renderSetting = QgsRenderContext()
renderSetting.setFlag(QgsRenderContext.Antialiasing, False)
renderSetting.setFlag(QgsRenderContext.DrawSymbolBounds, False)
renderSetting.setFlag(QgsRenderContext.ForceVectorOutput, False)
renderSetting.setFlag(QgsRenderContext.UseAdvancedEffects, False)
renderSetting.setFlag(QgsRenderContext.UseRenderingOptimization, False)
arrEffectStack = arrowSymbol.paintEffect()
effList = arrEffectStack.effectList()
for i in range(0, len(effList)):
if type(effList[i]) == QgsDrawSourceEffect:
effList[i].setDrawMode(1)
arrEffectStack.setEnabled(True)
lineSymbol.appendSymbolLayer(arrowSymbol)
return lineSymbol
def changeSymbol(self):
"""Change symbology from seleted layer"""
if isinstance(self.lyr.renderer(),QgsSingleSymbolRenderer) == True:
symbLyr = QgsArrowSymbolLayer()
symbLyr.setIsCurved(False)
symbLyr.setIsRepeated(True)
self.lyr.renderer().symbol().changeSymbolLayer(0, symbLyr)
self.lyr.triggerRepaint()
self.iface.layerTreeView().refreshLayerSymbology(self.lyr.id())
else:
QMessageBox.warning(None,"Can't change symbology","The layer must be single symbol")
def test_subsymbol(self):
"""
Test rendering a generator in a subsymbol of another symbol
"""
sym = QgsLineSymbol()
arrow = QgsArrowSymbolLayer()
arrow.setIsRepeated(False)
arrow.setArrowStartWidth(10)
arrow.setArrowWidth(5)
arrow.setHeadLength(20)
arrow.setHeadThickness(10)
sym.changeSymbolLayer(0, arrow)
self.lines_layer.renderer().setSymbol(sym)
# here "$geometry" must refer to the created ARROW shape, NOT the original feature line geometry!
generator_layer = QgsGeometryGeneratorSymbolLayer.create(
{'geometryModifier': 'buffer($geometry, 3)'})
generator_layer.setSymbolType(QgsSymbol.Fill)
generator_layer.setUnits(QgsUnitTypes.RenderMillimeters)
self.assertIsNotNone(generator_layer.subSymbol())
generator_layer.subSymbol().symbolLayer(0).setColor(
QColor(255, 255, 255))
generator_layer.subSymbol().symbolLayer(0).setStrokeColor(
QColor(0, 0, 0))
generator_layer.subSymbol().symbolLayer(0).setStrokeWidth(2)
sub_symbol = QgsFillSymbol()
sub_symbol.changeSymbolLayer(0, generator_layer)
arrow.setSubSymbol(sub_symbol)
rendered_layers = [self.lines_layer]
self.mapsettings.setLayers(rendered_layers)
renderchecker = QgsMultiRenderChecker()
renderchecker.setMapSettings(self.mapsettings)
renderchecker.setControlName('expected_geometrygenerator_subsymbol')
res = renderchecker.runTest('geometrygenerator_subsymbol')
self.report += renderchecker.report()
self.assertTrue(res)
