sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-10-19 00:07:15 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1574 from mhugent/getFeatureExpression

Browse Source 
Get feature expression
This commit is contained in:
Jürgen Fischer 2014-09-02 17:20:33 +02:00
commit c5bcbaf4d9
2 changed files with 54 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
resources/function_help/getFeature Normal file
View File

@ -0,0 +1,6 @@
<h3>getFeature function</h3>
Returns the first feature of a layer matching a given attribute value
<h4>Syntax</h4>
<pre>getFeature( layer, attributeField, value )</pre>

48
src/core/qgsexpression.cpp
View File

@ -30,6 +30,7 @@
#include "qgsfeature.h" #include "qgsfeature.h"
#include "qgsgeometry.h" #include "qgsgeometry.h"
#include "qgslogger.h" #include "qgslogger.h"
#include "qgsmaplayerregistry.h"
#include "qgsogcutils.h" #include "qgsogcutils.h"
#include "qgsvectorlayer.h" #include "qgsvectorlayer.h"
#include "qgssymbollayerv2utils.h" #include "qgssymbollayerv2utils.h"
@ -1505,6 +1506,52 @@ static QVariant fcnSpecialColumn( const QVariantList& values, const QgsFeature*
return QgsExpression::specialColumn( varName ); return QgsExpression::specialColumn( varName );
} }
static QVariant fcnGetFeature( const QVariantList& values, const QgsFeature* f, QgsExpression* parent )
{
//arguments: 1. layer id / name, 2. key attribute, 3. eq value
QString layerString = getStringValue( values.at( 0 ), parent );
QgsVectorLayer* vl = qobject_cast<QgsVectorLayer*>( QgsMapLayerRegistry::instance()->mapLayer( layerString ) ); //search by id first
if ( !vl )
{
QList<QgsMapLayer *> layersByName = QgsMapLayerRegistry::instance()->mapLayersByName( layerString );
if ( layersByName.size() > 0 )
{
vl = qobject_cast<QgsVectorLayer*>( layersByName.at( 0 ) );
}
}
//no layer found
if ( !vl )
{
return QVariant();
}
QString attribute = getStringValue( values.at( 1 ), parent );
int attributeId = vl->fieldNameIndex( attribute );
if ( attributeId == -1 )
{
return QVariant();
}
const QVariant& attVal = values.at( 2 );
QgsFeatureRequest req;
if ( !parent->needsGeometry() )
{
req.setFlags( QgsFeatureRequest::NoGeometry );
}
QgsFeatureIterator fIt = vl->getFeatures( req );
QgsFeature fet;
while ( fIt.nextFeature( fet ) )
{
if ( fet.attribute( attributeId ) == attVal )
{
return QVariant::fromValue( fet );
}
}
return QVariant();
}
bool QgsExpression::registerFunction( QgsExpression::Function* function ) bool QgsExpression::registerFunction( QgsExpression::Function* function )
{ {
int fnIdx = functionIndex( function->name() ); int fnIdx = functionIndex( function->name() );
@ -1683,6 +1730,7 @@ const QList<QgsExpression::Function*> &QgsExpression::Functions()
<< new StaticFunction( "$currentfeature", 0, fcnFeature, "Record" ) << new StaticFunction( "$currentfeature", 0, fcnFeature, "Record" )
<< new StaticFunction( "$scale", 0, fcnScale, "Record" ) << new StaticFunction( "$scale", 0, fcnScale, "Record" )
<< new StaticFunction( "$uuid", 0, fcnUuid, "Record" ) << new StaticFunction( "$uuid", 0, fcnUuid, "Record" )
<< new StaticFunction( "getFeature", 3, fcnGetFeature, "Record" )
//return all attributes string for referencedColumns - this is caught by //return all attributes string for referencedColumns - this is caught by
// QgsFeatureRequest::setSubsetOfAttributes and causes all attributes to be fetched by the // QgsFeatureRequest::setSubsetOfAttributes and causes all attributes to be fetched by the