Implement sinkProperties for QgsProcessingFeatureBasedAlgorithm subclasses

This commit is contained in:
Nyall Dawson 2020-04-03 14:25:31 +10:00
parent 2d8da255a2
commit b2ca789ba4
3 changed files with 41 additions and 0 deletions

View File

@ -1191,6 +1191,11 @@ Read the source from ``parameters`` and ``context`` and set it
.. versionadded:: 3.4
%End
virtual QgsProcessingAlgorithm::VectorProperties sinkProperties( const QString &sink,
const QVariantMap &parameters,
QgsProcessingContext &context,
const QMap< QString, QgsProcessingAlgorithm::VectorProperties > &sourceProperties ) const;
};

View File

@ -1031,3 +1031,34 @@ void QgsProcessingFeatureBasedAlgorithm::prepareSource( const QVariantMap &param
}
}
QgsProcessingAlgorithm::VectorProperties QgsProcessingFeatureBasedAlgorithm::sinkProperties( const QString &sink, const QVariantMap &parameters, QgsProcessingContext &context, const QMap<QString, QgsProcessingAlgorithm::VectorProperties> &sourceProperties ) const
{
QgsProcessingAlgorithm::VectorProperties result;
if ( sink == QStringLiteral( "OUTPUT" ) )
{
if ( sourceProperties.value( QStringLiteral( "INPUT" ) ).availability == QgsProcessingAlgorithm::Available )
{
const VectorProperties inputProps = sourceProperties.value( QStringLiteral( "INPUT" ) );
result.fields = outputFields( inputProps.fields );
result.crs = outputCrs( inputProps.crs );
result.wkbType = outputWkbType( inputProps.wkbType );
result.availability = Available;
return result;
}
else
{
std::unique_ptr< QgsProcessingFeatureSource > source( parameterAsSource( parameters, QStringLiteral( "INPUT" ), context ) );
if ( source )
{
result.fields = outputFields( source->fields() );
result.crs = outputCrs( source->sourceCrs() );
result.wkbType = outputWkbType( source->wkbType() );
result.availability = Available;
return result;
}
}
}
return result;
}

View File

@ -1199,6 +1199,11 @@ class CORE_EXPORT QgsProcessingFeatureBasedAlgorithm : public QgsProcessingAlgor
*/
void prepareSource( const QVariantMap &parameters, QgsProcessingContext &context );
QgsProcessingAlgorithm::VectorProperties sinkProperties( const QString &sink,
const QVariantMap &parameters,
QgsProcessingContext &context,
const QMap< QString, QgsProcessingAlgorithm::VectorProperties > &sourceProperties ) const override;
private:
std::unique_ptr< QgsProcessingFeatureSource > mSource;