mirror of
https://github.com/qgis/QGIS.git
synced 2025-10-31 00:06:02 -04:00
97 lines
4.1 KiB
Python
97 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
ParameterGuiUtils.py
|
|
---------------------
|
|
Date : June 2017
|
|
Copyright : (C) 2017 by Nyall Dawson
|
|
Email : nyall dot dawson 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. *
|
|
* *
|
|
***************************************************************************
|
|
"""
|
|
|
|
|
|
__author__ = 'Nyall Dawson'
|
|
__date__ = 'June 2017'
|
|
__copyright__ = '(C) 2017, Nyall Dawson'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
from qgis.core import (QgsProcessing,
|
|
QgsProviderRegistry,
|
|
QgsProcessingFeatureSourceDefinition,
|
|
QgsVectorFileWriter,
|
|
QgsRasterFileWriter)
|
|
from qgis.PyQt.QtCore import QCoreApplication
|
|
from processing.tools import dataobjects
|
|
|
|
|
|
def tr(string, context=''):
|
|
if context == '':
|
|
context = 'Processing'
|
|
return QCoreApplication.translate(context, string)
|
|
|
|
|
|
def getFileFilter(param):
|
|
"""
|
|
Returns a suitable file filter pattern for the specified parameter definition
|
|
:param param:
|
|
:return:
|
|
"""
|
|
if param.type() == 'layer':
|
|
vectors = QgsProviderRegistry.instance().fileVectorFilters().split(';;')
|
|
vectors.pop(0)
|
|
rasters = QgsProviderRegistry.instance().fileRasterFilters().split(';;')
|
|
rasters.pop(0)
|
|
filters = set(vectors + rasters)
|
|
filters = sorted(filters)
|
|
return tr('All files (*.*)') + ';;' + ";;".join(filters)
|
|
elif param.type() == 'multilayer':
|
|
if param.layerType() == QgsProcessing.TypeRaster:
|
|
exts = QgsRasterFileWriter.supportedFormatExtensions()
|
|
elif param.layerType() == QgsProcessing.TypeFile:
|
|
return tr('All files (*.*)', 'QgsProcessingParameterMultipleLayers')
|
|
else:
|
|
exts = QgsVectorFileWriter.supportedFormatExtensions()
|
|
for i in range(len(exts)):
|
|
exts[i] = tr('{0} files (*.{1})', 'QgsProcessingParameterMultipleLayers').format(exts[i].upper(), exts[i].lower())
|
|
return tr('All files (*.*)') + ';;' + ';;'.join(exts)
|
|
elif param.type() == 'raster':
|
|
return QgsProviderRegistry.instance().fileRasterFilters()
|
|
elif param.type() == 'rasterDestination':
|
|
if param.provider() is not None:
|
|
exts = param.provider().supportedOutputRasterLayerExtensions()
|
|
else:
|
|
exts = QgsRasterFileWriter.supportedFormatExtensions()
|
|
for i in range(len(exts)):
|
|
exts[i] = tr('{0} files (*.{1})', 'ParameterRaster').format(exts[i].upper(), exts[i].lower())
|
|
return ';;'.join(exts) + ';;' + tr('All files (*.*)')
|
|
elif param.type() in ('sink', 'vectorDestination'):
|
|
if param.provider() is not None:
|
|
exts = param.provider().supportedOutputVectorLayerExtensions()
|
|
else:
|
|
exts = QgsVectorFileWriter.supportedFormatExtensions()
|
|
for i in range(len(exts)):
|
|
exts[i] = tr('{0} files (*.{1})', 'ParameterVector').format(exts[i].upper(), exts[i].lower())
|
|
return ';;'.join(exts) + ';;' + tr('All files (*.*)')
|
|
elif param.type() == 'source':
|
|
return QgsProviderRegistry.instance().fileVectorFilters()
|
|
elif param.type() == 'vector':
|
|
return QgsProviderRegistry.instance().fileVectorFilters()
|
|
elif param.type() == 'fileDestination':
|
|
return param.fileFilter() + ';;' + tr('All files (*.*)')
|
|
|
|
if param.defaultFileExtension():
|
|
return tr('Default extension') + ' (*.' + param.defaultFileExtension() + ')'
|
|
else:
|
|
return ''
|