Python SubElement.attrib['opacity']方法代码示例

# 需要导入模块: from xml.etree.ElementTree import SubElement [as 别名]
# 或者: from xml.etree.ElementTree.SubElement import attrib['opacity'] [as 别名]
    def sld2qml(self):
        try:
            sldDocument = minidom.parseString(self.dlg.sldText2.text().encode('utf-8'))

            # Namespaces (sld, se, none) are most important part. Firstly they must be handled.
            if len(sldDocument.getElementsByTagName('sld:RasterSymbolizer')) > 0:
                nameSpace = 'sld:'

            elif len(sldDocument.getElementsByTagName('se:RasterSymbolizer')) > 0:
                nameSpace = 'se:'

            else:
                nameSpace = ''  # no namespace.

            # main QML structure.
            qmlRoot = Element('qgis')
            pipe = SubElement(qmlRoot, 'pipe')

            rasterRenderer = SubElement(pipe, 'rasterrenderer')
            rasterTransparency = SubElement(rasterRenderer, 'rasterTransparency')

            # checking document for containing ColorMap tag. If does, it is a single-band raster.
            sldRasterType = sldDocument.getElementsByTagName(nameSpace + 'ColorMap')

            if len(sldRasterType) == 0:
                sldRedBand = sldDocument.getElementsByTagName(nameSpace + 'RedChannel')[0].getElementsByTagName(
                    nameSpace + 'SourceChannelName')[0].firstChild.nodeValue
                sldGreenBand = sldDocument.getElementsByTagName(nameSpace + 'GreenChannel')[0].getElementsByTagName(
                    nameSpace + 'SourceChannelName')[0].firstChild.nodeValue
                sldBlueBand = sldDocument.getElementsByTagName(nameSpace + 'BlueChannel')[0].getElementsByTagName(
                    nameSpace + 'SourceChannelName')[0].firstChild.nodeValue

                rasterRenderer.attrib['type'] = 'multibandcolor'
                rasterRenderer.attrib['redBand'] = sldRedBand
                rasterRenderer.attrib['greenBand'] = sldGreenBand
                rasterRenderer.attrib['blueBand'] = sldBlueBand

            else:
                rasterRenderer.attrib['type'] = 'singlebandpseudocolor'
                rasterRenderer.attrib['band'] = '1'
                rasterShader = SubElement(rasterRenderer, 'rastershader')

                colorRampShader = SubElement(rasterShader, 'colorrampshader')

                sldColorMapType = None
                sldColorMapType2 = None
                try:
                    sldColorMapType = sldDocument.getElementsByTagName(nameSpace + 'ColorMap')[0].attributes.has_key(
                        'type')  # sometimes "ColorMap" tag does not contain "type" attribute. This means it is a ramp color.
                    sldColorMapType2 = sldDocument.getElementsByTagName(nameSpace + 'ColorMap')[0].attributes[
                        'type'].value  # or getting raster map colortype by "type" atribute.
                except:
                    pass

                if sldColorMapType and sldColorMapType2 == 'intervals':
                    colorRampShader.attrib['colorRampType'] = 'DISCRETE'
                else:
                    colorRampShader.attrib['colorRampType'] = 'INTERPOLATED'

                sldColorValues = list()
                sldItemList = sldDocument.getElementsByTagName(nameSpace + 'ColorMapEntry')

                # some SLD documents don't have 'label' attribute so the problem is handled by this way.
                try:
                    for m in sldItemList:
                        sldColorValues.append(
                            [m.attributes['quantity'].value, m.attributes['label'].value, m.attributes['color'].value])
                except:
                    for m in sldItemList:
                        sldColorValues.append([m.attributes['quantity'].value, m.attributes['quantity'].value,
                                               m.attributes['color'].value])

                for s in range(len(sldColorValues)):
                    item = SubElement(colorRampShader, 'item')
                    item.attrib['alpha'] = '255'
                    item.attrib['value'] = sldColorValues[s][0]
                    item.attrib['label'] = sldColorValues[s][1]
                    item.attrib['color'] = sldColorValues[s][2]

            # some SLD documents don't have 'Opacity' tag so the problem is handled by this way.
            try:
                sldOpacity = sldDocument.getElementsByTagName(nameSpace + 'Opacity')[0].firstChild.nodeValue
            except:
                sldOpacity = '1'

            rasterRenderer.attrib['opacity'] = sldOpacity

            brightnessContrast = SubElement(pipe, 'brightnesscontrast')
            hueSaturation = SubElement(pipe, 'huesaturation')
            rasterResampler = SubElement(pipe, 'rasterresampler')
            blendMode = SubElement(qmlRoot, 'blendMode')
            blendMode.text = '0'
            textQML = minidom.parseString(tostring(qmlRoot)).toprettyxml(indent="    ")
            self.saveFile(textQML, 'QML')

        except Exception as sldTransformError:
            QMessageBox.critical(None, "Invalid XML", str(sldTransformError))

