mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
956c155e8f
pep8 --ignore=E111,E128,E201,E202,E203,E211,E221,E222,E225,E226,E227,E231,E241,E261,E265,E272,E302,E303,E501,E701 \
--exclude="ui_*.py,debian/*,python/ext-libs/*" \
.
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
Gridify.py
|
|
---------------------
|
|
Date : May 2010
|
|
Copyright : (C) 2010 by Michael Minn
|
|
Email : pyqgis at michaelminn dot com
|
|
***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
***************************************************************************
|
|
"""
|
|
|
|
__author__ = 'Michael Minn'
|
|
__date__ = 'May 2010'
|
|
__copyright__ = '(C) 2010, Michael Minn'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
from processing.core.GeoAlgorithm import GeoAlgorithm
|
|
from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
|
|
from processing.core.parameters import ParameterVector
|
|
from processing.core.outputs import OutputVector
|
|
|
|
from processing.tools import dataobjects, vector
|
|
|
|
class Merge(GeoAlgorithm):
|
|
LAYER1 = 'LAYER1'
|
|
LAYER2 = 'LAYER2'
|
|
OUTPUT = 'OUTPUT'
|
|
|
|
def defineCharacteristics(self):
|
|
self.name = 'Merge vector layers'
|
|
self.group = 'Vector general tools'
|
|
|
|
self.addParameter(ParameterVector(self.LAYER1,
|
|
self.tr('Input layer 1'), [ParameterVector.VECTOR_TYPE_ANY]))
|
|
self.addParameter(ParameterVector(self.LAYER2,
|
|
self.tr('Input layer 2'), [ParameterVector.VECTOR_TYPE_ANY]))
|
|
|
|
self.addOutput(OutputVector(self.OUTPUT, self.tr('Output')))
|
|
|
|
def processAlgorithm(self, progress):
|
|
layer1 = dataobjects.getObjectFromUri(
|
|
self.getParameterValue(self.LAYER1))
|
|
layer2 = dataobjects.getObjectFromUri(
|
|
self.getParameterValue(self.LAYER2))
|
|
|
|
if layer1.wkbType() != layer2.wkbType():
|
|
raise GeoAlgorithmExecutionException(
|
|
self.tr('Merged layers must have be same type of geometry'))
|
|
|
|
count = 0
|
|
fields = []
|
|
layers = [layer1, layer2]
|
|
for layer in layers:
|
|
count += layer.featureCount()
|
|
|
|
for sfield in layer.pendingFields():
|
|
found = None
|
|
for dfield in fields:
|
|
if dfield.name() == sfield.name() and \
|
|
dfield.type() == sfield.type():
|
|
found = dfield
|
|
break
|
|
|
|
if not found:
|
|
fields.append(sfield)
|
|
|
|
writer = self.getOutputFromName(self.OUTPUT).getVectorWriter(
|
|
fields, layer1.wkbType(), layer1.crs())
|
|
|
|
total = 100.0 / float(count)
|
|
count = 0
|
|
for layer in layers:
|
|
idx = {}
|
|
for dfield in fields:
|
|
i = 0
|
|
for sfield in layer.pendingFields():
|
|
if sfield.name() == dfield.name() and \
|
|
sfield.type() == dfield.type():
|
|
idx[dfield] = i
|
|
break
|
|
i += 1
|
|
|
|
features = vector.features(layer)
|
|
for f in features:
|
|
sAttributes = f.attributes()
|
|
dAttributes = []
|
|
for dfield in fields:
|
|
if dfield in idx:
|
|
dAttributes.append(sAttributes[idx[dfield]])
|
|
else:
|
|
dAttributes.append(dfield.type())
|
|
f.setAttributes(dAttributes)
|
|
writer.addFeature(f)
|
|
|
|
count += 1
|
|
progress.setPercentage(int(count * total))
|
|
|
|
del writer
|