def testQgsSimpleLineSymbolLayerV2(self):
'''
Create a new style from a .sld file and match test
'''
mTestName = QString('QgsSimpleLineSymbolLayerV2')
mFilePath = QDir.toNativeSeparators(
QString('%1/symbol_layer/%2.sld').arg(unitTestDataPath(),
mTestName))
mDoc = QDomDocument(mTestName)
mFile = QFile(mFilePath)
mFile.open(QIODevice.ReadOnly)
mDoc.setContent(mFile, True)
mFile.close()
mSymbolLayer = QgsSimpleLineSymbolLayerV2.createFromSld(
mDoc.elementsByTagName('LineSymbolizer').item(0).toElement())
mExpectedValue = type(QgsSimpleLineSymbolLayerV2())
mValue = type(mSymbolLayer)
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = QString(u'#aa007f')
mValue = mSymbolLayer.color().name()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = 1.26
mValue = mSymbolLayer.width()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = Qt.RoundCap
mValue = mSymbolLayer.penCapStyle()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = Qt.MiterJoin
mValue = mSymbolLayer.penJoinStyle()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = True
mValue = mSymbolLayer.useCustomDashPattern()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = [5.0, 2.0]
mValue = mSymbolLayer.customDashVector()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue, mValue)
assert mExpectedValue == mValue, mMessage
def make_svg_link_sym_renderer(self, icon_name, svg_size=7, line_width=0.2):
symbol = QgsLineSymbolV2().createSimple({})
symbol.deleteSymbolLayer(0)
# Line
line_sym_lay = QgsSimpleLineSymbolLayerV2()
line_sym_lay.setWidth(line_width)
symbol.appendSymbolLayer(line_sym_lay)
# Symbol
self.marker_sym = QgsMarkerSymbolV2.createSimple({})
self.marker_sym.deleteSymbolLayer(0)
# marker_sym_lay = QgsSimpleMarkerSymbolLayerV2()
current_dir = os.path.dirname(__file__)
svg_props = dict()
svg_props['name'] = os.path.join(current_dir, icon_name)
svg_props['size'] = str(svg_size)
marker_sym_lay = QgsSvgMarkerSymbolLayerV2().create(svg_props)
self.marker_sym.appendSymbolLayer(marker_sym_lay)
marker_line_sym_lay = QgsMarkerLineSymbolLayerV2()
marker_line_sym_lay.setSubSymbol(self.marker_sym) # Causes crash !!!
marker_line_sym_lay.setPlacement(QgsMarkerLineSymbolLayerV2.CentralPoint)
symbol.appendSymbolLayer(marker_line_sym_lay)
renderer = QgsSingleSymbolRendererV2(symbol)
return renderer
def make_simple_link_sym_renderer(self, width=0.2):
symbol = QgsLineSymbolV2().createSimple({})
symbol.deleteSymbolLayer(0)
# Line
line_sym_lay = QgsSimpleLineSymbolLayerV2()
line_sym_lay.setWidth(width)
symbol.appendSymbolLayer(line_sym_lay)
renderer = QgsSingleSymbolRendererV2(symbol)
return renderer
def testQgsSimpleLineSymbolLayerV2(self):
'''
Create a new style from a .sld file and match test
'''
mTestName = QString ('QgsSimpleLineSymbolLayerV2')
mFilePath = QDir.toNativeSeparators(
QString('%1/symbol_layer/%2.sld').arg(
unitTestDataPath(), mTestName))
mDoc = QDomDocument(mTestName)
mFile = QFile (mFilePath)
mFile.open(QIODevice.ReadOnly)
mDoc.setContent(mFile,True)
mFile.close()
mSymbolLayer = QgsSimpleLineSymbolLayerV2.createFromSld(
mDoc.elementsByTagName('LineSymbolizer').item(0).toElement())
mExpectedValue = type(QgsSimpleLineSymbolLayerV2())
mValue = type(mSymbolLayer)
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = QString(u'#aa007f')
mValue = mSymbolLayer.color().name()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = 1.26
mValue = mSymbolLayer.width()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = Qt.RoundCap
mValue = mSymbolLayer.penCapStyle()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = Qt.MiterJoin
mValue = mSymbolLayer.penJoinStyle()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = True
mValue = mSymbolLayer.useCustomDashPattern()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
mExpectedValue = [5.0, 2.0]
mValue = mSymbolLayer.customDashVector()
mMessage = 'Expected "%s" got "%s"' % (mExpectedValue,mValue)
assert mExpectedValue == mValue, mMessage
def symbolize(layer, oldlayer):
renderer = None
# See if we can figure out ho thick the line is to begin with:
# create a new single symbol renderer
symbol = QgsSymbolV2.defaultSymbol(layer.geometryType())
renderer = QgsSingleSymbolRendererV2(symbol)
# symbol_layer = QgsSimpleLineSymbolLayerV2.create(properties)
symbol_layer = QgsSimpleLineSymbolLayerV2(random_color(), 2.0)
# assign the symbol layer to the symbol
# TODO: on windows::: 'NoneType' object has no attribute 'appendSymbolLayer'
renderer.symbol().appendSymbolLayer(symbol_layer)
# assign the renderer to the layer
layer.setRendererV2(renderer)
def __getQgsLineSymbolLayer(self, msstyle):
"""docstring for __getLineSymbolLayer"""
props = {}
type_marker = ''
symbol = msstyle.get('symbol', False)
if symbol:
(type_marker, symbolname, msSymbol,
props) = self.__getMsSymbol(symbol, props)
opacity = self.__getMsOpacity(msstyle)
self.__getMsColor(msstyle, props, opacity, isline=True)
self.__getMsOutlinecolor(msstyle, props, opacity, isline=True)
self.__getMsLinecap(msstyle, props)
self.__getMsLinejoin(msstyle, props)
self.__getMsPattern(msstyle, props)
self.__getMsWidth(msstyle, props)
self.__getMsOutline(msstyle, props)
self.__getMsScale(msstyle, props)
offset = self.__getMsOffset(msstyle, props)
if not type_marker:
self.deleteProperties(props, _qgis.SIMPLE_LINE_SYMBOL_LAYER)
qgsSymbol = QgsSimpleLineSymbolLayerV2.create(props)
else:
props_parent = {}
self.__setMsOffsetXY(offset, props)
self.__getMsAngle(msstyle, props, props_parent)
gap = self.__getMsGap(msstyle)
if type_marker == _ms.MS_SYMBOL_ARROW:
#El estilo de relleno de la flecha, no soportado por mapserver
#[solid|horizontal|vertical|cross|b_diagonal|f_diagonal|diagonal_x|dense1|dense2|dense3|dense4|dense5|dense6|dense7|no]
if QGis.QGIS_VERSION_INT >= 21600:
props['style'] = 'solid'
props_parent['head_type'] = _qgis.MARKERS_ARROW[
symbolname][0]
props_parent['arrow_type'] = _qgis.MARKERS_ARROW[
symbolname][1]
props_parent['is_repeated'] = '1' if gap else '0'
self.deleteProperties(props,
_qgis.SIMPLE_FILL_SYMBOL_LAYER)
qgsSubSymbol = self.__getSubSymbol(
QgsFillSymbolV2, QgsSimpleFillSymbolLayerV2, QGis.Fill,
props)
self.deleteProperties(props_parent,
_qgis.ARROW_SYMBOL_LAYER)
qgsSymbol = QgsArrowSymbolLayer.create(props_parent)
else:
self.deleteProperties(props,
_qgis.SIMPLE_MARKER_SYMBOL_LAYER)
qgsSubSymbol = self.__getSubSymbol(
QgsMarkerSymbolV2, QgsSimpleMarkerSymbolLayerV2,
QGis.Point, props)
self.deleteProperties(props_parent,
_qgis.MARKER_LINE_SYMBOL_LAYER)
qgsSymbol = QgsMarkerLineSymbolLayerV2.create(props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
else:
size = self.__getMsSize(msstyle, props)
self.__getMarkerDisplacementAndRotate(msstyle, gap, size,
props_parent)
self.__getMsInitialGap(msstyle, gap, props_parent)
self.__getMsAnchorpoint(msSymbol, props)
qgsSubSymbol = self.__getQgsMarkerSubSymbol(
type_marker, msSymbol, size, props)
self.deleteProperties(props_parent,
_qgis.MARKER_LINE_SYMBOL_LAYER)
qgsSymbol = QgsMarkerLineSymbolLayerV2.create(props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
#print(props)
return qgsSymbol
def __getQgsPolygonSymbolLayer(self, msstyle):
props = {}
type_marker = ''
symbol = msstyle.get('symbol', False)
if symbol:
(type_marker, symbolname, msSymbol,
props) = self.__getMsSymbol(symbol, props, True)
opacity = self.__getMsOpacity(msstyle)
color = self.__getMsColor(msstyle, props, opacity)
self.__getMsOutlinecolor(msstyle, props, opacity)
self.__getMsLinecap(msstyle, props)
self.__getMsLinejoin(msstyle, props)
self.__getMsPattern(msstyle, props)
self.__getMsWidth(msstyle, props)
self.__getMsOutline(msstyle, props)
self.__getMsScale(msstyle, props)
offset = self.__getMsOffset(msstyle, props)
if not type_marker:
#-linea sencilla exterior-#
#-relleno simple (sin estilo de relleno preconfigurados)-#
if not color:
self.deleteProperties(props, _qgis.SIMPLE_LINE_SYMBOL_LAYER)
qgsSymbol = QgsSimpleLineSymbolLayerV2.create(props)
else:
self.deleteProperties(props, _qgis.SIMPLE_FILL_SYMBOL_LAYER)
qgsSymbol = QgsSimpleFillSymbolLayerV2.create(props)
#-relleno de gradientes-#
#el gradiente es para renderer por categorias continuas
#no aplica, porque aca es para gradiente de un simbolo
else:
props_parent = {}
self.__setMsOffsetXY(offset, props)
self.__getMsAngle(msstyle, props, props_parent)
gap = self.__getMsGap(msstyle)
size = self.__getMsSize(msstyle, props)
self.__getMarkerDisplacementAndRotate(msstyle, gap, size,
props_parent)
geomtransform = msstyle.get('geomtransform', '').lower()
if geomtransform == 'centroid':
#-relleno de centroides-#
self.__getMsAnchorpoint(msSymbol, props)
qgsSubSymbol = self.__getQgsMarkerSubSymbol(
type_marker, msSymbol, size, props)
#self.deleteProperties(props_parent, _qgis.CENTROID_FILL_SYMBOL_LAYER)
qgsSymbol = QgsCentroidFillSymbolLayerV2.create({})
qgsSymbol.setSubSymbol(qgsSubSymbol)
elif not color:
#-Linea exterior:linea de marcador-#
self.__getMsAnchorpoint(msSymbol, props)
qgsSubSymbol = self.__getQgsMarkerSubSymbol(
type_marker, msSymbol, size, props)
self.deleteProperties(props_parent,
_qgis.MARKER_LINE_SYMBOL_LAYER)
qgsSymbol = QgsMarkerLineSymbolLayerV2.create(props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
elif type_marker == _ms.MS_SYMBOL_HATCH:
if not props['use_custom_dash']:
#-relleno simple (con estilo de relleno)-#
self.deleteProperties(props,
_qgis.SIMPLE_LINE_SYMBOL_LAYER)
qgsSymbol = QgsSimpleLineSymbolLayerV2.create(props)
else:
#-Patron de relleno de linea-#
self.deleteProperties(props,
_qgis.SIMPLE_LINE_SYMBOL_LAYER)
qgsSubSymbol = self.__getSubSymbol(
QgsLineSymbolV2, QgsSimpleLineSymbolLayerV2, QGis.Line,
props)
self.deleteProperties(props_parent,
_qgis.LINE_PATTERN_FILL_SYMBOL_LAYER)
qgsSymbol = QgsLinePatternFillSymbolLayer.create(
props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
elif type_marker == _ms.MS_SYMBOL_PIXMAP:
#-relleno de imagen raster-#
self.deleteProperties(props, _qgis.RASTER_FILL_SYMBOL_LAYER)
qgsSymbol = QgsRasterFillSymbolLayer.create(props)
elif type_marker == _ms.MS_SYMBOL_SVG:
#-relleno SVG-#
self.deleteProperties(props, _qgis.SIMPLE_LINE_SYMBOL_LAYER)
qgsSubSymbol = self.__getSubSymbol(QgsLineSymbolV2,
QgsSimpleLineSymbolLayerV2,
QGis.Line, props)
self.deleteProperties(props_parent,
_qgis.SVG_FILL_SYMBOL_LAYER)
qgsSymbol = QgsSVGFillSymbolLayer.create(props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
else:
#-patron de relleno de puntos-#
self.__getMsAnchorpoint(msSymbol, props)
qgsSubSymbol = self.__getQgsMarkerSubSymbol(
type_marker, msSymbol, size, props)
self.deleteProperties(props_parent,
_qgis.POINT_PATTERN_FILL_SYMBOL_LAYER)
qgsSymbol = QgsPointPatternFillSymbolLayer.create(props_parent)
qgsSymbol.setSubSymbol(qgsSubSymbol)
#qgsSymbol.setAlpha((opacity*1.0)/100)
#print(props)
return qgsSymbol
def make_flow_sym_renderer(self, layer, field_name, ranges_colors=None):
feats = layer.getFeatures()
min_val = sys.float_info.max
max_val = -min_val
# Find values range
for feat in feats:
attr = feat.attribute(field_name)
val = float(attr)
if val < min_val:
min_val = val
if val > max_val:
max_val = val
# Define colors
if ranges_colors is None:
ranges_colors = []
colors = [
QColor(0, 255, 0),
QColor(128, 255, 0),
QColor(255, 255, 0),
QColor(255, 128, 0),
QColor(255, 0, 0)]
intv_nr = len(colors)
intv = (max_val - min_val) / intv_nr
for c in range(intv_nr):
vrange = [min_val + intv * c, min_val + intv * (c + 1)]
if c == len(colors) - 1:
vrange[1] = max_val
ranges_colors.append([vrange, colors[c]])
range_list = []
for range_col in ranges_colors:
r_min = range_col[0][0]
r_max = range_col[0][1]
title = str(r_min) + ' - ' + str(r_max)
range_list.append(symbology_from_range(layer, r_min, r_max, range_col[1], title))
renderer = QgsGraduatedSymbolRendererV2(field_name, range_list)
# Line
symbol = QgsLineSymbolV2().createSimple({})
symbol.deleteSymbolLayer(0)
line_sym_lay = QgsSimpleLineSymbolLayerV2()
line_sym_lay.setWidth(0.2)
symbol.appendSymbolLayer(line_sym_lay)
# Define arrows for flow and velocities
if u'Link flow' in layer.name() in layer.name():
self.marker_sym = QgsMarkerSymbolV2.createSimple({'name': 'triangle', 'color': 'black'})
data_def_angle = QgsDataDefined()
data_def_angle_exp =\
'case ' \
'when "' + field_name + '" >= 0 ' \
'then degrees(azimuth( start_point( $geometry), end_point($geometry))) ' \
'else ' \
'case ' \
'when degrees(azimuth( start_point( $geometry), end_point($geometry))) < 180 ' \
'then degrees(azimuth( start_point( $geometry), end_point($geometry))) + 180 ' \
'else ' \
'degrees(azimuth( start_point( $geometry), end_point($geometry))) - 180 ' \
'end ' \
'end'
data_def_angle.setExpressionString(data_def_angle_exp)
data_def_angle.setActive(True)
self.marker_sym.setDataDefinedAngle(data_def_angle)
# Size: 0 if attribute = 0
data_def_size = QgsDataDefined()
data_def_size_exp =\
'case ' \
'when "' + field_name + '" = 0 ' \
'then 0 ' \
'else ' \
'2 ' \
'end'
data_def_size.setExpressionString(data_def_size_exp)
data_def_size.setActive(True)
self.marker_sym.setDataDefinedSize(data_def_size)
marker_sym_lay = QgsMarkerLineSymbolLayerV2()
marker_sym_lay.setColor(QColor(0, 0, 0))
marker_sym_lay.setFillColor(QColor(0, 0, 0))
marker_sym_lay.setPlacement(QgsMarkerLineSymbolLayerV2.CentralPoint)
marker_sym_lay.setRotateMarker(False)
marker_sym_lay.setSubSymbol(self.marker_sym)
self.marker_sym.appendSymbolLayer(marker_sym_lay)
symbol.appendSymbolLayer(marker_sym_lay)
# renderer.setSourceSymbol(symbol)
renderer.updateSymbols(symbol)
return renderer