added missing file with JoinAttributes algorithm

volaya · volaya · commit a4a8b4cc76bc · 2012-12-24T13:00:31.000+01:00
diff --git a/python/plugins/sextante/algs/JoinAttributes.py b/python/plugins/sextante/algs/JoinAttributes.py
@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    JoinAttributes.py
+    ---------------------
+    Date                 : August 2012
+    Copyright            : (C) 2012 by Victor Olaya
+    Email                : volayaf at gmail 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.                                   *
+*                                                                         *
+***************************************************************************
+"""
+from sextante.core.SextanteVectorWriter import SextanteVectorWriter
+from sextante.parameters.ParameterTableField import ParameterTableField
+
+__author__ = 'Victor Olaya'
+__date__ = 'August 2012'
+__copyright__ = '(C) 2012, Victor Olaya'
+# This will get replaced with a git SHA1 when you do a git archive
+__revision__ = '$Format:%H$'
+
+from sextante.core.GeoAlgorithm import GeoAlgorithm
+from sextante.outputs.OutputVector import OutputVector
+from sextante.parameters.ParameterVector import ParameterVector
+from qgis.core import *
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+from sextante.core.QGisLayers import QGisLayers
+
+
+class JoinAttributes(GeoAlgorithm):
+    
+    OUTPUT_LAYER = "OUTPUT_LAYER"
+    INPUT_LAYER = "INPUT_LAYER"
+    INPUT_LAYER_2 = "INPUT_LAYER_2"
+    TABLE_FIELD = "TABLE_FIELD"
+    TABLE_FIELD_2 = "TABLE_FIELD_2"
+
+    def defineCharacteristics(self):
+        self.name = "Create new layer with selected features"
+        self.group = "Vector general tools"
+        self.addParameter(ParameterVector(self.INPUT_LAYER, "Input layer", ParameterVector.VECTOR_TYPE_ANY, False))
+        self.addParameter(ParameterVector(self.INPUT_LAYER_2, "Input layer 2", ParameterVector.VECTOR_TYPE_ANY, False))
+        self.addParameter(ParameterTableField(self.TABLE_FIELD, "Table field", self.INPUT_LAYER))
+        self.addParameter(ParameterTableField(self.TABLE_FIELD_2, "Table field 2", self.INPUT_LAYER_2))
+        self.addOutput(OutputVector(self.OUTPUT_LAYER, "Output layer"))
+
+    def processAlgorithm(self, progress):
+        input = self.getParameterValue(self.INPUT_LAYER)
+        input2 = self.getParameterValue(self.INPUT_LAYER_2)
+        output = self.getOutputFromName(self.OUTPUT_LAYER)
+        field = self.getParameterValue(self.TABLE_FIELD)
+        field2 = self.getParameterValue(self.TABLE_FIELD_2)                
+        
+        # Layer 1
+        layer = QGisLayers.getObjectFromUri(input)
+        provider = layer.dataProvider()
+        allAttrs = provider.attributeIndexes()
+        provider.select( allAttrs )
+        join_field1_index = provider.fieldNameIndex(field)
+        # Layer 2
+        layer2 = QGisLayers.getObjectFromUri(input2)
+        provider2 = layer2.dataProvider()
+        allAttrs = provider2.attributeIndexes()
+        provider2.select( allAttrs )
+        fields2 = provider2.fields()
+        join_field2_index = provider2.fieldNameIndex(field2)
+        
+        # Output
+        outFields = provider.fields()
+        for (i,f) in fields2.iteritems():
+            f.setName("x_"+f.name())
+            outFields[len(outFields)] = f
+        
+        writer = output.getVectorWriter(outFields, provider.geometryType(), provider.crs())
+        
+        inFeat = QgsFeature()
+        inFeat2 = QgsFeature()
+        outFeat = QgsFeature()
+        
+        # Create output vector layer with additional attribute
+        while provider.nextFeature(inFeat):
+            inGeom = inFeat.geometry()
+            atMap = inFeat.attributeMap()
+            join_value1 = atMap[join_field1_index].toString()        
+            while provider2.nextFeature(inFeat2):                
+                atMap2 = inFeat2.attributeMap()
+                join_value2 = atMap2[join_field2_index].toString()        
+                if join_value1 == join_value2:
+                    # create the new feature
+                    outFeat.setGeometry( inGeom )
+                    outFeat.setAttributeMap( atMap )
+                    l = len(provider.fields())
+                    for (i,a) in atMap2.iteritems():
+                        outFeat.addAttribute( l+i, a )
+        
+                    writer.addFeature( outFeat )
+        
+        del writer