# 需要导入模块: from xml.etree.ElementTree import SubElement [as 别名]
# 或者: from xml.etree.ElementTree.SubElement import attrib['opacity'] [as 别名]
    def qml2Sld(self):
        try:
            layer = self.allMapLayers[self.dlg.comboBox.currentIndex()]
        except:
            return

        qmlDocument = QDomDocument()
        root = qmlDocument.createElement('SLD4raster')
        qmlDocument.appendChild(root)
        qgisNode = qmlDocument.createElement('qgis')
        root.appendChild(qgisNode)

        errorMessage = None
        context = QgsReadWriteContext()
        layer.writeSymbology(qgisNode, qmlDocument, errorMessage, context)

        qmlString = minidom.parseString(qmlDocument.toString().encode('utf-8'))  # for non ASCII labels.

        sldRoot = Element('sld:StyledLayerDescriptor')
        sldRoot.attrib['xmlns'] = 'http://www.opengis.net/sld'
        sldRoot.attrib['xmlns:sld'] = 'http://www.opengis.net/sld'
        sldRoot.attrib['xmlns:ogc'] = 'http://www.opengis.net/ogc'
        sldRoot.attrib['xmlns:gml'] = 'http://www.opengis.net/gml'
        sldRoot.attrib['version'] = '1.0.0'

        userLayer = SubElement(sldRoot, 'sld:UserLayer')
        layerFeatureConstraints = SubElement(userLayer, 'sld:LayerFeatureConstraints')
        featureTypeConstraint = SubElement(layerFeatureConstraints, 'sld:FeatureTypeConstraint')
        userStyle = SubElement(userLayer, 'sld:UserStyle')

        styleName = SubElement(userStyle, 'sld:Name')
        styleName.text = self.dlg.comboBox.currentText()

        styleDescription = SubElement(userStyle, 'sld:Description')
        styleDescription.text = 'Generated by SLD4raster - https://cbsuygulama.wordpress.com/sld4raster'

        styleTitle = SubElement(userStyle, 'sld:Title')
        featureTypeStyle = SubElement(userStyle, 'sld:FeatureTypeStyle')
        featureName = SubElement(featureTypeStyle, 'sld:Name')
        featureRule = SubElement(featureTypeStyle, 'sld:Rule')
        rasterSymbolizer = SubElement(featureRule, 'sld:RasterSymbolizer')
        geometry = SubElement(rasterSymbolizer, 'sld:Geometry')
        ogcPropertyName = SubElement(geometry, 'ogc:PropertyName')
        ogcPropertyName.text = 'grid'
        opacity = SubElement(rasterSymbolizer, 'sld:Opacity')

        ###Getting raster type parameters
        rasterType = str(qmlString.getElementsByTagName('rasterrenderer')[0].attributes['type'].value)
        isGradient = 'gradient' in qmlString.getElementsByTagName('rasterrenderer')[0].attributes.keys()

        ###SLD for multiband raster
        if rasterType == 'multibandcolor':
            ###Getting RGB band order.
            redBand = str(qmlString.getElementsByTagName('rasterrenderer')[0].attributes['redBand'].value)
            greenBand = str(qmlString.getElementsByTagName('rasterrenderer')[0].attributes['greenBand'].value)
            blueBand = str(qmlString.getElementsByTagName('rasterrenderer')[0].attributes['blueBand'].value)

            channelSelection = SubElement(rasterSymbolizer, 'sld:ChannelSelection')

            redChannel = SubElement(channelSelection, 'sld:RedChannel')
            redSourceChannel = SubElement(redChannel, 'sld:SourceChannelName')
            redSourceChannel.text = redBand

            greenChannel = SubElement(channelSelection, 'sld:GreenChannel')
            greenSourceChannel = SubElement(greenChannel, 'sld:SourceChannelName')
            greenSourceChannel.text = greenBand

            blueChannel = SubElement(channelSelection, 'sld:BlueChannel')
            blueSourceChannel = SubElement(blueChannel, 'sld:SourceChannelName')
            blueSourceChannel.text = blueBand

        ###SLD for gradiented (black to white) raster
        elif isGradient:
            blackWhiteColor = ['#000000', '#FFFFFF']
            colorMap = SubElement(rasterSymbolizer, 'sld:ColorMap')
            gradientType = qmlString.getElementsByTagName('rasterrenderer')[0].attributes['gradient'].value
            blackWhiteValue = [qmlString.getElementsByTagName('minValue')[0].firstChild.nodeValue,
                               qmlString.getElementsByTagName('maxValue')[0].firstChild.nodeValue]

            ###Getting gradient color type
            if gradientType == 'WhiteToBlack':
                blackWhiteColor.reverse()

            for i in range(len(blackWhiteColor)):
                colorMapEntry = SubElement(colorMap, 'sld:ColorMapEntry')
                colorMapEntry.attrib['color'] = blackWhiteColor[i]
                colorMapEntry.attrib['opacity'] = '1.0'
                colorMapEntry.attrib['quantity'] = blackWhiteValue[i]

        ###SLD for singleband raster
        else:
            colorMap = SubElement(rasterSymbolizer, 'sld:ColorMap')

            ###Getting color ramp type
            colorType = str(qmlString.getElementsByTagName('colorrampshader')[0].attributes['colorRampType'].value)
            if colorType == 'DISCRETE':
                colorMap.attrib['type'] = "intervals"

            ###Getting color values
            colorValue = list()
#.........这里部分代码省略.........